~h j d dl mZ d dlZd dlmZ d dlmZmZ d dlm Z dgZ
erd dl m Z G d d Z
y) ) annotationsN)
TracebackType)
TYPE_CHECKINGAny)Tool MCPClient)StdioServerParametersc f e Zd ZdZ d d
dZd Z d d dZd
dZd
dZ d dZ y)r a_ Manages the connection to an MCP server and make its tools available to SmolAgents.
Note: tools can only be accessed after the connection has been started with the
`connect()` method, done during the init. If you don't use the context manager
we strongly encourage to use "try ... finally" to ensure the connection is cleaned up.
Args:
server_parameters (StdioServerParameters | dict[str, Any] | list[StdioServerParameters | dict[str, Any]]):
Configuration parameters to connect to the MCP server. Can be a list if you want to connect multiple MCPs at once.
- An instance of `mcp.StdioServerParameters` for connecting a Stdio MCP server via standard input/output using a subprocess.
- A `dict` with at least:
- "url": URL of the server.
- "transport": Transport protocol to use, one of:
- "streamable-http": (recommended) Streamable HTTP transport.
- "sse": Legacy HTTP+SSE transport (deprecated).
If "transport" is omitted, the legacy "sse" transport is assumed (a deprecation warning will be issued).
The HTTP+SSE transport is deprecated and future behavior will default to the Streamable HTTP transport.
Please pass explicitly the "transport" key.
adapter_kwargs (dict[str, Any], optional):
Additional keyword arguments to be passed directly to `MCPAdapt`.
Example:
```python
# fully managed context manager + stdio
with MCPClient(...) as tools:
# tools are now available
# context manager + Streamable HTTP transport:
with MCPClient({"url": "http://localhost:8000/mcp", "transport": "streamable-http"}) as tools:
# tools are now available
# manually manage the connection via the mcp_client object:
try:
mcp_client = MCPClient(...)
tools = mcp_client.get_tools()
# use your tools here.
finally:
mcp_client.disconnect()
```
Nc ^ ddl m} ddlm} t |t rG|j d }|!t j dt d}||d< |dvrt d | d
|xs i } || | fi || _ d | _
| j y # t $ r t d w xY w) Nr )MCPAdapt)SmolAgentsAdapterzLPlease install 'mcp' extra to use MCPClient: `pip install 'smolagents[mcp]'` transporta Passing a dict as server_parameters without specifying the 'transport' key is deprecated. For now, it defaults to the legacy 'sse' (HTTP+SSE) transport, but this default will change to 'streamable-http' in version 1.20. Please add the 'transport' key explicitly. sse> streamable-httpr zUnsupported transport: z7. Supported transports are 'streamable-http' and 'sse'.)
mcpadapt.corer mcpadapt.smolagents_adapterr
ModuleNotFoundError
isinstancedictgetwarningswarn
FutureWarning
ValueError_adapter_toolsconnect)selfserver_parametersadapter_kwargsr r
r s Q/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/smolagents/mcp_client.py__init__zMCPClient.__init__R s
v . E ' .)--k:I
h " " 1:!+. :: -i[8op (-2 !24E4GZ>Z
)- ) # v%&tu u vs B B,c B | j j | _ y)z3Connect to the MCP server and initialize the tools.N)r __enter__r r s r! r zMCPClient.connectp s "&--"9"9"; c > | j j ||| y)z Disconnect from the MCP serverN)r __exit__r exc_type exc_value
exc_tracebacks r!
disconnectzMCPClient.disconnectt s
xMBr& c H | j t d | j S )a The SmolAgents tools available from the MCP server.
Note: for now, this always returns the tools available at the creation of the session,
but it will in a future release return also new tools available from the MCP server if
any at call time.
Raises:
ValueError: If the MCP server tools is None (usually assuming the server is not started).
Returns:
list[Tool]: The SmolAgents tools available from the MCP server.
zbCouldn't retrieve tools from MCP server, run `mcp_client.connect()` first before accessing `tools`)r r r% s r! get_toolszMCPClient.get_tools} s* ;; t
{{r& c | j S )zConnect to the MCP server and return the tools directly.
Note that because of the `.connect` in the init, the mcp_client
is already connected at this point.
)r r% s r! r$ zMCPClient.__enter__ s {{r& c * | j ||| y)z Disconnect from the MCP server.N)r- r) s r! r( zMCPClient.__exit__ s
)];r& )N)r zY'StdioServerParameters' | dict[str, Any] | list['StdioServerParameters' | dict[str, Any]]r zdict[str, Any] | None)NNN)r* ztype[BaseException] | Noner+ zBaseException | Noner, zTracebackType | None)returnz
list[Tool])
__name__
__module__ __qualname____doc__r" r r- r/ r$ r( r& r! r r ! s .f 15 t . << 04*..2 C,C (C , C&<,< (< , r= s1 $ # % ! -3~< ~