From 3e0ede096e048ecca412a06e2b6f193645465905 Mon Sep 17 00:00:00 2001
From: shrimbly <falloon.willie@gmail.com>
Date: Fri, 27 Feb 2026 06:52:07 +1300
Subject: [PATCH] fix: remove develop-only node types from master and add type
 cast

- Remove router/switch/conditionalSwitch entries from WorkflowCanvas
  defaultDimensions and validation.ts (these types only exist on develop)
- Add ModelType import and cast in nodeDefaults.ts for type safety

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 src/components/WorkflowCanvas.tsx | 3 ---
 src/lib/quickstart/validation.ts  | 6 ------
 src/store/utils/nodeDefaults.ts   | 3 ++-
 3 files changed, 2 insertions(+), 10 deletions(-)

diff --git a/src/components/WorkflowCanvas.tsx b/src/components/WorkflowCanvas.tsx
index 28e42c67..2ae6d4fd 100644
--- a/src/components/WorkflowCanvas.tsx
+++ b/src/components/WorkflowCanvas.tsx
@@ -1152,9 +1152,6 @@ export function WorkflowCanvas() {
             easeCurve: { width: 340, height: 480 },
             videoTrim: { width: 360, height: 360 },
             videoFrameGrab: { width: 320, height: 320 },
-            router: { width: 200, height: 80 },
-            switch: { width: 220, height: 120 },
-            conditionalSwitch: { width: 260, height: 180 },
             glbViewer: { width: 360, height: 380 },
           };
           const dims = defaultDimensions[nodeType];
diff --git a/src/lib/quickstart/validation.ts b/src/lib/quickstart/validation.ts
index 39cbcb24..4afce636 100644
--- a/src/lib/quickstart/validation.ts
+++ b/src/lib/quickstart/validation.ts
@@ -387,12 +387,6 @@ function createDefaultNodeData(type: NodeType): WorkflowNodeData {
         status: "idle",
         error: null,
       };
-    case "router":
-      return {};
-    case "switch":
-      return { inputType: null, switches: [{ id: "sw-1", name: "Output 1", enabled: true }] };
-    case "conditionalSwitch":
-      return { incomingText: null, rules: [{ id: "rule-1", value: "", mode: "contains", label: "Rule 1", isMatched: false }] };
     case "glbViewer":
       return {
         glbUrl: null,
diff --git a/src/store/utils/nodeDefaults.ts b/src/store/utils/nodeDefaults.ts
index c3ab6832..84d079c1 100644
--- a/src/store/utils/nodeDefaults.ts
+++ b/src/store/utils/nodeDefaults.ts
@@ -1,5 +1,6 @@
 import {
   NodeType,
+  ModelType,
   ImageInputNodeData,
   AudioInputNodeData,
   AnnotationNodeData,
@@ -128,7 +129,7 @@ export const createDefaultNodeData = (type: NodeType): WorkflowNodeData => {
       if (nodeDefaults.generateImage?.selectedModel) {
         selectedModel = nodeDefaults.generateImage.selectedModel;
       } else {
-        const modelDisplayName = MODEL_DISPLAY_NAMES[legacyDefaults.model] || legacyDefaults.model;
+        const modelDisplayName = MODEL_DISPLAY_NAMES[legacyDefaults.model as ModelType] || legacyDefaults.model;
         selectedModel = {
           provider: "gemini",
           modelId: legacyDefaults.model,