wehub-resource-sync
49b9bb6724
Deploy Docs / deploy-docs (push) Failing after 1s
Conformance Tests / client-conformance (push) Failing after 3s
Conformance Tests / server-conformance (push) Failing after 1s
GitHub Actions Security Analysis / zizmor (push) Failing after 1s
CI / checks (push) Failing after 59m20s
CI / all-green (push) Waiting to run
28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
from mcp_types import ElicitRequest, ElicitRequestFormParams, ElicitResult, InputRequiredResult
|
|
|
|
from mcp.server.mcpserver import Context, MCPServer, RequestStateSecurity
|
|
|
|
CONFIRM = ElicitRequest(
|
|
params=ElicitRequestFormParams(
|
|
message="Issue this refund?",
|
|
requested_schema={"type": "object", "properties": {"ok": {"type": "boolean"}}, "required": ["ok"]},
|
|
)
|
|
)
|
|
|
|
|
|
def make_server(key: str) -> MCPServer:
|
|
"""Every worker process: the same key, and the same name."""
|
|
mcp = MCPServer("billing", request_state_security=RequestStateSecurity(keys=[key]))
|
|
|
|
@mcp.tool()
|
|
async def refund(amount: int, ctx: Context) -> str | InputRequiredResult:
|
|
"""Refund an amount, once a human has confirmed it."""
|
|
if ctx.input_responses is None:
|
|
return InputRequiredResult(input_requests={"ok": CONFIRM}, request_state=f"refund:{amount}")
|
|
answer = (ctx.input_responses or {}).get("ok")
|
|
if not isinstance(answer, ElicitResult) or answer.action != "accept" or not (answer.content or {}).get("ok"):
|
|
return "refund cancelled"
|
|
return f"refunded ${amount}"
|
|
|
|
return mcp
|