@@ -47,7 +47,7 @@ from pydantic_ai import Agent
47
47
from pydantic_ai.mcp import MCPServerSSE
48
48
49
49
server = MCPServerSSE(url = ' http://localhost:3001/sse' ) # (1)!
50
- agent = Agent(' openai:gpt-4o' , mcp_servers = [server]) # (2)!
50
+ agent = Agent(' openai:gpt-4o' , toolsets = [server]) # (2)!
51
51
52
52
53
53
async def main ():
@@ -118,7 +118,7 @@ from pydantic_ai import Agent
118
118
from pydantic_ai.mcp import MCPServerStreamableHTTP
119
119
120
120
server = MCPServerStreamableHTTP(' http://localhost:8000/mcp' ) # (1)!
121
- agent = Agent(' openai:gpt-4o' , mcp_servers = [server]) # (2)!
121
+ agent = Agent(' openai:gpt-4o' , toolsets = [server]) # (2)!
122
122
123
123
async def main ():
124
124
async with agent.run_toolsets(): # (3)!
@@ -156,7 +156,7 @@ server = MCPServerStdio( # (1)!
156
156
' stdio' ,
157
157
]
158
158
)
159
- agent = Agent(' openai:gpt-4o' , mcp_servers = [server])
159
+ agent = Agent(' openai:gpt-4o' , toolsets = [server])
160
160
161
161
162
162
async def main ():
@@ -200,7 +200,7 @@ server = MCPServerStdio('python', ['mcp_server.py'], process_tool_call=process_t
200
200
agent = Agent(
201
201
model = TestModel(call_tools = [' echo_deps' ]),
202
202
deps_type = int ,
203
- mcp_servers = [server]
203
+ toolsets = [server]
204
204
)
205
205
206
206
@@ -243,7 +243,7 @@ calculator_server = MCPServerSSE(
243
243
# Both servers might have a tool named 'get_data', but they'll be exposed as:
244
244
# - 'weather_get_data'
245
245
# - 'calc_get_data'
246
- agent = Agent(' openai:gpt-4o' , mcp_servers = [weather_server, calculator_server])
246
+ agent = Agent(' openai:gpt-4o' , toolsets = [weather_server, calculator_server])
247
247
```
248
248
249
249
### Example with Stdio Server
@@ -273,7 +273,7 @@ js_server = MCPServerStdio(
273
273
tool_prefix = ' js' # Tools will be prefixed with 'js_'
274
274
)
275
275
276
- agent = Agent(' openai:gpt-4o' , mcp_servers = [python_server, js_server])
276
+ agent = Agent(' openai:gpt-4o' , toolsets = [python_server, js_server])
277
277
```
278
278
279
279
When the model interacts with these servers, it will see the prefixed tool names, but the prefixes will be automatically handled when making tool calls.
@@ -360,7 +360,7 @@ from pydantic_ai import Agent
360
360
from pydantic_ai.mcp import MCPServerStdio
361
361
362
362
server = MCPServerStdio(command = ' python' , args = [' generate_svg.py' ])
363
- agent = Agent(' openai:gpt-4o' , mcp_servers = [server])
363
+ agent = Agent(' openai:gpt-4o' , toolsets = [server])
364
364
365
365
366
366
async def main ():
0 commit comments