feat:上下文超过一定长度就自动启动预处理压缩

Author: yefansky

src/cvbManager.ts

@@ -961,19 +961,19 @@ function func2() {
 
     const systemContent = "你是一个代码分析助手。给定一个文件的内容和用户的请求，识别并提取出对理解代码在请求上下文中的有价值的代码片段。注意输出的时候不要有 \`\`\`";
 
-    outputChannel.appendLine(`compress processing .. ${filePath}`);
+    outputChannel.appendLine(`compress processing .. ${filePath} [🚀start]`);
     try {
       const response = await callDeepSeekApi(requestContent, systemContent, undefined, true, undefined, signal, true);
       if (response) {
         const segments = response.split("===SEGMENT===").map(segment => segment.trim());
         const compressedContent = segments.join("\n//...CCVB\n");
         compressedFiles[filePath] = compressedContent;
-        outputChannel.appendLine(`compress processing .. ${filePath} [success]`);
+        outputChannel.appendLine(`compress processing .. ${filePath} [✅success]`);
       } else {
-        outputChannel.appendLine(`compress processing .. ${filePath} [failed]`);
+        outputChannel.appendLine(`compress processing .. ${filePath} [❌failed]`);
       }
     } catch (error) {
-      outputChannel.appendLine(`compress processing .. ${filePath} [failed: ${error}]`);
+      outputChannel.appendLine(`compress processing .. ${filePath} [⚠️error: ${error}]`);
     }
   };
 

src/extension.ts

@@ -2,7 +2,7 @@ import * as vscode from 'vscode';
 import * as path from 'path';
 import * as fs from 'fs';
 import { selectFiles } from './fileSelector';
-import { generateCvb, applyCvbToWorkspace, generateTimestamp, Cvb, TCVB, mergeCvb} from './cvbManager';
+import { generateCvb, applyCvbToWorkspace, generateTimestamp, Cvb, TCVB, mergeCvb, compressCvb} from './cvbManager';
 import { queryCodeReDesign, generateFilenameFromRequest, analyzeCode, callDeepSeekFixApi, GetLastMessageBody } from './deepseekApi';
 import { setupCvbAsMarkdown } from './cvbMarkdownHandler';
 import { registerCvbContextMenu } from './siderBar';
@@ -66,7 +66,16 @@ export async function doUploadCommand(cvbFilePath: string, userPrompt: string, o
         i++;
     }
 
-    const cvbContent = fs.readFileSync(cvbFilePath, 'utf-8');
+    let cvbContent = fs.readFileSync(cvbFilePath, 'utf-8');
+    const CVB_QUERY_LENGTH_LIMIT = 1024 * 2;
+    if(cvbContent.length > CVB_QUERY_LENGTH_LIMIT) {
+        const inputCvb = new Cvb(cvbContent);
+        if (!inputCvb.getMetaData("compressFrom")) {
+            currentOutputChannel?.appendLine("输入数据过于巨大,先进行压缩预处理...");
+            const compressedCvb = await compressCvb(inputCvb, userPrompt);
+            cvbContent = compressedCvb.toString();
+        }
+    }
 
     let apiResponse = await queryCodeReDesign(cvbContent, userPrompt, outputChannel, CurrentOperationController.signal);
     let processSuccess = true;