69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from typing import Any, cast
|
|
|
|
from wf_authoring import build_async_registry
|
|
from wf_core import RuntimeContext
|
|
from wf_mcp.capabilities import DiscoveredTool
|
|
from wf_mcp.models import AuthRecord, ConnectionConfig
|
|
from wf_mcp.runtime import ToolExecutor
|
|
from wf_mcp.sdk import ToolCallResult
|
|
from wf_mcp.workflow import wrap_discovered_tool
|
|
|
|
|
|
class RecordingAdapter:
|
|
"""Tiny adapter fake that records exactly what MCP payload would be sent."""
|
|
|
|
def __init__(self) -> None:
|
|
self.payloads: list[dict[str, Any]] = []
|
|
|
|
async def call_tool(
|
|
self,
|
|
connection: ConnectionConfig,
|
|
auth: AuthRecord | None,
|
|
tool_name: str,
|
|
payload: dict[str, Any],
|
|
) -> ToolCallResult:
|
|
self.payloads.append(payload)
|
|
return ToolCallResult(outcome="ok", output={})
|
|
|
|
|
|
def test_discovered_tool_wrapper_omits_unset_optional_arguments() -> None:
|
|
adapter = RecordingAdapter()
|
|
spec = wrap_discovered_tool(
|
|
connection=ConnectionConfig(
|
|
id="playwright.default",
|
|
server="playwright",
|
|
account="default",
|
|
),
|
|
auth=None,
|
|
executor=cast(ToolExecutor, adapter),
|
|
tool=DiscoveredTool(
|
|
name="browser_snapshot",
|
|
title=None,
|
|
description=None,
|
|
input_schema={
|
|
"type": "object",
|
|
"properties": {
|
|
"target": {"type": "string"},
|
|
"depth": {"type": "integer"},
|
|
},
|
|
},
|
|
output_schema={"type": "object", "properties": {}},
|
|
),
|
|
)
|
|
handler = build_async_registry(spec)[spec.name]
|
|
|
|
async def run_calls() -> None:
|
|
await handler({}, RuntimeContext(current_node_id="snapshot"))
|
|
await handler(
|
|
{"target": "main"},
|
|
RuntimeContext(current_node_id="snapshot"),
|
|
)
|
|
|
|
asyncio.run(run_calls())
|
|
|
|
assert adapter.payloads[0] == {}
|
|
assert adapter.payloads[1] == {"target": "main"}
|