a6dcb7756e
- Convert LiteLLM requests to streaming mode with stream_request() - Add streaming parser to handle live LLM output segments - Update TUI for real-time streaming content rendering - Add tracer methods for streaming content tracking - Clean function tags from streamed content to prevent display - Remove all truncation from tool renderers for full content visibility
30 lines
844 B
Python
30 lines
844 B
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 WebSearchRenderer(BaseToolRenderer):
|
|
tool_name: ClassVar[str] = "web_search"
|
|
css_classes: ClassVar[list[str]] = ["tool-call", "web-search-tool"]
|
|
|
|
@classmethod
|
|
def render(cls, tool_data: dict[str, Any]) -> Static:
|
|
args = tool_data.get("args", {})
|
|
query = args.get("query", "")
|
|
|
|
text = Text()
|
|
text.append("🌐 ")
|
|
text.append("Searching the web...", style="bold #60a5fa")
|
|
|
|
if query:
|
|
text.append("\n ")
|
|
text.append(query, style="dim")
|
|
|
|
css_classes = cls.get_css_classes("completed")
|
|
return Static(text, classes=css_classes)
|