alright uhh

src/wf_mcp/broker/artifact_tools.py

@@ -30,6 +30,7 @@ def register_artifact_tools(server: FastMCP, service: WfMcpService) -> None:
         outcomes: list[str],
         description: str | None = None,
         required_capabilities: dict[str, dict[str, Any]] | None = None,
+        source_bindings: dict[str, str] | None = None,
         created_from_catalog_version: str | None = None,
     ) -> dict[str, Any]:
         return await handlers.create_artifact_from_plan(
@@ -43,6 +44,7 @@ def register_artifact_tools(server: FastMCP, service: WfMcpService) -> None:
                 name: capability
                 for name, capability in (required_capabilities or {}).items()
             },
+            source_bindings=source_bindings,
             created_from_catalog_version=created_from_catalog_version,
         )
 

src/wf_mcp/workflow_surface/handlers.py

@@ -157,6 +157,7 @@ class WorkflowSurfaceHandlers:
         kind: ArtifactKind = "workflow",
         description: str | None = None,
         required_capabilities: dict[str, dict[str, Any]] | None = None,
+        source_bindings: dict[str, str] | None = None,
         created_from_catalog_version: str | None = None,
     ) -> dict[str, Any]:
         if self.service.artifact_store is None:
@@ -178,6 +179,7 @@ class WorkflowSurfaceHandlers:
                 name: RequiredCapability.model_validate(capability)
                 for name, capability in (required_capabilities or {}).items()
             },
+            source_bindings=source_bindings,
             created_from_catalog_version=created_from_catalog_version,
         )
         self.service.artifact_store.save_artifact(workflow_artifact)

src/wf_mcp/workflow_surface/tools.py

@@ -68,6 +68,7 @@ def register_workflow_tools(server: FastMCP[Any], service: WfMcpService) -> None
         required_capabilities: (
             Mapping[str, RequiredCapability | dict[str, Any]] | None
         ) = None,
+        source_bindings: Mapping[str, str] | None = None,
         created_from_catalog_version: str | None = None,
     ) -> dict[str, Any]:
         return await handlers.create_artifact_from_plan(
@@ -87,6 +88,7 @@ def register_workflow_tools(server: FastMCP[Any], service: WfMcpService) -> None
                 for name, capability in (required_capabilities or {}).items()
             }
             or None,
+            source_bindings=dict(source_bindings or {}),
             created_from_catalog_version=created_from_catalog_version,
         )