sudden code review

src/wf_mcp/broker/catalog.py

@@ -130,9 +130,7 @@ class CombinedCatalog:
                     "description": entry.description,
                     "outcomes": list(entry.outcomes),
                     "input_schema": entry.input_schema,
-                    "output_schema": workflow_output_schema_from_mcp_tool_schema(
-                        entry.output_schema
-                    ),
+                    "output_schema": entry.output_schema,
                 }
                 for entry in self.entries()
             ],

src/wf_mcp/broker/service/core.py

@@ -49,9 +49,8 @@ from ...models import (
     ConnectionConfig,
     RawWorkflowPlan,
 )
-from ...runtime import ToolExecutor
-from ...sdk.converters import workflow_output_schema_from_mcp_tool_schema
 from ...sdk import BackendAdapter
+from ...runtime import ToolExecutor
 from ...shared.errors import error_payload
 from ...shared.names import RESERVED_CONNECTION_IDS
 from ...storage import Store
@@ -876,7 +875,7 @@ class WfMcpService:
         """
         model_prefix = entry.qualified_name.replace(".", "_").replace("-", "_")
         input_model = _model_from_schema(f"{model_prefix}_Input", entry.input_schema)
-        output_schema = workflow_output_schema_from_mcp_tool_schema(entry.output_schema)
+        output_schema = entry.output_schema
         output_model = _model_from_schema(f"{model_prefix}_Output", output_schema)
 
         async def invoke_tool(payload: BaseModel) -> NodeReturn[BaseModel]: