PraisonAI 3.0.0__py3-none-any.whl

praisonai/cli/features/tui/widgets/queue_panel.py

@@ -0,0 +1,200 @@
+"""
+Queue Panel Widget for PraisonAI TUI.
+
+Displays queue status and allows queue management.
+"""
+
+from typing import List, Optional
+from dataclasses import dataclass
+
+try:
+    from textual.widget import Widget
+    from textual.widgets import Static, DataTable, Button
+    from textual.containers import Vertical
+    from textual.reactive import reactive
+    from textual.message import Message
+    from rich.text import Text
+    from rich.panel import Panel
+    TEXTUAL_AVAILABLE = True
+except ImportError:
+    TEXTUAL_AVAILABLE = False
+    Widget = object
+    Message = object
+
+
+@dataclass
+class QueueItem:
+    """A queue item for display."""
+    run_id: str
+    agent_name: str
+    input_preview: str
+    state: str
+    priority: str
+    wait_time: str
+
+
+if TEXTUAL_AVAILABLE:
+    class QueuePanelWidget(Vertical):
+        """
+        Queue panel widget.
+        
+        Displays:
+        - Queue summary (queued/running counts)
+        - List of queued runs
+        - Cancel/retry buttons
+        """
+        
+        DEFAULT_CSS = """
+        QueuePanelWidget {
+            height: 100%;
+            border: solid $secondary;
+            background: $surface;
+            padding: 0 1;
+        }
+        
+        QueuePanelWidget .queue-header {
+            height: 1;
+            background: $secondary;
+            padding: 0 1;
+        }
+        
+        QueuePanelWidget DataTable {
+            height: 1fr;
+        }
+        
+        QueuePanelWidget .queue-item-queued {
+            color: $warning;
+        }
+        
+        QueuePanelWidget .queue-item-running {
+            color: $success;
+        }
+        
+        QueuePanelWidget .queue-item-failed {
+            color: $error;
+        }
+        """
+        
+        class RunSelected(Message):
+            """Event when a run is selected."""
+            def __init__(self, run_id: str):
+                self.run_id = run_id
+                super().__init__()
+        
+        class CancelRequested(Message):
+            """Event when cancel is requested."""
+            def __init__(self, run_id: str):
+                self.run_id = run_id
+                super().__init__()
+        
+        class RetryRequested(Message):
+            """Event when retry is requested."""
+            def __init__(self, run_id: str):
+                self.run_id = run_id
+                super().__init__()
+        
+        queued_count: reactive[int] = reactive(0)
+        running_count: reactive[int] = reactive(0)
+        
+        def __init__(
+            self,
+            name: Optional[str] = None,
+            id: Optional[str] = None,
+            classes: Optional[str] = None,
+        ):
+            super().__init__(name=name, id=id, classes=classes)
+            self._items: List[QueueItem] = []
+            self._selected_run_id: Optional[str] = None
+        
+        def compose(self):
+            """Compose the widget."""
+            yield Static("Queue", id="queue-title", classes="queue-header")
+            yield DataTable(id="queue-table")
+        
+        def on_mount(self) -> None:
+            """Handle mount."""
+            table = self.query_one("#queue-table", DataTable)
+            table.add_columns("ID", "Agent", "Input", "State", "Priority")
+            table.cursor_type = "row"
+            self._update_header()
+        
+        def watch_queued_count(self, value: int) -> None:
+            self._update_header()
+        
+        def watch_running_count(self, value: int) -> None:
+            self._update_header()
+        
+        def _update_header(self) -> None:
+            """Update the header text."""
+            try:
+                header = self.query_one("#queue-title", Static)
+                text = Text()
+                text.append("Queue ", style="bold")
+                text.append(f"⏳ {self.queued_count} queued", style="yellow")
+                text.append(" │ ", style="dim")
+                text.append(f"▶ {self.running_count} running", style="green")
+                header.update(text)
+            except Exception:
+                pass
+        
+        def update_items(self, items: List[QueueItem]) -> None:
+            """Update the queue items."""
+            self._items = items
+            
+            table = self.query_one("#queue-table", DataTable)
+            table.clear()
+            
+            for item in items:
+                # Style based on state
+                state_style = {
+                    "queued": "yellow",
+                    "running": "green",
+                    "failed": "red",
+                    "cancelled": "dim",
+                    "succeeded": "cyan",
+                }.get(item.state.lower(), "")
+                
+                state_text = Text(item.state, style=state_style)
+                
+                # Truncate input preview
+                input_preview = item.input_preview[:30]
+                if len(item.input_preview) > 30:
+                    input_preview += "..."
+                
+                table.add_row(
+                    item.run_id[:8],
+                    item.agent_name,
+                    input_preview,
+                    state_text,
+                    item.priority,
+                    key=item.run_id,
+                )
+        
+        def on_data_table_row_selected(self, event: DataTable.RowSelected) -> None:
+            """Handle row selection."""
+            if event.row_key:
+                self._selected_run_id = str(event.row_key.value)
+                self.post_message(self.RunSelected(self._selected_run_id))
+        
+        def cancel_selected(self) -> None:
+            """Cancel the selected run."""
+            if self._selected_run_id:
+                self.post_message(self.CancelRequested(self._selected_run_id))
+        
+        def retry_selected(self) -> None:
+            """Retry the selected run."""
+            if self._selected_run_id:
+                self.post_message(self.RetryRequested(self._selected_run_id))
+        
+        @property
+        def selected_run_id(self) -> Optional[str]:
+            """Get the selected run ID."""
+            return self._selected_run_id
+
+else:
+    class QueuePanelWidget:
+        """Placeholder when Textual is not available."""
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "Textual is required for TUI. Install with: pip install praisonai[tui]"
+            )

praisonai/cli/features/tui/widgets/status.py

@@ -0,0 +1,167 @@
+"""
+Status Widget for PraisonAI TUI.
+
+Displays status bar with session info, model, tokens, and cost.
+"""
+
+from typing import Optional
+from dataclasses import dataclass
+
+try:
+    from textual.widget import Widget
+    from textual.widgets import Static
+    from textual.containers import Horizontal
+    from textual.reactive import reactive
+    from rich.text import Text
+    from rich.table import Table
+    TEXTUAL_AVAILABLE = True
+except ImportError:
+    TEXTUAL_AVAILABLE = False
+    Widget = object
+
+
+@dataclass
+class StatusInfo:
+    """Status information."""
+    session_id: str = ""
+    model: str = "gpt-4o-mini"
+    total_tokens: int = 0
+    total_cost: float = 0.0
+    queued_count: int = 0
+    running_count: int = 0
+    is_processing: bool = False
+    status_message: str = ""
+
+
+if TEXTUAL_AVAILABLE:
+    class StatusWidget(Static):
+        """
+        Status bar widget.
+        
+        Displays:
+        - Session ID
+        - Current model
+        - Token usage
+        - Cost
+        - Queue status
+        - Processing indicator
+        """
+        
+        DEFAULT_CSS = """
+        StatusWidget {
+            height: 1;
+            background: $primary;
+            color: $text;
+            padding: 0 1;
+        }
+        """
+        
+        session_id: reactive[str] = reactive("")
+        model: reactive[str] = reactive("gpt-4o-mini")
+        total_tokens: reactive[int] = reactive(0)
+        total_cost: reactive[float] = reactive(0.0)
+        queued_count: reactive[int] = reactive(0)
+        running_count: reactive[int] = reactive(0)
+        is_processing: reactive[bool] = reactive(False)
+        status_message: reactive[str] = reactive("")
+        
+        def __init__(
+            self,
+            name: Optional[str] = None,
+            id: Optional[str] = None,
+            classes: Optional[str] = None,
+        ):
+            super().__init__("", name=name, id=id, classes=classes)
+        
+        def on_mount(self) -> None:
+            """Handle mount."""
+            self._update_display()
+        
+        def watch_session_id(self, value: str) -> None:
+            self._update_display()
+        
+        def watch_model(self, value: str) -> None:
+            self._update_display()
+        
+        def watch_total_tokens(self, value: int) -> None:
+            self._update_display()
+        
+        def watch_total_cost(self, value: float) -> None:
+            self._update_display()
+        
+        def watch_queued_count(self, value: int) -> None:
+            self._update_display()
+        
+        def watch_running_count(self, value: int) -> None:
+            self._update_display()
+        
+        def watch_is_processing(self, value: bool) -> None:
+            self._update_display()
+        
+        def watch_status_message(self, value: str) -> None:
+            self._update_display()
+        
+        def _update_display(self) -> None:
+            """Update the status display."""
+            parts = []
+            
+            # App name
+            parts.append(("◉ PraisonAI", "bold cyan"))
+            
+            # Session
+            if self.session_id:
+                parts.append((f"Session: {self.session_id[:8]}", ""))
+            
+            # Model
+            parts.append((f"Model: {self.model}", ""))
+            
+            # Tokens
+            if self.total_tokens > 0:
+                parts.append((f"Tokens: {self.total_tokens:,}", ""))
+            
+            # Cost
+            if self.total_cost > 0:
+                parts.append((f"${self.total_cost:.4f}", "green"))
+            
+            # Queue status
+            if self.queued_count > 0 or self.running_count > 0:
+                queue_text = f"⏳ {self.queued_count} queued"
+                if self.running_count > 0:
+                    queue_text += f" │ ▶ {self.running_count} running"
+                parts.append((queue_text, "yellow"))
+            
+            # Processing indicator
+            if self.is_processing:
+                parts.append(("⟳ Processing...", "bold magenta"))
+            
+            # Status message
+            if self.status_message:
+                parts.append((self.status_message, "italic"))
+            
+            # Build text
+            text = Text()
+            for i, (content, style) in enumerate(parts):
+                if i > 0:
+                    text.append(" │ ", style="dim")
+                text.append(content, style=style)
+            
+            self.update(text)
+        
+        def update_info(self, info: StatusInfo) -> None:
+            """Update all status info at once."""
+            self.session_id = info.session_id
+            self.model = info.model
+            self.total_tokens = info.total_tokens
+            self.total_cost = info.total_cost
+            self.queued_count = info.queued_count
+            self.running_count = info.running_count
+            self.is_processing = info.is_processing
+            self.status_message = info.status_message
+
+else:
+    class StatusWidget:
+        """Placeholder when Textual is not available."""
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "Textual is required for TUI. Install with: pip install praisonai[tui]"
+            )

praisonai/cli/features/tui/widgets/tool_panel.py

@@ -0,0 +1,248 @@
+"""
+Tool Panel Widget for PraisonAI TUI.
+
+Displays tool execution status and approval dialogs.
+"""
+
+from typing import List, Optional
+from dataclasses import dataclass, field
+import time
+
+try:
+    from textual.widget import Widget
+    from textual.widgets import Static, ListView, ListItem
+    from textual.containers import Vertical
+    from textual.reactive import reactive
+    from textual.message import Message
+    from rich.text import Text
+    from rich.panel import Panel
+    TEXTUAL_AVAILABLE = True
+except ImportError:
+    TEXTUAL_AVAILABLE = False
+    Widget = object
+    Message = object
+
+
+@dataclass
+class ToolCall:
+    """A tool call for display."""
+    call_id: str
+    tool_name: str
+    args_preview: str
+    status: str  # "pending", "running", "completed", "failed", "approved", "rejected"
+    result_preview: Optional[str] = None
+    started_at: float = field(default_factory=time.time)
+    ended_at: Optional[float] = None
+    requires_approval: bool = False
+
+
+if TEXTUAL_AVAILABLE:
+    class ToolPanelWidget(Vertical):
+        """
+        Tool panel widget.
+        
+        Displays:
+        - Recent tool calls
+        - Tool execution status
+        - Approval requests
+        """
+        
+        DEFAULT_CSS = """
+        ToolPanelWidget {
+            height: 100%;
+            border: solid $accent;
+            background: $surface;
+            padding: 0 1;
+        }
+        
+        ToolPanelWidget .tool-header {
+            height: 1;
+            background: $accent;
+            padding: 0 1;
+        }
+        
+        ToolPanelWidget .tool-pending {
+            color: $warning;
+        }
+        
+        ToolPanelWidget .tool-running {
+            color: $primary;
+        }
+        
+        ToolPanelWidget .tool-completed {
+            color: $success;
+        }
+        
+        ToolPanelWidget .tool-failed {
+            color: $error;
+        }
+        
+        ToolPanelWidget .tool-approval {
+            background: $warning-darken-2;
+            border: solid $warning;
+            padding: 1;
+        }
+        """
+        
+        class ApprovalResponse(Message):
+            """Event when user responds to approval request."""
+            def __init__(self, call_id: str, approved: bool):
+                self.call_id = call_id
+                self.approved = approved
+                super().__init__()
+        
+        def __init__(
+            self,
+            max_items: int = 50,
+            name: Optional[str] = None,
+            id: Optional[str] = None,
+            classes: Optional[str] = None,
+        ):
+            super().__init__(name=name, id=id, classes=classes)
+            self._calls: List[ToolCall] = []
+            self._max_items = max_items
+            self._pending_approval: Optional[ToolCall] = None
+        
+        def compose(self):
+            """Compose the widget."""
+            yield Static("Tools", id="tool-title", classes="tool-header")
+            yield Vertical(id="tool-list")
+        
+        def on_mount(self) -> None:
+            """Handle mount."""
+            self._update_display()
+        
+        def add_call(self, call: ToolCall) -> None:
+            """Add a tool call."""
+            self._calls.append(call)
+            
+            # Trim old calls
+            if len(self._calls) > self._max_items:
+                self._calls = self._calls[-self._max_items:]
+            
+            # Check for approval
+            if call.requires_approval and call.status == "pending":
+                self._pending_approval = call
+            
+            self._update_display()
+        
+        def update_call(self, call_id: str, status: str, result: Optional[str] = None) -> None:
+            """Update a tool call status."""
+            for call in self._calls:
+                if call.call_id == call_id:
+                    call.status = status
+                    if result:
+                        call.result_preview = result[:100]
+                    if status in ("completed", "failed"):
+                        call.ended_at = time.time()
+                    break
+            
+            # Clear pending approval if this was it
+            if self._pending_approval and self._pending_approval.call_id == call_id:
+                self._pending_approval = None
+            
+            self._update_display()
+        
+        def _update_display(self) -> None:
+            """Update the display."""
+            try:
+                container = self.query_one("#tool-list", Vertical)
+            except Exception:
+                return
+            
+            # Clear existing
+            container.remove_children()
+            
+            # Show pending approval first
+            if self._pending_approval:
+                approval_text = Text()
+                approval_text.append("⚠ Approval Required\n", style="bold yellow")
+                approval_text.append(f"Tool: {self._pending_approval.tool_name}\n")
+                approval_text.append(f"Args: {self._pending_approval.args_preview}\n")
+                approval_text.append("[Y] Approve  [N] Reject", style="dim")
+                
+                container.mount(Static(
+                    Panel(approval_text, border_style="yellow"),
+                    classes="tool-approval"
+                ))
+            
+            # Show recent calls
+            for call in reversed(self._calls[-10:]):
+                if call == self._pending_approval:
+                    continue
+                
+                # Status icon
+                status_icons = {
+                    "pending": "⏳",
+                    "running": "⟳",
+                    "completed": "✓",
+                    "failed": "✗",
+                    "approved": "✓",
+                    "rejected": "✗",
+                }
+                icon = status_icons.get(call.status, "?")
+                
+                # Status style
+                status_styles = {
+                    "pending": "yellow",
+                    "running": "cyan",
+                    "completed": "green",
+                    "failed": "red",
+                    "approved": "green",
+                    "rejected": "red",
+                }
+                style = status_styles.get(call.status, "")
+                
+                text = Text()
+                text.append(f"{icon} ", style=style)
+                text.append(call.tool_name, style="bold")
+                
+                if call.result_preview:
+                    text.append(f" → {call.result_preview[:30]}...", style="dim")
+                
+                container.mount(Static(text, classes=f"tool-{call.status}"))
+        
+        def handle_approval_key(self, key: str) -> bool:
+            """
+            Handle approval key press.
+            
+            Returns True if handled.
+            """
+            if not self._pending_approval:
+                return False
+            
+            if key.lower() == "y":
+                self.post_message(self.ApprovalResponse(
+                    self._pending_approval.call_id,
+                    approved=True
+                ))
+                self.update_call(self._pending_approval.call_id, "approved")
+                return True
+            elif key.lower() == "n":
+                self.post_message(self.ApprovalResponse(
+                    self._pending_approval.call_id,
+                    approved=False
+                ))
+                self.update_call(self._pending_approval.call_id, "rejected")
+                return True
+            
+            return False
+        
+        @property
+        def has_pending_approval(self) -> bool:
+            """Check if there's a pending approval."""
+            return self._pending_approval is not None
+        
+        def clear(self) -> None:
+            """Clear all tool calls."""
+            self._calls.clear()
+            self._pending_approval = None
+            self._update_display()
+
+else:
+    class ToolPanelWidget:
+        """Placeholder when Textual is not available."""
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "Textual is required for TUI. Install with: pip install praisonai[tui]"
+            )