feat: 增强模型配置和版本自动升级功能

添加模型配置的 maxOutputTokens 支持并优化自动升级流程
改进用户取消操作时的提示信息
优化模型添加后的自动切换逻辑
缩短版本检查缓存时间至1小时

Author: echoVic

src/agent/Agent.ts

@@ -185,7 +185,7 @@ export class Agent {
         baseUrl: modelConfig.baseUrl,
         temperature: modelConfig.temperature ?? this.config.temperature,
         maxContextTokens: modelConfig.maxContextTokens ?? this.config.maxContextTokens, // 上下文窗口（压缩判断）
-        maxOutputTokens: this.config.maxOutputTokens, // 输出限制（API max_tokens）
+        maxOutputTokens: modelConfig.maxOutputTokens ?? this.config.maxOutputTokens, // 输出限制（API max_tokens）
         timeout: this.config.timeout,
       });
 
@@ -278,8 +278,8 @@ export class Agent {
           : await this.runLoop(enhancedMessage, context, loopOptions);
 
       if (!result.success) {
-        // 如果是用户中止，返回空字符串（不抛出异常）
-        if (result.error?.type === 'aborted') {
+        // 如果是用户中止或用户拒绝，返回空字符串（不抛出异常）
+        if (result.error?.type === 'aborted' || result.metadata?.shouldExitLoop) {
           return ''; // 返回空字符串，让调用方自行处理
         }
         // 其他错误则抛出异常
@@ -530,7 +530,6 @@ IMPORTANT: Execute according to the approved plan above. Follow the steps exactl
         // checkAndCompactInLoop 返回是否发生了压缩
         // 🆕 传入上一轮 LLM 返回的真实 prompt tokens（比估算更准确）
         const didCompact = await this.checkAndCompactInLoop(
-          messages,
           context,
           turnsCount,
           lastPromptTokens, // 首轮为 undefined，使用估算；后续轮次使用真实值
@@ -863,7 +862,7 @@ IMPORTANT: Execute according to the approved plan above. Follow the steps exactl
             }
             logger.debug('==================================\n');
 
-            // 🆕 检查是否应该退出循环（ExitPlanMode 返回时设置此标记）
+            // 🆕 检查是否应该退出循环（ExitPlanMode 或用户拒绝时设置此标记）
             if (result.metadata?.shouldExitLoop) {
               logger.debug('🚪 检测到退出循环标记，结束 Agent 循环');
 
@@ -1366,15 +1365,13 @@ IMPORTANT: Execute according to the approved plan above. Follow the steps exactl
    * 在 Agent 循环中检查并执行压缩
    * 仅使用 LLM 返回的真实 usage.promptTokens 进行判断（不再估算）
    *
-   * @param messages - 实际发送给 LLM 的消息数组
    * @param context - 聊天上下文
    * @param currentTurn - 当前轮次
    * @param actualPromptTokens - LLM 返回的真实 prompt tokens（必须，来自上一轮响应）
    * @param onCompacting - 压缩状态回调
    * @returns 是否发生了压缩
    */
   private async checkAndCompactInLoop(
-    messages: Message[],
     context: ChatContext,
     currentTurn: number,
     actualPromptTokens?: number,

src/agent/types.ts

@@ -169,7 +169,7 @@ export interface LoopResult {
     configuredMaxTurns?: number;
     actualMaxTurns?: number;
     hitSafetyLimit?: boolean;
-    shouldExitLoop?: boolean; // ExitPlanMode 设置此标记以退出循环
+    shouldExitLoop?: boolean; // ExitPlanMode 或用户拒绝时设置此标记以退出循环
     targetMode?: PermissionMode; // Plan 模式批准后的目标权限模式
     planContent?: string; // Plan 模式批准后的方案内容
   };

src/config/types.ts

@@ -59,6 +59,7 @@ export interface ModelConfig {
   // 可选：模型特定参数
   temperature?: number;
   maxContextTokens?: number; // 上下文窗口大小
+  maxOutputTokens?: number; // 输出 token 限制
   topP?: number;
   topK?: number;
 

src/services/VersionChecker.ts

@@ -18,8 +18,8 @@ const CACHE_DIR = path.join(
 );
 const CACHE_FILE = path.join(CACHE_DIR, 'version-cache.json');
 
-// 缓存有效期：24 小时
-const CACHE_TTL = 24 * 60 * 60 * 1000;
+// 缓存有效期：1 小时
+const CACHE_TTL = 1 * 60 * 60 * 1000;
 
 // npm registry URL
 const NPM_REGISTRY_URL = `https://registry.npmmirror.com/${PACKAGE_NAME}/latest`;
@@ -224,13 +224,58 @@ export function formatUpdateMessage(result: VersionCheckResult): string | null {
 }
 
 /**
- * 启动时版本检查（后台执行，不阻塞）
+ * 执行自动升级（后台进程，不阻塞主进程）
+ * @returns 升级提示消息
+ */
+async function performUpgrade(
+  currentVersion: string,
+  latestVersion: string
+): Promise<string> {
+  const { spawn } = await import('child_process');
+
+  try {
+    const updateCommand = `npm install -g blade-code@${latestVersion} --registry https://registry.npmjs.org`;
+
+    // 使用 spawn + detached + unref 在后台运行升级
+    // 这样主进程退出后，升级进程会继续运行完成安装
+    const updateProcess = spawn(updateCommand, {
+      stdio: 'ignore',
+      shell: true,
+      detached: true,
+    });
+    updateProcess.unref();
+
+    return `⬆️ 正在后台升级 ${currentVersion} → ${latestVersion}，下次启动生效`;
+  } catch {
+    return (
+      `\x1b[33m⚠️  Update available: ${currentVersion} → ${latestVersion}\x1b[0m\n` +
+      `   Run \x1b[36mnpm install -g ${PACKAGE_NAME}@latest\x1b[0m to update`
+    );
+  }
+}
+
+/**
+ * 启动时版本检查并自动升级
  *
- * @returns Promise<string | null> 更新提示消息，如果没有更新则返回 null
+ * @param autoUpgrade - 是否自动升级（默认 true）
+ * @returns Promise<string | null> 提示消息，如果没有更新则返回 null
  */
-export async function checkVersionOnStartup(): Promise<string | null> {
+export async function checkVersionOnStartup(
+  autoUpgrade = true
+): Promise<string | null> {
   try {
     const result = await checkVersion();
+
+    if (!result.hasUpdate || !result.latestVersion) {
+      return null;
+    }
+
+    // 自动升级
+    if (autoUpgrade) {
+      return await performUpgrade(result.currentVersion, result.latestVersion);
+    }
+
+    // 仅显示提示
     return formatUpdateMessage(result);
   } catch {
     return null;

src/store/slices/sessionSlice.ts

@@ -258,5 +258,6 @@ export const createSessionSlice: StateCreator<
         },
       }));
     },
+
   },
 });

src/tools/execution/PipelineStages.ts

@@ -359,7 +359,10 @@ export class ConfirmationStage implements PipelineStage {
           await confirmationHandler.requestConfirmation(confirmationDetails);
 
         if (!response.approved) {
-          execution.abort(`User rejected execution: ${response.reason || 'No reason provided'}`);
+          execution.abort(
+            `User rejected execution: ${response.reason || 'No reason provided'}`,
+            { shouldExitLoop: true }
+          );
           return;
         }
 

src/tools/types/ExecutionTypes.ts

@@ -96,7 +96,7 @@ export class ToolExecution {
     return this.aborted || (this.context.signal?.aborted ?? false);
   }
 
-  abort(reason?: string): void {
+  abort(reason?: string, options?: { shouldExitLoop?: boolean }): void {
     this.aborted = true;
     this.result = {
       success: false,
@@ -106,6 +106,7 @@ export class ToolExecution {
         type: ToolErrorType.EXECUTION_ERROR,
         message: reason || 'Execution aborted',
       },
+      metadata: options?.shouldExitLoop ? { shouldExitLoop: true } : undefined,
     };
   }
 

src/ui/App.tsx

@@ -126,11 +126,11 @@ export const AppWrapper: React.FC<AppProps> = (props) => {
         }
       }
 
-      // 8. 后台检查版本更新（不阻塞启动）
-      checkVersionOnStartup().then((updateMessage) => {
-        if (updateMessage) {
+      // 8. 后台检查版本更新并自动升级（不阻塞启动）
+      checkVersionOnStartup().then((message) => {
+        if (message) {
           console.log('');
-          console.log(updateMessage);
+          console.log(message);
           console.log('');
         }
       });

src/ui/components/BladeInterface.tsx

@@ -310,6 +310,13 @@ export const BladeInterface: React.FC<BladeInterfaceProps> = ({
     appActions.showModelEditWizard(model);
   });
 
+  const handleModelAddComplete = useMemoizedFn((addedConfig: SetupConfig) => {
+    sessionActions.addAssistantMessage(
+      `✅ 已添加模型配置: ${addedConfig.name}，并已切换到该模型`
+    );
+    closeModal();
+  });
+
   const handleModelEditComplete = useMemoizedFn((updatedConfig: SetupConfig) => {
     sessionActions.addAssistantMessage(`✅ 已更新模型配置: ${updatedConfig.name}`);
     closeModal();
@@ -537,7 +544,7 @@ export const BladeInterface: React.FC<BladeInterfaceProps> = ({
                 onComplete={
                   inlineModelWizardMode === 'edit'
                     ? handleModelEditComplete
-                    : closeModal
+                    : handleModelAddComplete
                 }
                 onCancel={closeModal}
               />

src/ui/components/ModelConfigWizard.tsx

@@ -464,8 +464,9 @@ export const ModelConfigWizard: React.FC<ModelConfigWizardProps> = ({
         // setup 模式：由父组件（BladeInterface）负责创建模型
         onComplete(setupConfig);
       } else if (mode === 'add') {
-        // add 模式：直接在这里创建模型，然后通知父组件关闭
-        await configActions().addModel(setupConfig);
+        // add 模式：直接在这里创建模型，然后自动切换到新模型
+        const newModel = await configActions().addModel(setupConfig);
+        await configActions().setCurrentModel(newModel.id);
         onComplete(setupConfig);
       } else {
         if (!modelId) {