iddlZddlZddlZddlmZmZddlmZddlmZddl Z ddl m Z ddl m Z ddlmZmZddlmZmZdd lmZdd lmZdd lmZdd lmZej8eZGd deZGdde Z y)N)AnyOptional)AsyncExitStack)cached_property)MCPAuthException) MCPClientBase) OAuthParamsAsyncOAuth2Client)MemoryObjectReceiveStreamMemoryObjectSendStream) sse_client)SessionMessage) OAuth2Token) BaseModelcVeZdZUdZeed<dZeeee fed<dZ e ed<dZ e ed<y) SSEServerParamsz6Parameters required for a MCPClient with SSE transporturlNheaderstimeouti,sse_read_timeout) __name__ __module__ __qualname____doc__str__annotations__rrdictrrfloatrU/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/mistralai/extra/mcp/sse.pyrrs6@ H(,GXd38n %,GU$e$r!rc :eZdZUdZeeed<eed< ddedeedeedee ffd Z e d efd Z defd Z d ed eeeffdZded eded e fdZdZde d dfdZd eeeffdZd efdZded eeej4eefeeffdZxZS) MCPClientSSEaLMCP client that uses sse for communication. The client provides authentication for OAuth2 protocol following the current MCP authorization spec: https://modelcontextprotocol.io/specification/2025-03-26/basic/authorization. This is possibly going to change in the future since the protocol has ongoing discussions. _oauth_params _sse_paramsN sse_paramsname oauth_params auth_tokencPt||||_||_||_y)N)r()super__init__r&r% _auth_token)selfr'r(r)r* __class__s r"r-zMCPClientSSE.__init__/s- d#%4@2<r!returncL|jjjdS)Nz/sse)r&rrstrip)r/s r"base_urlzMCPClientSSE.base_url;s##**622r!cn|j"tjd|j||_y)z/Update the oauth params and client accordingly.Nz$Overriding current oauth params for )r%loggerwarning_name)r/r)s r"set_oauth_paramszMCPClientSSE.set_oauth_params?s.    ) NNA$**N O)r! redirect_urlcK|j tdtj|j}|j |jj j |\}}||fSw)zrzsc  %% 7DR%-.%   8 $%i%B=Br!