fast-agent-mcp 0.2.44__py3-none-any.whl → 0.2.46__py3-none-any.whl

mcp_agent/mcp/__init__.py

@@ -0,0 +1,50 @@
+"""
+MCP (Model Context Protocol) integration components.
+"""
+
+from mcp.types import PromptMessage
+
+from .helpers import (
+    get_image_data,
+    get_resource_text,
+    get_resource_uri,
+    get_text,
+    is_image_content,
+    is_resource_content,
+    is_resource_link,
+    is_text_content,
+)
+from .interfaces import (
+    AgentProtocol,
+    AugmentedLLMProtocol,
+    MCPConnectionManagerProtocol,
+    ModelFactoryClassProtocol,
+    ModelT,
+    ServerConnection,
+    ServerRegistryProtocol,
+)
+from .prompt_message_multipart import PromptMessageMultipart
+
+__all__ = [
+    # Types from mcp.types
+    "PromptMessage",
+    # Multipart message handling
+    "PromptMessageMultipart",
+    # Protocol interfaces
+    "AugmentedLLMProtocol",
+    "AgentProtocol",
+    "MCPConnectionManagerProtocol",
+    "ServerRegistryProtocol",
+    "ServerConnection",
+    "ModelFactoryClassProtocol",
+    "ModelT",
+    # Helper functions
+    "get_text",
+    "get_image_data",
+    "get_resource_uri",
+    "is_text_content",
+    "is_image_content",
+    "is_resource_content",
+    "is_resource_link",
+    "get_resource_text",
+]

mcp_agent/mcp/helpers/__init__.py

@@ -1,3 +1,25 @@
 """
 Helper modules for working with MCP content.
-"""
+"""
+
+from .content_helpers import (
+    get_image_data,
+    get_resource_text,
+    get_resource_uri,
+    get_text,
+    is_image_content,
+    is_resource_content,
+    is_resource_link,
+    is_text_content,
+)
+
+__all__ = [
+    "get_text",
+    "get_image_data",
+    "get_resource_uri",
+    "is_text_content",
+    "is_image_content",
+    "is_resource_content",
+    "is_resource_link",
+    "get_resource_text",
+]

mcp_agent/mcp/interfaces.py

@@ -36,6 +36,17 @@ if TYPE_CHECKING:
     from mcp_agent.llm.usage_tracking import UsageAccumulator
 
 
+__all__ = [
+    "MCPConnectionManagerProtocol",
+    "ServerRegistryProtocol",
+    "ServerConnection",
+    "AugmentedLLMProtocol",
+    "AgentProtocol",
+    "ModelFactoryClassProtocol",
+    "ModelT",
+]
+
+
 @runtime_checkable
 class MCPConnectionManagerProtocol(Protocol):
     """Protocol for MCPConnectionManager functionality needed by ServerRegistry."""
@@ -101,7 +112,7 @@ class AugmentedLLMProtocol(Protocol):
 
     async def structured(
         self,
-        multipart_messages: List[PromptMessageMultipart],
+        multipart_messages: List[Union[PromptMessageMultipart, PromptMessage]],
         model: Type[ModelT],
         request_params: RequestParams | None = None,
     ) -> Tuple[ModelT | None, PromptMessageMultipart]:
@@ -110,7 +121,7 @@ class AugmentedLLMProtocol(Protocol):
 
     async def generate(
         self,
-        multipart_messages: List[PromptMessageMultipart],
+        multipart_messages: List[Union[PromptMessageMultipart, PromptMessage]],
         request_params: RequestParams | None = None,
     ) -> PromptMessageMultipart:
         """

mcp_agent/resources/examples/workflows/evaluator.py

@@ -18,7 +18,7 @@ fast = FastAgent("Evaluator-Optimizer")
     candidate details, and company information. Tailor the response to the company and job requirements.
     """,
     servers=["fetch"],
-    model="haiku3",
+    model="gpt-4.1-nano",
     use_history=True,
 )
 # Define evaluator agent
@@ -40,7 +40,7 @@ fast = FastAgent("Evaluator-Optimizer")
     Summarize your evaluation as a structured response with:
     - Overall quality rating.
     - Specific feedback and areas for improvement.""",
-    model="gpt-4.1",
+    model="sonnet",
 )
 # Define the evaluator-optimizer workflow
 @fast.evaluator_optimizer(

mcp_agent/resources/examples/workflows/router.py

@@ -43,7 +43,7 @@ SAMPLE_REQUESTS = [
 )
 @fast.router(
     name="route",
-    model="sonnet",
+    model="gpt-4.1",
     agents=["code_expert", "general_assistant", "fetcher"],
 )
 async def main() -> None:

mcp_agent/ui/console_display.py

@@ -1,6 +1,8 @@
+from json import JSONDecodeError
 from typing import Optional, Union
 
 from mcp.types import CallToolResult
+from rich.json import JSON
 from rich.panel import Panel
 from rich.text import Text
 
@@ -52,13 +54,15 @@ class ConsoleDisplay:
             elif len(content) == 1 and is_text_content(content[0]):
                 text_content = get_text(content[0])
                 char_count = len(text_content) if text_content else 0
-                status = f"Text Only ({char_count} chars)"
+                status = f"Text Only {char_count} chars"
             else:
                 text_count = sum(1 for item in content if is_text_content(item))
                 if text_count == len(content):
                     status = f"{len(content)} Text Blocks" if len(content) > 1 else "1 Text Block"
                 else:
-                    status = f"{len(content)} Content Blocks"
+                    status = (
+                        f"{len(content)} Content Blocks" if len(content) > 1 else "1 Content Block"
+                    )
 
         # Combined separator and status line
         left = f"[{block_color}]▎[/{block_color}][{text_color}]▶[/{text_color}]{f' [{block_color}]{name}[/{block_color}]' if name else ''}"
@@ -357,16 +361,22 @@ class ConsoleDisplay:
         right = f"[dim]{model}[/dim]" if model else ""
         self._create_combined_separator_status(left, right)
 
-        # Display content as markdown if it looks like markdown, otherwise as text
         if isinstance(message_text, str):
             content = message_text
-            #         if any(marker in content for marker in ["##", "**", "*", "`", "---", "###"]):
-            md = Markdown(content, code_theme=CODE_STYLE)
-            console.console.print(md, markup=self._markup)
-        #        else:
-        #           console.console.print(content, markup=self._markup)
+
+            # Try to detect and pretty print JSON
+            try:
+                import json
+
+                json.loads(content)
+                json = JSON(message_text)
+                console.console.print(json, markup=self._markup)
+            except (JSONDecodeError, TypeError, ValueError):
+                # Not JSON, treat as markdown
+                md = Markdown(content, code_theme=CODE_STYLE)
+                console.console.print(md, markup=self._markup)
         else:
-            # Handle Text objects directly
+            # Handle Rich Text objects directly
             console.console.print(message_text, markup=self._markup)
 
         # Bottom separator with server list: ─ [server1] [server2] ────────