Marcelo Trylesinski
27 lines
774 B
Python
27 lines
774 B
Python
from mcp_types import SamplingMessage, TextContent
|
|
|
|
from mcp.server.mcpserver import Context, MCPServer
|
|
|
|
mcp = MCPServer(name="Sampling Example")
|
|
|
|
|
|
@mcp.tool()
|
|
async def generate_poem(topic: str, ctx: Context) -> str:
|
|
"""Generate a poem using LLM sampling."""
|
|
prompt = f"Write a short poem about {topic}"
|
|
|
|
result = await ctx.session.create_message( # pyright: ignore[reportDeprecated]
|
|
messages=[
|
|
SamplingMessage(
|
|
role="user",
|
|
content=TextContent(type="text", text=prompt),
|
|
)
|
|
],
|
|
max_tokens=100,
|
|
)
|
|
|
|
# Since we're not passing tools param, result.content is single content
|
|
if result.content.type == "text":
|
|
return result.content.text
|
|
return str(result.content)
|