jinzd-ai-cli 0.1.18 → 0.1.19

package/dist/index.js

@@ -129,7 +129,7 @@ var EnvLoader = class {
 };
 
 // src/core/constants.ts
-var VERSION = "0.1.18";
+var VERSION = "0.1.19";
 var APP_NAME = "ai-cli";
 var CONFIG_DIR_NAME = ".aicli";
 var CONFIG_FILE_NAME = "config.json";
@@ -1033,6 +1033,13 @@ var ZhipuProvider = class extends OpenAICompatibleProvider {
 };
 
 // src/providers/kimi.ts
+var KIMI_TOOL_CALL_REMINDER = `
+
+\u3010\u26A0\uFE0F \u5DE5\u5177\u8C03\u7528\u5F3A\u5236\u89C4\u8303 - \u6781\u5176\u91CD\u8981\u3011
+\u8C03\u7528\u4EFB\u4F55\u5DE5\u5177\uFF08write_file\u3001bash\u3001read_file \u7B49\uFF09\u65F6\uFF0C\u5FC5\u987B\u4E14\u53EA\u80FD\u4F7F\u7528 API \u7684 function calling \u7ED3\u6784\u5316\u63A5\u53E3\u3002
+\u4E25\u7981\u5728\u56DE\u590D\u6587\u672C\u4E2D\u8F93\u51FA XML \u683C\u5F0F\u7684\u4F2A\u5DE5\u5177\u8C03\u7528\u6807\u7B7E\uFF08\u5982 <write_file>\u3001<bash>\u3001<read_file> \u7B49\uFF09\u3002
+\u6587\u672C\u4E2D\u7684 XML \u6807\u7B7E\u4E0D\u4F1A\u88AB\u7CFB\u7EDF\u6267\u884C\uFF0C\u4F1A\u5BFC\u81F4\u6587\u4EF6\u672A\u5199\u5165\u3001\u547D\u4EE4\u672A\u8FD0\u884C\uFF0C\u4EFB\u52A1\u5F7B\u5E95\u5931\u8D25\u3002
+\u6BCF\u6B21\u9700\u8981\u5199\u6587\u4EF6\uFF0C\u5FC5\u987B\u8C03\u7528 write_file \u5DE5\u5177 API\uFF0C\u4E0D\u5141\u8BB8\u4EFB\u4F55\u4F8B\u5916\u3002`;
 var KimiProvider = class extends OpenAICompatibleProvider {
   defaultBaseUrl = "https://api.moonshot.ai/v1";
   info = {
@@ -1096,6 +1103,86 @@ var KimiProvider = class extends OpenAICompatibleProvider {
       }
     ]
   };
+  /**
+   * 覆写 chatWithTools，叠加两道防线：
+   *
+   * 方案 B（预防）：在 system prompt 末尾追加工具调用规范提示，
+   *   指示 Kimi 必须使用 API function calling，禁止输出 XML 伪工具调用。
+   *
+   * 方案 A（兜底）：若 Kimi 仍然在文本中输出了 XML 格式工具调用，
+   *   自动检测并解析，转换为真实 ToolCall 对象送入执行器，
+   *   确保文件写入等操作不会静默丢失。
+   */
+  async chatWithTools(request, tools) {
+    const enhancedRequest = {
+      ...request,
+      systemPrompt: (request.systemPrompt ?? "") + KIMI_TOOL_CALL_REMINDER
+    };
+    const result = await super.chatWithTools(enhancedRequest, tools);
+    if ("content" in result && result.content) {
+      const xmlToolCalls = this.parseXmlToolCalls(result.content, tools);
+      if (xmlToolCalls.length > 0) {
+        process.stderr.write(
+          `[kimi] \u26A0 \u68C0\u6D4B\u5230 ${xmlToolCalls.length} \u4E2A XML \u4F2A\u5DE5\u5177\u8C03\u7528\uFF08Kimi \u672A\u4F7F\u7528 API \u673A\u5236\uFF09\uFF0C\u5DF2\u81EA\u52A8\u8F6C\u6362\u4E3A\u771F\u5B9E\u5DE5\u5177\u8C03\u7528\u6267\u884C\u3002
+`
+        );
+        return { toolCalls: xmlToolCalls, usage: result.usage };
+      }
+    }
+    return result;
+  }
+  /**
+   * 方案 A 核心逻辑：从 Kimi 的文本响应中提取 XML 格式伪工具调用。
+   *
+   * Kimi 输出的 XML 格式为：
+   *   <tool_name>
+   *   <param1>value1</param1>
+   *   <param2>value2</param2>
+   *   </tool_name>
+   *
+   * 使用 indexOf 而非正则，避免大文件内容（14KB+ Markdown）时的性能问题。
+   * 参数值的提取依据工具 schema 中已知的参数名，安全可靠。
+   */
+  parseXmlToolCalls(content, tools) {
+    if (!content.includes("</")) return [];
+    const toolCalls = [];
+    let callIndex = 0;
+    for (const tool of tools) {
+      const startTag = `<${tool.name}>`;
+      const endTag = `</${tool.name}>`;
+      const paramNames = Object.keys(tool.parameters);
+      let searchFrom = 0;
+      while (true) {
+        const blockStart = content.indexOf(startTag, searchFrom);
+        if (blockStart === -1) break;
+        const blockEnd = content.indexOf(endTag, blockStart + startTag.length);
+        if (blockEnd === -1) break;
+        const innerContent = content.slice(blockStart + startTag.length, blockEnd);
+        const args = {};
+        for (const paramName of paramNames) {
+          const openTag = `<${paramName}>`;
+          const closeTag = `</${paramName}>`;
+          const pStart = innerContent.indexOf(openTag);
+          if (pStart !== -1) {
+            const pEnd = innerContent.indexOf(closeTag, pStart + openTag.length);
+            if (pEnd !== -1) {
+              args[paramName] = innerContent.slice(pStart + openTag.length, pEnd);
+            }
+          }
+        }
+        if (Object.keys(args).length > 0) {
+          toolCalls.push({
+            // id 格式：xml-fallback-{时间戳}-{序号}，便于调试日志追踪
+            id: `xml-fallback-${Date.now()}-${callIndex++}`,
+            name: tool.name,
+            arguments: args
+          });
+        }
+        searchFrom = blockEnd + endTag.length;
+      }
+    }
+    return toolCalls;
+  }
 };
 
 // src/providers/custom.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jinzd-ai-cli",
-  "version": "0.1.18",
+  "version": "0.1.19",
   "description": "Cross-platform REPL-style AI CLI with multi-provider support",
   "type": "module",
   "main": "./dist/index.js",