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) Has been cancelled
58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
import mcp_types as types
|
|
import pytest
|
|
|
|
from mcp import Client
|
|
from mcp.server.mcpserver import Context, MCPServer
|
|
from mcp.server.mcpserver.tools.base import Tool
|
|
from mcp.shared.exceptions import MCPError
|
|
|
|
|
|
def test_context_detected_in_union_annotation():
|
|
def my_tool(x: int, ctx: Context | None) -> str:
|
|
raise NotImplementedError
|
|
|
|
tool = Tool.from_function(my_tool)
|
|
assert tool.context_kwarg == "ctx"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_mcperror_raised_from_a_tool_surfaces_as_a_top_level_jsonrpc_error_with_code_and_data_intact():
|
|
"""SDK-defined: ``MCPError`` carries JSON-RPC ``ErrorData(code, message, data)``
|
|
and means "respond with a protocol error". The tool wrapper re-raises it so
|
|
the kernel writes a top-level JSON-RPC error - ``code`` and ``data`` survive
|
|
the round-trip rather than being flattened into ``CallToolResult(isError=True)``."""
|
|
mcp = MCPServer(name="srv")
|
|
|
|
@mcp.tool()
|
|
async def needs_sampling() -> str:
|
|
raise MCPError(
|
|
types.MISSING_REQUIRED_CLIENT_CAPABILITY,
|
|
"sampling capability required",
|
|
data={"requiredCapabilities": ["sampling"]},
|
|
)
|
|
|
|
async with Client(mcp) as client:
|
|
with pytest.raises(MCPError) as exc_info:
|
|
await client.call_tool("needs_sampling", {})
|
|
|
|
assert exc_info.value.error.code == types.MISSING_REQUIRED_CLIENT_CAPABILITY
|
|
assert exc_info.value.error.data == {"requiredCapabilities": ["sampling"]}
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_non_mcperror_exception_raised_from_a_tool_is_wrapped_as_an_is_error_result():
|
|
"""SDK-defined: ordinary exceptions from a tool body are execution failures
|
|
the LLM should see, so they become ``CallToolResult(isError=True)`` rather
|
|
than a protocol-level JSON-RPC error. Pins the other arm of the same branch."""
|
|
mcp = MCPServer(name="srv")
|
|
|
|
@mcp.tool()
|
|
async def boom() -> str:
|
|
raise RuntimeError("execution failure")
|
|
|
|
async with Client(mcp) as client:
|
|
result = await client.call_tool("boom", {})
|
|
|
|
assert isinstance(result, types.CallToolResult)
|
|
assert result.is_error is True
|