Add auth to SseServerParameters and StreamableHttpParameters

Author: vipyne

src/mcp/client/session_group.py

@@ -16,7 +16,8 @@
 from typing import Any, TypeAlias
 
 import anyio
-from pydantic import BaseModel
+import httpx
+from pydantic import BaseModel, ConfigDict
 from typing_extensions import Self
 
 import mcp
@@ -30,6 +31,8 @@
 class SseServerParameters(BaseModel):
     """Parameters for intializing a sse_client."""
 
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
     # The endpoint URL.
     url: str
 
@@ -42,10 +45,15 @@ class SseServerParameters(BaseModel):
     # Timeout for SSE read operations.
     sse_read_timeout: float = 60 * 5
 
+    # Optional HTTPX authentication handler.
+    auth: httpx.Auth | None = None
+
 
 class StreamableHttpParameters(BaseModel):
     """Parameters for intializing a streamablehttp_client."""
 
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
     # The endpoint URL.
     url: str
 
@@ -61,6 +69,9 @@ class StreamableHttpParameters(BaseModel):
     # Close the client session when the transport closes.
     terminate_on_close: bool = True
 
+    # Optional HTTPX authentication handler.
+    auth: httpx.Auth | None = None
+
 
 ServerParameters: TypeAlias = StdioServerParameters | SseServerParameters | StreamableHttpParameters