fast-agent-mcp 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

mcp_agent/core/agent_app.py

@@ -2,25 +2,25 @@
 Main application wrapper for interacting with agents.
 """
 
-from typing import Optional, Dict, Union, TYPE_CHECKING
+from typing import TYPE_CHECKING, Dict, Optional, Union
 
 from mcp_agent.app import MCPApp
-from mcp_agent.mcp.prompt_message_multipart import PromptMessageMultipart
-from mcp_agent.progress_display import progress_display
-from mcp_agent.workflows.orchestrator.orchestrator import Orchestrator
-from mcp_agent.workflows.parallel.parallel_llm import ParallelLLM
-from mcp_agent.workflows.evaluator_optimizer.evaluator_optimizer import (
-    EvaluatorOptimizerLLM,
-)
 
 # Import proxies directly - they handle their own circular imports
 from mcp_agent.core.proxies import (
     BaseAgentProxy,
+    ChainProxy,
     LLMAgentProxy,
     RouterProxy,
-    ChainProxy,
     WorkflowProxy,
 )
+from mcp_agent.mcp.prompt_message_multipart import PromptMessageMultipart
+from mcp_agent.progress_display import progress_display
+from mcp_agent.workflows.evaluator_optimizer.evaluator_optimizer import (
+    EvaluatorOptimizerLLM,
+)
+from mcp_agent.workflows.orchestrator.orchestrator import Orchestrator
+from mcp_agent.workflows.parallel.parallel_llm import ParallelLLM
 
 # Handle possible circular imports with types
 if TYPE_CHECKING:
@@ -32,15 +32,13 @@ else:
 class AgentApp:
     """Main application wrapper"""
 
-    def __init__(self, app: MCPApp, agents: ProxyDict):
+    def __init__(self, app: MCPApp, agents: ProxyDict) -> None:
         self._app = app
         self._agents = agents
         # Optional: set default agent for direct calls
         self._default = next(iter(agents)) if agents else None
 
-    async def send_prompt(
-        self, prompt: PromptMessageMultipart, agent_name: Optional[str] = None
-    ) -> str:
+    async def send_prompt(self, prompt: PromptMessageMultipart, agent_name: Optional[str] = None) -> str:
         """
         Send a PromptMessageMultipart to an agent
 
@@ -218,15 +216,9 @@ class AgentApp:
                             found_prompts = False
                             for agent_name, agent_proxy in self._agents.items():
                                 # Check if agent has an mcp_aggregator (agent instance)
-                                if hasattr(agent_proxy, "_agent") and hasattr(
-                                    agent_proxy._agent, "list_prompts"
-                                ):
-                                    rich_print(
-                                        f"\n[bold]Fetching prompts for agent [cyan]{agent_name}[/cyan]...[/bold]"
-                                    )
-                                    prompt_servers = (
-                                        await agent_proxy._agent.list_prompts()
-                                    )
+                                if hasattr(agent_proxy, "_agent") and hasattr(agent_proxy._agent, "list_prompts"):
+                                    rich_print(f"\n[bold]Fetching prompts for agent [cyan]{agent_name}[/cyan]...[/bold]")
+                                    prompt_servers = await agent_proxy._agent.list_prompts()
 
                                     if prompt_servers:
                                         found_prompts = True
@@ -234,31 +226,15 @@ class AgentApp:
                                             server_name,
                                             prompts_info,
                                         ) in prompt_servers.items():
-                                            if (
-                                                prompts_info
-                                                and hasattr(prompts_info, "prompts")
-                                                and prompts_info.prompts
-                                            ):
-                                                rich_print(
-                                                    f"\n[bold cyan]{server_name}:[/bold cyan]"
-                                                )
+                                            if prompts_info and hasattr(prompts_info, "prompts") and prompts_info.prompts:
+                                                rich_print(f"\n[bold cyan]{server_name}:[/bold cyan]")
                                                 for prompt in prompts_info.prompts:
                                                     rich_print(f"  {prompt.name}")
-                                            elif (
-                                                isinstance(prompts_info, list)
-                                                and prompts_info
-                                            ):
-                                                rich_print(
-                                                    f"\n[bold cyan]{server_name}:[/bold cyan]"
-                                                )
+                                            elif isinstance(prompts_info, list) and prompts_info:
+                                                rich_print(f"\n[bold cyan]{server_name}:[/bold cyan]")
                                                 for prompt in prompts_info:
-                                                    if (
-                                                        isinstance(prompt, dict)
-                                                        and "name" in prompt
-                                                    ):
-                                                        rich_print(
-                                                            f"  {prompt['name']}"
-                                                        )
+                                                    if isinstance(prompt, dict) and "name" in prompt:
+                                                        rich_print(f"  {prompt['name']}")
                                                     else:
                                                         rich_print(f"  {prompt}")
 
@@ -269,8 +245,8 @@ class AgentApp:
                         continue
                     elif "select_prompt" in command_result:
                         from rich import print as rich_print
-                        from rich.table import Table
                         from rich.console import Console
+                        from rich.table import Table
 
                         console = Console()
 
@@ -278,12 +254,8 @@ class AgentApp:
                         current_proxy = self._agents[agent]
 
                         # Check if the agent has prompt capabilities
-                        if not hasattr(current_proxy, "_agent") or not hasattr(
-                            current_proxy._agent, "list_prompts"
-                        ):
-                            rich_print(
-                                f"[red]Current agent '{agent}' does not support prompts[/red]"
-                            )
+                        if not hasattr(current_proxy, "_agent") or not hasattr(current_proxy._agent, "list_prompts"):
+                            rich_print(f"[red]Current agent '{agent}' does not support prompts[/red]")
                             continue
 
                         try:
@@ -291,15 +263,11 @@ class AgentApp:
                             all_prompts = []
 
                             # Get prompts from the current agent
-                            rich_print(
-                                f"\n[bold]Fetching prompts for agent [cyan]{agent}[/cyan]...[/bold]"
-                            )
+                            rich_print(f"\n[bold]Fetching prompts for agent [cyan]{agent}[/cyan]...[/bold]")
                             prompt_servers = await current_proxy._agent.list_prompts()
 
                             if not prompt_servers:
-                                rich_print(
-                                    "[yellow]No prompts available for this agent[/yellow]"
-                                )
+                                rich_print("[yellow]No prompts available for this agent[/yellow]")
                                 continue
 
                             # Process retrieved prompts
@@ -319,9 +287,7 @@ class AgentApp:
                                 for prompt in prompts:
                                     # Basic prompt information
                                     prompt_name = getattr(prompt, "name", "Unknown")
-                                    description = getattr(
-                                        prompt, "description", "No description"
-                                    )
+                                    description = getattr(prompt, "description", "No description")
 
                                     # Extract argument information
                                     arg_names = []
@@ -339,9 +305,7 @@ class AgentApp:
                                                 arg_names.append(name)
 
                                                 # Store description if available
-                                                description = getattr(
-                                                    arg, "description", None
-                                                )
+                                                description = getattr(arg, "description", None)
                                                 if description:
                                                     arg_descriptions[name] = description
 
@@ -371,32 +335,20 @@ class AgentApp:
 
                             # If no prompts were found
                             if not all_prompts:
-                                rich_print(
-                                    "[yellow]No prompts available for this agent[/yellow]"
-                                )
+                                rich_print("[yellow]No prompts available for this agent[/yellow]")
                                 continue
 
                             # Sort prompts by server then name
                             all_prompts.sort(key=lambda p: (p["server"], p["name"]))
 
                             # Check if a specific prompt was requested
-                            if (
-                                "prompt_name" in command_result
-                                and command_result["prompt_name"]
-                            ):
+                            if "prompt_name" in command_result and command_result["prompt_name"]:
                                 requested_name = command_result["prompt_name"]
                                 # Find the prompt in our list (either by name or namespaced name)
-                                matching_prompts = [
-                                    p
-                                    for p in all_prompts
-                                    if p["name"] == requested_name
-                                    or p["namespaced_name"] == requested_name
-                                ]
+                                matching_prompts = [p for p in all_prompts if p["name"] == requested_name or p["namespaced_name"] == requested_name]
 
                                 if not matching_prompts:
-                                    rich_print(
-                                        f"[red]Prompt '{requested_name}' not found[/red]"
-                                    )
+                                    rich_print(f"[red]Prompt '{requested_name}' not found[/red]")
                                     rich_print("[yellow]Available prompts:[/yellow]")
                                     for p in all_prompts:
                                         rich_print(f"  {p['namespaced_name']}")
@@ -407,22 +359,16 @@ class AgentApp:
                                     selected_prompt = matching_prompts[0]
                                 else:
                                     # If multiple matches, show them and ask user to be more specific
-                                    rich_print(
-                                        f"[yellow]Multiple prompts match '{requested_name}':[/yellow]"
-                                    )
+                                    rich_print(f"[yellow]Multiple prompts match '{requested_name}':[/yellow]")
                                     for i, p in enumerate(matching_prompts):
-                                        rich_print(
-                                            f"  {i + 1}. {p['namespaced_name']} - {p['description']}"
-                                        )
+                                        rich_print(f"  {i + 1}. {p['namespaced_name']} - {p['description']}")
 
                                     # Ask user to select one
                                     from mcp_agent.core.enhanced_prompt import (
                                         get_selection_input,
                                     )
 
-                                    selection = await get_selection_input(
-                                        "Enter prompt number to select: ", default="1"
-                                    )
+                                    selection = await get_selection_input("Enter prompt number to select: ", default="1")
 
                                     try:
                                         idx = int(selection) - 1
@@ -432,9 +378,7 @@ class AgentApp:
                                             rich_print("[red]Invalid selection[/red]")
                                             continue
                                     except ValueError:
-                                        rich_print(
-                                            "[red]Invalid input, please enter a number[/red]"
-                                        )
+                                        rich_print("[red]Invalid input, please enter a number[/red]")
                                         continue
                             else:
                                 # Display prompt selection UI
@@ -455,9 +399,7 @@ class AgentApp:
                                     if required_args and optional_args:
                                         args_display = f"[bold]{len(required_args)}[/bold]+{len(optional_args)}"
                                     elif required_args:
-                                        args_display = (
-                                            f"[bold]{len(required_args)}[/bold]"
-                                        )
+                                        args_display = f"[bold]{len(required_args)}[/bold]"
                                     elif optional_args:
                                         args_display = f"{len(optional_args)} opt"
                                     else:
@@ -472,9 +414,7 @@ class AgentApp:
                                     )
 
                                 console.print(table)
-                                prompt_names = [
-                                    str(i + 1) for i in range(len(all_prompts))
-                                ]
+                                prompt_names = [str(i + 1) for i in range(len(all_prompts))]
 
                                 # Ask user to select a prompt
                                 from mcp_agent.core.enhanced_prompt import (
@@ -489,9 +429,7 @@ class AgentApp:
 
                                 # Make cancellation easier
                                 if not selection or selection.strip() == "":
-                                    rich_print(
-                                        "[yellow]Prompt selection cancelled[/yellow]"
-                                    )
+                                    rich_print("[yellow]Prompt selection cancelled[/yellow]")
                                     continue
 
                                 try:
@@ -502,17 +440,13 @@ class AgentApp:
                                         rich_print("[red]Invalid selection[/red]")
                                         continue
                                 except ValueError:
-                                    rich_print(
-                                        "[red]Invalid input, please enter a number[/red]"
-                                    )
+                                    rich_print("[red]Invalid input, please enter a number[/red]")
                                     continue
 
                             # Get our prompt arguments
                             required_args = selected_prompt["required_args"]
                             optional_args = selected_prompt["optional_args"]
-                            arg_descriptions = selected_prompt.get(
-                                "arg_descriptions", {}
-                            )
+                            arg_descriptions = selected_prompt.get("arg_descriptions", {})
 
                             # Always initialize arg_values
                             arg_values = {}
@@ -525,13 +459,9 @@ class AgentApp:
                                         f"\n[bold]Prompt [cyan]{selected_prompt['name']}[/cyan] requires {len(required_args)} arguments and has {len(optional_args)} optional arguments:[/bold]"
                                     )
                                 elif required_args:
-                                    rich_print(
-                                        f"\n[bold]Prompt [cyan]{selected_prompt['name']}[/cyan] requires {len(required_args)} arguments:[/bold]"
-                                    )
+                                    rich_print(f"\n[bold]Prompt [cyan]{selected_prompt['name']}[/cyan] requires {len(required_args)} arguments:[/bold]")
                                 elif optional_args:
-                                    rich_print(
-                                        f"\n[bold]Prompt [cyan]{selected_prompt['name']}[/cyan] has {len(optional_args)} optional arguments:[/bold]"
-                                    )
+                                    rich_print(f"\n[bold]Prompt [cyan]{selected_prompt['name']}[/cyan] has {len(optional_args)} optional arguments:[/bold]")
 
                                 # Collect required arguments
                                 for arg_name in required_args:
@@ -573,21 +503,15 @@ class AgentApp:
                                             arg_values[arg_name] = arg_value
 
                             # Apply the prompt with or without arguments
-                            rich_print(
-                                f"\n[bold]Applying prompt [cyan]{selected_prompt['namespaced_name']}[/cyan]...[/bold]"
-                            )
+                            rich_print(f"\n[bold]Applying prompt [cyan]{selected_prompt['namespaced_name']}[/cyan]...[/bold]")
 
                             # Call apply_prompt on the agent - always pass arg_values (empty dict if no args)
-                            await current_proxy._agent.apply_prompt(
-                                selected_prompt["namespaced_name"], arg_values
-                            )
+                            await current_proxy._agent.apply_prompt(selected_prompt["namespaced_name"], arg_values)
 
                         except Exception as e:
                             import traceback
 
-                            rich_print(
-                                f"[red]Error selecting or applying prompt: {e}[/red]"
-                            )
+                            rich_print(f"[red]Error selecting or applying prompt: {e}[/red]")
                             rich_print(f"[dim]{traceback.format_exc()}[/dim]")
                         continue
 

mcp_agent/core/agent_utils.py

@@ -2,7 +2,7 @@
 Utility functions for agent operations.
 """
 
-from typing import List, TYPE_CHECKING
+from typing import TYPE_CHECKING, List
 
 from mcp_agent.event_progress import ProgressAction
 
@@ -28,7 +28,7 @@ def unwrap_proxy(proxy: BaseAgentProxy) -> AgentOrWorkflow:
     Returns:
         The underlying Agent or workflow instance
     """
-    from mcp_agent.core.proxies import LLMAgentProxy, ChainProxy
+    from mcp_agent.core.proxies import ChainProxy, LLMAgentProxy
 
     if isinstance(proxy, LLMAgentProxy):
         return proxy._agent
@@ -38,9 +38,7 @@ def unwrap_proxy(proxy: BaseAgentProxy) -> AgentOrWorkflow:
     return proxy._workflow
 
 
-def get_agent_instances(
-    agent_names: List[str], active_agents: ProxyDict
-) -> List[AgentOrWorkflow]:
+def get_agent_instances(agent_names: List[str], active_agents: ProxyDict) -> List[AgentOrWorkflow]:
     """
     Get list of actual agent/workflow instances from a list of names.
 

mcp_agent/core/decorators.py

@@ -3,7 +3,8 @@ Decorators for FastAgent applications.
 Contains decorator definitions extracted from fastagent.py.
 """
 
-from typing import Callable, Dict, List, Optional, TypeVar, Literal
+from typing import Callable, Dict, List, Literal, Optional, TypeVar
+
 from mcp_agent.core.agent_types import AgentConfig, AgentType
 from mcp_agent.workflows.llm.augmented_llm import RequestParams
 
@@ -46,11 +47,7 @@ def _create_decorator(
         # Create base request params
         def decorator(func: Callable) -> Callable:
             # Create base request params
-            if (
-                request_params is not None
-                or model is not None
-                or use_history != default_use_history
-            ):
+            if request_params is not None or model is not None or use_history != default_use_history:
                 max_tokens = 4096 if agent_type == AgentType.BASIC else None
                 params_dict = {"use_history": use_history, "model": model}
                 if max_tokens:
@@ -142,9 +139,7 @@ def agent(
         @fast.agent("agent_name", instruction="Your instruction here")  # Using keyword arg
     """
     # Use positional argument if provided, otherwise use keyword argument
-    final_instruction = (
-        instruction_or_kwarg if instruction_or_kwarg is not None else instruction
-    )
+    final_instruction = instruction_or_kwarg if instruction_or_kwarg is not None else instruction
 
     decorator = self._create_decorator(
         AgentType.BASIC,
@@ -427,9 +422,7 @@ def chain(
     return decorator
 
 
-def passthrough(
-    self, name: str = "Passthrough", use_history: bool = True, **kwargs
-) -> Callable:
+def passthrough(self, name: str = "Passthrough", use_history: bool = True, **kwargs) -> Callable:
     """
     Decorator to create and register a passthrough agent.
     A passthrough agent simply returns any input message without modification.

mcp_agent/core/enhanced_prompt.py

@@ -2,15 +2,16 @@
 Enhanced prompt functionality with advanced prompt_toolkit features.
 """
 
-from typing import List, Optional
 from importlib.metadata import version
+from typing import List, Optional
+
 from prompt_toolkit import PromptSession
+from prompt_toolkit.completion import Completer, Completion, WordCompleter
+from prompt_toolkit.filters import Condition
 from prompt_toolkit.formatted_text import HTML
 from prompt_toolkit.history import InMemoryHistory
 from prompt_toolkit.key_binding import KeyBindings
-from prompt_toolkit.completion import Completer, Completion, WordCompleter
 from prompt_toolkit.lexers import PygmentsLexer
-from prompt_toolkit.filters import Condition
 from prompt_toolkit.styles import Style
 from pygments.lexers.python import PythonLexer
 from rich import print as rich_print
@@ -45,7 +46,7 @@ class AgentCompleter(Completer):
         commands: List[str] = None,
         agent_types: dict = None,
         is_human_input: bool = False,
-    ):
+    ) -> None:
         self.agents = agents
         # Map commands to their descriptions for better completion hints
         self.commands = {
@@ -61,9 +62,7 @@ class AgentCompleter(Completer):
         if is_human_input:
             self.commands.pop("agents")
             self.commands.pop("prompts")  # Remove prompts command in human input mode
-            self.commands.pop(
-                "prompt", None
-            )  # Remove prompt command in human input mode
+            self.commands.pop("prompt", None)  # Remove prompt command in human input mode
         self.agent_types = agent_types or {}
 
     def get_completions(self, document, complete_event):
@@ -103,23 +102,23 @@ def create_keybindings(on_toggle_multiline=None, app=None):
     kb = KeyBindings()
 
     @kb.add("c-m", filter=Condition(lambda: not in_multiline_mode))
-    def _(event):
+    def _(event) -> None:
         """Enter: accept input when not in multiline mode."""
         event.current_buffer.validate_and_handle()
 
     @kb.add("c-m", filter=Condition(lambda: in_multiline_mode))
-    def _(event):
+    def _(event) -> None:
         """Enter: insert newline when in multiline mode."""
         event.current_buffer.insert_text("\n")
 
     # Use c-j (Ctrl+J) as an alternative to represent Ctrl+Enter in multiline mode
     @kb.add("c-j", filter=Condition(lambda: in_multiline_mode))
-    def _(event):
+    def _(event) -> None:
         """Ctrl+J (equivalent to Ctrl+Enter): Submit in multiline mode."""
         event.current_buffer.validate_and_handle()
 
     @kb.add("c-t")
-    def _(event):
+    def _(event) -> None:
         """Ctrl+T: Toggle multiline mode."""
         global in_multiline_mode
         in_multiline_mode = not in_multiline_mode
@@ -138,7 +137,7 @@ def create_keybindings(on_toggle_multiline=None, app=None):
         # The toolbar will show the current mode
 
     @kb.add("c-l")
-    def _(event):
+    def _(event) -> None:
         """Ctrl+L: Clear the input buffer."""
         event.current_buffer.text = ""
 
@@ -187,7 +186,7 @@ async def get_enhanced_input(
         agent_histories[agent_name] = InMemoryHistory()
 
     # Define callback for multiline toggle
-    def on_multiline_toggle(enabled):
+    def on_multiline_toggle(enabled) -> None:
         nonlocal session
         if hasattr(session, "app") and session.app:
             session.app.invalidate()
@@ -209,9 +208,7 @@ async def get_enhanced_input(
             ("↑/↓", "History"),
         ]
 
-        newline = (
-            "Ctrl+<Enter>:Submit" if in_multiline_mode else "<Enter>:Submit"
-        )
+        newline = "Ctrl+<Enter>:Submit" if in_multiline_mode else "<Enter>:Submit"
 
         # Only show relevant shortcuts based on mode
         shortcuts = [(k, v) for k, v in shortcuts if v]
@@ -250,9 +247,7 @@ async def get_enhanced_input(
     )
 
     # Create key bindings with a reference to the app
-    bindings = create_keybindings(
-        on_toggle_multiline=on_multiline_toggle, app=session.app
-    )
+    bindings = create_keybindings(on_toggle_multiline=on_multiline_toggle, app=session.app)
     session.app.key_bindings = bindings
 
     # Create formatted prompt text
@@ -267,20 +262,14 @@ async def get_enhanced_input(
         if default == "STOP":
             rich_print("Enter a prompt, [red]STOP[/red] to finish")
             if default:
-                rich_print(
-                    f"Press <ENTER> to use the default prompt:\n[cyan]{default}[/cyan]"
-                )
+                rich_print(f"Press <ENTER> to use the default prompt:\n[cyan]{default}[/cyan]")
 
     # Mention available features but only on first usage globally
     if not help_message_shown:
         if is_human_input:
-            rich_print(
-                "[dim]Type /help for commands. Ctrl+T toggles multiline mode.[/dim]"
-            )
+            rich_print("[dim]Type /help for commands. Ctrl+T toggles multiline mode.[/dim]")
         else:
-            rich_print(
-                "[dim]Type /help for commands, @agent to switch agent. Ctrl+T toggles multiline mode.[/dim]"
-            )
+            rich_print("[dim]Type /help for commands, @agent to switch agent. Ctrl+T toggles multiline mode.[/dim]")
         rich_print()
         help_message_shown = True
 
@@ -366,9 +355,7 @@ async def get_selection_input(
 
         try:
             # Get user input
-            selection = await prompt_session.prompt_async(
-                prompt_text, default=default or ""
-            )
+            selection = await prompt_session.prompt_async(prompt_text, default=default or "")
 
             # Handle cancellation
             if allow_cancel and not selection.strip():
@@ -409,9 +396,7 @@ async def get_argument_input(
     if description:
         rich_print(f"  [dim]{arg_name}: {description}[/dim]")
 
-    prompt_text = HTML(
-        f"Enter value for <ansibrightcyan>{arg_name}</ansibrightcyan> {required_text}: "
-    )
+    prompt_text = HTML(f"Enter value for <ansibrightcyan>{arg_name}</ansibrightcyan> {required_text}: ")
 
     # Create prompt session
     prompt_session = PromptSession()
@@ -452,13 +437,9 @@ async def handle_special_commands(command, agent_app=None):
         rich_print("  /prompt <name> - Apply a specific prompt by name")
         rich_print("  @agent_name    - Switch to agent")
         rich_print("  STOP           - Return control back to the workflow")
-        rich_print(
-            "  EXIT           - Exit fast-agent, terminating any running workflows"
-        )
+        rich_print("  EXIT           - Exit fast-agent, terminating any running workflows")
         rich_print("\n[bold]Keyboard Shortcuts:[/bold]")
-        rich_print(
-            "  Enter          - Submit (normal mode) / New line (multiline mode)"
-        )
+        rich_print("  Enter          - Submit (normal mode) / New line (multiline mode)")
         rich_print("  Ctrl+Enter      - Always submit (in any mode)")
         rich_print("  Ctrl+T         - Toggle multiline mode")
         rich_print("  Ctrl+L         - Clear input")
@@ -489,14 +470,10 @@ async def handle_special_commands(command, agent_app=None):
             rich_print("\n[bold]Fetching available MCP prompts...[/bold]")
             return {"list_prompts": True}
         else:
-            rich_print(
-                "[yellow]Prompt listing is not available outside of an agent context[/yellow]"
-            )
+            rich_print("[yellow]Prompt listing is not available outside of an agent context[/yellow]")
             return True
 
-    elif command == "SELECT_PROMPT" or (
-        isinstance(command, str) and command.startswith("SELECT_PROMPT:")
-    ):
+    elif command == "SELECT_PROMPT" or (isinstance(command, str) and command.startswith("SELECT_PROMPT:")):
         # Handle prompt selection UI
         if agent_app:
             # If it's a specific prompt, extract the name
@@ -507,9 +484,7 @@ async def handle_special_commands(command, agent_app=None):
             # Return a dictionary with a select_prompt action to be handled by the caller
             return {"select_prompt": True, "prompt_name": prompt_name}
         else:
-            rich_print(
-                "[yellow]Prompt selection is not available outside of an agent context[/yellow]"
-            )
+            rich_print("[yellow]Prompt selection is not available outside of an agent context[/yellow]")
             return True
 
     elif isinstance(command, str) and command.startswith("SWITCH:"):
@@ -519,9 +494,7 @@ async def handle_special_commands(command, agent_app=None):
                 #                rich_print(f"[green]Switching to agent: {agent_name}[/green]")
                 return {"switch_agent": agent_name}
             else:
-                rich_print(
-                    "[yellow]Agent switching not available in this context[/yellow]"
-                )
+                rich_print("[yellow]Agent switching not available in this context[/yellow]")
         else:
             rich_print(f"[red]Unknown agent: {agent_name}[/red]")
         return True