Ahmed Allam
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from typing import Any, ClassVar
|
|
|
|
from textual.widgets import Static
|
|
|
|
from .base_renderer import BaseToolRenderer
|
|
from .registry import register_tool_renderer
|
|
|
|
|
|
@register_tool_renderer
|
|
class PythonRenderer(BaseToolRenderer):
|
|
tool_name: ClassVar[str] = "python_action"
|
|
css_classes: ClassVar[list[str]] = ["tool-call", "python-tool"]
|
|
|
|
@classmethod
|
|
def render(cls, tool_data: dict[str, Any]) -> Static:
|
|
args = tool_data.get("args", {})
|
|
|
|
action = args.get("action", "")
|
|
code = args.get("code", "")
|
|
|
|
header = "</> [bold #3b82f6]Python[/]"
|
|
|
|
if code and action in ["new_session", "execute"]:
|
|
code_display = code[:250] + "..." if len(code) > 250 else code
|
|
content_text = f"{header}\n [italic white]{cls.escape_markup(code_display)}[/]"
|
|
elif action == "close":
|
|
content_text = f"{header}\n [dim]Closing session...[/]"
|
|
elif action == "list_sessions":
|
|
content_text = f"{header}\n [dim]Listing sessions...[/]"
|
|
else:
|
|
content_text = f"{header}\n [dim]Running...[/]"
|
|
|
|
css_classes = cls.get_css_classes("completed")
|
|
return Static(content_text, classes=css_classes)
|