샘플링 둘러보기

1from mcp.server.fastmcp import FastMCP, Context
2from mcp.types import SamplingMessage, TextContent
3
4mcp = FastMCP(name="Demo Server")
5
6
7@mcp.tool()
8async def summarize(text_to_summarize: str, ctx: Context):
9    prompt = f"""
10        Please summarize the following text:
11        {text_to_summarize}
12    """
13
14    result = await ctx.session.create_message(
15        messages=[
16            SamplingMessage(
17                role="user", content=TextContent(type="text", text=prompt)
18            )
19        ],
20        max_tokens=4000,
21        system_prompt="You are a helpful research assistant.",
22    )
23
24    if result.content.type == "text":
25        return result.content.text
26    else:
27        raise ValueError("Sampling failed")
28
29
30if __name__ == "__main__":
31    mcp.run(transport="stdio")
32