massgen 0.0.3__py3-none-any.whl

massgen/frontend/displays/__init__.py

@@ -0,0 +1,24 @@
+"""
+MassGen Display Components
+
+Provides various display interfaces for MassGen coordination visualization.
+"""
+
+from .base_display import BaseDisplay
+from .terminal_display import TerminalDisplay
+from .simple_display import SimpleDisplay
+from .rich_terminal_display import (
+    RichTerminalDisplay,
+    is_rich_available,
+    create_rich_display,
+)
+
+__all__ = [
+    "BaseDisplay",
+    "TerminalDisplay",
+    "SimpleDisplay",
+    "RichTerminalDisplay",
+    "is_rich_available",
+    "is_textual_available",
+    "create_rich_display",
+]

massgen/frontend/displays/base_display.py

@@ -0,0 +1,83 @@
+"""
+Base Display Interface for MassGen Coordination
+
+Defines the interface that all display implementations must follow.
+"""
+
+from abc import ABC, abstractmethod
+from typing import Dict, List, Any, Optional
+
+
+class BaseDisplay(ABC):
+    """Abstract base class for MassGen coordination displays."""
+
+    def __init__(self, agent_ids: List[str], **kwargs):
+        """Initialize display with agent IDs and configuration."""
+        self.agent_ids = agent_ids
+        self.agent_outputs = {agent_id: [] for agent_id in agent_ids}
+        self.agent_status = {agent_id: "waiting" for agent_id in agent_ids}
+        self.orchestrator_events = []
+        self.config = kwargs
+
+    @abstractmethod
+    def initialize(self, question: str, log_filename: Optional[str] = None):
+        """Initialize the display with question and optional log file."""
+        pass
+
+    @abstractmethod
+    def update_agent_content(
+        self, agent_id: str, content: str, content_type: str = "thinking"
+    ):
+        """Update content for a specific agent.
+
+        Args:
+            agent_id: The agent whose content to update
+            content: The content to add/update
+            content_type: Type of content ("thinking", "tool", "status")
+        """
+        pass
+
+    @abstractmethod
+    def update_agent_status(self, agent_id: str, status: str):
+        """Update status for a specific agent.
+
+        Args:
+            agent_id: The agent whose status to update
+            status: New status ("waiting", "working", "completed")
+        """
+        pass
+
+    @abstractmethod
+    def add_orchestrator_event(self, event: str):
+        """Add an orchestrator coordination event.
+
+        Args:
+            event: The coordination event message
+        """
+        pass
+
+    @abstractmethod
+    def show_final_answer(self, answer: str):
+        """Display the final coordinated answer.
+
+        Args:
+            answer: The final coordinated answer
+        """
+        pass
+
+    @abstractmethod
+    def cleanup(self):
+        """Clean up display resources."""
+        pass
+
+    def get_agent_content(self, agent_id: str) -> List[str]:
+        """Get all content for a specific agent."""
+        return self.agent_outputs.get(agent_id, [])
+
+    def get_agent_status(self, agent_id: str) -> str:
+        """Get current status for a specific agent."""
+        return self.agent_status.get(agent_id, "unknown")
+
+    def get_orchestrator_events(self) -> List[str]:
+        """Get all orchestrator events."""
+        return self.orchestrator_events.copy()