wehub-resource-sync
49b9bb6724
Deploy Docs / deploy-docs (push) Failing after 1s
Conformance Tests / client-conformance (push) Failing after 3s
Conformance Tests / server-conformance (push) Failing after 1s
GitHub Actions Security Analysis / zizmor (push) Failing after 1s
CI / checks (push) Failing after 59m20s
CI / all-green (push) Waiting to run
36 lines
884 B
Python
36 lines
884 B
Python
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))
|
|
|
|
|
|
async def estimate_delivery(stock: Annotated[Stock, Resolve(check_stock)]) -> str:
|
|
return "tomorrow" if stock.copies > 0 else "in 2-3 weeks"
|
|
|
|
|
|
@mcp.tool()
|
|
async def order_book(
|
|
title: str,
|
|
stock: Annotated[Stock, Resolve(check_stock)],
|
|
delivery: Annotated[str, Resolve(estimate_delivery)],
|
|
) -> str:
|
|
"""Order a book from the shop."""
|
|
if stock.copies == 0:
|
|
return f"{title!r} is on backorder; it would arrive {delivery}."
|
|
return f"Ordered {title!r}; it arrives {delivery}."
|