FilesExpand file tree

v2.0.0rc1

/

tutorial006.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

48 lines (34 loc) · 1.34 KB

v2.0.0rc1

/

tutorial006.py

Copy path

Top

File metadata and controls

• Code
• Blame

48 lines (34 loc) · 1.34 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

from mcp_types import (

CallToolRequestParams,

CallToolResult,

ListToolsResult,

PaginatedRequestParams,

RequestParams,

TextContent,

Tool,

)

from pydantic import BaseModel

from mcp.server import Server, ServerRequestContext

SEARCH_BOOKS = Tool(

name="search_books",

description="Search the catalog by title or author.",

input_schema={

"type": "object",

"properties": {"query": {"type": "string"}, "limit": {"type": "integer"}},

"required": ["query", "limit"],

},

)

async def list_tools(ctx: ServerRequestContext, params: PaginatedRequestParams | None) -> ListToolsResult:

return ListToolsResult(tools=[SEARCH_BOOKS])

async def call_tool(ctx: ServerRequestContext, params: CallToolRequestParams) -> CallToolResult:

args = params.arguments or {}

text = f"Found 3 books matching {args['query']!r} (showing up to {args['limit']})."

return CallToolResult(content=[TextContent(type="text", text=text)])

class ReindexParams(RequestParams):

full: bool = False

class ReindexResult(BaseModel):

indexed: int

async def reindex(ctx: ServerRequestContext, params: ReindexParams) -> ReindexResult:

return ReindexResult(indexed=3)

server = Server("Bookshop", on_list_tools=list_tools, on_call_tool=call_tool)

server.add_request_handler("bookshop/reindex", ReindexParams, reindex)