xiaozhou-chat 1.0.2 → 1.0.3

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,9 @@ function showHelp() {
   console.log(`
 可用命令：
   /help           显示帮助
+  /config [set]   查看或修改配置
+  /system [prompt] 设置系统提示词
+  /load <file>    加载文件内容到上下文
   /history [N]    显示历史（可选最近 N 条）
   /clear          清空历史
   /export         导出历史为 Markdown
@@ -148,23 +151,49 @@ function showHelp() {
 initConfigFile();
 const config = loadConfig();
 
-const API_KEY =
+let API_KEY =
   config.apiKey ||
   args["api-key"] ||
   process.env.NEWAPI_API_KEY;
 
-const BASE_URL =
+let BASE_URL =
   config.baseUrl ||
   args["base-url"] ||
   process.env.NEWAPI_BASE_URL ||
   "https://api.newapi.pro/v1";
 
-const MODEL =
+let MODEL =
   config.model ||
   args["model"] ||
   process.env.NEWAPI_MODEL ||
   "gpt-3.5-turbo";
 
+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");
   process.exit(1);
@@ -175,7 +204,7 @@ let messages = loadHistory();
 const rl = readline.createInterface({
   input: process.stdin,
   output: process.stdout,
-  prompt: "你> "
+  prompt: "小周> "
 });
 
 readline.emitKeypressEvents(process.stdin);
@@ -199,7 +228,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 +252,7 @@ async function chatStream(userInput) {
     },
     body: JSON.stringify({
       model: MODEL,
-      messages,
+      messages: apiMessages,
       stream: true
     })
   });
@@ -220,22 +265,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;
@@ -260,6 +309,70 @@ 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.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.startsWith("/config")) {
+    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 +404,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.3",
   "description": "CLI chatbot based on NewAPI",
   "bin": {
     "xiaozhou-chat": "bin/cli.js"