fast-agent-mcp 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

fast_agent/mcp/prompts/prompt_load.py

@@ -12,8 +12,6 @@ from fast_agent.core.logging.logger import get_logger
 from fast_agent.mcp import mime_utils, resource_utils
 from fast_agent.mcp.prompts.prompt_template import (
     PromptContent,
-    PromptTemplate,
-    PromptTemplateLoader,
 )
 from fast_agent.types import PromptMessageExtended
 
@@ -100,58 +98,62 @@ def create_resource_message(
         return message_class(content=embedded_resource)
 
 
-def load_prompt(file: Path) -> List[PromptMessage]:
+def load_prompt(file: Path) -> List[PromptMessageExtended]:
     """
-    Load a prompt from a file and return as PromptMessage objects.
+    Load a prompt from a file and return as PromptMessageExtended objects.
 
     The loader uses file extension to determine the format:
-    - .json files are loaded as MCP SDK compatible GetPromptResult JSON format
-    - All other files are loaded using the template-based delimited format
+    - .json files are loaded using enhanced format that preserves tool_calls, channels, etc.
+    - All other files are loaded using the template-based delimited format with resource loading
 
     Args:
         file: Path to the prompt file
 
     Returns:
-        List of PromptMessage objects
+        List of PromptMessageExtended objects with full conversation state
     """
-    file_str = str(file).lower()
+    path_str = str(file).lower()
 
-    if file_str.endswith(".json"):
-        # Handle JSON format as GetPromptResult
-        import json
+    if path_str.endswith(".json"):
+        # JSON files use the serialization module directly
+        from fast_agent.mcp.prompt_serialization import load_messages
+        return load_messages(str(file))
+    else:
+        # Non-JSON files need template processing for resource loading
+        from fast_agent.mcp.prompts.prompt_template import PromptTemplateLoader
 
-        from mcp.types import GetPromptResult
+        loader = PromptTemplateLoader()
+        template = loader.load_from_file(file)
 
-        # Load JSON directly into GetPromptResult
-        with open(file, "r", encoding="utf-8") as f:
-            json_data = json.load(f)
+        # Render the template without arguments to get the messages
+        messages = create_messages_with_resources(
+            template.content_sections,
+            [file]  # Pass the file path for resource resolution
+        )
+
+        # Convert to PromptMessageExtended
+        return PromptMessageExtended.to_extended(messages)
 
-        # Parse as GetPromptResult object
-        result = GetPromptResult.model_validate(json_data)
 
-        # Return the messages directly
-        return result.messages
-    else:
-        # Template-based format (delimited text)
-        template: PromptTemplate = PromptTemplateLoader().load_from_file(file)
-        return create_messages_with_resources(template.content_sections, [file])
 
 
-def load_prompt_multipart(file: Path) -> List[PromptMessageExtended]:
+def load_prompt_as_get_prompt_result(file: Path):
     """
-    Load a prompt from a file and return as PromptMessageExtended objects.
+    Load a prompt from a file and convert to GetPromptResult format for MCP compatibility.
 
-    The loader uses file extension to determine the format:
-    - .json files are loaded as MCP SDK compatible GetPromptResult JSON format
-    - All other files are loaded using the template-based delimited format
+    This loses extended fields (tool_calls, channels, etc.) but provides
+    compatibility with MCP prompt servers.
 
     Args:
         file: Path to the prompt file
 
     Returns:
-        List of PromptMessageExtended objects
+        GetPromptResult object for MCP compatibility
     """
-    # First load as regular PromptMessage objects
+    from fast_agent.mcp.prompt_serialization import to_get_prompt_result
+
+    # Load with full data
     messages = load_prompt(file)
-    # Then convert to multipart messages
-    return PromptMessageExtended.to_extended(messages)
+
+    # Convert to GetPromptResult (loses extended fields)
+    return to_get_prompt_result(messages)

fast_agent/mcp/prompts/prompt_server.py

@@ -11,7 +11,7 @@ import base64
 import logging
 import sys
 from pathlib import Path
-from typing import Any, Awaitable, Callable, Dict, List, Optional
+from typing import Any, Awaitable, Callable, Dict, List, Optional, Union
 
 from mcp.server.fastmcp import FastMCP
 from mcp.server.fastmcp.prompts.base import (
@@ -38,6 +38,7 @@ from fast_agent.mcp.prompts.prompt_template import (
     PromptMetadata,
     PromptTemplateLoader,
 )
+from fast_agent.types import PromptMessageExtended
 
 # Configure logging
 logging.basicConfig(level=logging.ERROR)
@@ -47,13 +48,13 @@ logger = logging.getLogger("prompt_server")
 mcp = FastMCP("Prompt Server")
 
 
-def convert_to_fastmcp_messages(prompt_messages: List[PromptMessage]) -> List[Message]:
+def convert_to_fastmcp_messages(prompt_messages: List[Union[PromptMessage, PromptMessageExtended]]) -> List[Message]:
     """
-    Convert PromptMessage objects from prompt_load to FastMCP Message objects.
-    This adapter prevents double-wrapping of messages.
+    Convert PromptMessage or PromptMessageExtended objects to FastMCP Message objects.
+    This adapter prevents double-wrapping of messages and handles both types.
 
     Args:
-        prompt_messages: List of PromptMessage objects from prompt_load
+        prompt_messages: List of PromptMessage or PromptMessageExtended objects
 
     Returns:
         List of FastMCP Message objects
@@ -61,13 +62,27 @@ def convert_to_fastmcp_messages(prompt_messages: List[PromptMessage]) -> List[Me
     result = []
 
     for msg in prompt_messages:
-        if msg.role == "user":
-            result.append(UserMessage(content=msg.content))
-        elif msg.role == "assistant":
-            result.append(AssistantMessage(content=msg.content))
+        # Handle both PromptMessage and PromptMessageExtended
+        if hasattr(msg, 'from_multipart'):
+            # PromptMessageExtended - convert to regular PromptMessage format
+            flat_messages = msg.from_multipart()
+            for flat_msg in flat_messages:
+                if flat_msg.role == "user":
+                    result.append(UserMessage(content=flat_msg.content))
+                elif flat_msg.role == "assistant":
+                    result.append(AssistantMessage(content=flat_msg.content))
+                else:
+                    logger.warning(f"Unknown message role: {flat_msg.role}, defaulting to user")
+                    result.append(UserMessage(content=flat_msg.content))
         else:
-            logger.warning(f"Unknown message role: {msg.role}, defaulting to user")
-            result.append(UserMessage(content=msg.content))
+            # Regular PromptMessage - use directly
+            if msg.role == "user":
+                result.append(UserMessage(content=msg.content))
+            elif msg.role == "assistant":
+                result.append(AssistantMessage(content=msg.content))
+            else:
+                logger.warning(f"Unknown message role: {msg.role}, defaulting to user")
+                result.append(UserMessage(content=msg.content))
 
     return result
 

fast_agent/resources/setup/fastagent.config.yaml

@@ -7,10 +7,10 @@
 # Accepts aliases for Anthropic Models: haiku, haiku3, sonnet, sonnet35, opus, opus3
 # and OpenAI Models: gpt-4.1, gpt-4.1-mini, o1, o1-mini, o3-mini
 #
-# If not specified, defaults to "haiku".
+# If not specified, defaults to "gpt-5-mini.low".
 # Can be overriden with a command line switch --model=<model>, or within the Agent constructor.
 
-default_model: haiku
+default_model: gpt-5-mini.low
 # mcp-ui support: disabled, enabled or auto. "auto" opens the web browser on the asset automatically
 # mcp_ui_output_dir: ".fast-agent/ui"  # Where to write MCP-UI HTML files (relative to CWD if not absolute)
 # mcp_ui_mode: enabled

fast_agent/types/__init__.py

@@ -18,7 +18,9 @@ from fast_agent.mcp.helpers.content_helpers import (
 
 # Public message model used across providers and MCP integration
 from fast_agent.mcp.prompt_message_extended import PromptMessageExtended
-from fast_agent.types.llm_stop_reason import LlmStopReason
+
+# Stop reason enum - imported directly to avoid circular dependency
+from .llm_stop_reason import LlmStopReason
 
 __all__ = [
     # Enums / types

fast_agent/ui/enhanced_prompt.py

@@ -3,12 +3,13 @@ Enhanced prompt functionality with advanced prompt_toolkit features.
 """
 
 import asyncio
+import json
 import os
 import shlex
 import subprocess
 import tempfile
 from importlib.metadata import version
-from typing import List, Optional
+from typing import TYPE_CHECKING, List, Optional
 
 from prompt_toolkit import PromptSession
 from prompt_toolkit.completion import Completer, Completion, WordCompleter
@@ -20,9 +21,13 @@ from prompt_toolkit.styles import Style
 from rich import print as rich_print
 
 from fast_agent.agents.agent_types import AgentType
+from fast_agent.constants import FAST_AGENT_ERROR_CHANNEL, FAST_AGENT_REMOVED_METADATA_CHANNEL
 from fast_agent.core.exceptions import PromptExitError
 from fast_agent.llm.model_info import get_model_info
 
+if TYPE_CHECKING:
+    from fast_agent.core.agent_app import AgentApp
+
 # Get the application version
 try:
     app_version = version("fast-agent-mcp")
@@ -38,6 +43,30 @@ available_agents = set()
 # Keep track of multi-line mode state
 in_multiline_mode = False
 
+
+def _extract_alert_flags_from_meta(blocks) -> set[str]:
+    flags: set[str] = set()
+    for block in blocks or []:
+        text = getattr(block, "text", None)
+        if not text:
+            continue
+        try:
+            payload = json.loads(text)
+        except (TypeError, ValueError):
+            continue
+        if payload.get("type") != "fast-agent-removed":
+            continue
+        category = payload.get("category")
+        match category:
+            case "text":
+                flags.add("T")
+            case "document":
+                flags.add("D")
+            case "vision":
+                flags.add("V")
+    return flags
+
+
 # Track whether help text has been shown globally
 help_message_shown = False
 
@@ -45,20 +74,17 @@ help_message_shown = False
 _agent_info_shown = set()
 
 
-async def _display_agent_info_helper(agent_name: str, agent_provider: object) -> None:
+async def _display_agent_info_helper(agent_name: str, agent_provider: "AgentApp | None") -> None:
     """Helper function to display agent information."""
     # Only show once per agent
     if agent_name in _agent_info_shown:
         return
 
     try:
-        # Get agent info
-        if hasattr(agent_provider, "_agent"):
-            # This is an AgentApp - get the specific agent
-            agent = agent_provider._agent(agent_name)
-        else:
-            # This is a single agent
-            agent = agent_provider
+        # Get agent info from AgentApp
+        if agent_provider is None:
+            return
+        agent = agent_provider._agent(agent_name)
 
         # Get counts TODO -- add this to the type library or adjust the way aggregator/reporting works
         server_count = 0
@@ -148,7 +174,9 @@ async def _display_agent_info_helper(agent_name: str, agent_provider: object) ->
         pass
 
 
-async def _display_all_agents_with_hierarchy(available_agents: List[str], agent_provider) -> None:
+async def _display_all_agents_with_hierarchy(
+    available_agents: List[str], agent_provider: "AgentApp | None"
+) -> None:
     """Display all agents with tree structure for workflow agents."""
     # Track which agents are children to avoid displaying them twice
     child_agents = set()
@@ -156,10 +184,9 @@ async def _display_all_agents_with_hierarchy(available_agents: List[str], agent_
     # First pass: identify all child agents
     for agent_name in available_agents:
         try:
-            if hasattr(agent_provider, "_agent"):
-                agent = agent_provider._agent(agent_name)
-            else:
-                agent = agent_provider
+            if agent_provider is None:
+                continue
+            agent = agent_provider._agent(agent_name)
 
             if agent.agent_type == AgentType.PARALLEL:
                 if hasattr(agent, "fan_out_agents") and agent.fan_out_agents:
@@ -184,10 +211,9 @@ async def _display_all_agents_with_hierarchy(available_agents: List[str], agent_
             continue
 
         try:
-            if hasattr(agent_provider, "_agent"):
-                agent = agent_provider._agent(agent_name)
-            else:
-                agent = agent_provider
+            if agent_provider is None:
+                continue
+            agent = agent_provider._agent(agent_name)
 
             # Display parent agent
             await _display_agent_info_helper(agent_name, agent_provider)
@@ -202,7 +228,7 @@ async def _display_all_agents_with_hierarchy(available_agents: List[str], agent_
             continue
 
 
-async def _display_parallel_children(parallel_agent, agent_provider) -> None:
+async def _display_parallel_children(parallel_agent, agent_provider: "AgentApp | None") -> None:
     """Display child agents of a parallel agent in tree format."""
     children = []
 
@@ -222,7 +248,7 @@ async def _display_parallel_children(parallel_agent, agent_provider) -> None:
         await _display_child_agent_info(child_agent, prefix, agent_provider)
 
 
-async def _display_router_children(router_agent, agent_provider) -> None:
+async def _display_router_children(router_agent, agent_provider: "AgentApp | None") -> None:
     """Display child agents of a router agent in tree format."""
     children = []
 
@@ -239,7 +265,9 @@ async def _display_router_children(router_agent, agent_provider) -> None:
         await _display_child_agent_info(child_agent, prefix, agent_provider)
 
 
-async def _display_child_agent_info(child_agent, prefix: str, agent_provider) -> None:
+async def _display_child_agent_info(
+    child_agent, prefix: str, agent_provider: "AgentApp | None"
+) -> None:
     """Display info for a child agent with tree prefix."""
     try:
         # Get counts for child agent
@@ -425,7 +453,9 @@ def get_text_from_editor(initial_text: str = "") -> str:
     return edited_text.strip()  # Added strip() to remove trailing newlines often added by editors
 
 
-def create_keybindings(on_toggle_multiline=None, app=None, agent_provider=None, agent_name=None):
+def create_keybindings(
+    on_toggle_multiline=None, app=None, agent_provider: "AgentApp | None" = None, agent_name=None
+):
     """Create custom key bindings."""
     kb = KeyBindings()
 
@@ -504,30 +534,20 @@ def create_keybindings(on_toggle_multiline=None, app=None, agent_provider=None,
         """Ctrl+Y: Copy last assistant response to clipboard."""
         if kb.agent_provider and kb.current_agent_name:
             try:
-                # Get the agent
-                if hasattr(kb.agent_provider, "_agent"):
-                    agent = kb.agent_provider._agent(kb.current_agent_name)
-                else:
-                    agent = kb.agent_provider
-
-                # Get message history
-                if hasattr(agent, "_llm") and agent._llm and agent._llm.message_history:
-                    # Find last assistant message
-                    for msg in reversed(agent._llm.message_history):
-                        if msg.role == "assistant":
-                            content = msg.last_text()
-                            import pyperclip
-
-                            pyperclip.copy(content)
-                            rich_print("\n[green]✓ Copied to clipboard[/green]")
-                            return
-
-                else:
-                    pass
+                # Get the agent from AgentApp
+                agent = kb.agent_provider._agent(kb.current_agent_name)
+
+                # Find last assistant message
+                for msg in reversed(agent.message_history):
+                    if msg.role == "assistant":
+                        content = msg.last_text()
+                        import pyperclip
+
+                        pyperclip.copy(content)
+                        rich_print("\n[green]✓ Copied to clipboard[/green]")
+                        return
             except Exception:
                 pass
-        else:
-            pass
 
     return kb
 
@@ -542,7 +562,7 @@ async def get_enhanced_input(
     agent_types: dict[str, AgentType] = None,
     is_human_input: bool = False,
     toolbar_color: str = "ansiblue",
-    agent_provider: object = None,
+    agent_provider: "AgentApp | None" = None,
 ) -> str:
     """
     Enhanced input with advanced prompt_toolkit features.
@@ -557,7 +577,7 @@ async def get_enhanced_input(
         agent_types: Dictionary mapping agent names to their types for display
         is_human_input: Whether this is a human input request (disables agent selection features)
         toolbar_color: Color to use for the agent name in the toolbar (default: "ansiblue")
-        agent_provider: Optional agent provider for displaying agent info
+        agent_provider: Optional AgentApp for displaying agent info
 
     Returns:
         User input string
@@ -598,18 +618,22 @@ async def get_enhanced_input(
 
         shortcut_text = " | ".join(f"{key}:{action}" for key, action in shortcuts)
 
-        # Resolve model name and TDV from the current agent if available
+        # Resolve model name, turn counter, and TDV from the current agent if available
         model_display = None
         tdv_segment = None
+        turn_count = 0
         try:
-            agent_obj = (
-                agent_provider._agent(agent_name)
-                if agent_provider and hasattr(agent_provider, "_agent")
-                else agent_provider
-            )
-            if agent_obj and hasattr(agent_obj, "llm") and agent_obj.llm:
-                model_name = getattr(agent_obj.llm, "model_name", None)
-                if model_name:
+            if agent_provider:
+                agent = agent_provider._agent(agent_name)
+
+                # Get turn count from message history
+                for message in agent.message_history:
+                    if message.role == "user":
+                        turn_count += 1
+
+                # Get model name from LLM
+                if agent.llm and agent.llm.model_name:
+                    model_name = agent.llm.model_name
                     # Truncate model name to max 25 characters with ellipsis
                     max_len = 25
                     if len(model_name) > max_len:
@@ -619,20 +643,35 @@ async def get_enhanced_input(
                         model_display = model_name
 
                 # Build TDV capability segment based on model database
-                info = get_model_info(agent_obj)
+                info = get_model_info(agent)
                 # Default to text-only if info resolution fails for any reason
                 t, d, v = (True, False, False)
                 if info:
                     t, d, v = info.tdv_flags
 
+                # Check for alert flags in user messages
+                alert_flags: set[str] = set()
+                error_seen = False
+                for message in agent.message_history:
+                    if message.channels:
+                        if message.channels.get(FAST_AGENT_ERROR_CHANNEL):
+                            error_seen = True
+                    if message.role == "user" and message.channels:
+                        meta_blocks = message.channels.get(FAST_AGENT_REMOVED_METADATA_CHANNEL, [])
+                        alert_flags.update(_extract_alert_flags_from_meta(meta_blocks))
+
+                if error_seen and not alert_flags:
+                    alert_flags.add("T")
+
                 def _style_flag(letter: str, supported: bool) -> str:
                     # Enabled uses the same color as NORMAL mode (ansigreen), disabled is dim
+                    if letter in alert_flags:
+                        return f"<style fg='ansired' bg='ansiblack'>{letter}</style>"
+
                     enabled_color = "ansigreen"
-                    return (
-                        f"<style fg='{enabled_color}' bg='ansiblack'>{letter}</style>"
-                        if supported
-                        else f"<style fg='ansiblack' bg='ansiwhite'>{letter}</style>"
-                    )
+                    if supported:
+                        return f"<style fg='{enabled_color}' bg='ansiblack'>{letter}</style>"
+                    return f"<style fg='ansiblack' bg='ansiwhite'>{letter}</style>"
 
                 tdv_segment = f"{_style_flag('T', t)}{_style_flag('D', d)}{_style_flag('V', v)}"
         except Exception:
@@ -640,7 +679,7 @@ async def get_enhanced_input(
             model_display = None
             tdv_segment = None
 
-        # Build dynamic middle segments: model (in green) and optional shortcuts
+        # Build dynamic middle segments: model (in green), turn counter, and optional shortcuts
         middle_segments = []
         if model_display:
             # Model chip + inline TDV flags
@@ -650,6 +689,10 @@ async def get_enhanced_input(
                 )
             else:
                 middle_segments.append(f"<style bg='ansigreen'>{model_display}</style>")
+
+        # Add turn counter (formatted as 3 digits)
+        middle_segments.append(f"{turn_count:03d}")
+
         if shortcut_text:
             middle_segments.append(shortcut_text)
         middle = " | ".join(middle_segments)
@@ -761,7 +804,9 @@ async def get_enhanced_input(
             elif cmd in ("save_history", "save"):
                 # Return a structured action for the interactive loop to handle
                 # Prefer programmatic saving via HistoryExporter; fall back to magic-string there if needed
-                filename = cmd_parts[1].strip() if len(cmd_parts) > 1 and cmd_parts[1].strip() else None
+                filename = (
+                    cmd_parts[1].strip() if len(cmd_parts) > 1 and cmd_parts[1].strip() else None
+                )
                 return {"save_history": True, "filename": filename}
             elif cmd == "prompt":
                 # Handle /prompt with no arguments as interactive mode
@@ -941,7 +986,9 @@ async def handle_special_commands(command, agent_app=None):
         rich_print("  /usage         - Show current usage statistics")
         rich_print("  /markdown      - Show last assistant message without markdown formatting")
         rich_print("  /save_history <filename> - Save current chat history to a file")
-        rich_print("      [dim]Tip: Use a .json extension for MCP-compatible JSON; any other extension saves Markdown.[/dim]")
+        rich_print(
+            "      [dim]Tip: Use a .json extension for MCP-compatible JSON; any other extension saves Markdown.[/dim]"
+        )
         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")

fast_agent/ui/interactive_prompt.py

@@ -14,14 +14,16 @@ Usage:
     )
 """
 
-from typing import Awaitable, Callable, Dict, List, Mapping, Optional, Protocol, Union
+from typing import TYPE_CHECKING, Awaitable, Callable, Dict, List, Optional, Union
+
+if TYPE_CHECKING:
+    from fast_agent.core.agent_app import AgentApp
 
 from mcp.types import Prompt, PromptMessage
 from rich import print as rich_print
 
 from fast_agent.agents.agent_types import AgentType
 from fast_agent.history.history_exporter import HistoryExporter
-from fast_agent.interfaces import AgentProtocol
 from fast_agent.mcp.mcp_aggregator import SEP
 from fast_agent.types import PromptMessageExtended
 from fast_agent.ui.enhanced_prompt import (
@@ -41,36 +43,6 @@ SendFunc = Callable[[Union[str, PromptMessage, PromptMessageExtended], str], Awa
 AgentGetter = Callable[[str], Optional[object]]
 
 
-class PromptProvider(Protocol):
-    """Protocol for objects that can provide prompt functionality."""
-
-    async def list_prompts(
-        self, namespace: Optional[str] = None, agent_name: Optional[str] = None
-    ) -> Mapping[str, List[Prompt]]:
-        """List available prompts."""
-        ...
-
-    async def apply_prompt(
-        self,
-        prompt_name: str,
-        prompt_title: Optional[str] = None,
-        arguments: Optional[Dict[str, str]] = None,
-        agent_name: Optional[str] = None,
-        as_template: bool = False,
-        **kwargs,
-    ) -> str:
-        """Apply a prompt."""
-        ...
-
-    def _agent(self, agent_name: str) -> AgentProtocol:
-        """Return the concrete agent by name (AgentApp provides this)."""
-        ...
-
-    def _show_turn_usage(self, agent_name: str) -> None:
-        """Display usage for a given agent after a turn."""
-        ...
-
-
 class InteractivePrompt:
     """
     Provides interactive prompt functionality that works with any agent implementation.
@@ -91,7 +63,7 @@ class InteractivePrompt:
         send_func: SendFunc,
         default_agent: str,
         available_agents: List[str],
-        prompt_provider: PromptProvider,
+        prompt_provider: "AgentApp",
         default: str = "",
     ) -> str:
         """
@@ -101,7 +73,7 @@ class InteractivePrompt:
             send_func: Function to send messages to agents
             default_agent: Name of the default agent to use
             available_agents: List of available agent names
-            prompt_provider: Optional provider that implements list_prompts and apply_prompt
+            prompt_provider: AgentApp instance for accessing agents and prompts
             default: Default message to use when user presses enter
 
         Returns:
@@ -290,7 +262,7 @@ class InteractivePrompt:
         rich_print()
 
     async def _get_all_prompts(
-        self, prompt_provider: PromptProvider, agent_name: Optional[str] = None
+        self, prompt_provider: "AgentApp", agent_name: Optional[str] = None
     ):
         """
         Get a list of all available prompts.
@@ -370,7 +342,7 @@ class InteractivePrompt:
             rich_print(f"[dim]{traceback.format_exc()}[/dim]")
             return []
 
-    async def _list_prompts(self, prompt_provider: PromptProvider, agent_name: str) -> None:
+    async def _list_prompts(self, prompt_provider: "AgentApp", agent_name: str) -> None:
         """
         List available prompts for an agent.
 
@@ -474,7 +446,7 @@ class InteractivePrompt:
 
     async def _select_prompt(
         self,
-        prompt_provider: PromptProvider,
+        prompt_provider: "AgentApp",
         agent_name: str,
         requested_name: Optional[str] = None,
         send_func: Optional[SendFunc] = None,
@@ -808,7 +780,7 @@ class InteractivePrompt:
             rich_print(f"[red]Error selecting or applying prompt: {e}[/red]")
             rich_print(f"[dim]{traceback.format_exc()}[/dim]")
 
-    async def _list_tools(self, prompt_provider: PromptProvider, agent_name: str) -> None:
+    async def _list_tools(self, prompt_provider: "AgentApp", agent_name: str) -> None:
         """
         List available tools for an agent.
 
@@ -908,7 +880,7 @@ class InteractivePrompt:
             rich_print(f"[red]Error listing tools: {e}[/red]")
             rich_print(f"[dim]{traceback.format_exc()}[/dim]")
 
-    async def _show_usage(self, prompt_provider: PromptProvider, agent_name: str) -> None:
+    async def _show_usage(self, prompt_provider: "AgentApp", agent_name: str) -> None:
         """
         Show usage statistics for the current agent(s) in a colorful table format.
 
@@ -930,7 +902,7 @@ class InteractivePrompt:
         except Exception as e:
             rich_print(f"[red]Error showing usage: {e}[/red]")
 
-    async def _show_system(self, prompt_provider: PromptProvider, agent_name: str) -> None:
+    async def _show_system(self, prompt_provider: "AgentApp", agent_name: str) -> None:
         """
         Show the current system prompt for the agent.
 
@@ -980,7 +952,7 @@ class InteractivePrompt:
             rich_print(f"[red]Error showing system prompt: {e}[/red]")
             rich_print(f"[dim]{traceback.format_exc()}[/dim]")
 
-    async def _show_markdown(self, prompt_provider: PromptProvider, agent_name: str) -> None:
+    async def _show_markdown(self, prompt_provider: "AgentApp", agent_name: str) -> None:
         """
         Show the last assistant message without markdown formatting.