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

{fast_agent_mcp-0.2.28.dist-info → fast_agent_mcp-0.2.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fast-agent-mcp
-Version: 0.2.28
+Version: 0.2.29
 Summary: Define, Prompt and Test MCP enabled Agents and Workflows
 Author-email: Shaun Smith <fastagent@llmindset.co.uk>
 License:                                  Apache License
@@ -213,16 +213,17 @@ Requires-Dist: a2a-types>=0.1.0
 Requires-Dist: aiohttp>=3.11.13
 Requires-Dist: anthropic>=0.49.0
 Requires-Dist: azure-identity>=1.14.0
+Requires-Dist: deprecated>=1.2.18
 Requires-Dist: fastapi>=0.115.6
 Requires-Dist: google-genai
-Requires-Dist: mcp==1.9.1
+Requires-Dist: mcp==1.9.3
 Requires-Dist: openai>=1.63.2
 Requires-Dist: opentelemetry-distro>=0.50b0
 Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.29.0
-Requires-Dist: opentelemetry-instrumentation-anthropic>=0.39.3; python_version >= '3.10' and python_version < '4.0'
+Requires-Dist: opentelemetry-instrumentation-anthropic>=0.40.7; python_version >= '3.10' and python_version < '4.0'
 Requires-Dist: opentelemetry-instrumentation-google-genai>=0.2b0
-Requires-Dist: opentelemetry-instrumentation-mcp>=0.40.3; python_version >= '3.10' and python_version < '4.0'
-Requires-Dist: opentelemetry-instrumentation-openai>=0.39.3; python_version >= '3.10' and python_version < '4.0'
+Requires-Dist: opentelemetry-instrumentation-mcp>=0.40.7; python_version >= '3.10' and python_version < '4.0'
+Requires-Dist: opentelemetry-instrumentation-openai>=0.0.40.7; python_version >= '3.10' and python_version < '4.0'
 Requires-Dist: prompt-toolkit>=3.0.50
 Requires-Dist: pydantic-settings>=2.7.0
 Requires-Dist: pydantic>=2.10.4
@@ -286,11 +287,13 @@ Start by installing the [uv package manager](https://docs.astral.sh/uv/) for Pyt
 
 ```bash
 uv pip install fast-agent-mcp          # install fast-agent!
-
-uv run fast-agent setup                # create an example agent and config files
+fast-agent go                          # start an interactive session
+fast-agent go https://hf.co/mcp        # with a remote MCP
+fast-agent go --model=generic.qwen2.5  # use ollama qwen 2.5
+fast-agent setup                       # create an example agent and config files
 uv run agent.py                        # run your first agent
 uv run agent.py --model=o3-mini.low    # specify a model
-uv run fast-agent quickstart workflow  # create "building effective agents" examples
+fast-agent quickstart workflow  # create "building effective agents" examples
 ```
 
 Other quickstart examples include a Researcher Agent (with Evaluator-Optimizer workflow) and Data Analysis Agent (similar to the ChatGPT experience), demonstrating MCP Roots support.

{fast_agent_mcp-0.2.28.dist-info → fast_agent_mcp-0.2.29.dist-info}/RECORD

@@ -1,11 +1,11 @@
 mcp_agent/__init__.py,sha256=18T0AG0W9sJhTY38O9GFFOzliDhxx9p87CvRyti9zbw,1620
-mcp_agent/app.py,sha256=WRsiUdwy_9IAnaGRDwuLm7pzgQpt2wgsg10vBOpfcwM,5539
+mcp_agent/app.py,sha256=3mtHP1nRQcRaKhhxgTmCOv00alh70nT7UxNA8bN47QE,5560
 mcp_agent/config.py,sha256=ITwLZ-Wzn8I2xYOMDP9XvNwZTLzzUbvQNnnna7PxflQ,13438
 mcp_agent/console.py,sha256=Gjf2QLFumwG1Lav__c07X_kZxxEUSkzV-1_-YbAwcwo,813
-mcp_agent/context.py,sha256=W9a91UVrpaT-GgoCtJp_ccHXnhmr8HaATDw7srm9NpU,7559
+mcp_agent/context.py,sha256=H7JbaZ_8SzzTagLmIgUPUPxX5370C5qjQAsasFPZG2Y,7510
 mcp_agent/context_dependent.py,sha256=QXfhw3RaQCKfscEEBRGuZ3sdMWqkgShz2jJ1ivGGX1I,1455
 mcp_agent/event_progress.py,sha256=040lrCCclcOuryi07YGSej25kTQF5_JMXY12Yj-3u1U,2773
-mcp_agent/mcp_server_registry.py,sha256=cZJJEXvN_FEUlTFWPtr2q2mRcDjPGBWVmRifc1TnLa0,12042
+mcp_agent/mcp_server_registry.py,sha256=b3iSb-0ULYc5yUG2KHav41WGwSYWiJCGQsOwWHWByxo,12346
 mcp_agent/progress_display.py,sha256=GeJU9VUt6qKsFVymG688hCMVCsAygG9ifiiEb5IcbN4,361
 mcp_agent/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_agent/agents/agent.py,sha256=EAYlcP1qqI1D0_CS808I806z1048FBjZQxxpcCZPeIU,3154
@@ -26,7 +26,7 @@ mcp_agent/cli/commands/check_config.py,sha256=KJbXUFx5Qih3lb_r-Fcx_uAjgHhgD7qqPe
 mcp_agent/cli/commands/go.py,sha256=LIsOJQuTdfCUcNm7JT-NQDU8cI-GCnYwYjN2VOWxvqs,8658
 mcp_agent/cli/commands/quickstart.py,sha256=SM3CHMzDgvTxIpKjFuX9BrS_N1vRoXNBDaO90aWx1Rk,14586
 mcp_agent/cli/commands/setup.py,sha256=eOEd4TL-b0DaDeSJMGOfNOsTEItoZ67W88eTP4aP-bo,6482
-mcp_agent/cli/commands/url_parser.py,sha256=7QL9bp9tO7w0cPnwhbpt8GwjbOJ1Rrry1o71uVJhSss,5655
+mcp_agent/cli/commands/url_parser.py,sha256=5VdtcHRHzi67YignStVbz7u-rcvNNErw9oJLAUFOtEY,5855
 mcp_agent/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_agent/core/agent_app.py,sha256=aVvOzMrXZ3TfRGyAsnvcrMMYZxBf8Saa0UuHiA7DV0w,9922
 mcp_agent/core/agent_types.py,sha256=bQVQMTwKH7qHIJsNglj4C_d6PNFBBzC_0RIkcENSII4,1459
@@ -36,7 +36,7 @@ mcp_agent/core/enhanced_prompt.py,sha256=bzvcengS7XzHWB7NWhyxHM3hhO2HI4zP5DbGXAO
 mcp_agent/core/error_handling.py,sha256=xoyS2kLe0eG0bj2eSJCJ2odIhGUve2SbDR7jP-A-uRw,624
 mcp_agent/core/exceptions.py,sha256=ENAD_qGG67foxy6vDkIvc-lgopIUQy6O7zvNPpPXaQg,2289
 mcp_agent/core/fastagent.py,sha256=uS_NSXeniUYFu6xce8OHGJ9PbEYNU-gm1XVpa1r0rZc,22893
-mcp_agent/core/interactive_prompt.py,sha256=QuPdJQTBwXnnKyp52QjeVGyaXMiMm3fZK6g1ly2X7v4,24696
+mcp_agent/core/interactive_prompt.py,sha256=JKEU1Gvq6zXaLniDPx8wll08ZTC6g1rQflL7khmnhs8,24710
 mcp_agent/core/mcp_content.py,sha256=2D7KHY9mG_vxoDwFLKvsPQV9VRIzHItM7V-jcEnACh8,8878
 mcp_agent/core/prompt.py,sha256=qnintOUGEoDPYLI9bu9G2OlgVMCe5ZPUZilgMzydXhc,7919
 mcp_agent/core/request_params.py,sha256=qmFWZXeYEJyYw2IwonyrTnZWxQG7qX6bKpOPcqETa60,1603
@@ -64,14 +64,14 @@ mcp_agent/llm/providers/__init__.py,sha256=heVxtmuqFJOnjjxHz4bWSqTAxXoN1E8twC_gQ
 mcp_agent/llm/providers/anthropic_utils.py,sha256=vYDN5G5jKMhD2CQg8veJYab7tvvzYkDMq8M1g_hUAQg,3275
 mcp_agent/llm/providers/augmented_llm_anthropic.py,sha256=gK_IvllVBNJUUrSfpgFpdhM-d4liCt0MLq7d2lXS7RI,15510
 mcp_agent/llm/providers/augmented_llm_azure.py,sha256=VPrD6lNrEw6EdYUTa9MDvHDNIPjJU5CG5xnKCM3JYdA,5878
-mcp_agent/llm/providers/augmented_llm_deepseek.py,sha256=NiZK5nv91ZS2VgVFXpbsFNFYLsLcppcbo_RstlRMd7I,1145
+mcp_agent/llm/providers/augmented_llm_deepseek.py,sha256=GFLzITAsUPUXpQ_SKFfAvxYb1kCk1tlcjMnkHnEHNxY,3228
 mcp_agent/llm/providers/augmented_llm_generic.py,sha256=5Uq8ZBhcFuQTt7koP_5ykolREh2iWu8zKhNbh3pM9lQ,1210
 mcp_agent/llm/providers/augmented_llm_google_native.py,sha256=Axk6oKH5ctB6rXGnCjRKVkJq6O7rRqlD7aJ2He6UuZ8,20406
 mcp_agent/llm/providers/augmented_llm_google_oai.py,sha256=cO4dvjTl9ymqEurCOo5nP09ATfXVjgkuk1yZAlWpS1s,1137
 mcp_agent/llm/providers/augmented_llm_openai.py,sha256=5CFHKayjm-aeCBpohIK3WelAEuX7_LDGZIKnWR_rq-s,14577
 mcp_agent/llm/providers/augmented_llm_openrouter.py,sha256=V_TlVKm92GHBxYIo6gpvH_6cAaIdppS25Tz6x5T7LW0,2341
 mcp_agent/llm/providers/augmented_llm_tensorzero.py,sha256=Mol_Wzj_ZtccW-LMw0oFwWUt1m1yfofloay9QYNP23c,20729
-mcp_agent/llm/providers/google_converter.py,sha256=bA0oYdB6tfRX_iuwTr8xTBYWlzNNskwYIS3Y9aFyEbo,16643
+mcp_agent/llm/providers/google_converter.py,sha256=zsqxJJ636WzCL2K6w-yB94O8bdNR6mo8f5mQEnUJFyg,16831
 mcp_agent/llm/providers/multipart_converter_anthropic.py,sha256=t5lHYGfFUacJldnrVtMNW-8gEMoto8Y7hJkDrnyZR-Y,16650
 mcp_agent/llm/providers/multipart_converter_openai.py,sha256=XPIulWntNpZWNGWrc240StPzok2RqrDAV7OigDwQ1uU,15850
 mcp_agent/llm/providers/multipart_converter_tensorzero.py,sha256=BFTdyVk42HZskDAuTHicfDTUJq89d1fz8C9nAOuHxlE,8646
@@ -87,11 +87,12 @@ mcp_agent/logging/logger.py,sha256=l02OGX_c5FOyH0rspd4ZvnkJcbb0FahhUhlh2KI8mqE,1
 mcp_agent/logging/rich_progress.py,sha256=oY9fjb4Tyw6887v8sgO6EGIK4lnmIoR3NNxhA_-Ln_M,4893
 mcp_agent/logging/transport.py,sha256=m8YsLLu5T8eof_ndpLQs4gHOzqqEL98xsVwBwDsBfxI,17335
 mcp_agent/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mcp_agent/mcp/common.py,sha256=DiWLH9rxWvCgkKRsHQehY9mDhQl9gki1-q7LVUflDvI,425
+mcp_agent/mcp/common.py,sha256=MpSC0fLO21RcDz4VApah4C8_LisVGz7OXkR17Xw-9mY,431
 mcp_agent/mcp/gen_client.py,sha256=fAVwFVCgSamw4PwoWOV4wrK9TABx1S_zZv8BctRyF2k,3030
+mcp_agent/mcp/hf_auth.py,sha256=-ZefZ7Vh_hQngVYKUT1BDya84gxbwpRiWqxaHEb3k5E,2430
 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=1zlBZaGRvCHuflrTd4dUeC4BzcPXxfPuciulJB0DIFI,7900
+mcp_agent/mcp/mcp_agent_client_session.py,sha256=V17Lj21rMGIKKVAIyNx5l5gmC8jQuohjJGpRcoCXfVA,6862
 mcp_agent/mcp/mcp_aggregator.py,sha256=Mdmr-6gNlrcofHzhHZloz1QVbC5ZAnCSPNFY5fwm-Bs,47075
 mcp_agent/mcp/mcp_connection_manager.py,sha256=5JekxOJsB46spHsiXt7pyRPicg8TGHMiSJRtXRW2JB8,17074
 mcp_agent/mcp/mime_utils.py,sha256=difepNR_gpb4MpMLkBRAoyhDk-AjXUHTiqKvT_VwS1o,1805
@@ -153,8 +154,8 @@ mcp_agent/resources/examples/workflows/router.py,sha256=E4x_-c3l4YW9w1i4ARcDtkde
 mcp_agent/resources/examples/workflows/short_story.txt,sha256=X3y_1AyhLFN2AKzCKvucJtDgAFIJfnlbsbGZO5bBWu0,1187
 mcp_agent/tools/tool_definition.py,sha256=L3Pxl-uLEXqlVoo-bYuFTFALeI-2pIU44YgFhsTKEtM,398
 mcp_agent/ui/console_display.py,sha256=UKqax5V2TC0hkZZORmmd6UqUk0DGX7A25E3h1k9f42k,10982
-fast_agent_mcp-0.2.28.dist-info/METADATA,sha256=DbIbErJ9uZh6NnL6Tl-OD4ihPgzSRXR_ExDpafUwGKQ,30581
-fast_agent_mcp-0.2.28.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fast_agent_mcp-0.2.28.dist-info/entry_points.txt,sha256=bRniFM5zk3Kix5z7scX0gf9VnmGQ2Cz_Q1Gh7Ir4W00,186
-fast_agent_mcp-0.2.28.dist-info/licenses/LICENSE,sha256=cN3FxDURL9XuzE5mhK9L2paZo82LTfjwCYVT7e3j0e4,10939
-fast_agent_mcp-0.2.28.dist-info/RECORD,,
+fast_agent_mcp-0.2.29.dist-info/METADATA,sha256=diqb8oCcIBC11MDR4NLqMVQVLy_jmCP7x5lfvuviRYE,30799
+fast_agent_mcp-0.2.29.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fast_agent_mcp-0.2.29.dist-info/entry_points.txt,sha256=bRniFM5zk3Kix5z7scX0gf9VnmGQ2Cz_Q1Gh7Ir4W00,186
+fast_agent_mcp-0.2.29.dist-info/licenses/LICENSE,sha256=cN3FxDURL9XuzE5mhK9L2paZo82LTfjwCYVT7e3j0e4,10939
+fast_agent_mcp-0.2.29.dist-info/RECORD,,

mcp_agent/app.py

@@ -119,7 +119,7 @@ class MCPApp:
         if self._initialized:
             return
 
-        self._context = await initialize_context(self._config_or_path)
+        self._context = await initialize_context(self._config_or_path, store_globally=True)
 
         # Set the properties that were passed in the constructor
         self._context.human_input_handler = self._human_input_callback

mcp_agent/cli/commands/url_parser.py

@@ -8,6 +8,8 @@ import re
 from typing import Dict, List, Literal, Tuple
 from urllib.parse import urlparse
 
+from mcp_agent.mcp.hf_auth import add_hf_auth_header
+
 
 def parse_server_url(
     url: str,
@@ -131,7 +133,11 @@ def parse_server_urls(
     result = []
     for url in url_list:
         server_name, transport_type, parsed_url = parse_server_url(url)
-        result.append((server_name, transport_type, parsed_url, headers))
+        
+        # Apply HuggingFace authentication if appropriate
+        final_headers = add_hf_auth_header(parsed_url, headers)
+        
+        result.append((server_name, transport_type, parsed_url, final_headers))
 
     return result
 

mcp_agent/context.py

@@ -12,7 +12,8 @@ from opentelemetry import trace
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.instrumentation.anthropic import AnthropicInstrumentor
 from opentelemetry.instrumentation.google_genai import GoogleGenAiSdkInstrumentor
-from opentelemetry.instrumentation.mcp import McpInstrumentor
+
+# from opentelemetry.instrumentation.mcp import McpInstrumentor
 from opentelemetry.instrumentation.openai import OpenAIInstrumentor
 from opentelemetry.propagate import set_global_textmap
 from opentelemetry.sdk.resources import Resource
@@ -114,7 +115,9 @@ async def configure_otel(config: "Settings") -> None:
     AnthropicInstrumentor().instrument()
     OpenAIInstrumentor().instrument()
     GoogleGenAiSdkInstrumentor().instrument()
-    McpInstrumentor().instrument()
+
+
+#    McpInstrumentor().instrument()
 
 
 async def configure_logger(config: "Settings") -> None:
@@ -198,7 +201,6 @@ _global_context: Context | None = None
 def get_current_context() -> Context:
     """
     Synchronous initializer/getter for global application context.
-    For async usage, use aget_current_context instead.
     """
     global _global_context
     if _global_context is None:

mcp_agent/core/interactive_prompt.py

@@ -351,7 +351,7 @@ class InteractivePrompt:
                 for prompt in prompts:
                     # Get basic prompt info
                     prompt_name = getattr(prompt, "name", "Unknown")
-                    description = getattr(prompt, "description", "No description")
+                    prompt_description = getattr(prompt, "description", "No description")
 
                     # Extract argument information
                     arg_names = []
@@ -387,7 +387,7 @@ class InteractivePrompt:
                             "server": server_name,
                             "name": prompt_name,
                             "namespaced_name": namespaced_name,
-                            "description": description,
+                            "description": prompt_description,
                             "arg_count": len(arg_names),
                             "arg_names": arg_names,
                             "required_args": required_args,

mcp_agent/llm/providers/augmented_llm_deepseek.py

@@ -1,6 +1,10 @@
+from typing import List, Tuple, Type
+
 from mcp_agent.core.request_params import RequestParams
 from mcp_agent.llm.provider_types import Provider
 from mcp_agent.llm.providers.augmented_llm_openai import OpenAIAugmentedLLM
+from mcp_agent.mcp.interfaces import ModelT
+from mcp_agent.mcp.prompt_message_multipart import PromptMessageMultipart
 
 DEEPSEEK_BASE_URL = "https://api.deepseek.com"
 DEFAULT_DEEPSEEK_MODEL = "deepseekchat"  # current Deepseek only has two type models
@@ -28,3 +32,48 @@ class DeepSeekAugmentedLLM(OpenAIAugmentedLLM):
             base_url = self.context.config.deepseek.base_url
 
         return base_url if base_url else DEEPSEEK_BASE_URL
+
+    async def _apply_prompt_provider_specific_structured(
+        self,
+        multipart_messages: List[PromptMessageMultipart],
+        model: Type[ModelT],
+        request_params: RequestParams | None = None,
+    ) -> Tuple[ModelT | None, PromptMessageMultipart]:  # noqa: F821
+        request_params = self.get_request_params(request_params)
+
+        request_params.response_format = {"type": "json_object"}
+
+        # Get the full schema and extract just the properties
+        full_schema = model.model_json_schema()
+        properties = full_schema.get("properties", {})
+        required_fields = full_schema.get("required", [])
+
+        # Create a cleaner format description
+        format_description = "{\n"
+        for field_name, field_info in properties.items():
+            field_type = field_info.get("type", "string")
+            description = field_info.get("description", "")
+            format_description += f'  "{field_name}": "{field_type}"'
+            if description:
+                format_description += f"  // {description}"
+            if field_name in required_fields:
+                format_description += "  // REQUIRED"
+            format_description += "\n"
+        format_description += "}"
+
+        multipart_messages[-1].add_text(
+            f"""YOU MUST RESPOND WITH A JSON OBJECT IN EXACTLY THIS FORMAT:
+            {format_description}
+
+            IMPORTANT RULES:
+            - Respond ONLY with the JSON object, no other text
+            - Do NOT include "properties" or "schema" wrappers
+            - Do NOT use code fences or markdown
+            - The response must be valid JSON that matches the format above
+            - All required fields must be included"""
+        )
+
+        result: PromptMessageMultipart = await self._apply_prompt_provider_specific(
+            multipart_messages, request_params
+        )
+        return self._structured_from_multipart(result, model)

mcp_agent/llm/providers/google_converter.py

@@ -166,6 +166,10 @@ class GoogleConverter:
         fast_agent_parts: List[
             TextContent | ImageContent | EmbeddedResource | CallToolRequestParams
         ] = []
+
+        if content is None or not hasattr(content, 'parts') or content.parts is None:
+                    return [] # Google API response 'content' object is None. Cannot extract parts.
+
         for part in content.parts:
             if part.text:
                 fast_agent_parts.append(TextContent(type="text", text=part.text))

mcp_agent/mcp/common.py

@@ -8,9 +8,9 @@ SEP = "-"
 
 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}"
+    return f"{server_name}{SEP}{resource_name}"[:64]
 
 
 def is_namespaced_name(name: str) -> bool:
     """Check if a name is already namespaced"""
-    return SEP in name
+    return SEP in name

mcp_agent/mcp/hf_auth.py

@@ -0,0 +1,87 @@
+"""HuggingFace authentication utilities for MCP connections."""
+
+import os
+from typing import Dict, Optional
+from urllib.parse import urlparse
+
+
+def is_huggingface_url(url: str) -> bool:
+    """
+    Check if a URL is a HuggingFace URL that should receive HF_TOKEN authentication.
+    
+    Args:
+        url: The URL to check
+        
+    Returns:
+        True if the URL is a HuggingFace URL, False otherwise
+    """
+    try:
+        parsed = urlparse(url)
+        hostname = parsed.hostname
+        if hostname is None:
+            return False
+        
+        # Check for HuggingFace domains
+        return hostname in {"hf.co", "huggingface.co"}
+    except Exception:
+        return False
+
+
+def get_hf_token_from_env() -> Optional[str]:
+    """
+    Get the HuggingFace token from the HF_TOKEN environment variable.
+    
+    Returns:
+        The HF_TOKEN value if set, None otherwise
+    """
+    return os.environ.get("HF_TOKEN")
+
+
+def should_add_hf_auth(url: str, existing_headers: Optional[Dict[str, str]]) -> bool:
+    """
+    Determine if HuggingFace authentication should be added to the headers.
+    
+    Args:
+        url: The URL to check
+        existing_headers: Existing headers dictionary (may be None)
+        
+    Returns:
+        True if HF auth should be added, False otherwise
+    """
+    # Only add HF auth if:
+    # 1. URL is a HuggingFace URL
+    # 2. No existing 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
+        
+    return get_hf_token_from_env() is not None
+
+
+def add_hf_auth_header(url: str, headers: Optional[Dict[str, str]]) -> Optional[Dict[str, str]]:
+    """
+    Add HuggingFace authentication header if appropriate.
+    
+    Args:
+        url: The URL to check
+        headers: Existing headers dictionary (may be None)
+        
+    Returns:
+        Updated headers dictionary with HF auth if appropriate, or original headers
+    """
+    if not should_add_hf_auth(url, headers):
+        return headers
+        
+    hf_token = get_hf_token_from_env()
+    if hf_token is None:
+        return headers
+        
+    # Create new headers dict or copy existing one
+    result_headers = dict(headers) if headers else {}
+    result_headers["Authorization"] = f"Bearer {hf_token}"
+    
+    return result_headers

mcp_agent/mcp/mcp_agent_client_session.py

@@ -7,15 +7,13 @@ from datetime import timedelta
 from typing import TYPE_CHECKING
 
 from mcp import ClientSession, ServerNotification
+from mcp.shared.message import MessageMetadata
 from mcp.shared.session import (
     ProgressFnT,
     ReceiveResultT,
-    RequestId,
-    SendNotificationT,
     SendRequestT,
-    SendResultT,
 )
-from mcp.types import ErrorData, Implementation, ListRootsResult, Root, ToolListChangedNotification
+from mcp.types import Implementation, ListRootsResult, Root, ToolListChangedNotification
 from pydantic import FileUrl
 
 from mcp_agent.context_dependent import ContextDependent
@@ -76,12 +74,16 @@ class MCPAgentClientSession(ClientSession, ContextDependent):
 
         # Only register callbacks if the server_config has the relevant settings
         list_roots_cb = list_roots if (self.server_config and self.server_config.roots) else None
-        
+
         # Register sampling callback if either:
         # 1. Sampling is explicitly configured, OR
         # 2. Application-level auto_sampling is enabled
         sampling_cb = None
-        if self.server_config and hasattr(self.server_config, "sampling") and self.server_config.sampling:
+        if (
+            self.server_config
+            and hasattr(self.server_config, "sampling")
+            and self.server_config.sampling
+        ):
             # Explicit sampling configuration
             sampling_cb = sample
         elif self._should_enable_auto_sampling():
@@ -100,9 +102,10 @@ class MCPAgentClientSession(ClientSession, ContextDependent):
         """Check if auto_sampling is enabled at the application level."""
         try:
             from mcp_agent.context import get_current_context
+
             context = get_current_context()
             if context and context.config:
-                return getattr(context.config, 'auto_sampling', True)
+                return getattr(context.config, "auto_sampling", True)
         except Exception:
             pass
         return True  # Default to True if can't access config
@@ -112,6 +115,7 @@ class MCPAgentClientSession(ClientSession, ContextDependent):
         request: SendRequestT,
         result_type: type[ReceiveResultT],
         request_read_timeout_seconds: timedelta | None = None,
+        metadata: MessageMetadata | None = None,
         progress_callback: ProgressFnT | None = None,
     ) -> ReceiveResultT:
         logger.debug("send_request: request=", data=request.model_dump())
@@ -120,32 +124,18 @@ class MCPAgentClientSession(ClientSession, ContextDependent):
                 request=request,
                 result_type=result_type,
                 request_read_timeout_seconds=request_read_timeout_seconds,
+                metadata=metadata,
                 progress_callback=progress_callback,
-                metadata=None,
             )
-            logger.debug("send_request: response=", data=result.model_dump())
+            logger.debug(
+                "send_request: response=",
+                data=result.model_dump() if result is not None else "no response returned",
+            )
             return result
         except Exception as e:
             logger.error(f"send_request failed: {str(e)}")
             raise
 
-    async def send_notification(self, notification: SendNotificationT) -> None:
-        logger.debug("send_notification:", data=notification.model_dump())
-        try:
-            return await super().send_notification(notification)
-        except Exception as e:
-            logger.error("send_notification failed", data=e)
-            raise
-
-    async def _send_response(
-        self, request_id: RequestId, response: SendResultT | ErrorData
-    ) -> None:
-        logger.debug(
-            f"send_response: request_id={request_id}, response=",
-            data=response.model_dump(),
-        )
-        return await super()._send_response(request_id, response)
-
     async def _received_notification(self, notification: ServerNotification) -> None:
         """
         Can be overridden by subclasses to handle a notification without needing
@@ -189,17 +179,3 @@ class MCPAgentClientSession(ClientSession, ContextDependent):
             await self._tool_list_changed_callback(server_name)
         except Exception as e:
             logger.error(f"Error in tool list changed callback: {e}")
-
-    async def send_progress_notification(
-        self, progress_token: str | int, progress: float, total: float | None = None
-    ) -> None:
-        """
-        Sends a progress notification for a request that is currently being
-        processed.
-        """
-        logger.debug(
-            "send_progress_notification: progress_token={progress_token}, progress={progress}, total={total}"
-        )
-        return await super().send_progress_notification(
-            progress_token=progress_token, progress=progress, total=total
-        )

mcp_agent/mcp_server_registry.py

@@ -27,6 +27,7 @@ from mcp_agent.config import (
     get_settings,
 )
 from mcp_agent.logging.logger import get_logger
+from mcp_agent.mcp.hf_auth import add_hf_auth_header
 from mcp_agent.mcp.logger_textio import get_stderr_handler
 from mcp_agent.mcp.mcp_connection_manager import (
     MCPConnectionManager,
@@ -176,11 +177,14 @@ class ServerRegistry:
             if not config.url:
                 raise ValueError(f"URL is required for SSE transport: {server_name}")
 
+            # Apply HuggingFace authentication if appropriate
+            headers = add_hf_auth_header(config.url, config.headers)
+
             # Use sse_client to get the read and write streams
             async with _add_none_to_context(
                 sse_client(
                     config.url,
-                    config.headers,
+                    headers,
                     sse_read_timeout=config.read_transport_sse_timeout_seconds,
                 )
             ) as (read_stream, write_stream, _):
@@ -198,9 +202,12 @@ class ServerRegistry:
                         logger.debug(f"{server_name}: Closed session to server")
         elif config.transport == "http":
             if not config.url:
-                raise ValueError(f"URL is required for SSE transport: {server_name}")
+                raise ValueError(f"URL is required for HTTP transport: {server_name}")
+
+            # Apply HuggingFace authentication if appropriate
+            headers = add_hf_auth_header(config.url, config.headers)
 
-            async with streamablehttp_client(config.url, config.headers) as (
+            async with streamablehttp_client(config.url, headers) as (
                 read_stream,
                 write_stream,
                 _,