iZ dZddlmZmZddlZgdZdZdZGddeZ dd e e e fdzd ejdzd ejdzd ejfd Zy)z@Utilities for creating standardized httpx AsyncClient instances.)AnyProtocolN)create_mcp_http_clientMCP_DEFAULT_TIMEOUTMCP_DEFAULT_SSE_READ_TIMEOUTg>@gr@c eZdZ ddeeefdzdej dzdejdzdejfdZ y)McpHttpClientFactoryNheaderstimeoutauthreturncy)N)selfr r r s W/root/aria/mcps/aria-brain/venv/lib/python3.12/site-packages/mcp/shared/_httpx_utils.py__call__zMcpHttpClientFactory.__call__s  NNN) __name__ __module__ __qualname__dictstrhttpxTimeoutAuth AsyncClientrrrrr r s]*.(,"&  c3h$& % jj4      rr r r r r cddi}|#tjtt|d<n||d<|||d<|||d<tjdi|S)a 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") follow_redirectsT)readr r r r)rrrrr)r r r kwargss rrrsp^ DF !MM*=D`ay#y#y v    &v &&rr)__doc__typingrrr__all__rrr rrrrrrrrrr%sF [$ 8 &*$("@' #s(^d "@' ]]T !@' **t @'  @'r