chore: import upstream snapshot with attribution
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) Has been cancelled
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) Has been cancelled
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
"""Auto-answer form and URL elicitations and assert the tool result reflects them."""
|
||||
|
||||
import mcp_types as types
|
||||
|
||||
from mcp.client import Client, ClientRequestContext
|
||||
from stories._harness import Target, run_client
|
||||
|
||||
|
||||
async def on_elicit(context: ClientRequestContext, params: types.ElicitRequestParams) -> types.ElicitResult:
|
||||
if isinstance(params, types.ElicitRequestURLParams):
|
||||
# A real client would ask consent and open params.url in a browser, returning
|
||||
# `accept` right away; the server's notifications/elicitation/complete arrives
|
||||
# afterward (once the out-of-band flow finishes) for the client to correlate.
|
||||
assert params.url.startswith("https://example.com/")
|
||||
return types.ElicitResult(action="accept")
|
||||
assert "username" in params.requested_schema["properties"]
|
||||
return types.ElicitResult(action="accept", content={"username": "alice", "plan": "pro"})
|
||||
|
||||
|
||||
async def main(target: Target, *, mode: str = "auto") -> None:
|
||||
async with Client(target, mode=mode, elicitation_callback=on_elicit) as client:
|
||||
registered = await client.call_tool("register_user", {})
|
||||
assert isinstance(registered.content[0], types.TextContent)
|
||||
assert registered.content[0].text == "registered alice (plan: pro)", registered
|
||||
|
||||
linked = await client.call_tool("link_account", {"provider": "github"})
|
||||
assert isinstance(linked.content[0], types.TextContent)
|
||||
assert linked.content[0].text == "linked github", linked
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
run_client(main)
|
||||
Reference in New Issue
Block a user