fast-agent-mcp 0.2.27__py3-none-any.whl → 0.2.28__py3-none-any.whl

mcp_agent/mcp_server_registry.py

@@ -70,9 +70,15 @@ class ServerRegistry:
             config (Settings): The Settings object containing the server configurations.
             config_path (str): Path to the YAML configuration file.
         """
-        self.registry = (
-            self.load_registry_from_file(config_path) if config is None else config.mcp.servers
-        )
+        if config is None:
+            self.registry = self.load_registry_from_file(config_path)
+        elif config.mcp is not None and hasattr(config.mcp, 'servers') and config.mcp.servers is not None:
+            # Ensure config.mcp exists, has a 'servers' attribute, and it's not None
+            self.registry = config.mcp.servers
+        else:
+            # Default to an empty dictionary if config.mcp is None or has no 'servers'
+            self.registry = {}
+
         self.init_hooks: Dict[str, InitHookCallable] = {}
         self.connection_manager = MCPConnectionManager(self)
 
@@ -88,8 +94,13 @@ class ServerRegistry:
         Raises:
             ValueError: If the configuration is invalid.
         """
+        servers = {} 
 
-        servers = get_settings(config_path).mcp.servers or {}
+        settings = get_settings(config_path)
+        
+        if settings.mcp is not None and hasattr(settings.mcp, 'servers') and settings.mcp.servers is not None:
+            return settings.mcp.servers
+        
         return servers
 
     @asynccontextmanager

mcp_agent/tools/tool_definition.py

@@ -0,0 +1,14 @@
+from dataclasses import dataclass, field
+from typing import Any, Dict, Optional
+
+
+@dataclass
+class ToolDefinition:
+    """
+    Represents a definition of a tool available to the agent.
+    """
+
+    name: str
+    description: Optional[str] = None
+    inputSchema: Dict[str, Any] = field(default_factory=dict)
+    # Add other relevant fields if necessary based on how tools are defined in fast-agent