python-sdk/docs_src/client_callbacks/tutorial003.py at main · stophobia/python-sdk · GitHub

31 lines (23 loc) · 1.02 KB

from pydantic import BaseModel

from mcp.server import MCPServer

from mcp.server.mcpserver import Context

from mcp.types import ClientCapabilities, ElicitationCapability, RootsCapability, SamplingCapability

mcp = MCPServer("Library")

class CardHolder(BaseModel):

name: str

@mcp.tool()

async def issue_card(ctx: Context) -> str:

"""Issue a new library card."""

answer = await ctx.elicit("What name should go on the card?", schema=CardHolder)

if answer.action == "accept":

return f"Card issued to {answer.data.name}."

return "No card issued."

@mcp.tool()

def client_features(ctx: Context) -> list[str]:

"""Which optional features the connected client declared."""

declared = {

"elicitation": ClientCapabilities(elicitation=ElicitationCapability()),

"sampling": ClientCapabilities(sampling=SamplingCapability()),

"roots": ClientCapabilities(roots=RootsCapability()),

}

return [name for name, capability in declared.items() if ctx.session.check_client_capability(capability)]

Expand file tree