fix: current model switcher for pro max (#370)

2025-12-23 13:04:57 -08:00
parent a9ab40bd6e
commit 4ef232ff58
1 changed files with 6 additions and 3 deletions
--- 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);
                }}
              />
            )}