feat: 限制修复次数 && 保存对话过程

Author: yefansky

src/deepseekApi.ts

@@ -41,6 +41,10 @@ function getDeepSeekModelConfig(): { modelName: string, apiBaseURL: string, apiK
 
 let lastMessageBody : OpenAI.ChatCompletionMessageParam[];
 
+export function GetLastMessageBody() : OpenAI.ChatCompletionMessageParam[] {
+    return lastMessageBody;
+}
+
 /**
  * 调用 DeepSeek API
  * @param userContent 用户输入内容
@@ -307,9 +311,9 @@ export async function queryCodeReDesign(
         列出每个关键修改点所在的文件路径
     修改方案:
         文件路径1:
-            修改的关键代码,注意只输出关键修改,不要太长, 不要加载无用的上下文。不要输出没有改动部分的代码
+            描述修改点，避免用大块代码,注意只输出关键修改,不要太长, 不要加载无用的上下文。不要输出没有改动部分的代码
         文件路径2:
-            修改的关键代码，同上
+            描述修改点，同上
         …
     最后检查:
         对以上输出的方案大纲进行反思，重新阅读输入代码，结合以上方案大纲，逐条检查有没有和原文对不上的地方。检查方案是否完备、文件路径是否正确，设计思路是否无误，如有问题请提出修正意见

src/extension.ts

@@ -3,7 +3,7 @@ import * as path from 'path';
 import * as fs from 'fs';
 import { selectFiles } from './fileSelector';
 import { generateCvb, applyCvbToWorkspace, generateTimestamp, Cvb, TCVB, mergeCvb} from './cvbManager';
-import { queryCodeReDesign, generateFilenameFromRequest, analyzeCode, callDeepSeekFixApi } from './deepseekApi';
+import { queryCodeReDesign, generateFilenameFromRequest, analyzeCode, callDeepSeekFixApi, GetLastMessageBody } from './deepseekApi';
 import { setupCvbAsMarkdown } from './cvbMarkdownHandler';
 import { registerCvbContextMenu } from './siderBar';
 
@@ -57,6 +57,7 @@ export async function doUploadCommand(cvbFilePath: string, userPrompt: string, o
 
     let apiResponse = await queryCodeReDesign(cvbContent, userPrompt, outputChannel, getCurrentOperationController().signal);
     let processSuccess = true;
+    let attemptCount = 0;
     do {
         try {
             if (apiResponse) {
@@ -75,8 +76,28 @@ export async function doUploadCommand(cvbFilePath: string, userPrompt: string, o
             vscode.window.showInformationMessage(`API response have error ${err.message}, try fix ...`);
             apiResponse = await callDeepSeekFixApi(err.message, outputChannel, true, getCurrentOperationController().signal);
             processSuccess = false;
+            attemptCount++;
         }
-    } while (!processSuccess);
+    } while (!processSuccess && attemptCount < 3);
+
+    const lastMessageBody = GetLastMessageBody();
+
+    if (lastMessageBody && lastMessageBody.length > 2) {
+        const timestamp = generateTimestamp();
+        const summary = await generateFilenameFromRequest(userPrompt);
+        const mdFileName = `${timestamp}_${summary}.md`;
+        const mdFilePath = path.join(tmpDir, mdFileName);
+    
+        // 创建新数组，第一条消息替换为 { "role": "user", "content": userPrompt }
+        const modifiedMessages = [{ role: "user", content: userPrompt }, ...lastMessageBody.slice(2)];
+    
+        const mdContent = modifiedMessages.map(msg => {
+            return `**${msg.role}**:\n\n${msg.content}\n\n`;
+        }).join('\n');
+    
+        fs.writeFileSync(mdFilePath, mdContent, 'utf-8');
+        vscode.window.showInformationMessage(`Conversation log saved as: ${mdFilePath}`);
+    }    
 
     clearCurrentOperationController();
 }

src/siderBar.ts

@@ -97,10 +97,10 @@ class CvbViewProvider implements vscode.TreeDataProvider<CvbFile> {
       if (fs.existsSync(targetFolder)) {
         const files = fs.readdirSync(targetFolder);
         files.forEach(file => {
-          if (file.endsWith('.cvb')) {
-            const filePath = path.join(targetFolder, file);
-            cvbFiles.push(new CvbFile(file, vscode.Uri.file(filePath)));
-          }
+          if (file.endsWith('.cvb') || file.endsWith('.md')) {
+          const filePath = path.join(targetFolder, file);
+          cvbFiles.push(new CvbFile(file, vscode.Uri.file(filePath)));
+      }
         });
       }