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

fmt

Browse Source
This commit is contained in:
lda
2026-05-20 23:29:53 +07:00 Verified
parent bc8dc138c7
commit ecfb64691e
36 changed files with 1191 additions and 1368 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/demo_workflow.py
+2 -4
View File
@@ -9,8 +9,7 @@ DemoHandler = Callable[[dict[str, object], RuntimeContext], dict[str, object]]
def build_demo_workflow() -> Workflow:
return Workflow.model_validate(
{
return Workflow.model_validate({
"name": "drive_summary_demo",
"input_schema": {
"type": "object",
@@ -204,8 +203,7 @@ def build_demo_workflow() -> Workflow:
{"from": "send_email", "outcome": "sent", "to": END},
{"from": "skip_email", "outcome": "ok", "to": END},
],
}
)
})
def drive_list_files(
examples/mcp_tool_workflow.py
+2 -4
View File
@@ -34,8 +34,7 @@ async def run_example() -> dict[str, object]:
await service.refresh_connection_catalog("fixture.personal")
plan = RawWorkflowPlan.model_validate(
{
plan = RawWorkflowPlan.model_validate({
"name": "mcp_echo_workflow",
"input_schema": {
"type": "object",
@@ -73,8 +72,7 @@ async def run_example() -> dict[str, object]:
{"from": "echo", "outcome": "error", "to": "raise_mcp_error"},
{"from": "raise_mcp_error", "outcome": "ok", "to": END},
],
}
)
})
run = await service.run_workflow_from_plan(plan, {"text": "hello from MCP"})
return {
examples/raw_canonical_workflow.py
+2 -4
View File
@@ -6,8 +6,7 @@ from wf_core.run_state import RunState
def build_raw_canonical_workflow() -> Workflow:
"""Build a raw core workflow using the canonical post-migration shape."""
return Workflow.model_validate(
{
return Workflow.model_validate({
"name": "raw_canonical_echo",
"input_schema": {
"type": "object",
@@ -61,8 +60,7 @@ def build_raw_canonical_workflow() -> Workflow:
}
],
"edges": [{"from": "format", "outcome": "ok", "to": END}],
}
)
})
def build_raw_canonical_registry():
src/wf_artifacts/catalog.py
+2 -4
View File
@@ -38,12 +38,10 @@ def artifact_catalog_entry(
diagnostics: list[DependencyDiagnostic] | tuple[DependencyDiagnostic, ...] = (),
) -> WorkflowArtifactCatalogEntry:
"""Project an artifact as a catalog entry without exposing its internal plan."""
required_sources = sorted(
{
required_sources = sorted({
capability.logical_source
for capability in artifact.required_capability_map().values()
}
)
})
return WorkflowArtifactCatalogEntry(
name=artifact_node_name(artifact),
artifact_id=artifact.id,
src/wf_artifacts/drafts/models.py
+9 -3
View File
@@ -7,9 +7,15 @@ from pydantic import BaseModel, ConfigDict, Field, model_validator
from wf_core.models.conditions import Condition
JsonObject = dict[str, Any]
STEP_KIND_KEYS = frozenset(
{"use", "foreach", "interrupt", "join", "when", "choose", "match"}
)
STEP_KIND_KEYS = frozenset({
"use",
"foreach",
"interrupt",
"join",
"when",
"choose",
"match",
})
class DraftUseStep(BaseModel):
src/wf_authoring/builder/core.py
+4 -8
View File
@@ -254,16 +254,14 @@ class WorkflowBuilder:
mode: Literal["serial", "parallel"] = "serial",
on_item_error: Literal["fail", "collect", "skip"] = "fail",
) -> ForeachNode:
node = ForeachNode.model_validate(
{
node = ForeachNode.model_validate({
"id": id or self._next_step_id(f"foreach_{slug_id(as_)}"),
"type": "foreach",
"over": coerce_path(over),
"as": as_,
"mode": mode,
"on_item_error": on_item_error,
}
)
})
self.nodes.append(node)
return node
@@ -297,13 +295,11 @@ class WorkflowBuilder:
source = self._resolve_branch_ref(from_)
target = self._resolve_branch_ref(to)
self.edges.append(
Edge.model_validate(
{
Edge.model_validate({
"from": step_id(source),
"outcome": outcome,
"to": step_id(target),
}
)
})
)
return source, target
src/wf_core/models/schemas.py
+2 -4
View File
@@ -265,15 +265,13 @@ def _iter_state_field_declarations(
_attach_root_schema_context(validation_schema, root_schema)
yield (
path,
StateFieldDecl.model_validate(
{
StateFieldDecl.model_validate({
"path": StatePath.of(path),
"schema": SchemaRef.model_validate(validation_schema),
"reducer": reducer,
"trace": trace,
"default": default,
}
),
}),
)
child_properties = resolved_schema.get("properties")
if isinstance(child_properties, Mapping):
src/wf_mcp/broker/service/core.py
+3 -7
View File
@@ -319,8 +319,7 @@ class WfMcpService:
statuses: list[dict[str, Any]] = []
for connection in self.connections.list_all():
snapshot = self.store.load_catalog(connection.id)
statuses.append(
{
statuses.append({
"connection_id": connection.id,
"server": connection.server,
"account": connection.account,
@@ -333,12 +332,9 @@ class WfMcpService:
if snapshot is None
else snapshot.max_age_seconds,
"node_count": 0 if snapshot is None else len(snapshot.nodes),
"resource_count": 0
if snapshot is None
else len(snapshot.resources),
"resource_count": 0 if snapshot is None else len(snapshot.resources),
"prompt_count": 0 if snapshot is None else len(snapshot.prompts),
}
)
})
return statuses
def list_resources(
src/wf_mcp/cli.py
+9 -19
View File
@@ -94,26 +94,20 @@ async def _refresh_all(service, connection_id: str | None) -> list[dict[str, Any
try:
await service.refresh_connection_catalog(target_id)
snapshot = service.get_connection_snapshot(target_id)
results.append(
{
results.append({
"connection_id": target_id,
"refreshed": snapshot is not None,
"node_count": 0 if snapshot is None else len(snapshot.nodes),
"resource_count": 0
if snapshot is None
else len(snapshot.resources),
"resource_count": 0 if snapshot is None else len(snapshot.resources),
"prompt_count": 0 if snapshot is None else len(snapshot.prompts),
}
)
})
except Exception as exc:
results.append(
{
results.append({
"connection_id": target_id,
"refreshed": False,
"error_type": type(exc).__name__,
"error": str(exc),
}
)
})
return results
@@ -138,8 +132,7 @@ def main(argv: list[str] | None = None) -> int:
service = _service_from_config(args.config)
if args.command == "connections":
_json_dump(
[
_json_dump([
{
"id": connection.id,
"server": connection.server,
@@ -148,8 +141,7 @@ def main(argv: list[str] | None = None) -> int:
"metadata": connection.metadata,
}
for connection in service.connections.list_all()
]
)
])
return 0
if args.command == "status":
@@ -162,12 +154,10 @@ def main(argv: list[str] | None = None) -> int:
if args.command == "refresh":
results = asyncio.run(_refresh_all(service, args.connection_id))
_json_dump(
{
_json_dump({
"results": results,
"catalog": service.get_catalog().as_payload(),
}
)
})
if any(not result["refreshed"] for result in results):
return 1
return 0
src/wf_mcp/proxy_config.py
+2 -4
View File
@@ -38,12 +38,10 @@ def connection_to_fastmcp_server_config(
def broker_config_to_fastmcp_config(config: BrokerConfig) -> MCPConfig:
"""Convert broker config into FastMCP's multi-server config object."""
validate_proxy_config(config)
return MCPConfig.from_dict(
{
return MCPConfig.from_dict({
"mcpServers": {
connection.id: connection_to_fastmcp_server_config(connection)
for connection in config.connections
if connection.enabled
}
}
)
})
src/wf_mcp/proxy_results/resource_link_transform.py
+2 -4
View File
@@ -46,13 +46,11 @@ class ResourceLinkRewritingTool(Tool):
rewrite_uri: Callable[[str], str],
) -> ResourceLinkRewritingTool:
"""Copy one tool's public schema while replacing only execution."""
return cls.model_validate(
{
return cls.model_validate({
**tool.model_dump(),
"parent_tool": tool,
"rewrite_uri": rewrite_uri,
}
)
})
class ResourceLinkNamespace(Transform):
src/wf_mcp/workflow_surface/handlers.py
+11 -19
View File
@@ -214,8 +214,7 @@ class WorkflowSurfaceHandlers:
query=query,
):
continue
rows.append(
{
rows.append({
"name": name,
"source_id": "workflow",
"kind": "wrapper_artifact",
@@ -227,8 +226,7 @@ class WorkflowSurfaceHandlers:
"is_async": True,
"input_fields": _schema_field_names(artifact.input_schema),
"output_fields": _schema_field_names(artifact.output_schema),
}
)
})
return rows
def _wrapper_capability_detail(
@@ -447,12 +445,10 @@ class WorkflowSurfaceHandlers:
},
)
)
required_sources = sorted(
{
required_sources = sorted({
capability.logical_source
for capability in workflow_artifact.required_capability_map().values()
}
)
})
return {
"artifact_id": workflow_artifact.id,
"version": workflow_artifact.version,
@@ -909,16 +905,14 @@ def _available_sources(service: WfMcpService) -> list[AvailableSource]:
if (capability_name := _capability_name(spec.name)) is not None
if (detail := node_spec_details.get(spec.name)) is not None
}
capabilities.update(
{
capabilities.update({
capability_name: AvailableCapability(
name=capability_name,
kind="reducer",
)
for reducer in source.capabilities.reducers.values()
if (capability_name := _capability_name(reducer.name)) is not None
}
)
})
sources.append(
AvailableSource(
id=source.id,
@@ -982,9 +976,9 @@ def _observed_node_specs(service: WfMcpService) -> dict[str, NodeSpecInventory]:
observed: dict[str, NodeSpecInventory] = {}
for source in service.capability_sources.values():
inventory = source.as_inventory()
observed.update(
{detail.name: detail for detail in inventory.capabilities.node_spec_details}
)
observed.update({
detail.name: detail for detail in inventory.capabilities.node_spec_details
})
return observed
@@ -1063,8 +1057,7 @@ def _artifact_capability_id(artifact: WorkflowArtifact) -> str:
def _raw_plan_from_artifact(artifact: WorkflowArtifact) -> RawWorkflowPlan:
"""Validate the stored plan shape expected by the broker workflow runner."""
return RawWorkflowPlan.model_validate(
{
return RawWorkflowPlan.model_validate({
"name": _plan_field(artifact, "name"),
"input_schema": _plan_field(artifact, "input_schema"),
"state_schema": _plan_field(artifact, "state_schema"),
@@ -1072,8 +1065,7 @@ def _raw_plan_from_artifact(artifact: WorkflowArtifact) -> RawWorkflowPlan:
"start": _plan_field(artifact, "start"),
"nodes": _plan_field(artifact, "nodes"),
"edges": _plan_field(artifact, "edges"),
}
)
})
def _plan_field(artifact: WorkflowArtifact, field_name: str) -> Any:
tests/artifacts/test_draft_adapter.py
+10 -20
View File
@@ -10,8 +10,7 @@ from wf_core.models.steps import InputValueBinding
def test_adapter_lowers_keyed_use_steps_and_routes_through_builder() -> None:
draft = WorkflowDraft.model_validate(
{
draft = WorkflowDraft.model_validate({
"name": "echo",
"input_schema": {},
"state_schema": {"fields": {}},
@@ -19,8 +18,7 @@ def test_adapter_lowers_keyed_use_steps_and_routes_through_builder() -> None:
"start": "echo",
"steps": {"echo": {"use": "demo.echo"}},
"routes": {"echo": {"ok": "__end__"}},
}
)
})
workflow = build_workflow_from_draft(draft)
@@ -34,8 +32,7 @@ def test_adapter_lowers_keyed_use_steps_and_routes_through_builder() -> None:
def test_adapter_lowers_static_inputs_for_constant_like_steps() -> None:
draft = WorkflowDraft.model_validate(
{
draft = WorkflowDraft.model_validate({
"name": "constant",
"input_schema": {},
"state_schema": {"fields": {"message": {"type": "string"}}},
@@ -49,8 +46,7 @@ def test_adapter_lowers_static_inputs_for_constant_like_steps() -> None:
}
},
"routes": {"constant": {"ok": "__end__"}},
}
)
})
workflow = build_workflow_from_draft(draft)
node = workflow.nodes[0]
@@ -92,8 +88,7 @@ def test_invalid_literal_input_map_does_not_fall_through_to_join() -> None:
def test_adapter_lowers_when_step_through_builder() -> None:
draft = WorkflowDraft.model_validate(
{
draft = WorkflowDraft.model_validate({
"name": "when_example",
"input_schema": {},
"state_schema": {"fields": {}},
@@ -114,8 +109,7 @@ def test_adapter_lowers_when_step_through_builder() -> None:
"echo": {"use": "demo.echo"},
},
"routes": {"echo": {"ok": "__end__"}},
}
)
})
workflow = build_workflow_from_draft(draft)
condition = workflow.nodes[0]
@@ -130,8 +124,7 @@ def test_adapter_lowers_when_step_through_builder() -> None:
def test_adapter_lowers_choose_step_through_builder() -> None:
draft = WorkflowDraft.model_validate(
{
draft = WorkflowDraft.model_validate({
"name": "choose_example",
"input_schema": {},
"state_schema": {"fields": {}},
@@ -167,8 +160,7 @@ def test_adapter_lowers_choose_step_through_builder() -> None:
"high": {"ok": "__end__"},
"fallback": {"ok": "__end__"},
},
}
)
})
workflow = build_workflow_from_draft(draft)
condition_ids = [
@@ -186,8 +178,7 @@ def test_adapter_lowers_choose_step_through_builder() -> None:
def test_adapter_lowers_match_step_through_builder() -> None:
draft = WorkflowDraft.model_validate(
{
draft = WorkflowDraft.model_validate({
"name": "match_example",
"input_schema": {},
"state_schema": {"fields": {}},
@@ -211,8 +202,7 @@ def test_adapter_lowers_match_step_through_builder() -> None:
"ready": {"ok": "__end__"},
"waiting": {"ok": "__end__"},
},
}
)
})
workflow = build_workflow_from_draft(draft)
condition_ids = [
tests/artifacts/test_draft_models.py
+6 -12
View File
@@ -36,8 +36,7 @@ def test_draft_step_requires_exactly_one_kind_key() -> None:
def test_workflow_draft_accepts_when_step() -> None:
draft = WorkflowDraft.model_validate(
{
draft = WorkflowDraft.model_validate({
**_keyed_echo_draft(),
"start": "decide",
"steps": {
@@ -54,15 +53,13 @@ def test_workflow_draft_accepts_when_step() -> None:
}
},
},
}
)
})
assert isinstance(draft.steps["decide"], DraftWhenStep)
def test_workflow_draft_accepts_choose_step() -> None:
draft = WorkflowDraft.model_validate(
{
draft = WorkflowDraft.model_validate({
**_keyed_echo_draft(),
"start": "choose_next",
"steps": {
@@ -82,15 +79,13 @@ def test_workflow_draft_accepts_choose_step() -> None:
}
},
},
}
)
})
assert isinstance(draft.steps["choose_next"], DraftChooseStep)
def test_workflow_draft_accepts_match_step() -> None:
draft = WorkflowDraft.model_validate(
{
draft = WorkflowDraft.model_validate({
**_keyed_echo_draft(),
"start": "match_status",
"steps": {
@@ -106,8 +101,7 @@ def test_workflow_draft_accepts_match_step() -> None:
}
},
},
}
)
})
assert isinstance(draft.steps["match_status"], DraftMatchStep)
tests/artifacts/test_models.py
+4 -8
View File
@@ -66,8 +66,7 @@ def test_workflow_artifact_can_be_marked_as_wrapper_intent() -> None:
def test_workflow_artifact_accepts_legacy_required_capability_map_and_dumps_list() -> (
None
):
artifact = WorkflowArtifact.model_validate(
{
artifact = WorkflowArtifact.model_validate({
"id": "legacy_capabilities",
"version": 1,
"title": "Legacy Capabilities",
@@ -83,8 +82,7 @@ def test_workflow_artifact_accepts_legacy_required_capability_map_and_dumps_list
"observed_concrete_source": "demo.personal",
}
},
}
)
})
dumped = artifact.model_dump(mode="json")
required = dumped["required_capabilities"][0]
@@ -120,14 +118,12 @@ def test_workflow_deployment_binds_logical_sources_to_concrete_sources() -> None
def test_workflow_deployment_accepts_legacy_binding_map_and_dumps_list() -> None:
deployment = WorkflowDeployment.model_validate(
{
deployment = WorkflowDeployment.model_validate({
"id": "legacy_bindings.personal",
"artifact_id": "legacy_bindings",
"artifact_version": 1,
"bindings": {"demo": "demo.personal"},
}
)
})
dumped = deployment.model_dump(mode="json")
binding = dumped["bindings"][0]
tests/artifacts/test_store.py
+4 -8
View File
@@ -73,8 +73,7 @@ def test_file_store_loads_legacy_artifact_and_rewrites_canonical_shape(
artifact_dir.mkdir(parents=True)
artifact_path = artifact_dir / "1.json"
artifact_path.write_text(
json.dumps(
{
json.dumps({
"id": "legacy_capabilities",
"version": 1,
"title": "Legacy Capabilities",
@@ -88,8 +87,7 @@ def test_file_store_loads_legacy_artifact_and_rewrites_canonical_shape(
"input_schema_hash": "sha256:input",
}
},
}
),
}),
encoding="utf-8",
)
@@ -110,14 +108,12 @@ def test_file_store_loads_legacy_deployment_and_rewrites_canonical_shape(
store = FileWorkflowArtifactStore(tmp_path)
deployment_path = store.deployments_dir / "legacy_bindings.personal.json"
deployment_path.write_text(
json.dumps(
{
json.dumps({
"id": "legacy_bindings.personal",
"artifact_id": "legacy_bindings",
"artifact_version": 1,
"bindings": {"demo": "demo.personal"},
}
),
}),
encoding="utf-8",
)
tests/authoring/test_demo_workflow.py
+2 -4
View File
@@ -373,14 +373,12 @@ def test_foreach_stress_with_many_documents() -> None:
assert len(run.state["documents"]) == document_count
assert len(run.state["item_summaries"]) == document_count
assert (
len(
[
len([
frame
for frame in run.frames.values()
if frame.kind == "foreach_iteration"
and frame.status == FrameStatus.COMPLETED
]
)
])
== document_count
)
assert len([entry for entry in run.trace if entry.step_type == "foreach"]) == (
tests/authoring/test_ops.py
+4 -8
View File
@@ -39,15 +39,13 @@ def _build_first_workflow(use_safe_first: bool = False):
builder = WorkflowBuilder(
name="first_demo",
input_schema=SchemaRef(type="object"),
state_schema=StateSchema.model_validate(
{
state_schema=StateSchema.model_validate({
"type": "object",
"properties": {
"items": {"type": "array"},
"item": {"type": ["string", "null"]},
},
}
),
}),
output_schema=SchemaRef(type="object"),
start="pick_first",
)
@@ -65,16 +63,14 @@ def _build_first_maybe_workflow():
builder = WorkflowBuilder(
name="first_maybe_demo",
input_schema=SchemaRef(type="object"),
state_schema=StateSchema.model_validate(
{
state_schema=StateSchema.model_validate({
"type": "object",
"properties": {
"items": {"type": "array"},
"item": {"type": ["string", "null"]},
"missing": {"type": "boolean"},
},
}
),
}),
output_schema=SchemaRef(type="object"),
start="pick_first",
)
tests/core/test_atomic_state_patches.py
+8 -14
View File
@@ -126,8 +126,7 @@ def test_output_bindings_validate_exact_state_schema_before_mutation() -> None:
def test_output_bindings_validate_declared_parent_schema_before_mutation() -> None:
workflow = _workflow_from_state_schema(
StateSchema.model_validate(
{
StateSchema.model_validate({
"type": "object",
"properties": {
"person": {
@@ -136,8 +135,7 @@ def test_output_bindings_validate_declared_parent_schema_before_mutation() -> No
"additionalProperties": False,
}
},
}
)
})
)
state = {"person": {"name": "old"}}
@@ -206,9 +204,9 @@ def _workflow_with_node() -> Workflow:
return Workflow(
name="canonical_output",
input_schema=SchemaRef(type="object", properties={}),
state_schema=StateSchema.from_field_map(
{"person.name": StateField(type="string")}
),
state_schema=StateSchema.from_field_map({
"person.name": StateField(type="string")
}),
output_schema=SchemaRef(
type="object", properties={"person": {"type": "object"}}
),
@@ -225,16 +223,12 @@ def _workflow_with_node() -> Workflow:
],
start="rename",
nodes=[
NodeUse.model_validate(
{
NodeUse.model_validate({
"id": "rename",
"type": "node",
"node": "rename",
"output": [
{"source": "person.name", "target": "state.person.name"}
],
}
)
"output": [{"source": "person.name", "target": "state.person.name"}],
})
],
edges=[Edge.model_validate({"from": "rename", "outcome": "ok", "to": END})],
)
tests/core/test_canonical_node_bindings.py
+28 -38
View File
@@ -6,8 +6,7 @@ from wf_core.paths import GraphSourcePath, LocalPath, StatePath
def test_node_use_accepts_canonical_input_and_output_bindings():
node = NodeUse.model_validate(
{
node = NodeUse.model_validate({
"id": "echo",
"type": "node",
"node": "echo",
@@ -16,8 +15,7 @@ def test_node_use_accepts_canonical_input_and_output_bindings():
{"target": "mode", "value": None},
],
"output": [{"source": "echoed", "target": "state.echoed"}],
}
)
})
path_binding = node.input[0]
assert isinstance(path_binding, InputPathBinding)
@@ -34,16 +32,14 @@ def test_node_use_accepts_canonical_input_and_output_bindings():
def test_node_use_converts_old_maps_to_canonical_bindings():
node = NodeUse.model_validate(
{
node = NodeUse.model_validate({
"id": "echo",
"type": "node",
"node": "echo",
"in_map": {"input.message": "message"},
"input_values": {"mode": "fast"},
"out_map": {"echoed": "state.echoed"},
}
)
})
dumped = node.model_dump(mode="json")
assert "in_map" not in dumped
@@ -58,15 +54,13 @@ def test_node_use_converts_old_maps_to_canonical_bindings():
def test_node_use_serializes_canonical_binding_paths_as_strings_in_all_dump_modes():
node = NodeUse.model_validate(
{
node = NodeUse.model_validate({
"id": "echo",
"type": "node",
"node": "echo",
"input": [{"target": "message", "path": "input.message"}],
"output": [{"source": "echoed", "target": "state.echoed"}],
}
)
})
python_dumped = node.model_dump()
json_dumped = node.model_dump(mode="json")
@@ -83,39 +77,33 @@ def test_node_use_serializes_canonical_binding_paths_as_strings_in_all_dump_mode
def test_node_use_rejects_mixed_old_and_new_binding_styles():
with pytest.raises(ValidationError):
NodeUse.model_validate(
{
NodeUse.model_validate({
"id": "echo",
"type": "node",
"node": "echo",
"input": [{"target": "message", "path": "input.message"}],
"in_map": {"input.other": "other"},
}
)
})
def test_input_binding_rejects_path_and_value_together():
with pytest.raises(ValidationError):
NodeUse.model_validate(
{
NodeUse.model_validate({
"id": "bad",
"type": "node",
"node": "bad",
"input": [{"target": "message", "path": "input.message", "value": "x"}],
}
)
})
def test_input_binding_rejects_neither_path_nor_value():
with pytest.raises(ValidationError):
NodeUse.model_validate(
{
NodeUse.model_validate({
"id": "bad",
"type": "node",
"node": "bad",
"input": [{"target": "message"}],
}
)
})
@pytest.mark.parametrize(
@@ -127,9 +115,12 @@ def test_input_binding_rejects_neither_path_nor_value():
)
def test_bindings_reject_extra_fields(field: str, binding: dict[str, object]):
with pytest.raises(ValidationError):
NodeUse.model_validate(
{"id": "bad", "type": "node", "node": "bad", field: [binding]}
)
NodeUse.model_validate({
"id": "bad",
"type": "node",
"node": "bad",
field: [binding],
})
@pytest.mark.parametrize(
@@ -142,21 +133,22 @@ def test_bindings_reject_extra_fields(field: str, binding: dict[str, object]):
)
def test_deprecated_maps_reject_non_mapping_values(field: str, value: object):
with pytest.raises(ValidationError):
NodeUse.model_validate(
{"id": "bad", "type": "node", "node": "bad", field: value}
)
NodeUse.model_validate({
"id": "bad",
"type": "node",
"node": "bad",
field: value,
})
def test_deprecated_conversion_preserves_input_value_then_in_map_order():
node = NodeUse.model_validate(
{
node = NodeUse.model_validate({
"id": "ordered",
"type": "node",
"node": "ordered",
"input_values": {"first": 1, "second": 2},
"in_map": {"input.third": "third", "state.fourth": "fourth"},
}
)
})
dumped_input = node.model_dump(mode="json")["input"]
assert dumped_input[0]["target"] == "first"
@@ -170,14 +162,12 @@ def test_deprecated_conversion_preserves_input_value_then_in_map_order():
def test_deprecated_input_value_preserves_explicit_null():
node = NodeUse.model_validate(
{
node = NodeUse.model_validate({
"id": "null",
"type": "node",
"node": "null",
"input_values": {"maybe": None},
}
)
})
value_binding = node.input[0]
assert isinstance(value_binding, InputValueBinding)
tests/core/test_mapping_validation.py
+12 -9
View File
@@ -178,9 +178,10 @@ def _workflow(
return Workflow(
name="mapping_validation",
input_schema=SchemaRef.model_validate(
{"type": "object", "properties": {"person": {"type": "object"}}}
),
input_schema=SchemaRef.model_validate({
"type": "object",
"properties": {"person": {"type": "object"}},
}),
state_schema=StateSchema.from_field_map(
state_fields or {"person": StateField(type="object")}
),
@@ -188,12 +189,14 @@ def _workflow(
node_defs=[
NodeDef(
name="tool",
input_schema=SchemaRef.model_validate(
{"type": "object", "properties": {"user": {"type": "object"}}}
),
output_schema=SchemaRef.model_validate(
{"type": "object", "properties": {"user": {"type": "object"}}}
),
input_schema=SchemaRef.model_validate({
"type": "object",
"properties": {"user": {"type": "object"}},
}),
output_schema=SchemaRef.model_validate({
"type": "object",
"properties": {"user": {"type": "object"}},
}),
outcomes=["ok"],
)
],
tests/core/test_nested_mappings.py
+22 -44
View File
@@ -19,8 +19,7 @@ from wf_core import (
def test_canonical_bindings_resolve_input_values_paths_and_explicit_null() -> None:
workflow = Workflow.model_validate(
{
workflow = Workflow.model_validate({
"name": "canonical",
"input_schema": {
"type": "object",
@@ -65,8 +64,7 @@ def test_canonical_bindings_resolve_input_values_paths_and_explicit_null() -> No
}
],
"edges": [{"from": "echo", "outcome": "ok", "to": END}],
}
)
})
run = execute_workflow(
workflow,
{"message": "hi"},
@@ -130,12 +128,10 @@ def test_missing_nested_node_output_path_fails() -> None:
def test_root_node_local_paths_map_whole_input_and_output_payloads() -> None:
workflow = Workflow(
name="root_mapping",
input_schema=SchemaRef.model_validate(
{
input_schema=SchemaRef.model_validate({
"type": "object",
"properties": {"rates": {"type": "object"}},
}
),
}),
state_schema=StateSchema.from_field_map({"rates": StateField(type="object")}),
output_schema=SchemaRef(type="object", properties={}),
node_defs=[
@@ -156,15 +152,13 @@ def test_root_node_local_paths_map_whole_input_and_output_payloads() -> None:
nodes=[
cast(
Any,
NodeUse.model_validate(
{
NodeUse.model_validate({
"id": "force",
"type": "node",
"node": "force_rates",
"in_map": {"input.rates": "."},
"out_map": {".": "state.rates"},
}
),
}),
)
],
edges=[Edge.model_validate({"from": "force", "outcome": "ok", "to": END})],
@@ -196,20 +190,16 @@ def test_static_input_values_are_merged_into_node_local_input() -> None:
node_defs=[
NodeDef(
name="constant",
input_schema=SchemaRef.model_validate(
{
input_schema=SchemaRef.model_validate({
"type": "object",
"properties": {"value": {"type": "string"}},
"required": ["value"],
}
),
output_schema=SchemaRef.model_validate(
{
}),
output_schema=SchemaRef.model_validate({
"type": "object",
"properties": {"value": {"type": "string"}},
"required": ["value"],
}
),
}),
outcomes=["ok"],
)
],
@@ -217,15 +207,13 @@ def test_static_input_values_are_merged_into_node_local_input() -> None:
nodes=[
cast(
Any,
NodeUse.model_validate(
{
NodeUse.model_validate({
"id": "constant",
"type": "node",
"node": "constant",
"input_values": {"value": "CLICKED"},
"out_map": {"value": "state.message"},
}
),
}),
)
],
edges=[Edge.model_validate({"from": "constant", "outcome": "ok", "to": END})],
@@ -244,40 +232,32 @@ def test_static_input_values_are_merged_into_node_local_input() -> None:
def _nested_mapping_workflow() -> Workflow:
return Workflow(
name="nested_mapping",
input_schema=SchemaRef.model_validate(
{
input_schema=SchemaRef.model_validate({
"type": "object",
"properties": {
"person": {"type": "object"},
"digital": {"type": "object"},
},
}
),
state_schema=StateSchema.from_field_map(
{
}),
state_schema=StateSchema.from_field_map({
"person": StateField(type="object"),
"experience": StateField(type="object"),
}
),
}),
output_schema=SchemaRef(type="object", properties={}),
node_defs=[
NodeDef(
name="big_tool",
input_schema=SchemaRef.model_validate(
{
input_schema=SchemaRef.model_validate({
"type": "object",
"properties": {"user": {"type": "object"}},
}
),
output_schema=SchemaRef.model_validate(
{
}),
output_schema=SchemaRef.model_validate({
"type": "object",
"properties": {
"user": {"type": "object"},
"job": {"type": "object"},
},
}
),
}),
outcomes=["ok"],
)
],
@@ -285,8 +265,7 @@ def _nested_mapping_workflow() -> Workflow:
nodes=[
cast(
Any,
NodeUse.model_validate(
{
NodeUse.model_validate({
"id": "big",
"type": "node",
"node": "big_tool",
@@ -299,8 +278,7 @@ def _nested_mapping_workflow() -> Workflow:
"user.gender": "state.person.gender",
"job.years": "state.experience.years",
},
}
),
}),
)
],
edges=[Edge.model_validate({"from": "big", "outcome": "ok", "to": END})],
tests/core/test_nested_state_paths.py
+26 -41
View File
@@ -31,8 +31,7 @@ def test_exact_nested_state_path_uses_declared_reducer() -> None:
def test_state_schema_accepts_legacy_field_list_and_dumps_json_schema() -> None:
schema = StateSchema.model_validate(
{
schema = StateSchema.model_validate({
"fields": [
{"path": "state.person", "type": "object"},
{
@@ -41,8 +40,7 @@ def test_state_schema_accepts_legacy_field_list_and_dumps_json_schema() -> None:
"reducer": "wf.std.replace",
},
]
}
)
})
assert schema.fields[0].path == StatePath.of("person")
assert schema.field_map()["person.name"].type == "string"
@@ -52,8 +50,7 @@ def test_state_schema_accepts_legacy_field_list_and_dumps_json_schema() -> None:
def test_state_schema_uses_json_schema_properties_as_canonical_shape() -> None:
schema = StateSchema.model_validate(
{
schema = StateSchema.model_validate({
"type": "object",
"properties": {
"person": {
@@ -68,8 +65,7 @@ def test_state_schema_uses_json_schema_properties_as_canonical_shape() -> None:
},
"count": {"type": "integer", "reducer": "wf.std.add"},
},
}
)
})
fields = schema.field_map()
assert fields["person.name"].validation_schema.type == "string"
@@ -79,14 +75,12 @@ def test_state_schema_uses_json_schema_properties_as_canonical_shape() -> None:
def test_state_schema_rejects_invalid_reducer_extension_keyword() -> None:
try:
StateSchema.model_validate(
{
StateSchema.model_validate({
"type": "object",
"properties": {
"count": {"type": "integer", "reducer": {"bad": True}},
},
}
)
})
except ValueError as exc:
assert "invalid reducer for state field 'count'" in str(exc)
else:
@@ -94,16 +88,14 @@ def test_state_schema_rejects_invalid_reducer_extension_keyword() -> None:
def test_state_schema_accepts_canonical_schema_field() -> None:
schema = StateSchema.model_validate(
{
schema = StateSchema.model_validate({
"fields": [
{
"path": "state.person.name",
"schema": {"type": "string", "title": "Person Name"},
}
]
}
)
})
field = schema.field_map()["person.name"]
assert field.validation_schema.type == "string"
@@ -119,15 +111,13 @@ def test_state_schema_accepts_deprecated_dict_shape_and_dumps_list() -> None:
def test_state_schema_accepts_deprecated_dict_value_with_schema_key() -> None:
schema = StateSchema.model_validate(
{
schema = StateSchema.model_validate({
"fields": {
"person.name": {
"schema": {"type": "string", "description": "Display name"},
}
}
}
)
})
assert schema.field_map()["person.name"].validation_schema.type == "string"
@@ -139,26 +129,27 @@ def test_state_schema_accepts_json_schema_field_without_type() -> None:
def test_state_schema_accepts_deprecated_state_prefixed_dict_keys() -> None:
schema = StateSchema.model_validate(
{"fields": {"state.person.name": {"type": "string"}}}
)
schema = StateSchema.model_validate({
"fields": {"state.person.name": {"type": "string"}}
})
assert schema.field_map()["person.name"].path == StatePath.of("person.name")
def test_state_field_decl_model_dump_serializes_path_as_string() -> None:
field = StateFieldDecl.model_validate(
{"path": "state.person.name", "type": "string"}
)
field = StateFieldDecl.model_validate({
"path": "state.person.name",
"type": "string",
})
assert field.model_dump()["path"] == "state.person.name"
assert field.model_dump(mode="json")["path"] == "state.person.name"
def test_state_schema_model_dump_serializes_paths_as_strings() -> None:
schema = StateSchema.model_validate(
{"fields": [{"path": "state.person.name", "type": "string"}]}
)
schema = StateSchema.model_validate({
"fields": [{"path": "state.person.name", "type": "string"}]
})
dumped = schema.model_dump(mode="json")
assert dumped["properties"]["person"]["properties"]["name"]["type"] == "string"
@@ -167,14 +158,12 @@ def test_state_schema_model_dump_serializes_paths_as_strings() -> None:
def test_state_schema_rejects_duplicate_field_paths() -> None:
try:
StateSchema.model_validate(
{
StateSchema.model_validate({
"fields": [
{"path": "state.person.name", "type": "string"},
{"path": "state.person.name", "type": "string"},
]
}
)
})
except ValueError as exc:
assert "duplicate state field path 'person.name'" in str(exc)
else:
@@ -183,8 +172,7 @@ def test_state_schema_rejects_duplicate_field_paths() -> None:
def test_exact_nested_state_path_uses_reducer_from_json_schema_property() -> None:
workflow = _workflow_from_state_schema(
StateSchema.model_validate(
{
StateSchema.model_validate({
"type": "object",
"properties": {
"person": {
@@ -194,8 +182,7 @@ def test_exact_nested_state_path_uses_reducer_from_json_schema_property() -> Non
},
}
},
}
)
})
)
state = {"person": {"tags": ["seed"]}}
@@ -205,14 +192,12 @@ def test_exact_nested_state_path_uses_reducer_from_json_schema_property() -> Non
def test_state_schema_field_map_uses_rootless_keys() -> None:
schema = StateSchema.model_validate(
{
schema = StateSchema.model_validate({
"fields": [
{"path": "state.person.name", "type": "string"},
{"path": "state.person.tags", "type": "array"},
]
}
)
})
fields = schema.field_map()
assert fields["person.name"].path == StatePath.of("person.name")
tests/core/test_path_values.py
+13 -19
View File
@@ -121,29 +121,23 @@ def test_pydantic_revalidates_existing_path_objects() -> None:
object.__setattr__(local, "parts", ("items[0]",))
with pytest.raises(ValidationError):
Payload.model_validate(
{
Payload.model_validate({
"source": source,
"target": StatePath.of("person"),
"local": LocalPath.root(),
}
)
})
with pytest.raises(ValidationError):
Payload.model_validate(
{
Payload.model_validate({
"source": GraphSourcePath.input("user"),
"target": target,
"local": LocalPath.root(),
}
)
})
with pytest.raises(ValidationError):
Payload.model_validate(
{
Payload.model_validate({
"source": GraphSourcePath.input("user"),
"target": StatePath.of("person"),
"local": local,
}
)
})
def test_pydantic_accepts_path_strings_and_serializes_strings() -> None:
@@ -152,9 +146,11 @@ def test_pydantic_accepts_path_strings_and_serializes_strings() -> None:
target: StatePath
local: LocalPath
payload = Payload.model_validate(
{"source": "input.user", "target": "state.person", "local": "user"}
)
payload = Payload.model_validate({
"source": "input.user",
"target": "state.person",
"local": "user",
})
assert payload.source == GraphSourcePath.input("user")
assert payload.target == StatePath.of("person")
@@ -184,13 +180,11 @@ def test_pydantic_accepts_existing_path_objects() -> None:
target: StatePath
local: LocalPath
payload = Payload.model_validate(
{
payload = Payload.model_validate({
"source": GraphSourcePath.state("person"),
"target": StatePath.of("person.name"),
"local": LocalPath.root(),
}
)
})
assert str(payload.source) == "state.person"
assert str(payload.target) == "state.person.name"
tests/core/test_schema_validation.py
+20 -35
View File
@@ -10,16 +10,14 @@ from wf_core.runtime.ops.schemas import validate_payload_against_schema
def test_schema_validation_rejects_wrong_property_type() -> None:
schema = SchemaRef.model_validate(
{
schema = SchemaRef.model_validate({
"type": "object",
"properties": {
"name": {"type": "string"},
"count": {"type": "integer"},
},
"required": ["name", "count"],
}
)
})
with pytest.raises(WorkflowExecutionError, match=r"count.*not of type 'integer'"):
validate_payload_against_schema(
@@ -30,8 +28,7 @@ def test_schema_validation_rejects_wrong_property_type() -> None:
def test_schema_validation_rejects_nested_missing_required_field() -> None:
schema = SchemaRef.model_validate(
{
schema = SchemaRef.model_validate({
"type": "object",
"properties": {
"profile": {
@@ -41,8 +38,7 @@ def test_schema_validation_rejects_nested_missing_required_field() -> None:
}
},
"required": ["profile"],
}
)
})
with pytest.raises(WorkflowExecutionError, match=r"profile.*email.*required"):
validate_payload_against_schema(
@@ -53,22 +49,19 @@ def test_schema_validation_rejects_nested_missing_required_field() -> None:
def test_schema_validation_accepts_valid_payload() -> None:
schema = SchemaRef.model_validate(
{
schema = SchemaRef.model_validate({
"type": "object",
"properties": {
"tags": {"type": "array", "items": {"type": "string"}},
},
"required": ["tags"],
}
)
})
validate_payload_against_schema(schema, {"tags": ["a", "b"]}, "node input")
def test_schema_ref_accepts_and_preserves_schema_with_defs_and_ref() -> None:
schema = SchemaRef.model_validate(
{
schema = SchemaRef.model_validate({
"$schema": "https://json-schema.org/draft/2020-12/schema",
"$defs": {
"tag": {
@@ -80,8 +73,7 @@ def test_schema_ref_accepts_and_preserves_schema_with_defs_and_ref() -> None:
"type": "object",
"properties": {"tag": {"$ref": "#/$defs/tag"}},
"required": ["tag"],
}
)
})
dumped = schema.model_dump(mode="json")
@@ -97,13 +89,11 @@ def test_schema_ref_rejects_invalid_json_schema_shape() -> None:
def test_schema_ref_defaults_to_draft_2020_12_without_schema_keyword() -> None:
schema = SchemaRef.model_validate(
{
schema = SchemaRef.model_validate({
"type": "object",
"properties": {"count": {"type": "integer"}},
"required": ["count"],
}
)
})
dumped = schema.model_dump(mode="json")
@@ -113,13 +103,11 @@ def test_schema_ref_defaults_to_draft_2020_12_without_schema_keyword() -> None:
def test_schema_ref_preserves_extra_json_schema_keywords() -> None:
schema = SchemaRef.model_validate(
{
schema = SchemaRef.model_validate({
"type": "object",
"properties": {"name": {"type": "string"}},
"additionalProperties": False,
}
)
})
dumped = schema.model_dump(mode="json")
@@ -137,9 +125,10 @@ def test_schema_ref_dump_omits_none_fields_and_stays_valid_json_schema() -> None
def test_state_field_decl_dump_omits_nested_schema_none_fields() -> None:
field = StateFieldDecl.model_validate(
{"path": "state.person", "schema": {"type": "object"}}
)
field = StateFieldDecl.model_validate({
"path": "state.person",
"schema": {"type": "object"},
})
dumped = field.model_dump(mode="json")
@@ -151,8 +140,7 @@ def test_state_field_decl_dump_omits_nested_schema_none_fields() -> None:
def test_state_schema_dump_is_valid_json_schema_with_reducer_keyword() -> None:
from wf_core import StateSchema
schema = StateSchema.model_validate(
{
schema = StateSchema.model_validate({
"type": "object",
"properties": {
"count": {
@@ -161,8 +149,7 @@ def test_state_schema_dump_is_valid_json_schema_with_reducer_keyword() -> None:
"reducer": "wf.std.add",
}
},
}
)
})
dumped = schema.model_dump(mode="json")
assert dumped["type"] == "object"
@@ -174,8 +161,7 @@ def test_state_schema_dump_is_valid_json_schema_with_reducer_keyword() -> None:
def test_state_field_validation_schema_preserves_root_defs_for_local_refs() -> None:
from wf_core import StateSchema
schema = StateSchema.model_validate(
{
schema = StateSchema.model_validate({
"type": "object",
"$defs": {
"PoolByCategory": {
@@ -190,8 +176,7 @@ def test_state_field_validation_schema_preserves_root_defs_for_local_refs() -> N
"items": {"$ref": "#/$defs/PoolByCategory"},
}
},
}
)
})
field_schema = schema.field_map()["current_pools"].validation_schema
tests/fixtures/mcp_echo_server.py
Vendored
+2 -4
View File
@@ -26,14 +26,12 @@ async def echo_tool(
async def resource_link_tool() -> list[mcp_types.ResourceLink]:
"""Return a link to a fixture resource so proxy URI rewriting is testable."""
return [
mcp_types.ResourceLink.model_validate(
{
mcp_types.ResourceLink.model_validate({
"type": "resource_link",
"name": "resource.welcome",
"uri": "fixture://docs/welcome",
"mimeType": "text/plain",
}
)
})
]
tests/refs/test_platform_refs.py
+2 -4
View File
@@ -33,12 +33,10 @@ def test_platform_refs_validate_and_serialize_through_pydantic() -> None:
source: SourceRef
capability: CapabilityRef
payload = Payload.model_validate(
{
payload = Payload.model_validate({
"source": "demo.personal",
"capability": "demo.personal.echo_tool",
}
)
})
assert payload.source == SourceRef.parse("demo.personal")
assert payload.capability == CapabilityRef.parse("demo.personal.echo_tool")
tests/rewrite/actions.py
+15 -27
View File
@@ -109,37 +109,33 @@ class RateChange:
@node(name="force 6* rating")
@staticmethod
def r80(r: Rates) -> Rates:
return Rates.model_validate(
{
return Rates.model_validate({
"rates": {
"r_1": 0,
"r_10": 0,
"r_80": r.rates["r_80"],
"r_240": r.rates["r_240"],
}
}
)
})
@node(name="force banner rating")
@staticmethod
def r240(_: Nothing) -> Rates:
return Rates.model_validate(
{"rates": {"r_1": 0, "r_10": 0, "r_80": 0, "r_240": 1}}
)
return Rates.model_validate({
"rates": {"r_1": 0, "r_10": 0, "r_80": 0, "r_240": 1}
})
@node(name="force 5*+ rating")
@staticmethod
def r10(r: Rates) -> Rates:
return Rates.model_validate(
{
return Rates.model_validate({
"rates": {
"r_1": 0,
"r_10": r.rates["r_10"],
"r_80": r.rates["r_80"],
"r_240": r.rates["r_240"],
}
}
)
})
@node(name="buff 6* rating")
@staticmethod
@@ -157,16 +153,14 @@ class RateChange:
r80 = br["r_80"] * (1 + rpn)
r10 = br["r_10"] # use initial rates because i dont know how this works
r1 = 1 - r240 - r80 - r10
return Rates.model_validate(
{
return Rates.model_validate({
"rates": {
"r_1": r1,
"r_10": r10,
"r_80": r80,
"r_240": r240,
}
}
)
})
@node(name="reset rating")
@staticmethod
@@ -178,8 +172,7 @@ class CounterUp:
@node(name="counter 6* reset")
@staticmethod
def c80(_: Nothing) -> Counters:
return Counters.model_validate(
{
return Counters.model_validate({
"counter": {
"c_80": 0,
"c_10": 0,
@@ -187,19 +180,16 @@ class CounterUp:
"simple_counter": 0,
# this is influenced by the add reducer.
# its top level. it doesnt reset. its a miracle. i hate this.
}
)
})
@node(name="counter 5* reset")
@staticmethod
def c10(c: Counters) -> Counters:
c80 = c.counter["c_80"]
return Counters.model_validate(
{
return Counters.model_validate({
"counter": {"c_10": 0, "c_80": c80}, # merge with or_!
"simple_counter": 0,
}
)
})
@node(name="counting up")
@staticmethod
@@ -238,12 +228,10 @@ def roll(state: CurrentPools) -> ThisStorage:
r = state.current_pools
(t,) = random.choices(r, weights=[*map(lambda p: p["rates"], r)])
this = Entity(category=t["category"], name=random.choice(t["pool"]))
return ThisStorage.model_validate(
{
return ThisStorage.model_validate({
"this": this,
"storage": [this], # I NEED MERGE
}
)
})
@node(outcomes=("240", "80", "10", "1")) # missed this! good job.
tests/rewrite/test_this_shit_works.py
+2 -4
View File
@@ -72,8 +72,7 @@ def test():
assert d.status == RunStatus.COMPLETED, "oops"
state = State.model_validate(d.state)
pprint(state.storage)
pprint(
[
pprint([
t
for t in d.trace
if t.node_id
@@ -81,8 +80,7 @@ def test():
"counter_up",
"tick",
)
]
)
])
pprint(d.state)
assert any(
i["name"] in context["pool"]["n_240"]
tests/wf_mcp/test_admin_surface.py
+2 -4
View File
@@ -84,15 +84,13 @@ class FakeManager:
metadata: dict[str, Any] | None = None,
enabled: bool = True,
) -> dict[str, Any]:
self.added.append(
{
self.added.append({
"connection_id": connection_id,
"server": server,
"account": account,
"metadata": metadata,
"enabled": enabled,
}
)
})
return {"action": "add_connection", "ok": True}
def update_connection(
tests/wf_mcp/test_broker_server.py
+2 -4
View File
@@ -32,8 +32,7 @@ def test_load_broker_config_resolves_relative_store_root() -> None:
tmp_path.mkdir(parents=True, exist_ok=True)
config_path = tmp_path / "wf_mcp.config.json"
config_path.write_text(
json.dumps(
{
json.dumps({
"store_root": ".broker-store",
"connections": [
{
@@ -42,8 +41,7 @@ def test_load_broker_config_resolves_relative_store_root() -> None:
"account": "personal",
}
],
}
),
}),
encoding="utf-8",
)
tests/wf_mcp/test_cli.py
+19 -27
View File
@@ -14,8 +14,7 @@ from .test_support import local_temp_root
def _write_config(path: Path) -> None:
path.write_text(
json.dumps(
{
json.dumps({
"store_root": ".wf_mcp_store",
"connections": [
{
@@ -24,17 +23,20 @@ def _write_config(path: Path) -> None:
"account": "personal",
}
],
}
),
}),
encoding="utf-8",
)
def test_build_parser_accepts_serve_transport() -> None:
parser = build_parser()
args = parser.parse_args(
["--config", "wf_mcp.config.json", "serve", "--transport", "streamable_http"]
)
args = parser.parse_args([
"--config",
"wf_mcp.config.json",
"serve",
"--transport",
"streamable_http",
])
assert args.command == "serve"
assert args.transport == "streamable_http"
@@ -45,8 +47,7 @@ def test_build_parser_accepts_serve_transport() -> None:
def test_build_parser_accepts_proxy_compatibility_flags() -> None:
parser = build_parser()
args = parser.parse_args(
[
args = parser.parse_args([
"--config",
"wf_mcp.config.json",
"serve",
@@ -54,8 +55,7 @@ def test_build_parser_accepts_proxy_compatibility_flags() -> None:
"--prompts-as-tools",
"--search-tools",
"--safe-tool-names",
]
)
])
assert args.command == "serve"
assert args.resources_as_tools is True
@@ -68,27 +68,23 @@ def test_build_parser_rejects_legacy_mode_flag() -> None:
parser = build_parser()
with pytest.raises(SystemExit):
parser.parse_args(
[
parser.parse_args([
"--config",
"wf_mcp.config.json",
"serve",
"--mode",
"unified",
]
)
])
def test_build_parser_accepts_no_admin_tools_flag() -> None:
parser = build_parser()
args = parser.parse_args(
[
args = parser.parse_args([
"--config",
"wf_mcp.config.json",
"serve",
"--no-admin-tools",
]
)
])
assert args.command == "serve"
assert args.admin_tools is False
@@ -158,8 +154,7 @@ def test_load_broker_config_normalizes_typed_stdio_metadata() -> None:
tmp_path.mkdir(parents=True, exist_ok=True)
config_path = tmp_path / "wf_mcp.config.json"
config_path.write_text(
json.dumps(
{
json.dumps({
"store_root": ".wf_mcp_store",
"connections": [
{
@@ -173,8 +168,7 @@ def test_load_broker_config_normalizes_typed_stdio_metadata() -> None:
},
}
],
}
),
}),
encoding="utf-8",
)
@@ -194,8 +188,7 @@ def test_load_broker_config_rejects_bad_metadata_shape() -> None:
tmp_path.mkdir(parents=True, exist_ok=True)
config_path = tmp_path / "wf_mcp.config.json"
config_path.write_text(
json.dumps(
{
json.dumps({
"connections": [
{
"id": "demo.personal",
@@ -204,8 +197,7 @@ def test_load_broker_config_rejects_bad_metadata_shape() -> None:
"metadata": {"transport": "stdio", "args": "server.py"},
}
],
}
),
}),
encoding="utf-8",
)
tests/wf_mcp/test_proxy.py
+8 -16
View File
@@ -389,8 +389,7 @@ def test_proxy_admin_tools_mutate_config_file() -> None:
tmp_path.mkdir(parents=True, exist_ok=True)
config_path = tmp_path / "wf_mcp.config.json"
config_path.write_text(
json.dumps(
{
json.dumps({
"store_root": ".wf_mcp_store",
"connections": [
{
@@ -400,8 +399,7 @@ def test_proxy_admin_tools_mutate_config_file() -> None:
"enabled": False,
}
],
}
),
}),
encoding="utf-8",
)
config = load_broker_config(config_path)
@@ -486,12 +484,10 @@ def test_proxy_admin_reload_remounts_connections() -> None:
tmp_path.mkdir(parents=True, exist_ok=True)
config_path = tmp_path / "wf_mcp.config.json"
config_path.write_text(
json.dumps(
{
json.dumps({
"store_root": ".wf_mcp_store",
"connections": [],
}
),
}),
encoding="utf-8",
)
config = load_broker_config(config_path)
@@ -548,12 +544,10 @@ def test_proxy_admin_reload_sends_list_changed_notifications() -> None:
tmp_path.mkdir(parents=True, exist_ok=True)
config_path = tmp_path / "wf_mcp.config.json"
config_path.write_text(
json.dumps(
{
json.dumps({
"store_root": ".wf_mcp_store",
"connections": [],
}
),
}),
encoding="utf-8",
)
config = load_broker_config(config_path)
@@ -582,12 +576,10 @@ def test_proxy_config_mutation_does_not_notify_before_reload() -> None:
tmp_path.mkdir(parents=True, exist_ok=True)
config_path = tmp_path / "wf_mcp.config.json"
config_path.write_text(
json.dumps(
{
json.dumps({
"store_root": ".wf_mcp_store",
"connections": [],
}
),
}),
encoding="utf-8",
)
config = load_broker_config(config_path)
tests/wf_mcp/test_proxy_results.py
+2 -4
View File
@@ -65,11 +65,9 @@ def test_rewrites_resource_links_inside_call_tool_result() -> None:
def _resource_link(uri: str) -> mcp_types.ResourceLink:
"""Build ResourceLink through validation because Pydantic accepts URI strings."""
return mcp_types.ResourceLink.model_validate(
{
return mcp_types.ResourceLink.model_validate({
"type": "resource_link",
"name": "dynamic-text",
"uri": uri,
"mimeType": "text/plain",
}
)
})
tests/wf_mcp/test_server.py
+2 -4
View File
@@ -526,8 +526,7 @@ def test_server_reload_syncs_service_connection_source_enabled_state() -> None:
tmp_path.mkdir(parents=True, exist_ok=True)
config_path = tmp_path / "wf_mcp.config.json"
config_path.write_text(
json.dumps(
{
json.dumps({
"store_root": ".wf_mcp_store",
"connections": [
{
@@ -542,8 +541,7 @@ def test_server_reload_syncs_service_connection_source_enabled_state() -> None:
},
}
],
}
),
}),
encoding="utf-8",
)
config = load_broker_config(config_path)