84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
"""Utilities for creating standardized httpx AsyncClient instances."""
|
|
|
|
from typing import Any, Protocol
|
|
|
|
import httpx
|
|
|
|
__all__ = ["create_mcp_http_client"]
|
|
|
|
|
|
class McpHttpClientFactory(Protocol):
|
|
def __call__(
|
|
self,
|
|
headers: dict[str, str] | None = None,
|
|
timeout: httpx.Timeout | None = None,
|
|
auth: httpx.Auth | None = None,
|
|
) -> httpx.AsyncClient: ...
|
|
|
|
|
|
def create_mcp_http_client(
|
|
headers: dict[str, str] | None = None,
|
|
timeout: httpx.Timeout | None = None,
|
|
auth: httpx.Auth | None = None,
|
|
) -> httpx.AsyncClient:
|
|
"""Create a standardized httpx AsyncClient with MCP defaults.
|
|
|
|
This function provides common defaults used throughout the MCP codebase:
|
|
- follow_redirects=True (always enabled)
|
|
- Default timeout of 30 seconds if not specified
|
|
|
|
Args:
|
|
headers: Optional headers to include with all requests.
|
|
timeout: Request timeout as httpx.Timeout object.
|
|
Defaults to 30 seconds if not specified.
|
|
auth: Optional authentication handler.
|
|
|
|
Returns:
|
|
Configured httpx.AsyncClient instance with MCP defaults.
|
|
|
|
Note:
|
|
The returned AsyncClient must be used as a context manager to ensure
|
|
proper cleanup of connections.
|
|
|
|
Examples:
|
|
# Basic usage with MCP defaults
|
|
async with create_mcp_http_client() as client:
|
|
response = await client.get("https://api.example.com")
|
|
|
|
# With custom headers
|
|
headers = {"Authorization": "Bearer token"}
|
|
async with create_mcp_http_client(headers) as client:
|
|
response = await client.get("/endpoint")
|
|
|
|
# With both custom headers and timeout
|
|
timeout = httpx.Timeout(60.0, read=300.0)
|
|
async with create_mcp_http_client(headers, timeout) as client:
|
|
response = await client.get("/long-request")
|
|
|
|
# With authentication
|
|
from httpx import BasicAuth
|
|
auth = BasicAuth(username="user", password="pass")
|
|
async with create_mcp_http_client(headers, timeout, auth) as client:
|
|
response = await client.get("/protected-endpoint")
|
|
"""
|
|
# Set MCP defaults
|
|
kwargs: dict[str, Any] = {
|
|
"follow_redirects": True,
|
|
}
|
|
|
|
# Handle timeout
|
|
if timeout is None:
|
|
kwargs["timeout"] = httpx.Timeout(30.0)
|
|
else:
|
|
kwargs["timeout"] = timeout
|
|
|
|
# Handle headers
|
|
if headers is not None:
|
|
kwargs["headers"] = headers
|
|
|
|
# Handle authentication
|
|
if auth is not None:
|
|
kwargs["auth"] = auth
|
|
|
|
return httpx.AsyncClient(**kwargs)
|