From 4154486169bb6b7f4ffa9444ce3d9137cbd90e0e Mon Sep 17 00:00:00 2001
From: shrimbly <falloon.willie@gmail.com>
Date: Fri, 27 Feb 2026 23:16:39 +1300
Subject: [PATCH] feat: enable Gemini in video node's provider list

Add geminiApiKey to useProviderApiKeys hook. Update GenerateVideoNode to
include Gemini as a provider when API key is configured, and pass the
X-Gemini-API-Key header when fetching models.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 src/components/nodes/GenerateVideoNode.tsx | 15 +++++++++++----
 src/store/workflowStore.ts                 |  1 +
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/components/nodes/GenerateVideoNode.tsx b/src/components/nodes/GenerateVideoNode.tsx
index c9f8f2a5..dbcd2d86 100644
--- a/src/components/nodes/GenerateVideoNode.tsx
+++ b/src/components/nodes/GenerateVideoNode.tsx
@@ -25,7 +25,7 @@ export function GenerateVideoNode({ id, data, selected }: NodeProps<GenerateVide
   const commentNavigation = useCommentNavigation(id);
   const updateNodeData = useWorkflowStore((state) => state.updateNodeData);
   // Use stable selector for API keys to prevent unnecessary re-fetches
-  const { replicateApiKey, falApiKey, kieApiKey, replicateEnabled, kieEnabled } = useProviderApiKeys();
+  const { geminiApiKey, replicateApiKey, falApiKey, kieApiKey, replicateEnabled, kieEnabled } = useProviderApiKeys();
   const generationsPath = useWorkflowStore((state) => state.generationsPath);
   const [externalModels, setExternalModels] = useState<ProviderModel[]>([]);
   const [isLoadingModels, setIsLoadingModels] = useState(false);
@@ -37,9 +37,13 @@ export function GenerateVideoNode({ id, data, selected }: NodeProps<GenerateVide
   // Get the current selected provider (default to fal since Gemini doesn't do video)
   const currentProvider: ProviderType = nodeData.selectedModel?.provider || "fal";
 
-  // Get enabled providers (exclude Gemini since it doesn't do video)
+  // Get enabled providers
   const enabledProviders = useMemo(() => {
     const providers: { id: ProviderType; name: string }[] = [];
+    // Gemini available when API key is configured (settings or env var)
+    if (geminiApiKey) {
+      providers.push({ id: "gemini", name: "Gemini" });
+    }
     // fal.ai is always available (works without key but rate limited)
     providers.push({ id: "fal", name: "fal.ai" });
     // Add Replicate if configured
@@ -51,7 +55,7 @@ export function GenerateVideoNode({ id, data, selected }: NodeProps<GenerateVide
       providers.push({ id: "kie", name: "Kie.ai" });
     }
     return providers;
-  }, [replicateEnabled, replicateApiKey, kieEnabled, kieApiKey]);
+  }, [geminiApiKey, replicateEnabled, replicateApiKey, kieEnabled, kieApiKey]);
 
   // Fetch models from external providers when provider changes
   const fetchModels = useCallback(async () => {
@@ -60,6 +64,9 @@ export function GenerateVideoNode({ id, data, selected }: NodeProps<GenerateVide
     try {
       const capabilities = VIDEO_CAPABILITIES.join(",");
       const headers: HeadersInit = {};
+      if (geminiApiKey) {
+        headers["X-Gemini-API-Key"] = geminiApiKey;
+      }
       if (replicateApiKey) {
         headers["X-Replicate-Key"] = replicateApiKey;
       }
@@ -91,7 +98,7 @@ export function GenerateVideoNode({ id, data, selected }: NodeProps<GenerateVide
     } finally {
       setIsLoadingModels(false);
     }
-  }, [currentProvider, replicateApiKey, falApiKey, kieApiKey]);
+  }, [currentProvider, geminiApiKey, replicateApiKey, falApiKey, kieApiKey]);
 
   useEffect(() => {
     fetchModels();
diff --git a/src/store/workflowStore.ts b/src/store/workflowStore.ts
index 58ac2cf1..b119ecc7 100644
--- a/src/store/workflowStore.ts
+++ b/src/store/workflowStore.ts
@@ -2136,6 +2136,7 @@ export const useWorkflowStore = create<WorkflowStore>()(workflowStoreImpl);
 export function useProviderApiKeys() {
   return useWorkflowStore(
     useShallow((state) => ({
+      geminiApiKey: state.providerSettings.providers.gemini?.apiKey ?? null,
       replicateApiKey: state.providerSettings.providers.replicate?.apiKey ?? null,
       falApiKey: state.providerSettings.providers.fal?.apiKey ?? null,
       kieApiKey: state.providerSettings.providers.kie?.apiKey ?? null,