lda.chat/lda-wf
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: expose capability step updates to mcp

Browse Source
This commit is contained in:
lda
2026-07-27 02:19:26 +07:00 Verified
parent bf8be6b3dc
commit a0d76fdfb2
7 changed files with 267 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/wf_mcp/server/test_config.py
+10
View File
@@ -62,6 +62,7 @@ def test_server_exposes_upstream_admin_and_workflow_tools() -> None:
assert "wf.workflow.set_workflow_output_map" in names
assert "wf.workflow.bind" in names
assert "wf.workflow.add_step_from_capability" in names
assert "wf.workflow.update_capability_step" in names
assert "wf.workflow.remove_draft_route" in names
assert "wf.workflow.remove_draft_step" in names
assert "wf.workflow.remove_draft_binding" in names
@@ -169,6 +170,15 @@ def test_server_exposes_upstream_admin_and_workflow_tools() -> None:
add_step_request = add_step_schema["properties"]["request"]
assert "capability_name" in add_step_request["properties"]
assert "bind_outputs" in add_step_request["properties"]
assert "input_bindings" in add_step_request["properties"]
assert "desc" in add_step_request["properties"]
assert "retry" in add_step_request["properties"]
assert "timeout_seconds" in add_step_request["properties"]
update_step_schema = tools_by_name[
"wf.workflow.update_capability_step"
].inputSchema
update_step_request = update_step_schema["properties"]["request"]
assert "update" in update_step_request["properties"]
from_capability_output = tools_by_name[
"wf.workflow.create_draft_workspace_from_capability"
].outputSchema
tests/wf_mcp/server/test_tools.py
+110
View File
@@ -55,6 +55,7 @@ async def test_server_search_mode_pins_stable_control_and_workflow_tools() -> No
assert "wf.workflow.set_workflow_output_bindings" in names
assert "wf.workflow.set_workflow_output_map" in names
assert "wf.workflow.bind" in names
assert "wf.workflow.update_capability_step" in names
assert "wf.workflow.remove_draft_route" in names
assert "wf.workflow.remove_draft_step" in names
assert "wf.workflow.remove_draft_binding" in names
@@ -208,6 +209,115 @@ async def test_registered_workflow_output_bindings_tool_preserves_union_order(
]
@pytest.mark.asyncio
async def test_registered_capability_tools_delegate_presence_aware_requests(
tmp_path: Path,
monkeypatch: pytest.MonkeyPatch,
) -> None:
class RecordingWorkflowHandler:
def __init__(self) -> None:
self.update_calls: list[dict[str, Any]] = []
self.add_calls: list[dict[str, Any]] = []
async def update_capability_step(self, **kwargs: Any) -> dict[str, Any]:
self.update_calls.append(kwargs)
return {
"workspace_id": kwargs["workspace_id"],
"revision": kwargs["revision"] + 1,
"status": "valid",
"diagnostics": [],
"summary": {},
}
async def add_step_from_capability(self, **kwargs: Any) -> dict[str, Any]:
self.add_calls.append(kwargs)
return {
"workspace_id": kwargs["workspace_id"],
"revision": kwargs["revision"] + 1,
"status": "valid",
"diagnostics": [],
"summary": {},
}
recorder = RecordingWorkflowHandler()
monkeypatch.setattr(
"wf_mcp.workflow_surface.tools.WorkflowApi",
lambda _context: recorder,
)
service = WfMcpService(
store=FileStore(tmp_path / "capability_tool_store"),
artifact_store=FileWorkflowArtifactStore(
tmp_path / "capability_tool_artifacts"
),
draft_workspace_store=FileDraftWorkspaceStore(
tmp_path / "capability_tool_drafts"
),
)
server = FastMCP("capability-update-tool-test")
register_workflow_tools(server, service)
async with Client(FastMCPTransport(server)) as client:
updated = await client.call_tool(
"wf.workflow.update_capability_step",
{
"request": {
"workspace_id": "report",
"revision": 4,
"step_id": "publish",
"update": {
"desc": None,
"input": [
{
"value": "markdown",
"target": "request.format",
}
],
},
}
},
)
added = await client.call_tool(
"wf.workflow.add_step_from_capability",
{
"request": {
"workspace_id": "report",
"revision": 5,
"step_id": "archive",
"capability_name": "local.report.archive",
"input_bindings": [
{
"path": "state.report.title",
"target": "request.title",
},
{"value": "pdf", "target": "request.format"},
],
"desc": "Archive report",
"retry": 0,
"timeout_seconds": 20,
}
},
)
assert structured(updated)["revision"] == 5
assert structured(added)["revision"] == 6
assert len(recorder.update_calls) == 1
update = recorder.update_calls[0]["update"]
assert update.model_fields_set == {"desc", "input"}
assert isinstance(update.input[0], InputValueBinding)
assert len(recorder.add_calls) == 1
add_call = recorder.add_calls[0]
assert add_call["input_map"] is None
assert [
binding.model_dump(mode="json") for binding in add_call["input_bindings"]
] == [
{"path": "state.report.title", "target": "request.title"},
{"value": "pdf", "target": "request.format"},
]
assert add_call["desc"] == "Archive report"
assert add_call["retry"] == 0
assert add_call["timeout_seconds"] == 20
@pytest.mark.asyncio
async def test_server_search_mode_can_use_safe_tool_names() -> None:
config = server_config()
tests/wf_mcp/workflow_surface/test_drafts.py
+87
View File
@@ -18,10 +18,12 @@ from wf_mcp.models import ConnectionConfig
from wf_mcp.storage import FileStore
from wf_mcp.workflow_surface import WorkflowSurfaceHandlers
from wf_mcp.workflow_surface.models import (
AddStepFromCapabilityRequest,
CreateMinimalDraftWorkspaceRequest,
SetStepInputBindingsRequest,
SetStepOutputBindingsRequest,
SetWorkflowOutputBindingsRequest,
UpdateCapabilityStepRequest,
)
from ..test_support import echo_tool
@@ -33,6 +35,91 @@ from .conftest import (
)
def test_update_capability_step_request_preserves_field_presence_and_binding_type() -> (
None
):
request = UpdateCapabilityStepRequest.model_validate(
{
"workspace_id": "report",
"revision": 4,
"step_id": "publish",
"update": {
"desc": None,
"input": [
{"value": "markdown", "target": "request.format"},
],
},
}
)
assert request.update.model_fields_set == {"desc", "input"}
assert request.update.desc is None
assert request.update.input is not None
assert isinstance(request.update.input[0], InputValueBinding)
@pytest.mark.parametrize(
"update",
[
{},
{"input": None},
{"retry": -1},
{"timeout_seconds": 0},
{"unknown": "field"},
],
)
def test_update_capability_step_request_rejects_invalid_patch(update) -> None:
with pytest.raises(ValidationError):
UpdateCapabilityStepRequest.model_validate(
{
"workspace_id": "report",
"revision": 4,
"step_id": "publish",
"update": update,
}
)
def test_add_step_from_capability_request_preserves_creation_parity() -> None:
request = AddStepFromCapabilityRequest.model_validate(
{
"workspace_id": "report",
"revision": 3,
"step_id": "publish",
"capability_name": "local.report.publish",
"input_bindings": [
{"path": "state.report.title", "target": "request.title"},
{"value": "markdown", "target": "request.format"},
],
"desc": "Publish report",
"retry": 0,
"timeout_seconds": 30,
}
)
assert request.input_map is None
assert request.input_bindings is not None
assert isinstance(request.input_bindings[0], InputPathBinding)
assert isinstance(request.input_bindings[1], InputValueBinding)
assert request.retry == 0
def test_add_step_from_capability_request_rejects_both_input_forms() -> None:
with pytest.raises(ValidationError, match="mutually exclusive"):
AddStepFromCapabilityRequest.model_validate(
{
"workspace_id": "report",
"revision": 3,
"step_id": "publish",
"capability_name": "local.report.publish",
"input_map": {"state.title": "request.title"},
"input_bindings": [
{"value": "markdown", "target": "request.format"},
],
}
)
def test_workflow_surface_rejects_unknown_draft_route_outcome_when_spec_is_known(
tmp_path: Path,
) -> None: