fast-agent-mcp 0.2.41__py3-none-any.whl → 0.2.42__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