fix:解决聊天过程中切换r1和非r1模型，以及中途中断插入话题等情况下的崩溃。增加聊天的健壮性

Author: yefansky

src/apiTools.ts

@@ -32,6 +32,7 @@ export interface Tool {
 }
 
 export const toolRegistry: Map<string, Tool> = new Map();
+export const webSearchRegistry : Map<string, Tool> = new Map();
 
 /**
  * 注册一个工具到全局工具表
@@ -41,6 +42,10 @@ export function registerTool(tool: Tool) {
     toolRegistry.set(tool.name, tool);
 }
 
+export function registerWebSearchTool(tool: Tool) {
+    webSearchRegistry.set(tool.name, tool);
+}
+
 /**
  * Retrieves an array of all registered tools from the tool registry.
  * @returns An array of Tool objects.
@@ -49,6 +54,10 @@ export function getAllTools(): Tool[] {
     return Array.from(toolRegistry.values());
 }
 
+export function getWebSearchTools(): Tool[] {
+    return Array.from(webSearchRegistry.values());
+}
+
 export async function handleNativeFunctionCalling(
     openai: OpenAI,
     modelName: string,
@@ -297,7 +306,7 @@ async function fetchPageContent(url: string): Promise<string> {
 }
 
 // 1. 搜索网络
-export const searchTool: Tool = {
+export const webSearchTool: Tool = {
     name: 'web_search',
     description: `执行网络搜索并返回前5个结果的摘要。适用于需要获取外部信息、验证数据或了解最新动态的情况。
 用户提供这个工具通常是对你的知识储备或判断持怀疑态度，希望通过网络搜索获取更权威或更新的信息。
@@ -349,7 +358,8 @@ export const searchTool: Tool = {
         }
     },
 };
-registerTool(searchTool);
+registerTool(webSearchTool);
+registerWebSearchTool(webSearchTool);
 
 // 获取网页正文内容的工具
 export const getWebpageContent: Tool = {
@@ -426,6 +436,7 @@ export const getWebpageContent: Tool = {
 };
 
 registerTool(getWebpageContent);
+registerWebSearchTool(getWebpageContent);
 
 // 列举指定路径下文件和目录的工具
 export const listDirectory: Tool = {

src/chatPanel.ts

@@ -146,7 +146,11 @@ export class ChatPanel {
                         <button id="stop" style="display:none;">Stop</button>
                         <div class="web-search">
                             <input type="checkbox" id="web-search">
-                            <label for="web-search">Agent模式(可联网)</label>
+                            <label for="web-search">联网搜索</label>
+                        </div>
+                        <div class="agent-mode">
+                            <input type="checkbox" id="agent-mode">
+                            <label for="agent-mode">Agent模式</label>
                         </div>
                     </div>
                 </div>
@@ -179,6 +183,7 @@ export class ChatPanel {
             const stopButton = document.getElementById('stop');
             const newSessionButton = document.getElementById('new-session');
             const webSearchCheckbox = document.getElementById('web-search');
+            const agentModeCheckbox = document.getElementById('agent-mode');
             const mermaidToggle = document.getElementById('mermaid-toggle');
     
             // 加载外部脚本
@@ -272,11 +277,18 @@ export class ChatPanel {
     }
 
     private async callModelApi(message: any, webviewOutputChannel: WebviewOutputChannel): Promise<string | null> {
-        const tools = message.webSearch ? apiTools.getAllTools() : null;
+        let tools = null;
         const normalSystemPrompt = "用markdown输出。如果有数学公式要用$$包裹，每条一行不要换行。如果有流程图(Mermaid)里的每个字符串都要用引号包裹。";
-        const systemPrompt = message.webSearch 
-            ?  this.loadAgentPrompt() + normalSystemPrompt 
-            : normalSystemPrompt;
+        let systemPrompt = normalSystemPrompt;
+
+        if (message.webSearch) {
+            tools = apiTools.getWebSearchTools();
+            systemPrompt += "每次回答问题前,先观察信息是否足够, 如果不够，先用tool_call进行网络搜索。不要盲目自信， 不要臆测不确定的信息。";
+        }
+        if (message.agentMode) {
+            tools = apiTools.getAllTools();
+            systemPrompt += this.loadAgentPrompt();
+        }
 
         return await callDeepSeekApi(
             this.conversation.map(msg => ({ role: msg.role, content: msg.content })),

src/deepseekApi.ts

@@ -142,6 +142,35 @@ export async function processDeepSeekResponse(
     return { chunkResponse, finishReason };
 }
 
+function fixMessages(messages: OpenAI.ChatCompletionMessageParam[]): OpenAI.ChatCompletionMessageParam[] {
+    const result = [...messages]; // Create a copy to avoid mutating the original
+    let i = 0;
+  
+    // Traverse the array, but check length dynamically since we may insert elements
+    while (i < result.length - 1) {
+      const current = result[i];
+      const next = result[i + 1];
+  
+      // Check if current and next have the same role
+      if (current.role === next.role) {
+        // Insert the opposite role with content "ok"
+        const newMessage : OpenAI.ChatCompletionMessageParam = {
+          role: current.role === 'user' ? 'assistant' : 'user',
+          content: 'ok'
+        };
+        // Insert at index i + 1
+        result.splice(i + 1, 0, newMessage);
+        // Move to the next pair, since we just inserted
+        i += 2;
+      } else {
+        // Move to the next message if no insertion is needed
+        i += 1;
+      }
+    }
+  
+    return result;
+  }
+
 /**
  * 调用 DeepSeek API，支持 Function Calling
  * @param userContent 用户输入内容，可以是字符串或消息数组
@@ -194,39 +223,39 @@ export async function callDeepSeekApi(
         let maxToken = 1024 * 8;
         let temperature = 0;
 
-        let systemPromot : OpenAI.ChatCompletionMessageParam = {role : "system",  content: systemContent};
-
-        if (/r1|reasoner/i.test(modelName)) {
+        const isR1Mode = /r1|reasoner/i.test(modelName);
+        if (isR1Mode) {
             temperature = 0.6;
-            systemPromot = {role : "user",  content: systemContent};
         }
 
-        // 构造消息体
+        // 构造消息体，先将所有内容放入 messages_body
         let messages_body: OpenAI.ChatCompletionMessageParam[] = [];
-        if (Array.isArray(userContent)) {
-            if (systemPromot.role === 'user') { 
-                userContent[0].content = systemPromot.content + "\n\n" + userContent[0].content; 
-            }
-            else{ 
-                messages_body.push(systemPromot); 
-            }
 
-            const role = (userContent[0].role === 'user') ? 'user' : 'assistant';
-            messages_body.push({ role, content: userContent[0].content });
+        // 将 userContent 转换为数组并加入 messages_body
+        const userContentArray = typeof userContent === 'string' 
+            ? [{ role: 'user', content: userContent }] 
+            : userContent;
+        for (const msg of userContentArray) {
+            const role = msg.role === 'user' ? 'user' : 'assistant';
+            messages_body.push({ role, content: msg.content });
+        }
 
-            // 如果 userContent 是数组，按交替方式生成消息
-            for (let i = 1; i < userContent.length; i++) {
-                const role = (userContent[i].role === 'user') ? 'user' : 'assistant';
-                messages_body.push({ role, content: userContent[i].content });
+        if (messages_body.length > 0) {
+            if (messages_body[0].role === 'user') {
+                messages_body[0].content = (messages_body[0].content as string).replace(/<system>.*?<\/system>/gs, ''); // 替换 <system> 标签为空串
+            } else {
+                messages_body.shift(); // 移除头部系统消息
             }
-        } 
-        else {
-            if (systemPromot.role === 'user') { userContent = systemPromot.content + "\n\n" + userContent; }
-            else{ messages_body.push(systemPromot); }
+        }
 
-            // 如果是单个字符串，默认是 'user' 角色
-            messages_body.push({ role: 'user', content: userContent });
+        if (isR1Mode) {
+            messages_body[0].content = `<system>${systemContent}</system>` +  messages_body[0].content;
         }
+        else {
+            messages_body.unshift({role: 'system', content: `<system>${systemContent}</system>`});
+        }
+
+        messages_body = fixMessages(messages_body);
 
         vscode.window.showInformationMessage('开始上传DeepSeek API');
 

src/resources/agentPrompt.txt

@@ -1,8 +1,7 @@
-每次回答问题前,不要太过自信, 先用tool收集一下信息,比如可以上网搜索或者查询记忆。
-第一步一定是先用读取记忆的tool检查以往的记忆里有没有相关信息。然后再看是否需要上网搜索。最后再看是否需要调用其他工具。一个tool没有返回就别急着推理下一步。
-每次只调用一个工具，然后结束输出等返回。然后再调用下一个工具。
-你要先理解任务，搞清楚哪些信息你是不知道的， 用记忆结构查询记忆，如果能查到就采用，
-然后指定任务计划，规划好分几个步骤，用tool去逐步去推进。
-等你觉得信息获得足够了再开始决策和输出结论。
-每次任务开始，你要先调取老记忆，看看以前有没有执行过类似任务，吸取经验。
-每次对话结束，如果获得了新的知识或者有感悟，或者我纠正了你某些内容，你可以用工具存储这些记忆。
+每次回答问题前,先观察信息是否足够, 如果不够，先用tool_call收集一下信息,优先级是： 调取记忆 > 本地文件读取 > 网络搜索。
+如果是比较复杂的需求，那要分步骤来。第一步一定是先用读取记忆的tool检查以往的记忆里有没有相关信息。然后再看是否需要上网搜索。最后再看是否需要调用其他tool_call。一个tool没有返回就别急着推理下一步。
+每次只调用一个tool_call，然后结束输出等返回。然后再调用下一个tool_call。
+你要先理解任务，搞清楚哪些信息你是不知道的， 用记忆tool_call查询过往记忆，查到的信息你自己判断是否和主题有关，无关的信息就无视掉，
+等你觉得信息足够了再开始决策和输出结论。
+你可以从老记忆中看看以前有没有执行过类似任务，吸取经验。
+每次对话结束，如果获得了新的知识或者有感悟，或者我纠正了你某些内容，你可以用tool_call存储这些记忆。

src/resources/chatPanelScript.js

@@ -208,7 +208,7 @@ function separateThinkContent(input) {
     const parts = input.split(/(<think>[\s\S]*?<\/think>)/g);
 
     parts.forEach(part => {
-        if (!part) return;
+        if (!part) {return; }
 
         if (part.startsWith('<think>') && part.endsWith('</think>')) {
             segments.push({
@@ -345,7 +345,7 @@ function setupInputHandlers() {
 function sendMessage() {
     const text = input.value.trim();
     if (text) {
-        vscode.postMessage({ command: 'sendMessage', text, webSearch: webSearchCheckbox.checked });
+        vscode.postMessage({ command: 'sendMessage', text, webSearch: webSearchCheckbox.checked, agentMode: agentModeCheckbox.checked });
         input.value = '';
     }
 }