Max Isbey
1b74b06753
Cut comment and docstring volume roughly in half across src, tests, examples, and docs_src: removed comments that restate the adjacent code, leftover development narration, section banners, and self-evident Args/Returns blocks, and compressed the remaining docstrings to a Google-style summary line plus only the detail that earns its place. Kept (and tightened) the load-bearing content: Raises sections, deprecation and version-availability notes, spec/RFC/issue references, why-comments for non-obvious decisions, and all coverage pragmas. The generated mcp_types.v* wire modules are untouched.
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
"""Supply a canned sampling_callback and assert its text round-trips through the tool."""
|
|
|
|
from mcp_types import CreateMessageRequestParams, CreateMessageResult, TextContent
|
|
|
|
from mcp.client import Client, ClientRequestContext
|
|
from stories._harness import Target, run_client
|
|
|
|
|
|
async def on_sample(context: ClientRequestContext, params: CreateMessageRequestParams) -> CreateMessageResult:
|
|
# A real host would call its LLM provider here; a canned answer keeps the round-trip assertable.
|
|
return CreateMessageResult(
|
|
role="assistant",
|
|
content=TextContent(text="[canned summary]"),
|
|
model="stub-model",
|
|
stop_reason="endTurn",
|
|
)
|
|
|
|
|
|
async def main(target: Target, *, mode: str = "auto") -> None:
|
|
async with Client(target, mode=mode, sampling_callback=on_sample) as client:
|
|
result = await client.call_tool("summarize", {"text": "hello world"})
|
|
|
|
assert not result.is_error, result
|
|
assert isinstance(result.content[0], TextContent)
|
|
assert result.content[0].text == "[canned summary]", result.content[0].text
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_client(main)
|