o i @s\ddlmZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZddlmZmZmZddlmZmZddlmZdd lmZdd lmZmZddlZddl Z ddl!Z!dd l"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)dd l!m*Z*m+Z+ddl,m-Z-m.Z.m/Z/ddl0m1Z1m2Z2ddl3m4Z4ddl5m6Z6m7Z7z:ddl8m9Z:ddl;mZ>ddl?m@Z@ddlAmBZBmCZCddlDmEZEmFZFddlGmHZHddlImJZJddlKmLZLWneMyZNzeMdeNdZN[NwwddlOmPZPmQZQmRZRmSZSmTZTdZUe!jVe!j+We!j+Xe!j+YdZZe[dZ\Gd d!d!e6eeZ]Gd"d#d#e]Z^Gd$d%d%e]Z_Gd&d'd'e_Z`e/d(Gd)d*d*e`ZaGd+d,d,e_ZbeceSjdBeeecefBefeBeeceSjdBeeecefBefeBBZg eeceeecefeeecefdBgeegfZh ee1eeheceeecefgeegfZi d=d1d2ZjGd3d4d4e&Zkd>d6d7Zld?d;d<ZmdS)@) annotationsN)ABCabstractmethod)Lock) AsyncIterator AwaitableCallableSequence)AbstractAsyncContextManagerAsyncExitStackasynccontextmanager)fieldreplace) timedelta)Path) AnnotatedAny)MemoryObjectReceiveStreamMemoryObjectSendStream) BaseModel DiscriminatorFieldTag) CoreSchema core_schema)Self assert_never deprecated) RunContextToolDefinition) model_request)AbstractToolset ToolsetTool)types) ClientSessionElicitationFnT LoggingFnT sse_client)StdioServerParameters stdio_client)GetSessionIdCallbackstreamablehttp_client)RequestContext)McpError)SessionMessageuPlease install the `mcp` package to use the MCP server, you can use the `mcp` optional group — `pip install "pydantic-ai-slim[mcp]"`)_mcp_utils exceptionsmessagesmodels) MCPServerMCPServerStdio MCPServerHTTP MCPServerSSEMCPServerStreamableHTTPload_mcp_servers)schemaz\$\{([^}:]+)(:-([^}]*))?\}cseZdZUdZded< ded< ded< ded < ded < d ed < d ed< ded< ded< dZded< ded<eddZded<ded<ded<ded<d ed!<d"ed#<d$ed%<    & '  (  ) dudd*dvd,d-Zd.d/Z e e dwd2d3Z e dxd4d5Zejdyd7d5Ze dzfd9d: Ze dzd;d<Ze d{d=d>Zd|d@dAZ d}d~dHdIZddOdPZddRdSZddVdWZddYdZZdd]d^Ze dd_d`ZddfdgZddkdlZddpdqZddsdtZZS)r6zuBase class for attaching agents to MCP servers. See for more information. str | None tool_prefixmcp_types.LoggingLevel | None log_levelLoggingFnT | None log_handlerfloattimeout read_timeoutProcessToolCallback | Noneprocess_tool_callboolallow_samplingmodels.Model | Nonesampling_modelint max_retriesNElicitationFnT | Noneelicitation_callback_idF)comparer _enter_lock_running_countzAsyncExitStack | None _exit_stackr%_clientz5MemoryObjectReceiveStream[SessionMessage | Exception] _read_streamz&MemoryObjectSendStream[SessionMessage] _write_streammcp_types.Implementation _server_info,Tr idr]c  CsR||_||_||_||_||_||_||_||_| |_| |_ | p!||_ | dSN) r>r@rBrDrErGrIrKrMrOrP __post_init__) selfr>r@rBrDrErGrIrKrMrOr]raU/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/mcp.py__init__~s  zMCPServer.__init__cCst|_d|_d|_dS)Nr)rrRrSrTr`rararbr_s zMCPServer.__post_init__returnsAsyncIterator[tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage]]]cCs td)z&Create the streams for the MCP server.z1MCP Server subclasses must implement this method.)NotImplementedErrorrdrararbclient_streamss zMCPServer.client_streamscCs|jSr^rPrdrararbr]sz MCPServer.idvaluecCs ||_dSr^rir`rjrararbr]s strcs|jrtjSt|Sr^)r]superlabelreprrd __class__rarbrnszMCPServer.labelcCsdS)Nz8Set the `tool_prefix` attribute to avoid name conflicts.rardrararbtool_name_conflict_hintz!MCPServer.tool_name_conflict_hintcCs*t|dddurtd|jjd|jS)zDAccess the information send by the MCP server during initialization.rYNzThe `z8.server_info` is only instantiated after initialization.)getattrAttributeErrorrq__name__rYrdrararb server_infos zMCPServer.server_infolist[mcp_types.Tool]c sT|4IdH|jIdH}WdIdH|jS1IdHs"wY|jS)zRetrieve tools that are currently active on the server. Note: - We don't cache tools as they might change. - We also don't subscribe to the server to avoid complexity. N)rU list_toolstools)r`resultrararbryszMCPServer.list_toolsnameargsdict[str, Any]metadatadict[str, Any] | None ToolResultc s`4IdHEz&jttjdtj|||r tjjdi|ndddtjIdH}Wnt yA}zt |j j d}~wwWdIdHn 1IdHsRwY|jrsd}|jrldd|jD}d|}t |pqd|j}rtd d |jDst|trt|d krd |vr|d S|Sfd d|jDIdH} t| d kr| dS| S)aKCall a tool on the server. Args: name: The name of the tool to call. args: The arguments to pass to the tool. metadata: Request-level metadata (optional) Returns: The result of the tool call. Raises: ModelRetry: If the tool call fails. Nz tools/call)r| arguments_meta)methodparamscSsg|] }t|tjr|jqSra) isinstance mcp_types TextContenttext.0partrararb sz.MCPServer.direct_call_tool.. zMCP tool call failedcss|] }t|tj VqdSr^)rrrrrararb s z-MCPServer.direct_call_tool..r r{csg|] }|IdHqSr^)_map_tool_result_partrrdrarbrsrra)rU send_requestr ClientRequestCallToolRequestCallToolRequestParams RequestParamsMetaCallToolResultr/r3 ModelRetryerrormessageisErrorcontentjoinstructuredContentanyrdictlen) r`r|r}rr{er text_parts structuredmappedrardrbdirect_call_toolsH  ( zMCPServer.direct_call_tool tool_argsctxRunContext[Any]toolToolsetTool[Any]csZ|jr||jd}t||d}|jdur$|||j||IdHS|||IdHS)N_) tool_name)r> removeprefixrrGr)r`r|rrrrararb call_tools  zMCPServer.call_tooldict[str, ToolsetTool[Any]]cs fddIdHDS)Nc sji|]1}jrjd|jn|jrt|j|j|j|jr(|jnd|j p-dddqS)rN)metar output_schema)r| descriptionparameters_json_schemar) r>r|tool_for_tool_defrr inputSchemarr model_dump outputSchema)rmcp_toolr|r`rarb "s  z'MCPServer.get_tools..)ry)r`rrarrb get_tools!s  zMCPServer.get_toolstool_defrcCst|||jtdS)N)toolsetrrMargs_validator)r#rMTOOL_SCHEMA_VALIDATOR)r`rrararbr3s zMCPServer.tool_for_tool_defrc s\|j4IdH|jdkrt4IdHo}||IdH\|_|_t|j|j|jr/|j nd|j |j t |j dd}||IdH|_t|j"|jIdH}|j|_|j}rf|j|IdHWdn1spwY||_WdIdHn 1IdHswY|jd7_WdIdH|S1IdHswY|S)a*Enter the MCP server context. This will initialize the connection to the server. If this server is an [`MCPServerStdio`][pydantic_ai.mcp.MCPServerStdio], the server will first be started as a subprocess. This is a no-op if the MCP server has already been entered. Nr)seconds) read_stream write_streamsampling_callbackrOlogging_callbackread_timeout_secondsr )rRrSr enter_async_contextrhrVrWr%rI_sampling_callbackrOrBrrErUanyio fail_afterrD initialize serverInforYr@set_logging_levelpop_allrT)r` exit_stackclientr{r@rararb __aenter__;s8    (zMCPServer.__aenter__r bool | Nonec s|jdkr td|j4IdH(|jd8_|jdkr.|jdur.|jIdHd|_WdIdHdS1IdHs?wYdS)Nrz5MCPServer.__aexit__ called more times than __aenter__r )rS ValueErrorrRrTaclose)r`r}rararb __aexit__[s .zMCPServer.__aexit__cCs t|jS)z#Check if the MCP server is running.)rHrSrdrararb is_runningds zMCPServer.is_runningcontext"RequestContext[ClientSession, Any]r$mcp_types.CreateMessageRequestParams3mcp_types.CreateMessageResult | mcp_types.ErrorDatac s|jdur tdt|}t}|j}r||d<|j}r%||d<|j}r.||d<t |j||dIdH}t j dt ||jj dS) zMCP sampling callback.NzSampling model is not set max_tokens temperaturestop_sequences)model_settings assistant)rolermodel)rKrr1map_from_mcp_paramsr5 ModelSettings maxTokensr stopSequencesr!rCreateMessageResultmap_from_model_response model_name) r`rr pai_messagesrrrrmodel_responserararbris"     zMCPServer._sampling_callbackrmcp_types.ContentBlock9str | messages.BinaryContent | dict[str, Any] | list[Any]cst|tjr"|j}|dr zt|WStyY|Sw|St|tjr4t j t |j |jdSt|tjrFt j t |j |jdSt|tjrT|j}|St|tjr}j|jIdH}t|jdkrs|jdSfdd|jDSt|dS)N)[{data media_typer rcsg|]}|qSra) _get_content)rresourcerdrarbrsz3MCPServer._map_tool_result_part..)rrrr startswith pydantic_core from_jsonr ImageContentr4 BinaryContentbase64 b64decodermimeType AudioContentEmbeddedResourcerr ResourceLinkrU read_resourceurircontentsr)r`rrrresource_resultrardrbrs8          zMCPServer._map_tool_result_partr?mcp_types.TextResourceContents | mcp_types.BlobResourceContentsstr | messages.BinaryContentcCsFt|tjr |jSt|tjrtjt|j |j pddSt |dS)Nzapplication/octet-streamr) rrTextResourceContentsrBlobResourceContentsr4rrrblobrr)r`rrararbrs   zMCPServer._get_contentobjectcCs"t|to|j|jko|j|jkSr^)rr6r]r>rkrararb__eq__"zMCPServer.__eq__) NNNrZr[NTNr N)r>r=r@r?rBrArDrCrErCrGrFrIrHrKrJrMrLrOrNr]r=rerf)rer=)rjr=rerl)rerX)rerxr^)r|rlr}r~rrrer) r|rlrr~rrrrrer)rrrer)rrrer)rer)r}rrer)rerH)rrrrrer)rrrer)rrrerrjrrerH) rv __module__ __qualname____doc____annotations__rOr rRrcr_rr rhpropertyr]setterrnrrrwryrrrrrrrrrrr  __classcell__rararprbr6@s         ;        " r6cseZdZUdZded< ded< ded< ded < d ed <d ed <ded<ded<ded<ded<ded<ded<ded<dZded<ddddddddd dd!ddd" d6fd$d%Zed7d+d,Ze d8d.d/Z d9d0d1Z d:fd4d5 Z Z S);r7aJRuns an MCP server in a subprocess and communicates with it over stdin/stdout. This class implements the stdio transport from the MCP specification. See for more information. !!! note Using this class as an async context manager will start the server as a subprocess when entering the context, and stop it when exiting the context. Example: ```python {py="3.10"} from pydantic_ai import Agent from pydantic_ai.mcp import MCPServerStdio server = MCPServerStdio( # (1)! 'uv', args=['run', 'mcp-run-python', 'stdio'], timeout=10 ) agent = Agent('openai:gpt-4o', toolsets=[server]) ``` 1. See [MCP Run Python](https://github.com/pydantic/mcp-run-python) for more information. rlcommand Sequence[str]r}dict[str, str] | Noneenvstr | Path | Nonecwdr=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMNrNrOrZr[Tr ) rrr>r@rBrDrErGrIrKrMrOr]r]c  s>||_||_||_||_tj||||| | | | | ||d dS)a]Build a new MCP server. Args: command: The command to run. args: The arguments to pass to the command. env: The environment variables to set in the subprocess. cwd: The working directory to use when spawning the process. tool_prefix: A prefix to add to all tools that are registered with the server. log_level: The log level to set when connecting to the server, if any. log_handler: A handler for logging messages from the server. timeout: The timeout in seconds to wait for the client to initialize. read_timeout: Maximum time in seconds to wait for new messages before timing out. process_tool_call: Hook to customize tool calling and optionally pass extra metadata. allow_sampling: Whether to allow MCP sampling through this client. sampling_model: The model to use for sampling. max_retries: The maximum number of times to retry a tool call. elicitation_callback: Callback function to handle elicitation requests from the server. id: An optional unique ID for the MCP server. An MCP server needs to have an ID in order to be used in a durable execution environment like Temporal, in which case the ID will be used to identify the server's activities within the workflow. r\N)rr}rrrmrc)r`rr}rrr>r@rBrDrErGrIrKrMrOr]rprarbrcs"& zMCPServerStdio.__init__rr__rerc CsRtddtttttttjtttdddS)NcStdi|SNra)r7dctrararb"z=MCPServerStdio.__get_pydantic_core_schema__..Frequired)rr}r)r no_info_after_validator_functiontyped_dict_schematyped_dict_field str_schema list_schema dict_schemaclsrrrararb__get_pydantic_core_schema__s z+MCPServerStdio.__get_pydantic_core_schema__rfc Cspt|jt|j|j|jd}t|d4IdH\}}||fVWdIdHdS1IdHs1wYdS)N)rr}rr)server)r*rlistr}rrr+)r`r-rrrararbrh/s   .zMCPServerStdio.client_streamscCsJd|jd|jg}|jr|d|j|jjdd|dS)Nzcommand=zargs=id=(, ))rr}r]appendrqrvrr` repr_argsrararb__repr__<s  zMCPServerStdio.__repr__rjrcsFt|o"t|to"|j|jko"|j|jko"|j|jko"|j|jkSr^)rmr rr7rr}rrrkrprarbr Es     zMCPServerStdio.__eq__)rrlr}rrrrrr>r=r@r?rBrArDrCrErCrGrFrIrHrKrJrMrLrOrNr]r=rrrrrerr r r )rvrrrrrOrc classmethodr,r rhr6r rrararprbr7sP  9    r7cseZdZUded< ded< ded< ded<d ed <d ed <d ed<d ed<ded<ded<ded<ded<dZded<dddddddddddddd d/fd$d%Zeed0d(d)Ze d1d+d,Z d2d-d.Z Z S)3_MCPServerHTTPrlurlrheadershttpx.AsyncClient | None http_clientr=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMNrNrOrZTr ) r;r=r]r>r@rBrDrErGrIrKrMrOrr] float | None_deprecated_kwargsrc  sd|vr| dur tdtjdtdd|d} t|| dur$d} ||_||_||_ t j ||||| | | | | ||d dS) aMBuild a new MCP server. Args: url: The URL of the endpoint on the MCP server. headers: Optional HTTP headers to be sent with each request to the endpoint. http_client: An `httpx.AsyncClient` to use with the endpoint. id: An optional unique ID for the MCP server. An MCP server needs to have an ID in order to be used in a durable execution environment like Temporal, in which case the ID will be used to identify the server's activities within the workflow. tool_prefix: A prefix to add to all tools that are registered with the server. log_level: The log level to set when connecting to the server, if any. log_handler: A handler for logging messages from the server. timeout: The timeout in seconds to wait for the client to initialize. read_timeout: Maximum time in seconds to wait for new messages before timing out. process_tool_call: Hook to customize tool calling and optionally pass extra metadata. allow_sampling: Whether to allow MCP sampling through this client. sampling_model: The model to use for sampling. max_retries: The maximum number of times to retry a tool call. elicitation_callback: Callback function to handle elicitation requests from the server. sse_read_timeoutNzE'read_timeout' and 'sse_read_timeout' cannot be set at the same time.z='sse_read_timeout' is deprecated, use 'read_timeout' instead.) stacklevelr[r\) TypeErrorwarningswarnDeprecationWarningpopr2validate_empty_kwargsr:r;r=rmrc)r`r:r;r=r]r>r@rBrDrErGrIrKrMrOr?rprarbrcs4%   z_MCPServerHTTP.__init__re+Callable[..., AbstractAsyncContextManager[tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage], GetSessionIdCallback],] | AbstractAsyncContextManager[tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage]]]]cCsdSr^rardrararb_transport_clientsz _MCPServerHTTP._transport_clientrfc sjr jr tdtjjjjjd}jdurR ddfd d }||d 4IdH^}}}||fVWdIdHdS1IdHsKwYdS|jd4IdH^}}}||fVWdIdHdS1IdHsvwYdS)Nz3`http_client` is mutually exclusive with `headers`.)r:rDr@r;rrDhttpx.Timeout | Noneauthhttpx.Auth | Nonerehttpx.AsyncClientcsjdusJjSr^)r=)r;rDrLrdrarbhttpx_client_factorysz;_MCPServerHTTP.client_streams..httpx_client_factory)rO)r;)NNN)r;rrDrKrLrMrerN) r=r;r functoolspartialrJr:rDrE)r`transport_client_partialrOrrrrardrbrhs.   . .z_MCPServerHTTP.client_streamscCs@d|jg}|jr|d|j|jjdd|dS)Nzurl=r/r0r1r2)r:r]r3rqrvrr4rararbr6s z_MCPServerHTTP.__repr__)r:rlr;rr=r<r]r=r>r=r@r?rBrArDrCrEr>rGrFrIrHrKrJrMrLrOrNr?r)rerIr r ) rvrrrrOrcrrrJr rhr6rrararprbr9PsJ   E &r9c<eZdZdZedddZed d Zdfdd ZZ S)r9aAn MCP server that connects over streamable HTTP connections. This class implements the SSE transport from the MCP specification. See for more information. !!! note Using this class as an async context manager will create a new pool of HTTP connections to connect to a server which should already be running. Example: ```python {py="3.10"} from pydantic_ai import Agent from pydantic_ai.mcp import MCPServerSSE server = MCPServerSSE('http://localhost:3001/sse') agent = Agent('openai:gpt-4o', toolsets=[server]) ``` rrrrerc C@tddttttjtttdddS)NcSrr)r9rrararbr !r!z;MCPServerSSE.__get_pydantic_core_schema__..Fr"r:r;rr$r%r&r'r)r*rararbr, z)MCPServerSSE.__get_pydantic_core_schema__cCtSr^r(rdrararbrJ,rszMCPServerSSE._transport_clientrjrrHc"t|ot|to|j|jkSr^)rmr rr9r:rkrprarbr 0r zMCPServerSSE.__eq__r7r  rvrrrr8r,rrJr rrararprbr9   r9zDThe `MCPServerHTTP` class is deprecated, use `MCPServerSSE` instead.c@seZdZdZdS)r8aAn MCP server that connects over HTTP using the old SSE transport. This class implements the SSE transport from the MCP specification. See for more information. !!! note Using this class as an async context manager will create a new pool of HTTP connections to connect to a server which should already be running. Example: ```python {py="3.10" test="skip"} from pydantic_ai import Agent from pydantic_ai.mcp import MCPServerHTTP server = MCPServerHTTP('http://localhost:3001/sse') agent = Agent('openai:gpt-4o', toolsets=[server]) ``` N)rvrrrrarararbr84sr8crS)r:aAn MCP server that connects over HTTP using the Streamable HTTP transport. This class implements the Streamable HTTP transport from the MCP specification. See for more information. !!! note Using this class as an async context manager will create a new pool of HTTP connections to connect to a server which should already be running. Example: ```python {py="3.10"} from pydantic_ai import Agent from pydantic_ai.mcp import MCPServerStreamableHTTP server = MCPServerStreamableHTTP('http://localhost:8000/mcp') agent = Agent('openai:gpt-4o', toolsets=[server]) ``` rrrrerc CrT)NcSrr)r:rrararbr ar!zFMCPServerStreamableHTTP.__get_pydantic_core_schema__..Fr"rUrVr*rararbr,^rWz4MCPServerStreamableHTTP.__get_pydantic_core_schema__cCrXr^)r-rdrararbrJlrsz)MCPServerStreamableHTTP._transport_clientrjrrHcrYr^)rmr rr:r:rkrprarbr pr zMCPServerStreamableHTTP.__eq__r7r rZrararprbr:Jr[r:rjr~rer=cCs"d|vr|ddr dSdSdS)Nr:z/ssessezstreamable-httpstdio)endswith)rjrararb_mcp_server_discriminators r_c@seZdZUdZded<dS)MCPServerConfigzConfiguration for MCP servers.zAnnotated[dict[str, Annotated[Annotated[MCPServerStdio, Tag('stdio')] | Annotated[MCPServerStreamableHTTP, Tag('streamable-http')] | Annotated[MCPServerSSE, Tag('sse')], Discriminator(_mcp_server_discriminator)]], Field(alias='mcpServers')] mcp_serversN)rvrrrrrarararbr`s  r`rcCs\t|trd dd}t||}|St|tr dd|DSt|tr,d d |DS|S) aRecursively expand environment variables in a JSON structure. Environment variables can be referenced using `${VAR_NAME}` syntax, or `${VAR_NAME:-default}` syntax to provide a default value if the variable is not set. Args: value: The value to expand (can be str, dict, list, or other JSON types). Returns: The value with all environment variables expanded. Raises: ValueError: If an environment variable is not defined and no default value is provided. match re.Match[str]rerlcSsZ|d}|ddu}|r|dnd}|tjvrtj|S|r%|p$dStd|d)Nr rAzEnvironment variable ${z} is not defined)grouposenvironr)rbvar_name has_default default_valuerararb replace_matchs   z'_expand_env_vars..replace_matchcSsi|] \}}|t|qSra_expand_env_vars)rkvrararbrsz$_expand_env_vars..cSsg|]}t|qSrarm)ritemrararbrsz$_expand_env_vars..N)rbrcrerl)rrl_ENV_VAR_PATTERNsubritemsr.)rjrlrararbrns     rn config_path str | Path=list[MCPServerStdio | MCPServerStreamableHTTP | MCPServerSSE]cCsrt|}|std|dt|}t|}t|}g}|j D]\}}||_ ||_ | |q'|S)aLoad MCP servers from a configuration file. Environment variables can be referenced in the configuration file using: - `${VAR_NAME}` syntax - expands to the value of VAR_NAME, raises error if not defined - `${VAR_NAME:-default}` syntax - expands to VAR_NAME if set, otherwise uses the default value Args: config_path: The path to the configuration file. Returns: A list of MCP servers. Raises: FileNotFoundError: If the configuration file does not exist. ValidationError: If the configuration file does not match the schema. ValueError: If an environment variable referenced in the configuration is not defined and no default value is provided. z Config file z not found)rexistsFileNotFoundErrorrr read_bytesrnr`model_validaterartr]r>r3)ru config_dataexpanded_config_dataconfigserversr|r-rararbr;s  r;)rjr~rer=)rjrrer)rurvrerw)n __future__rrrPrgrerDabcrrasynciorcollections.abcrrrr contextlibr r r dataclassesr rdatetimerpathlibrtypingrrrhttpxranyio.streams.memoryrrpydanticrrrrrrtyping_extensionsrrrpydantic_ai.toolsrrdirectr!toolsets.abstractr"r#mcpr$rmcp.client.sessionr%r&r'mcp.client.sser)mcp.client.stdior*r+mcp.client.streamable_httpr,r-mcp.shared.contextr.mcp.shared.exceptionsr/mcp.shared.messager0 ImportError _import_errorrer1r2r3r4r5__all__SchemaValidatorr)r' any_schemarcompilerrr6r7r9r9r8r:rlrrr.r CallToolFuncProcessToolCallbackr_r`rnr;rarararbs          t;*+  *   ,