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
33 lines
988 B
Python
33 lines
988 B
Python
from mcp_types import Completion, CompletionArgument, CompletionContext, PromptReference, ResourceTemplateReference
|
|
|
|
from mcp import Client
|
|
from mcp.server import MCPServer
|
|
|
|
mcp = MCPServer("Bookshop")
|
|
|
|
GENRES = ["fiction", "non-fiction", "poetry"]
|
|
|
|
|
|
@mcp.prompt()
|
|
def recommend(genre: str) -> str:
|
|
"""Ask for a recommendation in a genre."""
|
|
return f"Recommend one {genre} book from the catalog and say why."
|
|
|
|
|
|
@mcp.completion()
|
|
async def complete_genre(
|
|
ref: PromptReference | ResourceTemplateReference,
|
|
argument: CompletionArgument,
|
|
context: CompletionContext | None,
|
|
) -> Completion | None:
|
|
return Completion(values=[genre for genre in GENRES if genre.startswith(argument.value)])
|
|
|
|
|
|
async def main() -> None:
|
|
async with Client(mcp) as client:
|
|
result = await client.complete(
|
|
ref=PromptReference(type="ref/prompt", name="recommend"),
|
|
argument={"name": "genre", "value": "p"},
|
|
)
|
|
print(result.completion.values)
|