Icons for chat boxes.

lynxkite-lynxscribe/src/lynxkite_lynxscribe/agentic.py

@@ -22,12 +22,12 @@ op = ops.op_registration(
 )
 
 
-@op("Chat frontend", color="gray", outputs=[], view="service")
+@op("Chat frontend", color="gray", outputs=[], view="service", icon="message-circle-filled")
 def chat_frontend(agent: dict):
     return agent_from_dict(agent, default_model={"name": "openai", "model_name": "gpt-4.1-nano"})
 
 
-@op("Gradio chat", color="gray", outputs=[], view="gradio")
+@op("Gradio chat", color="gray", outputs=[], view="gradio", icon="message-circle-filled")
 def gradio_chat(agent: dict):
     import gradio as gr
 
@@ -57,7 +57,7 @@ def agent_from_dict(agent: dict, default_model: dict) -> "Agent":
     )
 
 
-@op("Agent")
+@op("Agent", icon="message-chatbot")
 def agent(
     tools: list[dict],
     *,
@@ -92,7 +92,7 @@ def agent(
     return params
 
 
-@op("Model: OpenAI", color="orange")
+@op("Model: OpenAI", color="orange", icon="server-spark")
 def openai_model(*, model_name: str, base_url: str = ""):
     """Use the OPENAI_API_KEY environment variable to provide authentication."""
     params = {"model": {"name": "openai", "model_name": model_name}}
@@ -101,7 +101,7 @@ def openai_model(*, model_name: str, base_url: str = ""):
     return params
 
 
-@op("MCP: Custom", color="green")
+@op("MCP: Custom", color="green", icon="tool")
 def custom_mcp(*, command: str, extra_prompt: ops.LongStr):
     if command.strip():
         return {"command": command.strip().split(), "extra_prompt": extra_prompt}
@@ -116,7 +116,7 @@ class MCPSearchEngine(str, enum.Enum):
     DuckDuckGo = "DuckDuckGo"
 
 
-@op("MCP: Search web", color="green")
+@op("MCP: Search web", color="green", icon="tool")
 def web_search(*, engine: MCPSearchEngine = MCPSearchEngine.Google):
     match engine:
         case MCPSearchEngine.Google:
@@ -127,17 +127,17 @@ def web_search(*, engine: MCPSearchEngine = MCPSearchEngine.Google):
             return {"command": ["uvx", "duckduckgo-mcp-server"]}
 
 
-@op("MCP: Financial data", color="green")
+@op("MCP: Financial data", color="green", icon="tool")
 def financial_data():
     return {"command": ["uvx", "investor-agent"]}
 
 
-@op("MCP: Calculator", color="green")
+@op("MCP: Calculator", color="green", icon="tool")
 def calculator():
     return {"command": ["uvx", "mcp-server-calculator"]}
 
 
-@op("MCP: Time", color="green")
+@op("MCP: Time", color="green", icon="tool")
 def time():
     return {"command": ["uvx", "mcp-server-time"]}