@qwen-code/qwen-code 0.0.13-nightly.4 → 0.0.13

package/dist/src/ui/hooks/useVisionAutoSwitch.js

@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import {} from '@google/genai';
-import { AuthType } from '@qwen-code/qwen-code-core';
+import { AuthType, ApprovalMode } from '@qwen-code/qwen-code-core';
 import { useCallback, useRef } from 'react';
 import { VisionSwitchOutcome } from '../components/ModelSwitchDialog.js';
 import { getDefaultVisionModel, isVisionModel, } from '../models/availableModels.js';
@@ -88,7 +88,7 @@ function checkImageFormatsSupport(parts) {
 /**
  * Determines if we should offer vision switch for the given parts, auth type, and current model
  */
-export function shouldOfferVisionSwitch(parts, authType, currentModel, visionModelPreviewEnabled = false) {
+export function shouldOfferVisionSwitch(parts, authType, currentModel, visionModelPreviewEnabled = true) {
     // Only trigger for qwen-oauth
     if (authType !== AuthType.QWEN_OAUTH) {
         return false;
@@ -114,10 +114,10 @@ export function processVisionSwitchOutcome(outcome) {
             return { modelOverride: vlModelId };
         case VisionSwitchOutcome.SwitchSessionToVL:
             return { persistSessionModel: vlModelId };
-        case VisionSwitchOutcome.DisallowWithGuidance:
-            return { showGuidance: true };
+        case VisionSwitchOutcome.ContinueWithCurrentModel:
+            return {}; // Continue with current model, no changes needed
         default:
-            return { showGuidance: true };
+            return {}; // Default to continuing with current model
     }
 }
 /**
@@ -132,7 +132,7 @@ export function getVisionSwitchGuidanceMessage() {
 /**
  * Custom hook for handling vision model auto-switching
  */
-export function useVisionAutoSwitch(config, addItem, visionModelPreviewEnabled = false, onVisionSwitchRequired) {
+export function useVisionAutoSwitch(config, addItem, visionModelPreviewEnabled = true, onVisionSwitchRequired) {
     const originalModelRef = useRef(null);
     const handleVisionSwitch = useCallback(async (query, userMessageTimestamp, isContinuation) => {
         // Skip vision switch handling for continuations or if no handler provided
@@ -158,20 +158,72 @@ export function useVisionAutoSwitch(config, addItem, visionModelPreviewEnabled =
         if (!shouldOfferVisionSwitch(query, contentGeneratorConfig.authType, config.getModel(), visionModelPreviewEnabled)) {
             return { shouldProceed: true };
         }
+        // In YOLO mode, automatically switch to vision model without user interaction
+        if (config.getApprovalMode() === ApprovalMode.YOLO) {
+            const vlModelId = getDefaultVisionModel();
+            originalModelRef.current = config.getModel();
+            await config.setModel(vlModelId, {
+                reason: 'vision_auto_switch',
+                context: 'YOLO mode auto-switch for image content',
+            });
+            return {
+                shouldProceed: true,
+                originalModel: originalModelRef.current,
+            };
+        }
+        // Check if there's a default VLM switch mode configured
+        const defaultVlmSwitchMode = config.getVlmSwitchMode();
+        if (defaultVlmSwitchMode) {
+            // Convert string value to VisionSwitchOutcome enum
+            let outcome;
+            switch (defaultVlmSwitchMode) {
+                case 'once':
+                    outcome = VisionSwitchOutcome.SwitchOnce;
+                    break;
+                case 'session':
+                    outcome = VisionSwitchOutcome.SwitchSessionToVL;
+                    break;
+                case 'persist':
+                    outcome = VisionSwitchOutcome.ContinueWithCurrentModel;
+                    break;
+                default:
+                    // Invalid value, fall back to prompting user
+                    outcome = VisionSwitchOutcome.ContinueWithCurrentModel;
+            }
+            // Process the default outcome
+            const visionSwitchResult = processVisionSwitchOutcome(outcome);
+            if (visionSwitchResult.modelOverride) {
+                // One-time model override
+                originalModelRef.current = config.getModel();
+                await config.setModel(visionSwitchResult.modelOverride, {
+                    reason: 'vision_auto_switch',
+                    context: `Default VLM switch mode: ${defaultVlmSwitchMode} (one-time override)`,
+                });
+                return {
+                    shouldProceed: true,
+                    originalModel: originalModelRef.current,
+                };
+            }
+            else if (visionSwitchResult.persistSessionModel) {
+                // Persistent session model change
+                await config.setModel(visionSwitchResult.persistSessionModel, {
+                    reason: 'vision_auto_switch',
+                    context: `Default VLM switch mode: ${defaultVlmSwitchMode} (session persistent)`,
+                });
+                return { shouldProceed: true };
+            }
+            // For ContinueWithCurrentModel or any other case, proceed with current model
+            return { shouldProceed: true };
+        }
         try {
             const visionSwitchResult = await onVisionSwitchRequired(query);
-            if (visionSwitchResult.showGuidance) {
-                // Show guidance and don't proceed with the request
-                addItem({
-                    type: MessageType.INFO,
-                    text: getVisionSwitchGuidanceMessage(),
-                }, userMessageTimestamp);
-                return { shouldProceed: false };
-            }
             if (visionSwitchResult.modelOverride) {
                 // One-time model override
                 originalModelRef.current = config.getModel();
-                config.setModel(visionSwitchResult.modelOverride);
+                await config.setModel(visionSwitchResult.modelOverride, {
+                    reason: 'vision_auto_switch',
+                    context: 'User-prompted vision switch (one-time override)',
+                });
                 return {
                     shouldProceed: true,
                     originalModel: originalModelRef.current,
@@ -179,9 +231,13 @@ export function useVisionAutoSwitch(config, addItem, visionModelPreviewEnabled =
             }
             else if (visionSwitchResult.persistSessionModel) {
                 // Persistent session model change
-                config.setModel(visionSwitchResult.persistSessionModel);
+                await config.setModel(visionSwitchResult.persistSessionModel, {
+                    reason: 'vision_auto_switch',
+                    context: 'User-prompted vision switch (session persistent)',
+                });
                 return { shouldProceed: true };
             }
+            // For ContinueWithCurrentModel or any other case, proceed with current model
             return { shouldProceed: true };
         }
         catch (_error) {
@@ -189,9 +245,12 @@ export function useVisionAutoSwitch(config, addItem, visionModelPreviewEnabled =
             return { shouldProceed: false };
         }
     }, [config, addItem, visionModelPreviewEnabled, onVisionSwitchRequired]);
-    const restoreOriginalModel = useCallback(() => {
+    const restoreOriginalModel = useCallback(async () => {
         if (originalModelRef.current) {
-            config.setModel(originalModelRef.current);
+            await config.setModel(originalModelRef.current, {
+                reason: 'vision_auto_switch',
+                context: 'Restoring original model after vision switch',
+            });
             originalModelRef.current = null;
         }
     }, [config]);

package/dist/src/ui/hooks/useVisionAutoSwitch.js.map

@@ -1 +1 @@
-{"version":3,"file":"useVisionAutoSwitch.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useVisionAutoSwitch.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAiC,MAAM,eAAe,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAe,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,EACL,qBAAqB,EACrB,aAAa,GACd,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,EACL,wBAAwB,EACxB,gCAAgC,GACjC,MAAM,2BAA2B,CAAC;AAEnC;;GAEG;AACH,SAAS,aAAa,CAAC,KAAoB;IACzC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACzB,oBAAoB;YACpB,IAAI,OAAO,IAAI,KAAK,QAAQ;gBAAE,OAAO,KAAK,CAAC;YAC3C,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,+DAA+D;IAC/D,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,IAAU;IAC7B,4CAA4C;IAC5C,IAAI,YAAY,IAAI,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,0CAA0C;IAC1C,IAAI,UAAU,IAAI,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,wBAAwB,CAAC,KAAoB;IAKpD,MAAM,oBAAoB,GAAa,EAAE,CAAC;IAC1C,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,qBAAqB,EAAE,KAAK;YAC5B,oBAAoB,EAAE,EAAE;SACzB,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAE1D,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,SAAS;QAEvC,IAAI,QAA4B,CAAC;QAEjC,mBAAmB;QACnB,IACE,YAAY,IAAI,IAAI;YACpB,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,CAAC,EAC/C,CAAC;YACD,SAAS,GAAG,IAAI,CAAC;YACjB,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;QACtC,CAAC;QAED,iBAAiB;QACjB,IAAI,UAAU,IAAI,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACxE,SAAS,GAAG,IAAI,CAAC;YACjB,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACpC,CAAC;QAED,sCAAsC;QACtC,IAAI,QAAQ,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpD,oBAAoB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,OAAO;QACL,SAAS;QACT,qBAAqB,EAAE,oBAAoB,CAAC,MAAM,GAAG,CAAC;QACtD,oBAAoB;KACrB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CACrC,KAAoB,EACpB,QAAkB,EAClB,YAAoB,EACpB,4BAAqC,KAAK;IAE1C,8BAA8B;IAC9B,IAAI,QAAQ,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;QACrC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,iEAAiE;IACjE,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAC/B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gEAAgE;IAChE,IAAI,aAAa,CAAC,YAAY,CAAC,EAAE,CAAC;QAChC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oDAAoD;IACpD,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AAWD;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACxC,OAA4B;IAE5B,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;IAE1C,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,mBAAmB,CAAC,UAAU;YACjC,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC;QAEtC,KAAK,mBAAmB,CAAC,iBAAiB;YACxC,OAAO,EAAE,mBAAmB,EAAE,SAAS,EAAE,CAAC;QAE5C,KAAK,mBAAmB,CAAC,oBAAoB;YAC3C,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QAEhC;YACE,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;IAClC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,8BAA8B;IAC5C,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;IAC1C,OAAO;mBACU,SAAS;6DACiC,CAAC;AAC9D,CAAC;AAUD;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,OAA2C,EAC3C,4BAAqC,KAAK,EAC1C,sBAIE;IAEF,MAAM,gBAAgB,GAAG,MAAM,CAAgB,IAAI,CAAC,CAAC;IAErD,MAAM,kBAAkB,GAAG,WAAW,CACpC,KAAK,EACH,KAAoB,EACpB,oBAA4B,EAC5B,cAAuB,EACc,EAAE;QACvC,0EAA0E;QAC1E,IAAI,cAAc,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAC9C,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAED,MAAM,sBAAsB,GAAG,MAAM,CAAC,yBAAyB,EAAE,CAAC;QAElE,mCAAmC;QACnC,IAAI,sBAAsB,EAAE,QAAQ,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC7D,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAED,mCAAmC;QACnC,MAAM,WAAW,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;QAEpD,uDAAuD;QACvD,IAAI,WAAW,CAAC,qBAAqB,EAAE,CAAC;YACtC,OAAO,CACL;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,gCAAgC,EAAE;aACzC,EACD,oBAAoB,CACrB,CAAC;YACF,6CAA6C;QAC/C,CAAC;QAED,mCAAmC;QACnC,IACE,CAAC,uBAAuB,CACtB,KAAK,EACL,sBAAsB,CAAC,QAAQ,EAC/B,MAAM,CAAC,QAAQ,EAAE,EACjB,yBAAyB,CAC1B,EACD,CAAC;YACD,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,kBAAkB,GAAG,MAAM,sBAAsB,CAAC,KAAK,CAAC,CAAC;YAE/D,IAAI,kBAAkB,CAAC,YAAY,EAAE,CAAC;gBACpC,mDAAmD;gBACnD,OAAO,CACL;oBACE,IAAI,EAAE,WAAW,CAAC,IAAI;oBACtB,IAAI,EAAE,8BAA8B,EAAE;iBACvC,EACD,oBAAoB,CACrB,CAAC;gBACF,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;YAClC,CAAC;YAED,IAAI,kBAAkB,CAAC,aAAa,EAAE,CAAC;gBACrC,0BAA0B;gBAC1B,gBAAgB,CAAC,OAAO,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAC7C,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,aAAa,CAAC,CAAC;gBAClD,OAAO;oBACL,aAAa,EAAE,IAAI;oBACnB,aAAa,EAAE,gBAAgB,CAAC,OAAO;iBACxC,CAAC;YACJ,CAAC;iBAAM,IAAI,kBAAkB,CAAC,mBAAmB,EAAE,CAAC;gBAClD,kCAAkC;gBAClC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;gBACxD,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC;YAED,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAAC,OAAO,MAAM,EAAE,CAAC;YAChB,kEAAkE;YAClE,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;QAClC,CAAC;IACH,CAAC,EACD,CAAC,MAAM,EAAE,OAAO,EAAE,yBAAyB,EAAE,sBAAsB,CAAC,CACrE,CAAC;IAEF,MAAM,oBAAoB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC5C,IAAI,gBAAgB,CAAC,OAAO,EAAE,CAAC;YAC7B,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAC1C,gBAAgB,CAAC,OAAO,GAAG,IAAI,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEb,OAAO;QACL,kBAAkB;QAClB,oBAAoB;KACrB,CAAC;AACJ,CAAC"}
+{"version":3,"file":"useVisionAutoSwitch.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useVisionAutoSwitch.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAiC,MAAM,eAAe,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAe,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAChF,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,EACL,qBAAqB,EACrB,aAAa,GACd,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,EACL,wBAAwB,EACxB,gCAAgC,GACjC,MAAM,2BAA2B,CAAC;AAEnC;;GAEG;AACH,SAAS,aAAa,CAAC,KAAoB;IACzC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACzB,oBAAoB;YACpB,IAAI,OAAO,IAAI,KAAK,QAAQ;gBAAE,OAAO,KAAK,CAAC;YAC3C,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,+DAA+D;IAC/D,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,IAAU;IAC7B,4CAA4C;IAC5C,IAAI,YAAY,IAAI,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,0CAA0C;IAC1C,IAAI,UAAU,IAAI,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,wBAAwB,CAAC,KAAoB;IAKpD,MAAM,oBAAoB,GAAa,EAAE,CAAC;IAC1C,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,qBAAqB,EAAE,KAAK;YAC5B,oBAAoB,EAAE,EAAE;SACzB,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAE1D,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,SAAS;QAEvC,IAAI,QAA4B,CAAC;QAEjC,mBAAmB;QACnB,IACE,YAAY,IAAI,IAAI;YACpB,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,CAAC,EAC/C,CAAC;YACD,SAAS,GAAG,IAAI,CAAC;YACjB,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;QACtC,CAAC;QAED,iBAAiB;QACjB,IAAI,UAAU,IAAI,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACxE,SAAS,GAAG,IAAI,CAAC;YACjB,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACpC,CAAC;QAED,sCAAsC;QACtC,IAAI,QAAQ,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpD,oBAAoB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,OAAO;QACL,SAAS;QACT,qBAAqB,EAAE,oBAAoB,CAAC,MAAM,GAAG,CAAC;QACtD,oBAAoB;KACrB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CACrC,KAAoB,EACpB,QAAkB,EAClB,YAAoB,EACpB,4BAAqC,IAAI;IAEzC,8BAA8B;IAC9B,IAAI,QAAQ,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;QACrC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,iEAAiE;IACjE,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAC/B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gEAAgE;IAChE,IAAI,aAAa,CAAC,YAAY,CAAC,EAAE,CAAC;QAChC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oDAAoD;IACpD,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AAWD;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACxC,OAA4B;IAE5B,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;IAE1C,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,mBAAmB,CAAC,UAAU;YACjC,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC;QAEtC,KAAK,mBAAmB,CAAC,iBAAiB;YACxC,OAAO,EAAE,mBAAmB,EAAE,SAAS,EAAE,CAAC;QAE5C,KAAK,mBAAmB,CAAC,wBAAwB;YAC/C,OAAO,EAAE,CAAC,CAAC,iDAAiD;QAE9D;YACE,OAAO,EAAE,CAAC,CAAC,2CAA2C;IAC1D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,8BAA8B;IAC5C,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;IAC1C,OAAO;mBACU,SAAS;6DACiC,CAAC;AAC9D,CAAC;AAUD;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,OAA2C,EAC3C,4BAAqC,IAAI,EACzC,sBAIE;IAEF,MAAM,gBAAgB,GAAG,MAAM,CAAgB,IAAI,CAAC,CAAC;IAErD,MAAM,kBAAkB,GAAG,WAAW,CACpC,KAAK,EACH,KAAoB,EACpB,oBAA4B,EAC5B,cAAuB,EACc,EAAE;QACvC,0EAA0E;QAC1E,IAAI,cAAc,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAC9C,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAED,MAAM,sBAAsB,GAAG,MAAM,CAAC,yBAAyB,EAAE,CAAC;QAElE,mCAAmC;QACnC,IAAI,sBAAsB,EAAE,QAAQ,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC7D,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAED,mCAAmC;QACnC,MAAM,WAAW,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;QAEpD,uDAAuD;QACvD,IAAI,WAAW,CAAC,qBAAqB,EAAE,CAAC;YACtC,OAAO,CACL;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,gCAAgC,EAAE;aACzC,EACD,oBAAoB,CACrB,CAAC;YACF,6CAA6C;QAC/C,CAAC;QAED,mCAAmC;QACnC,IACE,CAAC,uBAAuB,CACtB,KAAK,EACL,sBAAsB,CAAC,QAAQ,EAC/B,MAAM,CAAC,QAAQ,EAAE,EACjB,yBAAyB,CAC1B,EACD,CAAC;YACD,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAED,8EAA8E;QAC9E,IAAI,MAAM,CAAC,eAAe,EAAE,KAAK,YAAY,CAAC,IAAI,EAAE,CAAC;YACnD,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;YAC1C,gBAAgB,CAAC,OAAO,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;YAC7C,MAAM,MAAM,CAAC,QAAQ,CAAC,SAAS,EAAE;gBAC/B,MAAM,EAAE,oBAAoB;gBAC5B,OAAO,EAAE,yCAAyC;aACnD,CAAC,CAAC;YACH,OAAO;gBACL,aAAa,EAAE,IAAI;gBACnB,aAAa,EAAE,gBAAgB,CAAC,OAAO;aACxC,CAAC;QACJ,CAAC;QAED,wDAAwD;QACxD,MAAM,oBAAoB,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;QACvD,IAAI,oBAAoB,EAAE,CAAC;YACzB,mDAAmD;YACnD,IAAI,OAA4B,CAAC;YACjC,QAAQ,oBAAoB,EAAE,CAAC;gBAC7B,KAAK,MAAM;oBACT,OAAO,GAAG,mBAAmB,CAAC,UAAU,CAAC;oBACzC,MAAM;gBACR,KAAK,SAAS;oBACZ,OAAO,GAAG,mBAAmB,CAAC,iBAAiB,CAAC;oBAChD,MAAM;gBACR,KAAK,SAAS;oBACZ,OAAO,GAAG,mBAAmB,CAAC,wBAAwB,CAAC;oBACvD,MAAM;gBACR;oBACE,6CAA6C;oBAC7C,OAAO,GAAG,mBAAmB,CAAC,wBAAwB,CAAC;YAC3D,CAAC;YAED,8BAA8B;YAC9B,MAAM,kBAAkB,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;YAE/D,IAAI,kBAAkB,CAAC,aAAa,EAAE,CAAC;gBACrC,0BAA0B;gBAC1B,gBAAgB,CAAC,OAAO,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAC7C,MAAM,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,aAAa,EAAE;oBACtD,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE,4BAA4B,oBAAoB,sBAAsB;iBAChF,CAAC,CAAC;gBACH,OAAO;oBACL,aAAa,EAAE,IAAI;oBACnB,aAAa,EAAE,gBAAgB,CAAC,OAAO;iBACxC,CAAC;YACJ,CAAC;iBAAM,IAAI,kBAAkB,CAAC,mBAAmB,EAAE,CAAC;gBAClD,kCAAkC;gBAClC,MAAM,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,mBAAmB,EAAE;oBAC5D,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE,4BAA4B,oBAAoB,uBAAuB;iBACjF,CAAC,CAAC;gBACH,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC;YAED,6EAA6E;YAC7E,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,kBAAkB,GAAG,MAAM,sBAAsB,CAAC,KAAK,CAAC,CAAC;YAE/D,IAAI,kBAAkB,CAAC,aAAa,EAAE,CAAC;gBACrC,0BAA0B;gBAC1B,gBAAgB,CAAC,OAAO,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAC7C,MAAM,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,aAAa,EAAE;oBACtD,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE,iDAAiD;iBAC3D,CAAC,CAAC;gBACH,OAAO;oBACL,aAAa,EAAE,IAAI;oBACnB,aAAa,EAAE,gBAAgB,CAAC,OAAO;iBACxC,CAAC;YACJ,CAAC;iBAAM,IAAI,kBAAkB,CAAC,mBAAmB,EAAE,CAAC;gBAClD,kCAAkC;gBAClC,MAAM,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,mBAAmB,EAAE;oBAC5D,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE,kDAAkD;iBAC5D,CAAC,CAAC;gBACH,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC;YAED,6EAA6E;YAC7E,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;QAAC,OAAO,MAAM,EAAE,CAAC;YAChB,kEAAkE;YAClE,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;QAClC,CAAC;IACH,CAAC,EACD,CAAC,MAAM,EAAE,OAAO,EAAE,yBAAyB,EAAE,sBAAsB,CAAC,CACrE,CAAC;IAEF,MAAM,oBAAoB,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;QAClD,IAAI,gBAAgB,CAAC,OAAO,EAAE,CAAC;YAC7B,MAAM,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,OAAO,EAAE;gBAC9C,MAAM,EAAE,oBAAoB;gBAC5B,OAAO,EAAE,8CAA8C;aACxD,CAAC,CAAC;YACH,gBAAgB,CAAC,OAAO,GAAG,IAAI,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEb,OAAO;QACL,kBAAkB;QAClB,oBAAoB;KACrB,CAAC;AACJ,CAAC"}

package/dist/src/ui/hooks/useVisionAutoSwitch.test.js

@@ -6,7 +6,7 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import { describe, it, expect, vi, beforeEach } from 'vitest';
 import { renderHook, act } from '@testing-library/react';
-import { AuthType } from '@qwen-code/qwen-code-core';
+import { AuthType, ApprovalMode } from '@qwen-code/qwen-code-core';
 import { shouldOfferVisionSwitch, processVisionSwitchOutcome, getVisionSwitchGuidanceMessage, useVisionAutoSwitch, } from './useVisionAutoSwitch.js';
 import { VisionSwitchOutcome } from '../components/ModelSwitchDialog.js';
 import { MessageType } from '../types.js';
@@ -24,7 +24,7 @@ describe('useVisionAutoSwitch helpers', () => {
             const parts = [
                 { inlineData: { mimeType: 'image/png', data: '...' } },
             ];
-            const result = shouldOfferVisionSwitch(parts, AuthType.QWEN_OAUTH, 'qwen-vl-max-latest', true);
+            const result = shouldOfferVisionSwitch(parts, AuthType.QWEN_OAUTH, 'vision-model', true);
             expect(result).toBe(false);
         });
         it('returns true when image parts exist, QWEN_OAUTH, and model is not vision', () => {
@@ -59,6 +59,32 @@ describe('useVisionAutoSwitch helpers', () => {
             const result = shouldOfferVisionSwitch(parts, AuthType.QWEN_OAUTH, 'qwen3-coder-plus', false);
             expect(result).toBe(false);
         });
+        it('returns true when image parts exist in YOLO mode context', () => {
+            const parts = [
+                { inlineData: { mimeType: 'image/png', data: '...' } },
+            ];
+            const result = shouldOfferVisionSwitch(parts, AuthType.QWEN_OAUTH, 'qwen3-coder-plus', true);
+            expect(result).toBe(true);
+        });
+        it('returns false when no image parts exist in YOLO mode context', () => {
+            const parts = [{ text: 'just text' }];
+            const result = shouldOfferVisionSwitch(parts, AuthType.QWEN_OAUTH, 'qwen3-coder-plus', true);
+            expect(result).toBe(false);
+        });
+        it('returns false when already using vision model in YOLO mode context', () => {
+            const parts = [
+                { inlineData: { mimeType: 'image/png', data: '...' } },
+            ];
+            const result = shouldOfferVisionSwitch(parts, AuthType.QWEN_OAUTH, 'vision-model', true);
+            expect(result).toBe(false);
+        });
+        it('returns false when authType is not QWEN_OAUTH in YOLO mode context', () => {
+            const parts = [
+                { inlineData: { mimeType: 'image/png', data: '...' } },
+            ];
+            const result = shouldOfferVisionSwitch(parts, AuthType.USE_GEMINI, 'qwen3-coder-plus', true);
+            expect(result).toBe(false);
+        });
     });
     describe('processVisionSwitchOutcome', () => {
         it('maps SwitchOnce to a one-time model override', () => {
@@ -71,9 +97,9 @@ describe('useVisionAutoSwitch helpers', () => {
             const result = processVisionSwitchOutcome(VisionSwitchOutcome.SwitchSessionToVL);
             expect(result).toEqual({ persistSessionModel: vl });
         });
-        it('maps DisallowWithGuidance to showGuidance', () => {
-            const result = processVisionSwitchOutcome(VisionSwitchOutcome.DisallowWithGuidance);
-            expect(result).toEqual({ showGuidance: true });
+        it('maps ContinueWithCurrentModel to empty result', () => {
+            const result = processVisionSwitchOutcome(VisionSwitchOutcome.ContinueWithCurrentModel);
+            expect(result).toEqual({});
         });
     });
     describe('getVisionSwitchGuidanceMessage', () => {
@@ -87,13 +113,15 @@ describe('useVisionAutoSwitch helpers', () => {
     });
 });
 describe('useVisionAutoSwitch hook', () => {
-    const createMockConfig = (authType, initialModel) => {
+    const createMockConfig = (authType, initialModel, approvalMode = ApprovalMode.DEFAULT, vlmSwitchMode) => {
         let currentModel = initialModel;
         const mockConfig = {
             getModel: vi.fn(() => currentModel),
-            setModel: vi.fn((m) => {
+            setModel: vi.fn(async (m) => {
                 currentModel = m;
             }),
+            getApprovalMode: vi.fn(() => approvalMode),
+            getVlmSwitchMode: vi.fn(() => vlmSwitchMode),
             getContentGeneratorConfig: vi.fn(() => ({
                 authType,
                 model: currentModel,
@@ -147,11 +175,9 @@ describe('useVisionAutoSwitch hook', () => {
         expect(res).toEqual({ shouldProceed: true });
         expect(onVisionSwitchRequired).not.toHaveBeenCalled();
     });
-    it('shows guidance and blocks when dialog returns showGuidance', async () => {
+    it('continues with current model when dialog returns empty result', async () => {
         const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus');
-        const onVisionSwitchRequired = vi
-            .fn()
-            .mockResolvedValue({ showGuidance: true });
+        const onVisionSwitchRequired = vi.fn().mockResolvedValue({}); // Empty result for ContinueWithCurrentModel
         const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
         const parts = [
             { inlineData: { mimeType: 'image/png', data: '...' } },
@@ -161,8 +187,9 @@ describe('useVisionAutoSwitch hook', () => {
         await act(async () => {
             res = await result.current.handleVisionSwitch(parts, userTs, false);
         });
-        expect(addItem).toHaveBeenCalledWith({ type: MessageType.INFO, text: getVisionSwitchGuidanceMessage() }, userTs);
-        expect(res).toEqual({ shouldProceed: false });
+        // Should not add any guidance message
+        expect(addItem).not.toHaveBeenCalledWith({ type: MessageType.INFO, text: getVisionSwitchGuidanceMessage() }, userTs);
+        expect(res).toEqual({ shouldProceed: true });
         expect(config.setModel).not.toHaveBeenCalled();
     });
     it('applies a one-time override and returns originalModel, then restores', async () => {
@@ -170,7 +197,7 @@ describe('useVisionAutoSwitch hook', () => {
         const config = createMockConfig(AuthType.QWEN_OAUTH, initialModel);
         const onVisionSwitchRequired = vi
             .fn()
-            .mockResolvedValue({ modelOverride: 'qwen-vl-max-latest' });
+            .mockResolvedValue({ modelOverride: 'coder-model' });
         const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
         const parts = [
             { inlineData: { mimeType: 'image/png', data: '...' } },
@@ -180,18 +207,24 @@ describe('useVisionAutoSwitch hook', () => {
             res = await result.current.handleVisionSwitch(parts, 2020, false);
         });
         expect(res).toEqual({ shouldProceed: true, originalModel: initialModel });
-        expect(config.setModel).toHaveBeenCalledWith('qwen-vl-max-latest');
+        expect(config.setModel).toHaveBeenCalledWith('coder-model', {
+            reason: 'vision_auto_switch',
+            context: 'User-prompted vision switch (one-time override)',
+        });
         // Now restore
-        act(() => {
-            result.current.restoreOriginalModel();
+        await act(async () => {
+            await result.current.restoreOriginalModel();
+        });
+        expect(config.setModel).toHaveBeenLastCalledWith(initialModel, {
+            reason: 'vision_auto_switch',
+            context: 'Restoring original model after vision switch',
         });
-        expect(config.setModel).toHaveBeenLastCalledWith(initialModel);
     });
     it('persists session model when dialog requests persistence', async () => {
         const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus');
         const onVisionSwitchRequired = vi
             .fn()
-            .mockResolvedValue({ persistSessionModel: 'qwen-vl-max-latest' });
+            .mockResolvedValue({ persistSessionModel: 'coder-model' });
         const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
         const parts = [
             { inlineData: { mimeType: 'image/png', data: '...' } },
@@ -201,13 +234,16 @@ describe('useVisionAutoSwitch hook', () => {
             res = await result.current.handleVisionSwitch(parts, 3030, false);
         });
         expect(res).toEqual({ shouldProceed: true });
-        expect(config.setModel).toHaveBeenCalledWith('qwen-vl-max-latest');
+        expect(config.setModel).toHaveBeenCalledWith('coder-model', {
+            reason: 'vision_auto_switch',
+            context: 'User-prompted vision switch (session persistent)',
+        });
         // Restore should be a no-op since no one-time override was used
-        act(() => {
-            result.current.restoreOriginalModel();
+        await act(async () => {
+            await result.current.restoreOriginalModel();
         });
         // Last call should still be the persisted model set
-        expect(config.setModel.mock.calls.pop()?.[0]).toBe('qwen-vl-max-latest');
+        expect(config.setModel.mock.calls.pop()?.[0]).toBe('coder-model');
     });
     it('returns shouldProceed=true when dialog returns no special flags', async () => {
         const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus');
@@ -251,5 +287,214 @@ describe('useVisionAutoSwitch hook', () => {
         expect(res).toEqual({ shouldProceed: true });
         expect(onVisionSwitchRequired).not.toHaveBeenCalled();
     });
+    describe('YOLO mode behavior', () => {
+        it('automatically switches to vision model in YOLO mode without showing dialog', async () => {
+            const initialModel = 'qwen3-coder-plus';
+            const config = createMockConfig(AuthType.QWEN_OAUTH, initialModel, ApprovalMode.YOLO);
+            const onVisionSwitchRequired = vi.fn(); // Should not be called in YOLO mode
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/png', data: '...' } },
+            ];
+            let res;
+            await act(async () => {
+                res = await result.current.handleVisionSwitch(parts, 7070, false);
+            });
+            // Should automatically switch without calling the dialog
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+            expect(res).toEqual({
+                shouldProceed: true,
+                originalModel: initialModel,
+            });
+            expect(config.setModel).toHaveBeenCalledWith(getDefaultVisionModel(), {
+                reason: 'vision_auto_switch',
+                context: 'YOLO mode auto-switch for image content',
+            });
+        });
+        it('does not switch in YOLO mode when no images are present', async () => {
+            const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus', ApprovalMode.YOLO);
+            const onVisionSwitchRequired = vi.fn();
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [{ text: 'no images here' }];
+            let res;
+            await act(async () => {
+                res = await result.current.handleVisionSwitch(parts, 8080, false);
+            });
+            expect(res).toEqual({ shouldProceed: true });
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+            expect(config.setModel).not.toHaveBeenCalled();
+        });
+        it('does not switch in YOLO mode when already using vision model', async () => {
+            const config = createMockConfig(AuthType.QWEN_OAUTH, 'vision-model', ApprovalMode.YOLO);
+            const onVisionSwitchRequired = vi.fn();
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/png', data: '...' } },
+            ];
+            let res;
+            await act(async () => {
+                res = await result.current.handleVisionSwitch(parts, 9090, false);
+            });
+            expect(res).toEqual({ shouldProceed: true });
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+            expect(config.setModel).not.toHaveBeenCalled();
+        });
+        it('restores original model after YOLO mode auto-switch', async () => {
+            const initialModel = 'qwen3-coder-plus';
+            const config = createMockConfig(AuthType.QWEN_OAUTH, initialModel, ApprovalMode.YOLO);
+            const onVisionSwitchRequired = vi.fn();
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/png', data: '...' } },
+            ];
+            // First, trigger the auto-switch
+            await act(async () => {
+                await result.current.handleVisionSwitch(parts, 10100, false);
+            });
+            // Verify model was switched
+            expect(config.setModel).toHaveBeenCalledWith(getDefaultVisionModel(), {
+                reason: 'vision_auto_switch',
+                context: 'YOLO mode auto-switch for image content',
+            });
+            // Now restore the original model
+            await act(async () => {
+                await result.current.restoreOriginalModel();
+            });
+            // Verify model was restored
+            expect(config.setModel).toHaveBeenLastCalledWith(initialModel, {
+                reason: 'vision_auto_switch',
+                context: 'Restoring original model after vision switch',
+            });
+        });
+        it('does not switch in YOLO mode when authType is not QWEN_OAUTH', async () => {
+            const config = createMockConfig(AuthType.USE_GEMINI, 'qwen3-coder-plus', ApprovalMode.YOLO);
+            const onVisionSwitchRequired = vi.fn();
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/png', data: '...' } },
+            ];
+            let res;
+            await act(async () => {
+                res = await result.current.handleVisionSwitch(parts, 11110, false);
+            });
+            expect(res).toEqual({ shouldProceed: true });
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+            expect(config.setModel).not.toHaveBeenCalled();
+        });
+        it('does not switch in YOLO mode when visionModelPreviewEnabled is false', async () => {
+            const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus', ApprovalMode.YOLO);
+            const onVisionSwitchRequired = vi.fn();
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, false, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/png', data: '...' } },
+            ];
+            let res;
+            await act(async () => {
+                res = await result.current.handleVisionSwitch(parts, 12120, false);
+            });
+            expect(res).toEqual({ shouldProceed: true });
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+            expect(config.setModel).not.toHaveBeenCalled();
+        });
+        it('handles multiple image formats in YOLO mode', async () => {
+            const initialModel = 'qwen3-coder-plus';
+            const config = createMockConfig(AuthType.QWEN_OAUTH, initialModel, ApprovalMode.YOLO);
+            const onVisionSwitchRequired = vi.fn();
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { text: 'Here are some images:' },
+                { inlineData: { mimeType: 'image/jpeg', data: '...' } },
+                { fileData: { mimeType: 'image/png', fileUri: 'file://image.png' } },
+                { text: 'Please analyze them.' },
+            ];
+            let res;
+            await act(async () => {
+                res = await result.current.handleVisionSwitch(parts, 13130, false);
+            });
+            expect(res).toEqual({
+                shouldProceed: true,
+                originalModel: initialModel,
+            });
+            expect(config.setModel).toHaveBeenCalledWith(getDefaultVisionModel(), {
+                reason: 'vision_auto_switch',
+                context: 'YOLO mode auto-switch for image content',
+            });
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+        });
+    });
+    describe('VLM switch mode default behavior', () => {
+        it('should automatically switch once when vlmSwitchMode is "once"', async () => {
+            const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus', ApprovalMode.DEFAULT, 'once');
+            const onVisionSwitchRequired = vi.fn(); // Should not be called
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/jpeg', data: 'base64data' } },
+            ];
+            const switchResult = await result.current.handleVisionSwitch(parts, Date.now(), false);
+            expect(switchResult.shouldProceed).toBe(true);
+            expect(switchResult.originalModel).toBe('qwen3-coder-plus');
+            expect(config.setModel).toHaveBeenCalledWith('vision-model', {
+                reason: 'vision_auto_switch',
+                context: 'Default VLM switch mode: once (one-time override)',
+            });
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+        });
+        it('should switch session when vlmSwitchMode is "session"', async () => {
+            const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus', ApprovalMode.DEFAULT, 'session');
+            const onVisionSwitchRequired = vi.fn(); // Should not be called
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/jpeg', data: 'base64data' } },
+            ];
+            const switchResult = await result.current.handleVisionSwitch(parts, Date.now(), false);
+            expect(switchResult.shouldProceed).toBe(true);
+            expect(switchResult.originalModel).toBeUndefined(); // No original model for session switch
+            expect(config.setModel).toHaveBeenCalledWith('vision-model', {
+                reason: 'vision_auto_switch',
+                context: 'Default VLM switch mode: session (session persistent)',
+            });
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+        });
+        it('should continue with current model when vlmSwitchMode is "persist"', async () => {
+            const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus', ApprovalMode.DEFAULT, 'persist');
+            const onVisionSwitchRequired = vi.fn(); // Should not be called
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/jpeg', data: 'base64data' } },
+            ];
+            const switchResult = await result.current.handleVisionSwitch(parts, Date.now(), false);
+            expect(switchResult.shouldProceed).toBe(true);
+            expect(switchResult.originalModel).toBeUndefined();
+            expect(config.setModel).not.toHaveBeenCalled();
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+        });
+        it('should fall back to user prompt when vlmSwitchMode is not set', async () => {
+            const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus', ApprovalMode.DEFAULT, undefined);
+            const onVisionSwitchRequired = vi
+                .fn()
+                .mockResolvedValue({ modelOverride: 'vision-model' });
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/jpeg', data: 'base64data' } },
+            ];
+            const switchResult = await result.current.handleVisionSwitch(parts, Date.now(), false);
+            expect(switchResult.shouldProceed).toBe(true);
+            expect(onVisionSwitchRequired).toHaveBeenCalledWith(parts);
+        });
+        it('should fall back to persist behavior when vlmSwitchMode has invalid value', async () => {
+            const config = createMockConfig(AuthType.QWEN_OAUTH, 'qwen3-coder-plus', ApprovalMode.DEFAULT, 'invalid-value');
+            const onVisionSwitchRequired = vi.fn(); // Should not be called
+            const { result } = renderHook(() => useVisionAutoSwitch(config, addItem, true, onVisionSwitchRequired));
+            const parts = [
+                { inlineData: { mimeType: 'image/jpeg', data: 'base64data' } },
+            ];
+            const switchResult = await result.current.handleVisionSwitch(parts, Date.now(), false);
+            expect(switchResult.shouldProceed).toBe(true);
+            expect(switchResult.originalModel).toBeUndefined();
+            // For invalid values, it should continue with current model (persist behavior)
+            expect(config.setModel).not.toHaveBeenCalled();
+            expect(onVisionSwitchRequired).not.toHaveBeenCalled();
+        });
+    });
 });
 //# sourceMappingURL=useVisionAutoSwitch.test.js.map