Synapse

"""
Synapse: Context Window Summarizer & Token Compressor
Preserves critical conversation entities across unbounded multi-turn sessions.
"""
import re
from typing import List, Dict, Any
from pydantic import BaseModel, Field

class ConversationTurn(BaseModel):
    role: str
    content: str
    tokens: int

class CompressedContext(BaseModel):
    active_memory: str
    compressed_history_tokens: int
    retained_entities: Dict[str, str]

class SynapseContextCompressor:
    def __init__(self, target_budget: int = 4000):
        self.target_budget = target_budget

    def compress_history(self, turns: List[ConversationTurn]) -> CompressedContext:
        entities = {}
        summary_points = []

        # Extract structured state tokens (e.g. key=val, user requirements)
        for t in turns:
            matches = re.findall(r"([A-Za-z_]+)\s*=\s*['\"]([^'\"]+)['\"]", t.content)
            for k, v in matches:
                entities[k] = v
            if len(t.content) > 120:
                summary_points.append(f"{t.role}: {t.content[:100]}...")

        compressed_text = "STATE CONTEXT:\n" + "\n".join(f"- {k}: {v}" for k, v in entities.items())
        compressed_text += "\nHISTORY SUMMARY:\n" + "\n".join(summary_points)

        return CompressedContext(
            active_memory=compressed_text,
            compressed_history_tokens=len(compressed_text.split()) * 2,
            retained_entities=entities
        )