Pass InputRequiredResult through the MCPServer prompt and resource pipelines (#3020)
This commit is contained in:
parent
8f2c97b769
commit
8d0f928e40
1 changed files with 627 additions and 32 deletions
26
docs_src/mrtr/tutorial004.py
Normal file
26
docs_src/mrtr/tutorial004.py
Normal file
|
|
@ -0,0 +1,26 @@
|
|||
from mcp_types import ElicitRequest, ElicitRequestFormParams, ElicitResult, InputRequiredResult
|
||||
|
||||
from mcp.server.mcpserver import Context, MCPServer
|
||||
from mcp.server.mcpserver.prompts.base import UserMessage
|
||||
|
||||
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']}.")]
|
||||
Loading…
Reference in a new issue