Max
22 lines
682 B
Python
22 lines
682 B
Python
from typing import Annotated
|
|
|
|
from mcp.server import MCPServer
|
|
from mcp.server.mcpserver import Resolve, Sample
|
|
from mcp.types import CreateMessageResult, SamplingMessage, TextContent
|
|
|
|
mcp = MCPServer("Bookshop")
|
|
|
|
|
|
def draft_blurb(title: str) -> Sample:
|
|
prompt = f"Write a one-sentence blurb for the book {title!r}."
|
|
return Sample(
|
|
[SamplingMessage(role="user", content=TextContent(type="text", text=prompt))],
|
|
max_tokens=60,
|
|
)
|
|
|
|
|
|
@mcp.tool()
|
|
async def blurb(title: str, draft: Annotated[CreateMessageResult, Resolve(draft_blurb)]) -> str:
|
|
"""Draft a blurb for a book."""
|
|
return draft.content.text if draft.content.type == "text" else "No blurb."
|