Skip to content

Commit f3124c0

Browse files
committed
Lots of cleanup
1 parent 9dc684e commit f3124c0

31 files changed

+1281
-1106
lines changed

docs/mcp/client.md

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ from pydantic_ai import Agent
4747
from pydantic_ai.mcp import MCPServerSSE
4848

4949
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)!
5151

5252

5353
async def main():
@@ -118,7 +118,7 @@ from pydantic_ai import Agent
118118
from pydantic_ai.mcp import MCPServerStreamableHTTP
119119

120120
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)!
122122

123123
async def main():
124124
async with agent.run_toolsets(): # (3)!
@@ -156,7 +156,7 @@ server = MCPServerStdio( # (1)!
156156
'stdio',
157157
]
158158
)
159-
agent = Agent('openai:gpt-4o', mcp_servers=[server])
159+
agent = Agent('openai:gpt-4o', toolsets=[server])
160160

161161

162162
async def main():
@@ -200,7 +200,7 @@ server = MCPServerStdio('python', ['mcp_server.py'], process_tool_call=process_t
200200
agent = Agent(
201201
model=TestModel(call_tools=['echo_deps']),
202202
deps_type=int,
203-
mcp_servers=[server]
203+
toolsets=[server]
204204
)
205205

206206

@@ -243,7 +243,7 @@ calculator_server = MCPServerSSE(
243243
# Both servers might have a tool named 'get_data', but they'll be exposed as:
244244
# - 'weather_get_data'
245245
# - '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])
247247
```
248248

249249
### Example with Stdio Server
@@ -273,7 +273,7 @@ js_server = MCPServerStdio(
273273
tool_prefix='js' # Tools will be prefixed with 'js_'
274274
)
275275

276-
agent = Agent('openai:gpt-4o', mcp_servers=[python_server, js_server])
276+
agent = Agent('openai:gpt-4o', toolsets=[python_server, js_server])
277277
```
278278

279279
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
360360
from pydantic_ai.mcp import MCPServerStdio
361361

362362
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])
364364

365365

366366
async def main():

mcp-run-python/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ server = MCPServerStdio('deno',
5252
'jsr:@pydantic/mcp-run-python',
5353
'stdio',
5454
])
55-
agent = Agent('claude-3-5-haiku-latest', mcp_servers=[server])
55+
agent = Agent('claude-3-5-haiku-latest', toolsets=[server])
5656

5757

5858
async def main():

pydantic_ai_slim/pydantic_ai/_agent_graph.py

Lines changed: 104 additions & 101 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)