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,39 @@
|
||||
"""List prompts, autocomplete an argument, then render both prompts."""
|
||||
|
||||
from mcp_types import PromptReference, TextContent
|
||||
|
||||
from mcp.client import Client
|
||||
from stories._harness import Target, run_client
|
||||
|
||||
|
||||
async def main(target: Target, *, mode: str = "auto") -> None:
|
||||
async with Client(target, mode=mode) as client:
|
||||
listed = await client.list_prompts()
|
||||
by_name = {p.name: p for p in listed.prompts}
|
||||
assert set(by_name) == {"greet", "code_review"}
|
||||
assert by_name["greet"].arguments is not None
|
||||
assert [a.name for a in by_name["greet"].arguments] == ["name"]
|
||||
assert by_name["greet"].arguments[0].required is True
|
||||
assert by_name["code_review"].title == "Code Review"
|
||||
|
||||
completion = await client.complete(
|
||||
PromptReference(name="code_review"),
|
||||
argument={"name": "language", "value": "py"},
|
||||
)
|
||||
assert completion.completion.values == ["python", "pytorch"], completion
|
||||
|
||||
greeted = await client.get_prompt("greet", {"name": "Ada"})
|
||||
assert len(greeted.messages) == 1
|
||||
assert greeted.messages[0].role == "user"
|
||||
assert isinstance(greeted.messages[0].content, TextContent)
|
||||
assert "Ada" in greeted.messages[0].content.text
|
||||
|
||||
reviewed = await client.get_prompt("code_review", {"language": "rust", "code": "fn main() {}"})
|
||||
assert [m.role for m in reviewed.messages] == ["user", "assistant"]
|
||||
first = reviewed.messages[0].content
|
||||
assert isinstance(first, TextContent)
|
||||
assert "rust" in first.text and "fn main() {}" in first.text
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
run_client(main)
|
||||
Reference in New Issue
Block a user