Max
26 lines
949 B
Python
26 lines
949 B
Python
from mcp.server.mcpserver import Context, MCPServer
|
|
from mcp.server.mcpserver.prompts.base import UserMessage
|
|
from mcp.types import ElicitRequest, ElicitRequestFormParams, ElicitResult, InputRequiredResult
|
|
|
|
mcp = MCPServer("Briefing")
|
|
|
|
ASK_AUDIENCE = ElicitRequest(
|
|
params=ElicitRequestFormParams(
|
|
message="Who is the briefing for?",
|
|
requested_schema={
|
|
"type": "object",
|
|
"properties": {"audience": {"type": "string"}},
|
|
"required": ["audience"],
|
|
},
|
|
)
|
|
)
|
|
|
|
|
|
@mcp.prompt()
|
|
async def briefing(ctx: Context) -> list[UserMessage] | InputRequiredResult:
|
|
"""Draft a briefing tuned to its audience."""
|
|
answer = (ctx.input_responses or {}).get("audience")
|
|
if not isinstance(answer, ElicitResult) or answer.content is None:
|
|
return InputRequiredResult(input_requests={"audience": ASK_AUDIENCE})
|
|
return [UserMessage(f"Write a briefing for {answer.content['audience']}.")]
|