hdsp-jupyter-extension 2.0.26__py3-none-any.whl → 2.0.28__py3-none-any.whl

agent_server/langchain/middleware/subagent_events.py

@@ -42,35 +42,38 @@ class SubagentEvent:
         """Convert to debug message for UI display (legacy, for logging)."""
         if self.event_type == "subagent_start":
             desc_preview = self.description[:80] + "..." if self.description and len(self.description) > 80 else self.description
-            return f"Subagent - {self.subagent_name} - 작업 시작: {desc_preview}"
+            return f"Subagent-{self.subagent_name} 작업 시작: {desc_preview}"
         elif self.event_type == "subagent_tool_call":
-            return f"Subagent - {self.subagent_name} - Tool 실행: {self.tool_name}"
+            return f"Subagent-{self.subagent_name} Tool 실행: {self.tool_name}"
         elif self.event_type == "subagent_complete":
-            return f"Subagent - {self.subagent_name} - 완료"
+            return f"Subagent-{self.subagent_name} 완료"
         else:
-            return f"Subagent - {self.subagent_name} - {self.event_type}"
+            return f"Subagent-{self.subagent_name} {self.event_type}"
 
     def to_status_dict(self) -> Dict[str, Any]:
         """Convert to status dict with icon for SSE streaming."""
         if self.event_type == "subagent_start":
             desc_preview = self.description[:80] + "..." if self.description and len(self.description) > 80 else self.description
+            full_status = f"Subagent-{self.subagent_name} 작업 시작: {self.description}"
             return {
-                "status": f"Subagent - {self.subagent_name} - 작업 시작: {desc_preview}",
-                "icon": "subagentStart"
+                "status": f"Subagent-{self.subagent_name} 작업 시작: {desc_preview}",
+                "icon": "subagentStart",
+                "expandable": bool(self.description and len(self.description) > 80),
+                "full_text": full_status,
             }
         elif self.event_type == "subagent_tool_call":
             return {
-                "status": f"Subagent - {self.subagent_name} - Tool 실행: {self.tool_name}",
+                "status": f"Subagent-{self.subagent_name} Tool 실행: {self.tool_name}",
                 "icon": "tool"
             }
         elif self.event_type == "subagent_complete":
             return {
-                "status": f"Subagent - {self.subagent_name} - 완료",
+                "status": f"Subagent-{self.subagent_name} 완료",
                 "icon": "subagentComplete"
             }
         else:
             return {
-                "status": f"Subagent - {self.subagent_name} - {self.event_type}",
+                "status": f"Subagent-{self.subagent_name} {self.event_type}",
                 "icon": "info"
             }
 
@@ -169,3 +172,79 @@ def drain_subagent_events() -> List[SubagentEvent]:
 def get_pending_event_count() -> int:
     """Get the number of pending events in the queue."""
     return _event_queue.qsize()
+
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Summarization Events - For context compression status display
+# ═══════════════════════════════════════════════════════════════════════════
+
+@dataclass
+class SummarizationEvent:
+    """Represents a summarization event for UI display."""
+
+    event_type: str  # summarization_start, summarization_complete
+    message_count_before: Optional[int] = None
+    message_count_after: Optional[int] = None
+    timestamp: datetime = field(default_factory=datetime.now)
+
+    def to_status_dict(self) -> Dict[str, Any]:
+        """Convert to status dict with icon for SSE streaming."""
+        if self.event_type == "summarization_start":
+            return {
+                "status": "대화 컨텍스트 요약 중...",
+                "icon": "thinking"
+            }
+        elif self.event_type == "summarization_complete":
+            if self.message_count_before and self.message_count_after:
+                return {
+                    "status": f"대화가 자동으로 압축되었습니다. ({self.message_count_before} → {self.message_count_after} 메시지)",
+                    "icon": "check"
+                }
+            return {
+                "status": "대화가 자동으로 압축되었습니다.",
+                "icon": "check"
+            }
+        else:
+            return {
+                "status": f"Summarization {self.event_type}",
+                "icon": "info"
+            }
+
+
+# Separate queue for summarization events (to avoid mixing with subagent events)
+_summarization_queue: Queue = Queue()
+
+
+def emit_summarization_start() -> None:
+    """Emit a summarization start event."""
+    event = SummarizationEvent(event_type="summarization_start")
+    _summarization_queue.put(event)
+    logger.info("Summarization event: starting context compression")
+
+
+def emit_summarization_complete(before_count: int = 0, after_count: int = 0) -> None:
+    """Emit a summarization complete event."""
+    event = SummarizationEvent(
+        event_type="summarization_complete",
+        message_count_before=before_count,
+        message_count_after=after_count,
+    )
+    _summarization_queue.put(event)
+    logger.info(f"Summarization event: complete ({before_count} → {after_count} messages)")
+
+
+def drain_summarization_events() -> List[SummarizationEvent]:
+    """
+    Drain all pending summarization events from the queue.
+
+    Returns:
+        List of SummarizationEvent objects
+    """
+    events = []
+    while True:
+        try:
+            event = _summarization_queue.get_nowait()
+            events.append(event)
+        except Empty:
+            break
+    return events