ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-22 21:44:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fix output schema test

Browse source
This commit is contained in:
Jeremiah Lowin 2025-06-28 08:14:31 -04:00
commit b10bab9423
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/server/test_server_interactions.py
View file

@ -947,12 +947,14 @@ class TestToolOutputSchema:
assert result.data == {"message": "Hello, world!"}
async def test_output_schema_false_full_handshake(self):
"""Test that output_schema=False works through full client/server handshake."""
"""Test that output_schema=False works through full client/server
handshake. We test this by returning a scalar, which requires an output
schema to serialize."""
mcp = FastMCP()
@mcp.tool(output_schema=False) # type: ignore[arg-type]
def simple_tool() -> dict[str, str]:
return {"message": "Hello from disabled schema"}
def simple_tool() -> int:
return 42
async with Client(mcp) as client:
# List tools and verify output schema is None
@ -964,9 +966,7 @@ class TestToolOutputSchema:
result = await client.call_tool("simple_tool", {})
assert result.structured_content is None
assert result.data is None
assert json.loads(result.content[0].text) == { # type: ignore[attr-defined]
"message": "Hello from disabled schema"
}
assert result.content[0].text == "42" # type: ignore[attr-defined]
async def test_output_schema_explicit_object_full_handshake(self):
"""Test explicit object output schema through full client/server handshake."""