WIP

Author: cameroncooke

src/mcp/tools/workflow-discovery/manage_workflows.ts

@@ -25,10 +25,12 @@ export async function manage_workflowsLogic(
   const requestedSet = new Set(
     workflowNames.map((name) => name.trim().toLowerCase()).filter(Boolean),
   );
-  const nextWorkflows =
-    params.enable === false
-      ? currentWorkflows.filter((name) => !requestedSet.has(name.toLowerCase()))
-      : [...new Set([...currentWorkflows, ...workflowNames])];
+  let nextWorkflows: string[];
+  if (params.enable === false) {
+    nextWorkflows = currentWorkflows.filter((name) => !requestedSet.has(name.toLowerCase()));
+  } else {
+    nextWorkflows = [...new Set([...currentWorkflows, ...workflowNames])];
+  }
   const registryState = await applyWorkflowSelection(nextWorkflows);
 
   return createTextResponse(`Workflows enabled: ${registryState.enabledWorkflows.join(', ')}`);

src/utils/tool-registry.ts

@@ -41,18 +41,19 @@ export async function applyWorkflowSelection(workflowNames: string[]): Promise<R
   for (const workflow of selection.selectedWorkflows) {
     desiredWorkflows.add(workflow.directoryName);
     for (const tool of workflow.tools) {
-      desiredToolNames.add(tool.name);
-      if (!registryState.tools.has(tool.name)) {
+      const { name, description, schema, annotations, handler } = tool;
+      desiredToolNames.add(name);
+      if (!registryState.tools.has(name)) {
         const registeredTool = server.registerTool(
-          tool.name,
+          name,
           {
-            description: tool.description ?? '',
-            inputSchema: tool.schema,
-            annotations: tool.annotations,
+            description: description ?? '',
+            inputSchema: schema,
+            annotations,
           },
-          (args: unknown): Promise<ToolResponse> => tool.handler(args as Record<string, unknown>),
+          (args: unknown): Promise<ToolResponse> => handler(args as Record<string, unknown>),
         );
-        registryState.tools.set(tool.name, registeredTool);
+        registryState.tools.set(name, registeredTool);
       }
     }
   }

src/utils/workflow-selection.ts

@@ -49,17 +49,24 @@ export function resolveSelectedWorkflowNames(
     baseAutoSelected.push(DEBUG_WORKFLOW);
   }
 
-  const selectedNames =
-    normalizedNames.length > 0 ? [...new Set([...baseAutoSelected, ...normalizedNames])] : null;
-
-  // Filter selected name to only include workflows that match real workflows
-  const selectedWorkflowNames = selectedNames
-    ? selectedNames.filter((workflowName) => availableWorkflowNames.includes(workflowName))
-    : isWorkflowDiscoveryEnabled()
-      ? [...availableWorkflowNames]
-      : availableWorkflowNames.filter(
-          (workflowName) => workflowName !== WORKFLOW_DISCOVERY_WORKFLOW,
-        );
+  let selectedNames: WorkflowName[] | null = null;
+  if (normalizedNames.length > 0) {
+    selectedNames = [...new Set([...baseAutoSelected, ...normalizedNames])];
+  }
+
+  // Filter selected names to only include workflows that match real workflows.
+  let selectedWorkflowNames: WorkflowName[];
+  if (selectedNames) {
+    selectedWorkflowNames = selectedNames.filter((workflowName) =>
+      availableWorkflowNames.includes(workflowName),
+    );
+  } else if (isWorkflowDiscoveryEnabled()) {
+    selectedWorkflowNames = [...availableWorkflowNames];
+  } else {
+    selectedWorkflowNames = availableWorkflowNames.filter(
+      (workflowName) => workflowName !== WORKFLOW_DISCOVERY_WORKFLOW,
+    );
+  }
 
   return { selectedWorkflowNames, selectedNames };
 }