o h,@sddlZddlmZddlmZddlmZmZddlZddl Z ddl m Z ddl m Z mZddlmZddlmZddlmZdd lmZmZdd lmZeeZd ed efd dZedddedfd edeeefdBde de dede j!dBf ddZ"dS)N)asynccontextmanager)Any)urljoinurlparse) TaskStatus)MemoryObjectReceiveStreamMemoryObjectSendStream) aconnect_sse)SSEError)McpHttpClientFactorycreate_mcp_http_client)SessionMessageurlreturncCst|t|jS)N)rrpath)rrO/var/www/html/openai_agents/venv/lib/python3.10/site-packages/mcp/client/sse.pyremove_request_paramssri,headerstimeoutsse_read_timeouthttpx_client_factoryauthc std\}td\t4IdH}ztdt|||tj||dd4IdHytd4IdHYj tdtj fdt t ffd d }d t ffd d } ||IdH} td| || | z |fVW|jn|jwWdIdHn 1IdHswYWdIdHn 1IdHswYWIdHIdHnIdHIdHwWdIdHdS1IdHswYdS)a Client transport for SSE. `sse_read_timeout` determines how long (in seconds) the client will wait for a new event before disconnecting. All other HTTP operations are controlled by `timeout`. Args: url: The SSE endpoint URL. headers: Optional headers to include in requests. timeout: HTTP timeout for regular operations. sse_read_timeout: Timeout for SSE read operations. auth: Optional HTTPX authentication handler. rNzConnecting to SSE endpoint: )read)rrrGETzSSE connection established task_statusc szz2z3dHW}td|j|jdkrSt|j}td|t}t|}|j|jks?|j|jkrMd|}t |t || |qdkrzt j |j}td|Wnty}ztd|IdHWYd}~qd}~wwt|}|IdHq td|jq6Wn1ty} ztd | d} ~ wty}ztd |IdHWYd}~nd}~wwWIdHdSWIdHdSIdHw) NzReceived SSE event: endpointzReceived endpoint URL: z2Endpoint origin does not match connection origin: messagezReceived server message: zError parsing server messagezUnknown SSE event: zEncountered SSE exceptionzError in sse_reader) aiter_sseloggerdebugeventrdatarnetlocschemeerror ValueErrorstartedtypesJSONRPCMessagemodel_validate_json Exception exceptionsendr warningr aclose) rsse endpoint_url url_parsedendpoint_parsed error_msgrexcsession_messagesse_exc) event_sourceread_stream_writerrrr sse_readerEs^        $  +"zsse_client..sse_readerr2c szqzN4IdH;2z,3dHW}td|j||jjdddddIdH}|td|jq 6WdIdHn 1IdHsJwYWnty^tdYn wW IdHdSW IdHdS IdHw)NzSending client message: Tjson)by_aliasmode exclude_none)r<z"Client message sent successfully: zError in post_writer) r r!postr model_dumpraise_for_status status_coder,r-r0)r2r7response)client write_streamwrite_stream_readerrr post_writervs2 ( "zsse_client..post_writerz(Starting post writer with endpoint URL: )anyiocreate_memory_object_streamcreate_task_groupr r!rhttpxTimeoutr rDrBTASK_STATUS_IGNOREDrstrstart start_soon cancel_scopecancelr0) rrrrrr read_streamtgr;rHr2r)rEr9r:rrFrGr sse_clientsJ   1  *(Y.rV)#logging contextlibrtypingr urllib.parserrrIrL anyio.abcranyio.streams.memoryrr httpx_sser httpx_sse._exceptionsr mcp.typesr)mcp.shared._httpx_utilsr r mcp.shared.messager getLogger__name__r rOrdictfloatAuthrVrrrrsD