fast-agent-mcp 0.2.41__py3-none-any.whl → 0.2.43__py3-none-any.whl

mcp_agent/llm/providers/augmented_llm_openai.py

@@ -84,7 +84,9 @@ class OpenAIAugmentedLLM(AugmentedLLM[ChatCompletionMessageParam, ChatCompletion
         # TODO -- move this to model capabilities, add o4.
         chosen_model = self.default_request_params.model if self.default_request_params else None
         self._reasoning = chosen_model and (
-            chosen_model.startswith("o3") or chosen_model.startswith("o1")
+            chosen_model.startswith("o3")
+            or chosen_model.startswith("o1")
+            or chosen_model.startswith("o4")
         )
         if self._reasoning:
             self.logger.info(

mcp_agent/logging/logger.py

@@ -8,6 +8,7 @@ Logger module for the MCP Agent, which provides:
 """
 
 import asyncio
+import logging
 import threading
 import time
 from contextlib import asynccontextmanager, contextmanager
@@ -206,6 +207,12 @@ class LoggingConfig:
         if cls._initialized:
             return
 
+        # Suppress boto3/botocore logging to prevent flooding
+        logging.getLogger('boto3').setLevel(logging.WARNING)
+        logging.getLogger('botocore').setLevel(logging.WARNING)
+        logging.getLogger('urllib3').setLevel(logging.WARNING)
+        logging.getLogger('s3transfer').setLevel(logging.WARNING)
+
         bus = AsyncEventBus.get(transport=transport)
 
         # Add standard listeners

mcp_agent/mcp/hf_auth.py

@@ -69,14 +69,29 @@ def should_add_hf_auth(url: str, existing_headers: Optional[Dict[str, str]]) ->
     """
     # Only add HF auth if:
     # 1. URL is a HuggingFace URL
-    # 2. No existing Authorization header is set
+    # 2. No existing Authorization/X-HF-Authorization header is set
     # 3. HF_TOKEN environment variable is available
     
     if not is_huggingface_url(url):
         return False
         
-    if existing_headers and "Authorization" in existing_headers:
-        return False
+    if existing_headers:
+        # Check if this is a .hf.space domain
+        try:
+            parsed = urlparse(url)
+            hostname = parsed.hostname
+            if hostname and hostname.endswith(".hf.space"):
+                # For .hf.space, check for X-HF-Authorization header
+                if "X-HF-Authorization" in existing_headers:
+                    return False
+            else:
+                # For other HF domains, check for Authorization header
+                if "Authorization" in existing_headers:
+                    return False
+        except Exception:
+            # Fallback to checking Authorization header
+            if "Authorization" in existing_headers:
+                return False
         
     return get_hf_token_from_env() is not None
 
@@ -101,6 +116,19 @@ def add_hf_auth_header(url: str, headers: Optional[Dict[str, str]]) -> Optional[
         
     # Create new headers dict or copy existing one
     result_headers = dict(headers) if headers else {}
-    result_headers["Authorization"] = f"Bearer {hf_token}"
+    
+    # Check if this is a .hf.space domain
+    try:
+        parsed = urlparse(url)
+        hostname = parsed.hostname
+        if hostname and hostname.endswith(".hf.space"):
+            # Use X-HF-Authorization for .hf.space domains
+            result_headers["X-HF-Authorization"] = f"Bearer {hf_token}"
+        else:
+            # Use standard Authorization header for other HF domains
+            result_headers["Authorization"] = f"Bearer {hf_token}"
+    except Exception:
+        # Fallback to standard Authorization header
+        result_headers["Authorization"] = f"Bearer {hf_token}"
     
     return result_headers

mcp_agent/mcp/mcp_aggregator.py

@@ -258,7 +258,12 @@ class MCPAggregator(ContextDependent):
                 },
             )
 
-        async def fetch_tools(client: ClientSession):
+        async def fetch_tools(client: ClientSession, server_name: str) -> List[Tool]:
+            # Only fetch tools if the server supports them
+            if not await self.server_supports_feature(server_name, "tools"):
+                logger.debug(f"Server '{server_name}' does not support tools")
+                return []
+
             try:
                 result: ListToolsResult = await client.list_tools()
                 return result.tools or []
@@ -287,7 +292,7 @@ class MCPAggregator(ContextDependent):
                 server_connection = await self._persistent_connection_manager.get_server(
                     server_name, client_session_factory=MCPAgentClientSession
                 )
-                tools = await fetch_tools(server_connection.session)
+                tools = await fetch_tools(server_connection.session, server_name)
                 prompts = await fetch_prompts(server_connection.session, server_name)
             else:
                 # Create a factory function for the client session
@@ -326,7 +331,7 @@ class MCPAggregator(ContextDependent):
                     server_registry=self.context.server_registry,
                     client_session_factory=create_session,
                 ) as client:
-                    tools = await fetch_tools(client)
+                    tools = await fetch_tools(client, server_name)
                     prompts = await fetch_prompts(client, server_name)
 
             return server_name, tools, prompts
@@ -962,6 +967,11 @@ class MCPAggregator(ContextDependent):
             logger.error(f"Cannot refresh tools for unknown server '{server_name}'")
             return
 
+        # Check if server supports tools capability
+        if not await self.server_supports_feature(server_name, "tools"):
+            logger.debug(f"Server '{server_name}' does not support tools")
+            return
+
         await self.display.show_tool_update(aggregator=self, updated_server=server_name)
 
         async with self._refresh_lock:
@@ -1103,6 +1113,10 @@ class MCPAggregator(ContextDependent):
         Raises:
             Exception: If the resource couldn't be found or other error occurs
         """
+        # Check if server supports resources capability
+        if not await self.server_supports_feature(server_name, "resources"):
+            raise ValueError(f"Server '{server_name}' does not support resources")
+
         logger.info(
             "Requesting resource",
             data={
@@ -1163,6 +1177,11 @@ class MCPAggregator(ContextDependent):
             # Initialize empty list for this server
             results[s_name] = []
 
+            # Check if server supports resources capability
+            if not await self.server_supports_feature(s_name, "resources"):
+                logger.debug(f"Server '{s_name}' does not support resources")
+                continue
+
             try:
                 # Use the _execute_on_server method to call list_resources on the server
                 result = await self._execute_on_server(