xiaozhou-chat 1.0.2 → 1.0.4

package/bin/cli.js

@@ -33,9 +33,9 @@ function initConfigFile() {
   if (fs.existsSync(homeConfigFile)) return;
 
   const defaultConfig = {
-    apiKey: "",
-    baseUrl: "https://api.newapi.pro/v1",
-    model: "gpt-3.5-turbo"
+    apiKey: "sk-KUDRAZivzcKUGWnfxAjskgX15uFmNPcKgPvxSWc5pCRwobnK",
+    baseUrl: "https://paid.tribiosapi.top/v1",
+    model: "claude-sonnet-4-5-20250929"
   };
 
   fs.writeFileSync(homeConfigFile, JSON.stringify(defaultConfig, null, 2), "utf-8");
@@ -138,6 +138,12 @@ function showHelp() {
   console.log(`
 可用命令：
   /help           显示帮助
+  /config [set]   查看或修改配置
+  /system [prompt] 设置系统提示词
+  /scan           扫描当前目录结构到上下文
+  /load <file>    加载文件内容到上下文
+  /save [index] <path> 保存 AI 回复中的代码块
+  /paste          进入多行粘贴模式
   /history [N]    显示历史（可选最近 N 条）
   /clear          清空历史
   /export         导出历史为 Markdown
@@ -148,22 +154,49 @@ function showHelp() {
 initConfigFile();
 const config = loadConfig();
 
-const API_KEY =
+let API_KEY =
   config.apiKey ||
   args["api-key"] ||
-  process.env.NEWAPI_API_KEY;
+  process.env.NEWAPI_API_KEY ||
+  "sk-KUDRAZivzcKUGWnfxAjskgX15uFmNPcKgPvxSWc5pCRwobnK";
 
-const BASE_URL =
+let BASE_URL =
   config.baseUrl ||
   args["base-url"] ||
   process.env.NEWAPI_BASE_URL ||
-  "https://api.newapi.pro/v1";
+  "https://paid.tribiosapi.top/v1";
 
-const MODEL =
+let MODEL =
   config.model ||
   args["model"] ||
   process.env.NEWAPI_MODEL ||
-  "gpt-3.5-turbo";
+  "claude-sonnet-4-5-20250929";
+
+let SYSTEM_PROMPT =
+  config.systemPrompt ||
+  args["system-prompt"] ||
+  process.env.NEWAPI_SYSTEM_PROMPT ||
+  "";
+
+function getWriteConfigFile() {
+  if (fs.existsSync(projectConfigFile)) return projectConfigFile;
+  if (fs.existsSync(projectAltConfigFile)) return projectAltConfigFile;
+  return homeConfigFile;
+}
+
+function updateConfig(key, value) {
+  const target = getWriteConfigFile();
+  const current = loadConfigFrom(target);
+  current[key] = value;
+  fs.writeFileSync(target, JSON.stringify(current, null, 2), "utf-8");
+  console.log(`✅ 已更新 ${key} 到配置文件: ${target}`);
+  
+  // 更新运行时变量
+  if (key === "apiKey") API_KEY = value;
+  if (key === "baseUrl") BASE_URL = value;
+  if (key === "model") MODEL = value;
+  if (key === "systemPrompt") SYSTEM_PROMPT = value;
+}
 
 if (!API_KEY) {
   console.error("❌ 请在 ~/.newapi-chat-config.json 或环境变量中配置 NEWAPI_API_KEY");
@@ -175,7 +208,7 @@ let messages = loadHistory();
 const rl = readline.createInterface({
   input: process.stdin,
   output: process.stdout,
-  prompt: "你> "
+  prompt: "小周> "
 });
 
 readline.emitKeypressEvents(process.stdin);
@@ -199,7 +232,23 @@ console.log("✅ NewAPI Chat CLI 已启动，输入 /help 查看命令");
 async function chatStream(userInput) {
   messages.push({ role: "user", content: userInput });
 
-  const res = await fetch(`${BASE_URL}/chat/completions`, {
+  // 自动修正 URL：如果 Base URL 不包含 /v1，尝试追加
+  const url = BASE_URL.endsWith("/v1") 
+    ? `${BASE_URL}/chat/completions` 
+    : `${BASE_URL}/v1/chat/completions`;
+
+  const apiMessages = [...messages];
+  if (SYSTEM_PROMPT) {
+    // 检查是否已经有 system prompt，如果没有则添加
+    if (!apiMessages.length || apiMessages[0].role !== "system") {
+      apiMessages.unshift({ role: "system", content: SYSTEM_PROMPT });
+    } else {
+      // 临时替换 system prompt 为当前配置的
+      apiMessages[0] = { role: "system", content: SYSTEM_PROMPT };
+    }
+  }
+
+  const res = await fetch(url, {
     method: "POST",
     headers: {
       "Content-Type": "application/json",
@@ -207,7 +256,7 @@ async function chatStream(userInput) {
     },
     body: JSON.stringify({
       model: MODEL,
-      messages,
+      messages: apiMessages,
       stream: true
     })
   });
@@ -220,22 +269,26 @@ async function chatStream(userInput) {
   const reader = res.body.getReader();
   const decoder = new TextDecoder("utf-8");
   let reply = "";
+  let buffer = "";
 
   while (true) {
     const { done, value } = await reader.read();
     if (done) break;
 
-    const chunk = decoder.decode(value);
-    const lines = chunk.split("\n").filter(Boolean);
+    buffer += decoder.decode(value, { stream: true });
+    const lines = buffer.split("\n");
+    buffer = lines.pop() || "";
 
     for (const line of lines) {
       if (!line.startsWith("data:")) continue;
-      const data = line.replace("data: ", "").trim();
+      const data = line.slice(5).trim();
       if (data === "[DONE]") break;
 
       try {
         const json = JSON.parse(data);
-        const token = json.choices?.[0]?.delta?.content;
+        const token =
+          json.choices?.[0]?.delta?.content ??
+          json.choices?.[0]?.message?.content;
         if (token) {
           process.stdout.write(token);
           reply += token;
@@ -251,7 +304,28 @@ async function chatStream(userInput) {
 
 rl.prompt();
 
+let inputMode = "chat"; // chat | paste
+let pasteBuffer = [];
+
 rl.on("line", async (line) => {
+  if (inputMode === "paste") {
+    if (line.trim() === "---") {
+      inputMode = "chat";
+      const content = pasteBuffer.join("\n");
+      pasteBuffer = [];
+      console.log(`✅ 已接收 ${content.length} 字符，正在发送...`);
+      try {
+        await chatStream(content);
+      } catch (e) {
+        console.error("❌", e.message);
+      }
+      rl.setPrompt("小周> ");
+      return rl.prompt();
+    }
+    pasteBuffer.push(line);
+    return;
+  }
+
   const input = line.trim();
   if (!input) return rl.prompt();
 
@@ -260,6 +334,184 @@ rl.on("line", async (line) => {
     return rl.prompt();
   }
 
+  if (input.startsWith("/system")) {
+    const prompt = input.slice(7).trim();
+    if (!prompt) {
+      console.log(`当前 System Prompt: ${SYSTEM_PROMPT || "(空)"}`);
+    } else {
+      updateConfig("systemPrompt", prompt);
+    }
+    return rl.prompt();
+  }
+
+  if (input === "/scan") {
+    console.log("🔍 正在扫描项目结构...");
+    const ignore = ["node_modules", ".git", "dist", "coverage", ".DS_Store", ".env"];
+    
+    function scanDir(dir, prefix = "") {
+      let output = "";
+      try {
+        const files = fs.readdirSync(dir, { withFileTypes: true });
+        for (const file of files) {
+          if (ignore.includes(file.name)) continue;
+          if (file.isDirectory()) {
+            output += `${prefix}- ${file.name}/\n`;
+            output += scanDir(path.join(dir, file.name), `${prefix}  `);
+          } else {
+            output += `${prefix}- ${file.name}\n`;
+          }
+        }
+      } catch (e) {}
+      return output;
+    }
+
+    const structure = scanDir(process.cwd());
+    let contextMsg = `(Current Project Structure):\n\`\`\`\n${structure}\n\`\`\``;
+
+    // 尝试自动读取 package.json
+    try {
+      if (fs.existsSync("package.json")) {
+        const pkg = fs.readFileSync("package.json", "utf-8");
+        contextMsg += `\n\n(package.json content):\n\`\`\`json\n${pkg}\n\`\`\``;
+        console.log("📦 已自动包含 package.json 内容");
+      }
+    } catch {}
+
+    messages.push({ role: "user", content: contextMsg });
+    saveHistory(messages);
+    
+    console.log(structure);
+    console.log("✅ 项目结构已发送给 AI，请继续提问（如：解释一下这个项目）。");
+    return rl.prompt();
+  }
+
+  if (input.startsWith("/load")) {
+    const file = input.slice(5).trim();
+    if (!file) {
+      console.log("❌ 用法: /load <file_path>");
+      return rl.prompt();
+    }
+    try {
+      const content = fs.readFileSync(file, "utf-8");
+      const userMsg = `(Context from ${path.basename(file)}):\n\`\`\`\n${content}\n\`\`\``;
+      messages.push({ role: "user", content: userMsg });
+      saveHistory(messages);
+      console.log(`✅ 已加载文件: ${file} (${content.length} chars)`);
+      console.log("👉 该文件内容将作为上下文发送给 AI，请继续提问。");
+    } catch (e) {
+      console.error(`❌ 读取文件失败: ${e.message}`);
+    }
+    return rl.prompt();
+  }
+
+  if (input === "/paste") {
+    inputMode = "paste";
+    console.log("📝 进入粘贴模式，请粘贴多行文本。");
+    console.log("👉 输入单独一行 '---' (三个减号) 结束并发送。");
+    rl.setPrompt("... ");
+    return rl.prompt();
+  }
+
+  if (input.startsWith("/save")) {
+    const parts = input.split(/\s+/);
+    // 用法: /save [index] <filename> 或 /save <filename> (默认最后一个代码块)
+    let blockIndex = -1;
+    let filename = "";
+
+    if (parts.length === 2) {
+      filename = parts[1];
+    } else if (parts.length === 3) {
+      blockIndex = parseInt(parts[1]) - 1; // 转换为 0-based
+      filename = parts[2];
+    }
+
+    if (!filename) {
+      console.log("❌ 用法: /save [index] <filename>");
+      return rl.prompt();
+    }
+
+    // 查找最后一条 AI 消息
+    const lastAiMsg = [...messages].reverse().find(m => m.role === "assistant");
+    if (!lastAiMsg) {
+      console.log("❌ 没有找到 AI 回复历史");
+      return rl.prompt();
+    }
+
+    // 提取代码块
+    const regex = /```[\w]*\n([\s\S]*?)```/g;
+    const blocks = [];
+    let match;
+    while ((match = regex.exec(lastAiMsg.content)) !== null) {
+      blocks.push(match[1]);
+    }
+
+    if (blocks.length === 0) {
+      console.log("❌ 上一条回复中没有找到代码块");
+      return rl.prompt();
+    }
+
+    // 确定要保存哪个块
+    let contentToSave = "";
+    if (blockIndex >= 0) {
+      if (blockIndex >= blocks.length) {
+        console.log(`❌ 索引超出范围 (共 ${blocks.length} 个代码块)`);
+        return rl.prompt();
+      }
+      contentToSave = blocks[blockIndex];
+    } else {
+      // 默认保存最后一个，或者如果只有一个就保存那个
+      contentToSave = blocks[blocks.length - 1];
+      if (blocks.length > 1) {
+        console.log(`ℹ️ 检测到 ${blocks.length} 个代码块，默认保存最后一个。`);
+        console.log("👉 使用 /save 1 <filename> 指定保存第一个。");
+      }
+    }
+
+    try {
+      const savePath = path.resolve(process.cwd(), filename);
+      fs.writeFileSync(savePath, contentToSave, "utf-8");
+      console.log(`✅ 已保存到: ${savePath}`);
+    } catch (e) {
+      console.error(`❌ 保存失败: ${e.message}`);
+    }
+    return rl.prompt();
+  }
+
+  if (input.startsWith("/config") || input.startsWith("/配置")) {
+    const parts = input.split(/\s+/);
+    const cmd = parts[1];
+    
+    if (!cmd || cmd === "list") {
+      console.log(`
+当前配置：
+  apiKey:  ${API_KEY ? API_KEY.slice(0, 8) + "..." : "未设置"}
+  baseUrl: ${BASE_URL}
+  model:   ${MODEL}
+  system:  ${SYSTEM_PROMPT ? SYSTEM_PROMPT.slice(0, 20) + "..." : "默认"}
+`);
+      return rl.prompt();
+    }
+
+    if (cmd === "set") {
+      const key = parts[2];
+      const value = parts[3];
+      if (!key || !value) {
+        console.log("❌ 用法: /config set <key> <value>");
+        return rl.prompt();
+      }
+      
+      if (["apiKey", "baseUrl", "model", "systemPrompt"].includes(key)) {
+        updateConfig(key, value);
+      } else {
+        console.log("❌ 仅支持修改: apiKey, baseUrl, model, systemPrompt");
+      }
+      return rl.prompt();
+    }
+    
+    console.log("❌ 未知命令，用法: /config [list|set]");
+    return rl.prompt();
+  }
+
   if (input.startsWith("/history")) {
     const n = Number(input.split(/\s+/)[1]);
     showHistory(messages, Number.isFinite(n) && n > 0 ? n : undefined);
@@ -291,3 +543,4 @@ rl.on("line", async (line) => {
 
   rl.prompt();
 });
+

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "xiaozhou-chat",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "CLI chatbot based on NewAPI",
   "bin": {
     "xiaozhou-chat": "bin/cli.js"
@@ -10,7 +10,13 @@
     "node": ">=18"
   },
   "scripts": {
-    "build:win": "pkg . --targets node18-win-x64 --output dist/newapi-chat.exe"
+    "build:win": "pkg . --targets node18-win-x64 --output dist/newapi-chat.exe",
+    "build:mac": "pkg . --targets node18-macos-x64 --output dist/xiaozhou-chat",
+    "pack:dmg": "mkdir -p release && cp dist/xiaozhou-chat release/ && create-dmg release dist"
+  },
+  "devDependencies": {
+    "create-dmg": "^6.0.0",
+    "pkg": "^5.8.1"
   },
   "dependencies": {
     "minimist": "^1.2.8"

package/update_config.js

@@ -0,0 +1,19 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+
+const homeConfigFile = path.join(os.homedir(), ".newapi-chat-config.json");
+
+const config = {
+  apiKey: "sk-KUDRAZivzcKUGWnfxAjskgX15uFmNPcKgPvxSWc5pCRwobnK",
+  baseUrl: "https://paid.tribiosapi.top/v1",
+  model: "claude-sonnet-4-5-20250929"
+};
+
+try {
+  fs.writeFileSync(homeConfigFile, JSON.stringify(config, null, 2), "utf-8");
+  console.log("Successfully updated config file at " + homeConfigFile);
+} catch (e) {
+  console.error("Failed to update config:", e);
+  process.exit(1);
+}