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
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from typing import Any
|
|
|
|
from mcp_types import ListResourcesResult, PaginatedRequestParams, Resource
|
|
|
|
from mcp import Client
|
|
from mcp.server import Server, ServerRequestContext
|
|
|
|
BOOKS = [f"book-{n}" for n in range(1, 101)]
|
|
|
|
PAGE_SIZE = 10
|
|
|
|
|
|
async def list_books(ctx: ServerRequestContext[Any], params: PaginatedRequestParams | None) -> ListResourcesResult:
|
|
start = 0 if params is None or params.cursor is None else int(params.cursor)
|
|
end = start + PAGE_SIZE
|
|
page = [Resource(uri=f"books://catalog/{name}", name=name) for name in BOOKS[start:end]]
|
|
next_cursor = str(end) if end < len(BOOKS) else None
|
|
return ListResourcesResult(resources=page, next_cursor=next_cursor)
|
|
|
|
|
|
server = Server("Bookshop", on_list_resources=list_books)
|
|
|
|
|
|
async def main() -> None:
|
|
async with Client(server) as client:
|
|
resources: list[Resource] = []
|
|
cursor: str | None = None
|
|
while True:
|
|
page = await client.list_resources(cursor=cursor)
|
|
resources.extend(page.resources)
|
|
if page.next_cursor is None:
|
|
break
|
|
cursor = page.next_cursor
|
|
print(f"{len(resources)} resources")
|