xiaozhou-chat 1.0.4 → 1.0.7

package/lib/chat.js

@@ -0,0 +1,324 @@
+
+import { Spinner, StreamPrinter } from "./ui.js";
+import { sleep } from "./utils.js";
+
+// 尝试加载 Markdown 渲染库
+let marked;
+try {
+  marked = (await import('marked')).marked;
+  const TerminalRenderer = (await import('marked-terminal')).default;
+  marked.setOptions({ renderer: new TerminalRenderer() });
+} catch (e) {}
+
+export async function requestWithRetry(url, options, maxRetries = 3) {
+    let lastError;
+    for (let i = 0; i <= maxRetries; i++) {
+        try {
+            const res = await fetch(url, options);
+            if (!res.ok) {
+                const text = await res.text();
+                // 4xx errors: do not retry
+                if (res.status >= 400 && res.status < 500) {
+                     throw new Error(`API Error (${res.status}): ${text}`);
+                }
+                // 5xx errors: retry
+                throw new Error(`Server Error (${res.status}): ${text}`);
+            }
+            return res;
+        } catch (e) {
+            lastError = e;
+            if (e.name === 'AbortError') throw e;
+            if (i === maxRetries) break;
+            
+            const delay = 1000 * Math.pow(2, i);
+            // console.log(`⚠️  请求失败，${delay}ms 后重试...`); // Optional: callback for logging?
+            
+            if (options.signal?.aborted) throw new Error("Aborted during retry wait");
+            await sleep(delay);
+        }
+    }
+    throw lastError;
+}
+
+export async function chatStream(context, userInput = null, options = {}) {
+    const { 
+        messages, 
+        config, 
+        mcpClients, 
+        toolHandlers, // Function: (name, args) => result
+        signal 
+    } = context;
+    
+    const { isRecursion = false } = options;
+
+    if (userInput) {
+        messages.push({ role: "user", content: userInput });
+    }
+
+    // 构造 Tools 定义
+    // TODO: 从 mcpClients 和 builtInTools 获取
+    // 这里我们假设 context 传入了完整的 tools 列表，或者我们在外部组装
+    // 为了解耦，建议外部传入 tools array
+    const tools = context.tools || [];
+
+    const body = {
+        model: config.model,
+        messages: messages,
+        stream: true,
+        max_tokens: 8192
+    };
+
+    if (tools.length > 0) {
+        body.tools = tools;
+    }
+
+    if (config.systemPrompt) {
+        // Ensure system prompt is at the beginning
+        if (messages.length === 0 || messages[0].role !== "system") {
+             // Check if we should insert or replace? 
+             // Simplest: just don't mutate history permanently if it's transient?
+             // But usually system prompt is part of config. 
+             // Let's assume messages already handles system prompt or we unshift it here for the request ONLY.
+             // But standard is: messages[0] is system.
+             // We'll let the caller handle message structure, or handle it here.
+             // For CLI, we usually want to prepend system prompt if not present.
+             const sysMsg = { role: "system", content: config.systemPrompt };
+             if (messages.length > 0 && messages[0].role === "system") {
+                 // Replace existing? Or assume caller synced it?
+                 // Let's use a copy for the request to be safe
+                 body.messages = [sysMsg, ...messages.filter(m => m.role !== "system")];
+             } else {
+                 body.messages = [sysMsg, ...messages];
+             }
+        }
+    }
+
+    const spinner = new Spinner(isRecursion ? "AI 正在分析工具结果..." : "AI 正在思考...");
+    spinner.start();
+    
+    const printer = new StreamPrinter();
+
+    let requestUrl = `${config.baseUrl}/chat/completions`;
+    
+    // 自动尝试逻辑
+    let shouldRetryWithV1 = false;
+
+    try {
+        let res = await requestWithRetry(requestUrl, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${config.apiKey}`
+            },
+            body: JSON.stringify(body),
+            signal: signal
+        });
+
+        // 智能检测：如果返回的是 HTML (通常是 404 页或首页)，且 URL 没带 v1，可能是用户配错了
+        const contentType = res.headers.get("content-type");
+        if (contentType && contentType.includes("text/html")) {
+            if (!config.baseUrl.endsWith("/v1") && !config.baseUrl.endsWith("/v1/")) {
+                console.log("\n⚠️  检测到 API 返回 HTML，Base URL 可能缺少 '/v1'。");
+                console.log("🔄 正在尝试自动追加 '/v1' 重试...");
+                requestUrl = `${config.baseUrl}/v1/chat/completions`;
+                shouldRetryWithV1 = true;
+                
+                // 重试请求
+                res = await requestWithRetry(requestUrl, {
+                    method: "POST",
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: `Bearer ${config.apiKey}`
+                    },
+                    body: JSON.stringify(body),
+                    signal: signal
+                });
+            }
+        }
+
+        // 停止 Spinner，准备流式输出
+        spinner.stop();
+
+        if (shouldRetryWithV1 && res.ok && !res.headers.get("content-type")?.includes("text/html")) {
+             console.log(`✅ 自动修复成功！建议运行以下命令永久修改配置：`);
+             console.log(`👉 /config baseUrl ${config.baseUrl}/v1`);
+        }
+
+        if (!res.body) throw new Error("Response body is empty");
+        
+        const reader = res.body.getReader();
+        const decoder = new TextDecoder("utf-8");
+        let reply = "";
+        let buffer = "";
+        let currentToolCalls = {}; 
+        let usageInfo = null;
+        let hasReceivedContent = false;
+
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done) break;
+
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split("\n");
+            buffer = lines.pop() || "";
+
+            for (const line of lines) {
+                if (!line.trim()) continue;
+                
+                if (!line.startsWith("data:")) {
+                    // 尝试检测非流式错误返回
+                    try {
+                        const json = JSON.parse(line);
+                        if (json.error) {
+                            throw new Error(json.error.message || JSON.stringify(json.error));
+                        }
+                    } catch (e) {
+                        if (e.message.includes("JSON")) {
+                            // ignore json parse error, just bad format
+                        } else {
+                            throw e; // rethrow actual API error
+                        }
+                    }
+                    continue;
+                }
+
+                const data = line.slice(5).trim();
+                if (data === "[DONE]") break;
+
+                try {
+                    const json = JSON.parse(data);
+                    if (json.usage) {
+                        usageInfo = json.usage;
+                        continue;
+                    }
+                    const delta = json.choices?.[0]?.delta;
+                    
+                    if (delta?.content) {
+                        printer.add(delta.content);
+                        reply += delta.content;
+                        hasReceivedContent = true;
+                    }
+                    
+                    if (delta?.tool_calls) {
+                        hasReceivedContent = true;
+                        for (const tc of delta.tool_calls) {
+                            if (!currentToolCalls[tc.index]) {
+                                currentToolCalls[tc.index] = { 
+                                    id: tc.id, 
+                                    type: "function", 
+                                    function: { name: "", arguments: "" } 
+                                };
+                            }
+                            if (tc.function?.name) currentToolCalls[tc.index].function.name += tc.function.name;
+                            if (tc.function?.arguments) currentToolCalls[tc.index].function.arguments += tc.function.arguments;
+                        }
+                    }
+                } catch {}
+            }
+        }
+        
+        // 检查是否收到了有效内容
+        if (!hasReceivedContent && reply.length === 0) {
+            console.log(`\n🔍 调试信息: Status=${res.status}, Headers=${JSON.stringify([...res.headers.entries()])}`);
+            
+            // 检查 buffer 中是否残留了错误信息（针对非 SSE 格式的错误返回）
+            if (buffer.trim()) {
+                try {
+                    const json = JSON.parse(buffer);
+                    if (json.error) throw new Error(json.error.message || JSON.stringify(json.error));
+                } catch {}
+                throw new Error(`API 响应无法解析 (Raw: ${buffer.slice(0, 100)}...)`);
+            }
+            // 如果连 buffer 都是空的，但 status 是 200
+            throw new Error("API 返回了空内容 (Content-Length: 0)。请检查 Base URL 是否正确，或尝试更换 Model。");
+        }
+
+        await printer.waitIdle();
+        console.log(""); // newline
+
+        // Markdown Post-render
+        if (marked && (reply.includes("```") || reply.includes("**") || reply.includes("# "))) {
+             try { console.log(marked(reply)); } catch {}
+        }
+
+        // Add assistant message
+        const assistantMsg = { role: "assistant", content: reply };
+        const toolCalls = Object.values(currentToolCalls);
+        if (toolCalls.length > 0) {
+            assistantMsg.tool_calls = toolCalls;
+        }
+        messages.push(assistantMsg);
+
+        // Usage stats
+        if (usageInfo) {
+             const { prompt_tokens, completion_tokens, total_tokens } = usageInfo;
+             console.log(`\x1b[90m(Tokens: ${prompt_tokens} + ${completion_tokens} = ${total_tokens})\x1b[0m`);
+        }
+
+        // Handle Tool Calls
+        if (toolCalls.length > 0) {
+            for (const tc of toolCalls) {
+                const funcName = tc.function.name;
+                const argsStr = tc.function.arguments;
+                console.log(`🛠️  调用工具: ${funcName}(${argsStr})`);
+                
+                let result = null;
+                try {
+                    const args = JSON.parse(argsStr);
+                    // 1. Try built-in
+                    result = await toolHandlers(funcName, args);
+                    
+                    // 2. Try MCP
+                    if (!result && mcpClients) {
+                         // Find which client has this tool?
+                         // We assume caller knows or we iterate.
+                         // For now, simplify: mcpClients is a Map<name, client>
+                         // We need to know which client provides which tool.
+                         // Or we iterate all clients (slow?).
+                         // In original cli.js, it wasn't fully implemented for *execution* via MCP map iteration?
+                         // Ah, original cli.js didn't iterate mcpClients for execution in the snippet I saw.
+                         // It only had `builtInTools`.
+                         // But the todo said "integrate MCP support".
+                         // I should check if I missed MCP execution logic.
+                         // Assuming mcpClients have a `callTool` method.
+                         for (const client of mcpClients.values()) {
+                             try {
+                                 const mcpRes = await client.callTool(funcName, args);
+                                 if (mcpRes) {
+                                     result = JSON.stringify(mcpRes);
+                                     break;
+                                 }
+                             } catch {}
+                         }
+                    }
+                } catch (e) {
+                    result = `Error: ${e.message}`;
+                }
+
+                if (result === null) {
+                    result = "Error: Tool not found or failed.";
+                }
+
+                messages.push({
+                    role: "tool",
+                    tool_call_id: tc.id,
+                    content: result
+                });
+            }
+            
+            // Recursive call
+            return await chatStream(context, null, { isRecursion: true });
+        }
+
+    } catch (e) {
+    spinner.stop();
+    printer.stop();
+    if (e.name === 'AbortError' || e.message === "Aborted during retry wait") {
+        console.log("\n🛑 已中断生成");
+    } else {
+        // 确保错误信息换行打印
+        process.stdout.write("\n"); 
+        console.error("❌ 请求失败:", e.message || e);
+    }
+  }
+}

package/lib/config.js

@@ -0,0 +1,127 @@
+
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
+
+// 查找项目根目录
+export function findProjectRoot(startDir = process.cwd()) {
+  let dir = startDir;
+  while (true) {
+    if (fs.existsSync(path.join(dir, "package.json"))) {
+      return dir;
+    }
+    const parent = path.dirname(dir);
+    if (parent === dir) break;
+    dir = parent;
+  }
+  return startDir;
+}
+
+const projectRoot = findProjectRoot();
+const homeConfigFile = path.join(os.homedir(), ".newapi-chat-config.json");
+const projectConfigFile = path.join(projectRoot, ".newapi-chat-config.json");
+const projectAltConfigFile = path.join(projectRoot, "newapi-chat.config.json");
+
+// 初始化配置文件 (安全模式: 不写入默认 Key)
+export function initConfigFile() {
+  if (fs.existsSync(homeConfigFile)) return;
+
+  const defaultConfig = {
+    apiKey: "", // 用户需手动填入
+    baseUrl: "https://paid.tribiosapi.top/v1",
+    model: "claude-sonnet-4-5-20250929",
+    profiles: {
+      default: {
+        apiKey: "",
+        baseUrl: "https://paid.tribiosapi.top/v1",
+        model: "claude-sonnet-4-5-20250929"
+      }
+    },
+    currentProfile: "default"
+  };
+
+  fs.writeFileSync(homeConfigFile, JSON.stringify(defaultConfig, null, 2), "utf-8");
+  console.log(`✅ 已创建配置文件: ${homeConfigFile}`);
+  console.log("⚠️  请务必编辑该文件，填入你的 apiKey (sk-...)，否则无法使用。");
+}
+
+function loadConfigFrom(file) {
+  if (!fs.existsSync(file)) return {};
+  try {
+    return JSON.parse(fs.readFileSync(file, "utf-8"));
+  } catch {
+    return {};
+  }
+}
+
+// 加载配置 (Home < Project)
+export function loadConfig() {
+  const home = loadConfigFrom(homeConfigFile);
+  const project =
+    loadConfigFrom(projectConfigFile) ||
+    loadConfigFrom(projectAltConfigFile);
+
+  // Deep merge profiles if needed, but simple merge is okay for now
+  const config = { ...home, ...project };
+
+  // 确保 profiles 结构存在
+  if (!config.profiles) {
+    config.profiles = {
+      default: {
+        apiKey: config.apiKey || "",
+        baseUrl: config.baseUrl || "https://paid.tribiosapi.top/v1",
+        model: config.model || "claude-sonnet-4-5-20250929",
+        systemPrompt: config.systemPrompt
+      }
+    };
+    config.currentProfile = "default";
+  }
+
+  return config;
+}
+
+export function getWriteConfigFile() {
+  if (fs.existsSync(projectConfigFile)) return projectConfigFile;
+  if (fs.existsSync(projectAltConfigFile)) return projectAltConfigFile;
+  return homeConfigFile;
+}
+
+export function updateConfig(key, value) {
+  const target = getWriteConfigFile();
+  let current = loadConfigFrom(target);
+  
+  // 安全检查：如果是项目级配置，且正在写入 apiKey，给出警告或阻止
+  if ((target === projectConfigFile || target === projectAltConfigFile) && key === "apiKey") {
+      console.warn("⚠️  警告: 你正在将 API Key 写入项目级配置文件。这可能会被提交到版本控制系统！");
+  }
+
+  current[key] = value;
+  fs.writeFileSync(target, JSON.stringify(current, null, 2), "utf-8");
+  return current;
+}
+
+export function setProfileValue(profileName, key, value) {
+  const target = getWriteConfigFile();
+  let current = loadConfigFrom(target);
+  
+  if (!current.profiles) current.profiles = {};
+  if (!current.profiles[profileName]) current.profiles[profileName] = {};
+  
+  current.profiles[profileName][key] = value;
+  
+  fs.writeFileSync(target, JSON.stringify(current, null, 2), "utf-8");
+  return current;
+}
+
+export function getActiveConfig(config) {
+    const profileName = config.currentProfile || "default";
+    const profile = config.profiles?.[profileName] || {};
+    
+    // 优先级: Profile > Top Level > Env > Default
+    return {
+        apiKey: profile.apiKey || config.apiKey || process.env.NEWAPI_API_KEY || "",
+        baseUrl: profile.baseUrl || config.baseUrl || process.env.NEWAPI_BASE_URL || "https://paid.tribiosapi.top/v1",
+        model: profile.model || config.model || process.env.NEWAPI_MODEL || "claude-sonnet-4-5-20250929",
+        systemPrompt: profile.systemPrompt || config.systemPrompt || process.env.NEWAPI_SYSTEM_PROMPT || ""
+    };
+}

package/lib/history.js

@@ -0,0 +1,68 @@
+
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
+import { findProjectRoot } from "./config.js";
+
+const projectRoot = findProjectRoot();
+const homeHistoryFile = path.join(os.homedir(), ".newapi-chat-history.json");
+const projectHistoryFile = path.join(projectRoot, ".newapi-chat-history.json");
+
+function canWriteProjectDir() {
+  try {
+    fs.accessSync(projectRoot, fs.constants.W_OK);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+function getReadHistoryFile() {
+  return fs.existsSync(projectHistoryFile)
+    ? projectHistoryFile
+    : homeHistoryFile;
+}
+
+function getWriteHistoryFile() {
+  return canWriteProjectDir()
+    ? projectHistoryFile
+    : homeHistoryFile;
+}
+
+export function loadHistory() {
+  const target = getReadHistoryFile();
+  if (!fs.existsSync(target)) return [];
+  try {
+    return JSON.parse(fs.readFileSync(target, "utf-8"));
+  } catch {
+    return [];
+  }
+}
+
+export function saveHistory(messages) {
+  const target = getWriteHistoryFile();
+  // 简单的自动截断保护 (保留最近 500 条)
+  // 如果需要更复杂的压缩，由外部调用 compress
+  const toSave = messages.length > 500 ? messages.slice(-500) : messages;
+  fs.writeFileSync(target, JSON.stringify(toSave, null, 2));
+}
+
+export function clearHistory() {
+  const target = getWriteHistoryFile();
+  fs.writeFileSync(target, JSON.stringify([], null, 2));
+}
+
+export function exportHistory(messages) {
+  if (!messages.length) return "（暂无历史）";
+  
+  const md = messages.map((m) => {
+    const role = m.role === "user" ? "你" : "助手";
+    return `### ${role}\n\n${m.content}\n`;
+  }).join("\n");
+
+  const dir = canWriteProjectDir() ? projectRoot : os.homedir();
+  const file = path.join(dir, "newapi-chat-history.md");
+  
+  fs.writeFileSync(file, md, "utf-8");
+  return file;
+}

package/lib/mcp-lite.js

@@ -0,0 +1,128 @@
+import { spawn } from 'child_process';
+import readline from 'readline';
+
+export class MCPClient {
+  constructor(command, args, env = {}) {
+    this.command = command;
+    this.args = args;
+    this.env = env;
+    this.process = null;
+    this.requestId = 0;
+    this.pending = new Map();
+    this.rl = null;
+    this.initialized = false;
+    this.tools = [];
+  }
+
+  async connect() {
+    try {
+      console.log(`🔌 连接 MCP Server: ${this.command} ${this.args.join(' ')}`);
+      this.process = spawn(this.command, this.args, {
+        env: { ...process.env, ...this.env },
+        stdio: ['pipe', 'pipe', 'inherit'] // stderr 继承以便调试
+      });
+
+      this.rl = readline.createInterface({
+        input: this.process.stdout,
+        terminal: false
+      });
+
+      this.rl.on('line', (line) => {
+        if (!line.trim()) return;
+        try {
+          const msg = JSON.parse(line);
+          // console.log("Received:", JSON.stringify(msg).slice(0, 100)); // Debug
+
+          if (msg.id !== undefined && this.pending.has(msg.id)) {
+            const { resolve, reject } = this.pending.get(msg.id);
+            this.pending.delete(msg.id);
+            if (msg.error) {
+              reject(new Error(msg.error.message || JSON.stringify(msg.error)));
+            } else {
+              resolve(msg.result);
+            }
+          }
+        } catch (e) {
+          console.error("MCP JSON Parse Error:", e);
+        }
+      });
+      
+      this.process.on('error', (err) => {
+        console.error(`MCP Process Error (${this.command}):`, err);
+      });
+
+      this.process.on('exit', (code) => {
+        if (code !== 0) console.log(`MCP Server exited with code ${code}`);
+      });
+
+      // 初始化握手
+      await this.initialize();
+      this.initialized = true;
+      console.log("✅ MCP Server 连接成功");
+      
+      // 获取工具列表
+      await this.refreshTools();
+
+    } catch (error) {
+      console.error("MCP Connection Failed:", error);
+      throw error;
+    }
+  }
+
+  async request(method, params) {
+    if (!this.process) throw new Error("MCP Client not connected");
+    
+    const id = this.requestId++;
+    const msg = { jsonrpc: "2.0", id, method, params };
+    const jsonStr = JSON.stringify(msg);
+    // console.log("Sending:", jsonStr); // Debug
+    this.process.stdin.write(jsonStr + "\n");
+    
+    return new Promise((resolve, reject) => {
+      // 30秒超时
+      const timeout = setTimeout(() => {
+        if (this.pending.has(id)) {
+          this.pending.delete(id);
+          reject(new Error("MCP Request Timeout"));
+        }
+      }, 30000);
+
+      this.pending.set(id, { 
+        resolve: (res) => { clearTimeout(timeout); resolve(res); },
+        reject: (err) => { clearTimeout(timeout); reject(err); }
+      });
+    });
+  }
+
+  async initialize() {
+    return this.request("initialize", { 
+      protocolVersion: "2024-11-05", 
+      clientInfo: { name: "xiaozhou-chat", version: "1.0.0" },
+      capabilities: { tools: {} } 
+    });
+  }
+
+  async refreshTools() {
+    const res = await this.request("tools/list", {});
+    this.tools = res.tools || [];
+    console.log(`🛠️  加载了 ${this.tools.length} 个工具`);
+    return this.tools;
+  }
+  
+  async callTool(name, args) {
+    console.log(`🤖 调用工具: ${name}`);
+    const res = await this.request("tools/call", { name, arguments: args });
+    return res.content; // MCP 返回的是 content 数组
+  }
+
+  getOpenAITools() {
+    return this.tools.map(tool => ({
+      type: "function",
+      function: {
+        name: tool.name,
+        description: tool.description,
+        parameters: tool.inputSchema
+      }
+    }));
+  }
+}