141 lines
4.3 KiB
Python
141 lines
4.3 KiB
Python
from typing import Any, ClassVar
|
|
|
|
from rich.text import Text
|
|
from textual.widgets import Static
|
|
|
|
from .base_renderer import BaseToolRenderer
|
|
from .registry import register_tool_renderer
|
|
|
|
|
|
@register_tool_renderer
|
|
class ViewAgentGraphRenderer(BaseToolRenderer):
|
|
tool_name: ClassVar[str] = "view_agent_graph"
|
|
css_classes: ClassVar[list[str]] = ["tool-call", "agents-graph-tool"]
|
|
|
|
@classmethod
|
|
def render(cls, tool_data: dict[str, Any]) -> Static:
|
|
status = tool_data.get("status", "unknown")
|
|
|
|
text = Text()
|
|
text.append("◇ ", style="#a78bfa")
|
|
text.append("viewing agents graph", style="dim")
|
|
|
|
css_classes = cls.get_css_classes(status)
|
|
return Static(text, classes=css_classes)
|
|
|
|
|
|
@register_tool_renderer
|
|
class CreateAgentRenderer(BaseToolRenderer):
|
|
tool_name: ClassVar[str] = "create_agent"
|
|
css_classes: ClassVar[list[str]] = ["tool-call", "agents-graph-tool"]
|
|
|
|
@classmethod
|
|
def render(cls, tool_data: dict[str, Any]) -> Static:
|
|
args = tool_data.get("args", {})
|
|
status = tool_data.get("status", "unknown")
|
|
|
|
task = args.get("task", "")
|
|
name = args.get("name", "Agent")
|
|
|
|
text = Text()
|
|
text.append("◈ ", style="#a78bfa")
|
|
text.append("spawning ", style="dim")
|
|
text.append(name, style="bold #a78bfa")
|
|
|
|
if task:
|
|
text.append("\n ")
|
|
text.append(task, style="dim")
|
|
|
|
css_classes = cls.get_css_classes(status)
|
|
return Static(text, classes=css_classes)
|
|
|
|
|
|
@register_tool_renderer
|
|
class SendMessageToAgentRenderer(BaseToolRenderer):
|
|
tool_name: ClassVar[str] = "send_message_to_agent"
|
|
css_classes: ClassVar[list[str]] = ["tool-call", "agents-graph-tool"]
|
|
|
|
@classmethod
|
|
def render(cls, tool_data: dict[str, Any]) -> Static:
|
|
args = tool_data.get("args", {})
|
|
status = tool_data.get("status", "unknown")
|
|
|
|
message = args.get("message", "")
|
|
agent_id = args.get("agent_id", "")
|
|
|
|
text = Text()
|
|
text.append("→ ", style="#60a5fa")
|
|
if agent_id:
|
|
text.append(f"to {agent_id}", style="dim")
|
|
else:
|
|
text.append("sending message", style="dim")
|
|
|
|
if message:
|
|
text.append("\n ")
|
|
text.append(message, style="dim")
|
|
|
|
css_classes = cls.get_css_classes(status)
|
|
return Static(text, classes=css_classes)
|
|
|
|
|
|
@register_tool_renderer
|
|
class AgentFinishRenderer(BaseToolRenderer):
|
|
tool_name: ClassVar[str] = "agent_finish"
|
|
css_classes: ClassVar[list[str]] = ["tool-call", "agents-graph-tool"]
|
|
|
|
@classmethod
|
|
def render(cls, tool_data: dict[str, Any]) -> Static:
|
|
args = tool_data.get("args", {})
|
|
|
|
result_summary = args.get("result_summary", "")
|
|
findings = args.get("findings", [])
|
|
success = args.get("success", True)
|
|
|
|
text = Text()
|
|
|
|
if success:
|
|
text.append("◆ ", style="#22c55e")
|
|
text.append("Agent completed", style="bold #22c55e")
|
|
else:
|
|
text.append("◆ ", style="#ef4444")
|
|
text.append("Agent failed", style="bold #ef4444")
|
|
|
|
if result_summary:
|
|
text.append("\n ")
|
|
text.append(result_summary, style="bold")
|
|
|
|
if findings and isinstance(findings, list):
|
|
for finding in findings:
|
|
text.append("\n • ")
|
|
text.append(str(finding), style="dim")
|
|
else:
|
|
text.append("\n ")
|
|
text.append("Completing task...", style="dim")
|
|
|
|
css_classes = cls.get_css_classes("completed")
|
|
return Static(text, classes=css_classes)
|
|
|
|
|
|
@register_tool_renderer
|
|
class WaitForMessageRenderer(BaseToolRenderer):
|
|
tool_name: ClassVar[str] = "wait_for_message"
|
|
css_classes: ClassVar[list[str]] = ["tool-call", "agents-graph-tool"]
|
|
|
|
@classmethod
|
|
def render(cls, tool_data: dict[str, Any]) -> Static:
|
|
args = tool_data.get("args", {})
|
|
status = tool_data.get("status", "unknown")
|
|
|
|
reason = args.get("reason", "")
|
|
|
|
text = Text()
|
|
text.append("○ ", style="#6b7280")
|
|
text.append("waiting", style="dim")
|
|
|
|
if reason:
|
|
text.append("\n ")
|
|
text.append(reason, style="dim")
|
|
|
|
css_classes = cls.get_css_classes(status)
|
|
return Static(text, classes=css_classes)
|