o i#- @s@ddlmZddlZddlmZmZddlmZmZddl m Z ddl m Z ddl mZmZddlmZdd lmZmZz$dd lmZmZdd lmZdd lmZdd lmZmZddlm Z Wne!yqZ"ze!de"dZ"["wwddl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*e$Z+GdddeZ,Gddde,Z-Gddde,Z.dS)) annotationsN)ABCabstractmethod)AbstractAsyncContextManagerAsyncExitStack) timedelta)Path)AnyLiteral)urlparse)MemoryObjectReceiveStreamMemoryObjectSendStream) ClientSession stdio_client) sse_client)StdioServerParameters)GetSessionIdCallbackstreamablehttp_client)SessionMessagezThe 'mcp' package is required to run the MCP server integration but is not installed. To fix this, install the optional dependency: pip install 'livekit-agents[mcp]')RawFunctionTool ToolError function_toolget_function_infoget_raw_function_infois_function_toolis_raw_function_toolc@sdeZdZd"ddZed#dd Zd$d d Zd$d d Zd%ddZd&ddZ d$ddZ e d'dd Z d!S)( MCPServerclient_session_timeout_secondsfloatreturnNonecCs$d|_t|_||_d|_d|_dSNT)_clientr _exit_stack _read_timeout _cache_dirty _lk_tools)selfrr)X/var/www/html/livekit_bhavya/venv/lib/python3.10/site-packages/livekit/agents/llm/mcp.py__init__*s  zMCPServer.__init__boolcCs |jduSN)r#r(r)r)r* initialized2s zMCPServer.initializedcCs d|_dSr")r&r.r)r)r*invalidate_cache6s zMCPServer.invalidate_cachecsz:|j|IdH}|d|d}}|jt|||jr%t|jdnddIdH|_|jIdHd|_WdSt yJ| IdHw)Nrrseconds)read_timeout_secondsT) r$enter_async_contextclient_streamsrr%rr# initialize _initialized Exceptionaclose)r(streamsreceive_stream send_streamr)r)r*r69s&   zMCPServer.initialize list[MCPTool]cs^jdur tdjsjdurjSjIdH}fdd|jD}|_d_|S)NzMCPServer isn't initializedcs$g|]}|j|j|j|jqSr))_make_function_toolname description inputSchemameta).0toolr.r)r* Tsz(MCPServer.list_tools..F)r# RuntimeErrorr&r' list_toolstools)r(rHlk_toolsr)r.r*rGLs  zMCPServer.list_toolsr?strr@ str | None input_schemadict[str, Any]rBdict[str, Any] | NoneMCPToolcs4d fdd }||d}|r||d<t||d S) N raw_argumentsrMr r csjdur tdj|IdH}|jr&ddd|jD}t|t|jdkr4|jdSt|jdkrFt dd|jDStd d ) NzjTool invocation failed: internal service is unavailable. Please check that the MCPServer is still running. css(|]}t|dr |jnt|VqdS)textN)hasattrrRrJ)rCpartr)r)r* os  zFMCPServer._make_function_tool.._tool_called..rrcSsg|]}|qSr)) model_dump)rCitemr)r)r*rEyszGMCPServer._make_function_tool.._tool_called..zTool 'z^' completed without producing a result. This might indicate an issue with internal processing.) r#r call_toolisErrorjoincontentlenmodel_dump_jsonjsondumps)rP tool_result error_strr?r(r)r* _tool_calledds$   z3MCPServer._make_function_tool.._tool_called)r?r@ parametersrB) raw_schema)rPrMr r )r)r(r?r@rLrBrcrer)rbr*r>]s zMCPServer._make_function_toolcs4z|jIdHWd|_d|_dSd|_d|_wr-)r$r9r#r'r.r)r)r*r9s zMCPServer.acloseAbstractAsyncContextManager[tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage]] | tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage], GetSessionIdCallback]]cCsdSr-r)r.r)r)r*r5s zMCPServer.client_streamsN)rrr r!)r r,)r r!r r=) r?rJr@rKrLrMrBrNr rOr rf) __name__ __module__ __qualname__r+propertyr/r0r6rGr>r9rr5r)r)r)r*r)s       -rcsfeZdZdZ      d#d$fdd Zd%ddZd&ddZd'fdd Zd(dd Zd)d!d"Z Z S)* MCPServerHTTPa HTTP-based MCP server with configurable transport type and tool filtering. Args: url: The URL of the MCP server transport_type: Explicit transport type - "sse" or "streamable_http". If None, transport type is auto-detected from URL path: - URLs ending with 'sse' use Server-Sent Events (SSE) transport - URLs ending with 'mcp' use streamable HTTP transport - For other URLs, defaults to SSE transport for backward compatibility allowed_tools: Optional list of tool names to filter. If provided, only tools whose names are in this list will be available. If None, all tools from the server will be available. headers: Optional HTTP headers to include in requests timeout: Connection timeout in seconds (default: 5) sse_read_timeout: SSE read timeout in seconds (default: 300) client_session_timeout_seconds: Client session timeout in seconds (default: 5) Note: SSE transport is being deprecated in favor of streamable HTTP transport. See: https://github.com/modelcontextprotocol/modelcontextprotocol/pull/206 N,urlrJtransport_type(Literal['sse', 'streamable_http'] | None allowed_toolslist[str] | NoneheadersrNtimeoutrsse_read_timeoutrr r!csvtj|d||_||_||_||_|rt|nd|_|dur3|dvr,td|d|dk|_ dS| ||_ dS)Nr)ssestreamable_httpz8transport_type must be 'sse' or 'streamable_http', got ''rz) superr+rpru_timeout_sse_read_timeoutset_allowed_tools ValueError_use_streamable_http_should_use_streamable_http)r(rprqrsrurvrwr __class__r)r*r+s  zMCPServerHTTP.__init__r,cCs"t|}|jd}|dS)z Determine transport type based on URL path (for backward compatibility). Returns True for streamable HTTP if URL ends with 'mcp', False for SSE if URL ends with 'sse' or for backward compatibility. /z/mcp)r pathlowerrstripendswith)r(rp parsed_url path_lowerr)r)r*rs z)MCPServerHTTP._should_use_streamable_httprfcCsB|jrt|j|jt|jdt|jddSt|j|j|j|jdS)Nr1)rprurvrw)rrrprurr}r~rr.r)r)r*r5s   zMCPServerHTTP.client_streamsr=cs*tIdH}|jdur|S||S)zY List tools from the MCP server, filtered by allowed_tools if specified. N)r|rGr _filter_tools)r( all_toolsrr)r*rGs   zMCPServerHTTP.list_toolsrHcCs^|jdur|Sg}|D]!}t|rt|j}n t|r!t|j}nq ||jvr,||q |S)z= Filter tools by allowed_tools if specified. N)rrrr?rrappend)r(rHfiltered_toolsrD tool_namer)r)r*rs     zMCPServerHTTP._filter_toolscCs@|jrdnd}|jrdt|jnd}d|jd||dS)Nrzryz, allowed_tools=zMCPServerHTTP(url=z , transport=))rrlistrp)r(rq allowed_strr)r)r*__repr__szMCPServerHTTP.__repr__)NNNrnrorn)rprJrqrrrsrtrurNrvrrwrrrr r!)rprJr r,rhrg)rHr=r r=r rJ) rirjrk__doc__r+rr5rGrr __classcell__r)r)rr*rms   rmcs:eZdZ   ddfdd ZdddZdddZZS)MCPServerStdioNrncommandrJargs list[str]envdict[str, str] | Nonecwdstr | Path | Nonerrr r!cs*tj|d||_||_||_||_dS)Nrx)r|r+rrrr)r(rrrrrrr)r*r+&s  zMCPServerStdio.__init__AbstractAsyncContextManager[tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage]]]cCstt|j|j|j|jdS)N)rrrr)rrrrrrr.r)r)r*r54szMCPServerStdio.client_streamscCsd|jd|jd|jdS)NzMCPServerStdio(command=z, args=z, cwd=r)rrrr.r)r)r*r@szMCPServerStdio.__repr__)NNrn) rrJrrrrrrrrr r!)r rr)rirjrkr+r5rrr)r)rr*r%s  r)/ __future__rr^abcrr contextlibrrdatetimerpathlibrtypingr r urllib.parser anyio.streams.memoryr r mcprrmcp.client.ssermcp.client.stdiormcp.client.streamable_httprrmcp.shared.messager ImportErrore tool_contextrrrrrrrrOrrmrr)r)r)r*s:      $ x