From f2f856eec4d90ee2daccdc0770d391d78fac0440 Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Thu, 19 Mar 2026 00:40:08 +0800
Subject: [PATCH] fix: treat multimodal image models as image mode

---
 components/canvas/CanvasChatPanel.tsx | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/components/canvas/CanvasChatPanel.tsx b/components/canvas/CanvasChatPanel.tsx
index 478b9fd40..d97d80deb 100644
--- a/components/canvas/CanvasChatPanel.tsx
+++ b/components/canvas/CanvasChatPanel.tsx
@@ -2931,6 +2931,7 @@ export function CanvasChatPanel({
           model.id !== "auto" &&
           model.id !== modelsData?.agentModel?.id &&
           model.isMultimodal &&
+          !model.supportsImageGen &&
           !VIDEO_PROVIDER_TYPES.has(model.providerType)
       ),
     [allModels, modelsData?.agentModel?.id]
@@ -2945,6 +2946,7 @@ export function CanvasChatPanel({
     currentModel &&
       currentModel.id !== "auto" &&
       currentModel.isMultimodal &&
+      !currentModel.supportsImageGen &&
       !VIDEO_PROVIDER_TYPES.has(currentModel.providerType)
   );
   const isAgentModel =
@@ -3914,6 +3916,7 @@ export function CanvasChatPanel({
           requestModel &&
             requestModel.id !== "auto" &&
             requestModel.isMultimodal &&
+            !requestModel.supportsImageGen &&
             !requestIsVideoModel
         );
         hiddenPromptRef.current = null;
@@ -9978,6 +9981,7 @@ export function CanvasChatPanel({
                       nextModel &&
                         nextModel.id !== "auto" &&
                         nextModel.isMultimodal &&
+                        !nextModel.supportsImageGen &&
                         !nextIsVideoModel
                     );
                     currentModelIdRef.current = value;