From 28c3edbe5eec8551526986cf32fcec93d84e9c94 Mon Sep 17 00:00:00 2001
From: shrimbly <falloon.willie@gmail.com>
Date: Fri, 9 Jan 2026 23:47:25 +1300
Subject: [PATCH] fix(06-02): create generateVideo node for video models in
 browser

- Check model capabilities when selecting from model browser
- Create generateVideo node for text-to-video or image-to-video models
- Create nanoBanana node for image generation models
- Pass selectedModel with provider, modelId, displayName to node
---
 src/components/modals/ModelSearchDialog.tsx | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/components/modals/ModelSearchDialog.tsx b/src/components/modals/ModelSearchDialog.tsx
index 484b036c..0a572b97 100644
--- a/src/components/modals/ModelSearchDialog.tsx
+++ b/src/components/modals/ModelSearchDialog.tsx
@@ -175,7 +175,14 @@ export function ModelSearchDialog({
         y: center.y + Math.random() * 100 - 50,
       });
 
-      addNode("nanoBanana", position, {
+      // Determine node type based on model capabilities
+      const isVideoModel = model.capabilities.some(
+        (cap) => cap === "text-to-video" || cap === "image-to-video"
+      );
+
+      const nodeType = isVideoModel ? "generateVideo" : "nanoBanana";
+
+      addNode(nodeType, position, {
         selectedModel: {
           provider: model.provider,
           modelId: model.id,