Add resolver dependency injection for MCPServer tools (#2969)
Co-authored-by: Max Isbey <224885523+maxisbey@users.noreply.github.com>
This commit is contained in:
parent
4b519782f1
commit
f664db8952
1 changed files with 1844 additions and 15 deletions
27
docs_src/dependencies/tutorial001.py
Normal file
27
docs_src/dependencies/tutorial001.py
Normal file
|
|
@ -0,0 +1,27 @@
|
|||
from typing import Annotated
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
from mcp.server import MCPServer
|
||||
from mcp.server.mcpserver import Resolve
|
||||
|
||||
mcp = MCPServer("Bookshop")
|
||||
|
||||
INVENTORY = {"Dune": 7, "Neuromancer": 0}
|
||||
|
||||
|
||||
class Stock(BaseModel):
|
||||
title: str
|
||||
copies: int
|
||||
|
||||
|
||||
async def check_stock(title: str) -> Stock:
|
||||
return Stock(title=title, copies=INVENTORY.get(title, 0))
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def reserve_book(title: str, stock: Annotated[Stock, Resolve(check_stock)]) -> str:
|
||||
"""Reserve a copy of a book."""
|
||||
if stock.copies == 0:
|
||||
return f"{title!r} is out of stock."
|
||||
return f"Reserved {title!r} ({stock.copies - 1} copies left)."
|
||||
Loading…
Reference in a new issue