From 4ef232ff58e5d2845901d4c86fcd894842345346 Mon Sep 17 00:00:00 2001
From: jnjpng <jinjpeng@gmail.com>
Date: Tue, 23 Dec 2025 13:04:57 -0800
Subject: [PATCH] fix: current model switcher for pro max (#370)

---
 src/cli/App.tsx | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/cli/App.tsx b/src/cli/App.tsx
index 06d72db..5b1862e 100644
--- a/src/cli/App.tsx
+++ b/src/cli/App.tsx
@@ -5000,11 +5000,14 @@ Plan file path: ${planFilePath}`;
                   const { updateAgentLLMConfig } = await import(
                     "../agent/modify"
                   );
-                  const { getModelUpdateArgs } = await import("../agent/model");
+                  const { getModelUpdateArgs, getModelInfo } = await import(
+                    "../agent/model"
+                  );
                   const updateArgs = getModelUpdateArgs(modelHandle);
                   await updateAgentLLMConfig(agentId, modelHandle, updateArgs);
-                  // Update current model display
-                  setCurrentModelId(modelHandle);
+                  // Update current model display - use model id for correct "(current)" indicator
+                  const modelInfo = getModelInfo(modelHandle);
+                  setCurrentModelId(modelInfo?.id || modelHandle);
                 }}
               />
             )}