fast-agent-mcp 0.2.20__py3-none-any.whl → 0.2.21__py3-none-any.whl

{fast_agent_mcp-0.2.20.dist-info → fast_agent_mcp-0.2.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fast-agent-mcp
-Version: 0.2.20
+Version: 0.2.21
 Summary: Define, Prompt and Test MCP enabled Agents and Workflows
 Author-email: Shaun Smith <fastagent@llmindset.co.uk>, Sarmad Qadri <sarmad@lastmileai.dev>
 License:                                  Apache License
@@ -213,7 +213,7 @@ Requires-Dist: a2a-types>=0.1.0
 Requires-Dist: aiohttp>=3.11.13
 Requires-Dist: anthropic>=0.49.0
 Requires-Dist: fastapi>=0.115.6
-Requires-Dist: mcp==1.6.0
+Requires-Dist: mcp~=1.7.0
 Requires-Dist: openai>=1.63.2
 Requires-Dist: opentelemetry-distro>=0.50b0
 Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.29.0

{fast_agent_mcp-0.2.20.dist-info → fast_agent_mcp-0.2.21.dist-info}/RECORD

@@ -83,8 +83,8 @@ mcp_agent/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_agent/mcp/gen_client.py,sha256=fAVwFVCgSamw4PwoWOV4wrK9TABx1S_zZv8BctRyF2k,3030
 mcp_agent/mcp/interfaces.py,sha256=PAou8znAl2HgtvfCpLQOZFbKra9F72OcVRfBJbboNX8,6965
 mcp_agent/mcp/logger_textio.py,sha256=vljC1BtNTCxBAda9ExqNB-FwVNUZIuJT3h1nWmCjMws,3172
-mcp_agent/mcp/mcp_agent_client_session.py,sha256=RMYNltc2pDIzxwEJSS5589RbvPO0KWV4Y3jSyAmhKf0,4181
-mcp_agent/mcp/mcp_aggregator.py,sha256=c3UDWsTgHMcpHPx1p-vVru4y3eVO1jBQyLzwEMH2RHU,40237
+mcp_agent/mcp/mcp_agent_client_session.py,sha256=Ng7epBXq8BEA_3m1GX5LqwafgNUAMSzBugwN6N0VUWQ,4364
+mcp_agent/mcp/mcp_aggregator.py,sha256=lVSt0yp0CnaYjcHCWmluwBeFgl8JXHYEZk0MzXgrQzA,40110
 mcp_agent/mcp/mcp_connection_manager.py,sha256=L5Dk4cyarN_v2rfktkrfZJR4xUuD3yN_hUyQnKHBWgM,14044
 mcp_agent/mcp/mime_utils.py,sha256=difepNR_gpb4MpMLkBRAoyhDk-AjXUHTiqKvT_VwS1o,1805
 mcp_agent/mcp/prompt_message_multipart.py,sha256=BDwRdNwyWHb2q2bccDb2iR2VlORqVvkvoG3xYzcMpCE,4403
@@ -99,7 +99,7 @@ mcp_agent/mcp/prompts/__main__.py,sha256=gr1Tdz9fcK0EXjEuZg_BOnKUmvhYq5AH2lFZicV
 mcp_agent/mcp/prompts/prompt_constants.py,sha256=Q9W0t3rOXl2LHIG9wcghApUV2QZ1iICuo7SwVwHUf3c,566
 mcp_agent/mcp/prompts/prompt_helpers.py,sha256=Joqo2t09pTKDP-Wge3G-ozPEHikzjaqwV6GVk8hNR50,7534
 mcp_agent/mcp/prompts/prompt_load.py,sha256=Zo0FogqWFEG5FtF1d9ZH-RWsCSSMsi5FIEQHpJD8N7M,5404
-mcp_agent/mcp/prompts/prompt_server.py,sha256=SiUR2xYfd3vEpghnYRdzz2rFEMtAbCKx2xzUXgvz1g8,18501
+mcp_agent/mcp/prompts/prompt_server.py,sha256=DbuDcYCMbsbqwBeebpNGInAQ4-DP1Jjp49y8uZ-0XlY,18872
 mcp_agent/mcp/prompts/prompt_template.py,sha256=EejiqGkau8OizORNyKTUwUjrPof5V-hH1H_MBQoQfXw,15732
 mcp_agent/mcp_server/__init__.py,sha256=zBU51ITHIEPScd9nRafnhEddsWqXRPAAvHhkrbRI2_4,155
 mcp_agent/mcp_server/agent_server.py,sha256=s-nI0uTNWx4nYDDM_5GmuY5x6ZeFkymfNoCSuwuBRd8,19891
@@ -143,8 +143,8 @@ mcp_agent/resources/examples/workflows/parallel.py,sha256=DQ5vY5-h8Qa5QHcYjsWXhZ
 mcp_agent/resources/examples/workflows/router.py,sha256=E4x_-c3l4YW9w1i4ARcDtkdeqIdbWEGfsMzwLYpdbVc,1677
 mcp_agent/resources/examples/workflows/short_story.txt,sha256=X3y_1AyhLFN2AKzCKvucJtDgAFIJfnlbsbGZO5bBWu0,1187
 mcp_agent/ui/console_display.py,sha256=TVGDtJ37hc6UG0ei9g7ZPZZfFNeS1MYozt-Mx8HsPCk,9752
-fast_agent_mcp-0.2.20.dist-info/METADATA,sha256=RlU6MHHAJoP4xuuA8QsIspMEZfGdSKDo76so374wzA4,30142
-fast_agent_mcp-0.2.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fast_agent_mcp-0.2.20.dist-info/entry_points.txt,sha256=bRniFM5zk3Kix5z7scX0gf9VnmGQ2Cz_Q1Gh7Ir4W00,186
-fast_agent_mcp-0.2.20.dist-info/licenses/LICENSE,sha256=cN3FxDURL9XuzE5mhK9L2paZo82LTfjwCYVT7e3j0e4,10939
-fast_agent_mcp-0.2.20.dist-info/RECORD,,
+fast_agent_mcp-0.2.21.dist-info/METADATA,sha256=c-Aq5664t9xi4iW2mcDHUGaDL_jRUznn_nyeMVFJzgY,30142
+fast_agent_mcp-0.2.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fast_agent_mcp-0.2.21.dist-info/entry_points.txt,sha256=bRniFM5zk3Kix5z7scX0gf9VnmGQ2Cz_Q1Gh7Ir4W00,186
+fast_agent_mcp-0.2.21.dist-info/licenses/LICENSE,sha256=cN3FxDURL9XuzE5mhK9L2paZo82LTfjwCYVT7e3j0e4,10939
+fast_agent_mcp-0.2.21.dist-info/RECORD,,

mcp_agent/mcp/mcp_agent_client_session.py

@@ -3,6 +3,7 @@ A derived client session for the MCP Agent framework.
 It adds logging and supports sampling requests.
 """
 
+from datetime import timedelta
 from typing import TYPE_CHECKING, Optional
 
 from mcp import ClientSession
@@ -73,10 +74,13 @@ class MCPAgentClientSession(ClientSession, ContextDependent):
         self,
         request: SendRequestT,
         result_type: type[ReceiveResultT],
+        request_read_timeout_seconds: timedelta | None = None,
     ) -> ReceiveResultT:
         logger.debug("send_request: request=", data=request.model_dump())
         try:
-            result = await super().send_request(request, result_type)
+            result = await super().send_request(
+                request, result_type, request_read_timeout_seconds=request_read_timeout_seconds
+            )
             logger.debug("send_request: response=", data=result.model_dump())
             return result
         except Exception as e:

mcp_agent/mcp/mcp_aggregator.py

@@ -41,10 +41,12 @@ SEP = "-"
 T = TypeVar("T")
 R = TypeVar("R")
 
+
 def create_namespaced_name(server_name: str, resource_name: str) -> str:
     """Create a namespaced resource name from server and resource names"""
     return f"{server_name}{SEP}{resource_name}"
 
+
 def is_namespaced_name(name: str) -> bool:
     """Check if a name is already namespaced"""
     return SEP in name
@@ -325,14 +327,14 @@ class MCPAggregator(ContextDependent):
         except Exception as e:
             logger.debug(f"Error getting capabilities for server '{server_name}': {e}")
             return None
-            
+
     async def validate_server(self, server_name: str) -> bool:
         """
         Validate that a server exists in our server list.
-        
+
         Args:
             server_name: Name of the server to validate
-            
+
         Returns:
             True if the server exists, False otherwise
         """
@@ -340,25 +342,25 @@ class MCPAggregator(ContextDependent):
         if not valid:
             logger.debug(f"Server '{server_name}' not found")
         return valid
-    
+
     async def server_supports_feature(self, server_name: str, feature: str) -> bool:
         """
         Check if a server supports a specific feature.
-        
+
         Args:
             server_name: Name of the server to check
             feature: Feature to check for (e.g., "prompts", "resources")
-            
+
         Returns:
             True if the server supports the feature, False otherwise
         """
         if not await self.validate_server(server_name):
             return False
-            
+
         capabilities = await self.get_capabilities(server_name)
         if not capabilities:
             return False
-            
+
         return getattr(capabilities, feature, False)
 
     async def list_servers(self) -> List[str]:
@@ -465,26 +467,26 @@ class MCPAggregator(ContextDependent):
         if resource_type == "tool" and name in self._namespaced_tool_map:
             namespaced_tool = self._namespaced_tool_map[name]
             return namespaced_tool.server_name, namespaced_tool.tool.name
-            
+
         # Next, attempt to interpret as a namespaced name
         if is_namespaced_name(name):
             parts = name.split(SEP, 1)
             server_name, local_name = parts[0], parts[1]
-            
+
             # Validate that the parsed server actually exists
             if server_name in self.server_names:
                 return server_name, local_name
-            
+
             # If the server name doesn't exist, it might be a tool with a hyphen in its name
             # Fall through to the next checks
-            
+
         # For tools, search all servers for the tool by exact name match
         if resource_type == "tool":
             for server_name, tools in self._server_to_tool_map.items():
                 for namespaced_tool in tools:
                     if namespaced_tool.tool.name == name:
                         return server_name, name
-                        
+
         # For all other resource types, use the first server
         return (self.server_names[0] if self.server_names else None, name)
 
@@ -524,7 +526,10 @@ class MCPAggregator(ContextDependent):
                 operation_type="tool",
                 operation_name=local_tool_name,
                 method_name="call_tool",
-                method_args={"name": local_tool_name, "arguments": arguments},
+                method_args={
+                    "name": local_tool_name,
+                    "arguments": arguments,
+                },
                 error_factory=lambda msg: CallToolResult(
                     isError=True, content=[TextContent(type="text", text=msg)]
                 ),
@@ -558,7 +563,7 @@ class MCPAggregator(ContextDependent):
         elif is_namespaced_name(prompt_name):
             parts = prompt_name.split(SEP, 1)
             potential_server = parts[0]
-            
+
             # Only treat as namespaced if the server part is valid
             if potential_server in self.server_names:
                 server_name = potential_server
@@ -570,7 +575,7 @@ class MCPAggregator(ContextDependent):
         else:
             local_prompt_name = prompt_name
             # We'll search all servers below
-        
+
         # If we have a specific server to check
         if server_name:
             if not await self.validate_server(server_name):
@@ -579,7 +584,7 @@ class MCPAggregator(ContextDependent):
                     description=f"Error: Server '{server_name}' not found",
                     messages=[],
                 )
-                
+
             # Check if server supports prompts
             if not await self.server_supports_feature(server_name, "prompts"):
                 logger.debug(f"Server '{server_name}' does not support prompts")

mcp_agent/mcp/prompts/prompt_server.py

@@ -82,6 +82,7 @@ class PromptConfig(PromptMetadata):
     http_timeout: float = 10.0
     transport: str = "stdio"
     port: int = 8000
+    host: str = "0.0.0.0"
 
 
 # We'll maintain registries of all exposed resources and prompts
@@ -344,6 +345,12 @@ def parse_args():
         default=8000,
         help="Port to use for SSE transport (default: 8000)",
     )
+    parser.add_argument(
+        "--host",
+        type=str,
+        default="0.0.0.0",
+        help="Host to bind to for SSE transport (default: 0.0.0.0)",
+    )
     parser.add_argument(
         "--test", type=str, help="Test a specific prompt without starting the server"
     )
@@ -380,6 +387,7 @@ def initialize_config(args) -> PromptConfig:
         http_timeout=args.http_timeout,
         transport=args.transport,
         port=args.port,
+        host=args.host,
     )
 
 
@@ -497,7 +505,10 @@ async def async_main() -> int:
     if config.transport == "stdio":
         await mcp.run_stdio_async()
     else:  # sse
-        # TODO update to 2025-03-26 specification and test config.
+        # Set the host and port in settings before running the server
+        mcp.settings.host = config.host
+        mcp.settings.port = config.port
+        logger.info(f"Starting SSE server on {config.host}:{config.port}")
         await mcp.run_sse_async()
     return 0