Upload 19 files

anthropic-adapter.js

@@ -11,6 +11,7 @@ import { createContext, sendMessageStreaming, sendMultiChunkStreaming } from './
 import { anthropicToText, resolveModel, splitToChunks } from './message-convert.js';
 import { transformToAnthropicSSE, collectFullResponse, transformToAnthropicSSEWithTools, probeStream } from './stream-transform.js';
 import { parseToolCalls, toAnthropicToolUse } from './tool-prompt.js';
+import config from './config.js';
 
 const MAX_RETRY = 3;
 
@@ -61,6 +62,7 @@ export async function handleMessages(body, res, pool) {
   const stream = body.stream === true;
   const hasTools = body.tools && body.tools.length > 0;
   const isMultiChunk = chunks.length > 1;
+  const cost = config.modelCost?.[model] ?? config.defaultModelCost ?? 2;
 
   if (!text) {
     sendError(res, 400, 'No valid message content found');
@@ -115,7 +117,7 @@ export async function handleMessages(body, res, pool) {
             account = null;
           });
         }
-        probed.on('end', () => { if (account) pool.release(account, { success: true }); });
+        probed.on('end', () => { if (account) pool.release(account, { success: true, cost }); });
         probed.on('error', (err) => {
           if (!account) return;
           const msg = (err?.message || '').toLowerCase();
@@ -164,7 +166,7 @@ export async function handleMessages(body, res, pool) {
       if (result.cookies) account.cookies = result.cookies;
 
       const full = await collectFullResponse(result.stream);
-      pool.release(account, { success: true });
+      pool.release(account, { success: true, cost });
 
       // 检测工具调用
       const { hasToolCalls, toolCalls, textContent } = hasTools

chat.js

@@ -205,11 +205,34 @@ export async function getQuota(cookies) {
     });
     if (!resp.ok) return null;
     const data = resp.json();
-    // 可能返回 { answersCount, freeAnswersCount, ... }
-    return {
-      remaining: data.freeAnswersCount ?? data.answersCount ?? null,
-      raw: data,
-    };
+
+    // 从返回数据中提取剩余额度 — 兼容多种字段名
+    let remaining = null;
+    if (typeof data === 'object' && data !== null) {
+      // 尝试已知字段名
+      for (const key of ['freeAnswersCount', 'answersCount', 'remaining', 'count', 'free']) {
+        if (typeof data[key] === 'number') {
+          remaining = data[key];
+          break;
+        }
+      }
+      // 如果还是 null，遍历找第一个数字字段
+      if (remaining === null) {
+        for (const val of Object.values(data)) {
+          if (typeof val === 'number' && val >= 0) {
+            remaining = val;
+            break;
+          }
+        }
+      }
+    }
+
+    // 首次调试：打印完整返回以便确认字段
+    if (remaining === null) {
+      console.log(`[Quota] API 返回未知格式: ${JSON.stringify(data)}`);
+    }
+
+    return { remaining, raw: data };
   } catch {
     return null;
   }

config.js

@@ -89,6 +89,76 @@ export default {
     checkInterval: parseInt(env.CHECK_INTERVAL) || 300000,
   },
 
+  // ==================== 模型消耗量 (每次请求消耗的 requests 数) ====================
+  // chataibot.pro 不同模型扣费不同，用于本地扣减额度
+  modelCost: {
+    // 基础模型: 1 request
+    'gpt-3.5-turbo': 1,
+    'gpt-4o-mini': 1,
+    'gpt-4.1-nano': 1,
+    'gpt-4.1-mini': 1,
+    'gemini-flash': 1,
+    'gemini-3-flash': 1,
+    'deepseek': 1,
+    'deepseek-v3.2': 1,
+    'qwen3.5': 1,
+    'claude-3-haiku': 1,
+    'claude-4.5-haiku': 1,
+
+    // 中端模型: 2 requests
+    'gpt-4': 2,
+    'gpt-4-turbo': 2,
+    'gpt-4o': 2,
+    'gpt-4.1': 2,
+    'gemini-pro': 2,
+    'gemini-3-pro': 2,
+    'gemini-3.1-pro': 2,
+    'claude-3-sonnet': 2,
+    'qwen3.5-plus': 2,
+    'qwen3-max': 2,
+    'grok': 2,
+
+    // 高端模型: 4 requests
+    'claude-3-sonnet-high': 4,
+    'claude-4.6-sonnet': 4,
+    'claude-4.6-sonnet-high': 4,
+    'claude-3-opus': 4,
+    'claude-4.5-opus': 4,
+    'claude-4.6-opus': 4,
+    'gpt-5-pro': 4,
+    'gpt-5.1': 4,
+    'gpt-5.2': 4,
+    'gpt-5.4': 4,
+    'o1': 4,
+    'o1-preview': 4,
+    'o3': 4,
+    'o3-pro': 4,
+    'o4-mini': 4,
+    'perplexity-pro': 4,
+
+    // 搜索模型: 2 requests
+    'gpt-4o-search-preview': 2,
+    'gpt-4o-mini-search-preview': 2,
+    'gemini-2-flash-search': 2,
+    'gemini-3-pro-search': 2,
+    'gemini-3-flash-search': 2,
+
+    // 推理/高消耗: 4 requests
+    'gpt-5.1-high': 4,
+    'gpt-5.2-high': 4,
+    'gpt-5.4-high': 4,
+    'gpt-5.4-pro': 4,
+    'o1-mini': 2,
+    'o3-mini': 2,
+    'o3-mini-high': 4,
+    'o4-mini-high': 4,
+    'o4-mini-deep-research': 4,
+    'qwen3-thinking-2507': 2,
+    'perplexity': 2,
+  },
+  // 未知模型的默认消耗
+  defaultModelCost: 2,
+
   // ==================== 模型映射 ====================
   modelMapping: {
     // ---- OpenAI ----

openai-adapter.js

@@ -11,6 +11,7 @@ import { createContext, sendMessageStreaming, sendMultiChunkStreaming } from './
 import { openaiToText, resolveModel, splitToChunks } from './message-convert.js';
 import { transformToOpenAISSE, collectFullResponse, transformToOpenAISSEWithTools, probeStream } from './stream-transform.js';
 import { parseToolCalls, toOpenAIToolCalls } from './tool-prompt.js';
+import config from './config.js';
 
 const MAX_RETRY = 3;
 
@@ -73,6 +74,7 @@ export async function handleChatCompletions(body, res, pool) {
   const stream = body.stream === true;
   const hasTools = body.tools && body.tools.length > 0;
   const isMultiChunk = chunks.length > 1;
+  const cost = config.modelCost?.[model] ?? config.defaultModelCost ?? 2;
 
   if (!text) {
     sendError(res, 400, 'No valid message content found');
@@ -129,7 +131,7 @@ export async function handleChatCompletions(body, res, pool) {
             account = null;
           });
         }
-        probed.on('end', () => { if (account) pool.release(account, { success: true }); });
+        probed.on('end', () => { if (account) pool.release(account, { success: true, cost }); });
         probed.on('error', (err) => {
           if (!account) return;
           const msg = (err?.message || '').toLowerCase();
@@ -179,7 +181,7 @@ export async function handleChatCompletions(body, res, pool) {
       if (result.cookies) account.cookies = result.cookies;
 
       const full = await collectFullResponse(result.stream);
-      pool.release(account, { success: true });
+      pool.release(account, { success: true, cost });
 
       // 检测工具调用
       const { hasToolCalls, toolCalls, textContent } = hasTools

pool.js

@@ -27,12 +27,13 @@ const State = {
 export class AccountPool {
   constructor() {
     this.accounts = [];
-    this._registeringCount = 0;  // 当前正在注册的数量
-    this._maxConcurrentRegister = 10;  // 最大并行注册数
+    this._registeringCount = 0;
+    this._maxConcurrentRegister = 10;
     this._timer = null;
-    this._logs = [];        // 注册日志环形缓冲
-    this._maxLogs = 200;    // 最多保留 200 条
-    this._logId = 0;        // 自增 ID，方便前端增量拉取
+    this._logs = [];
+    this._maxLogs = 200;
+    this._logId = 0;
+    this._acquireLock = Promise.resolve();  // 互斥锁：防止并发 acquire 分配同一账号
   }
 
   /**
@@ -125,7 +126,11 @@ export class AccountPool {
         acc.state = State.ACTIVE;
         acc.lastCheckedAt = Date.now();
         const quota = await getQuota(acc.cookies);
-        if (quota) acc.remainingQuota = quota.remaining;
+        if (quota && typeof quota.remaining === 'number') {
+          acc.remainingQuota = quota.remaining;
+        }
+        // quota 未知时不赋值，保持 null — release 时用 cost 会跳过扣减
+        // 但 probeStream 会兜底检测 "Insufficient credits" 流错误
       } else {
         acc.state = State.NEEDS_LOGIN;
       }
@@ -166,9 +171,25 @@ export class AccountPool {
   }
 
   /**
-   * 获取一个可用账号 (Round-Robin + 额度感知)
+   * 获取一个可用账号 (互斥 + Round-Robin + 额度感知)
    */
-  async acquire() {
+  acquire() {
+    // 串行化: 每次 acquire 必须等前一个完成，防止并发分配同一账号
+    const prev = this._acquireLock;
+    let unlock;
+    this._acquireLock = new Promise(r => { unlock = r; });
+    return prev.then(() => this._doAcquire()).finally(unlock);
+  }
+
+  async _doAcquire() {
+    // 先清理卡死的 IN_USE 账号 (超过 5 分钟未释放)
+    const stuckTimeout = 5 * 60 * 1000;
+    for (const acc of this.accounts) {
+      if (acc.state === State.IN_USE && Date.now() - acc.lastUsedAt > stuckTimeout) {
+        acc.state = State.ACTIVE;
+        this._addLog(`回收卡死账号: ${acc.email}`, 'warn');
+      }
+    }
     // 优先选有额度的 (quota > 0 或 quota 未知)
     let candidates = this.accounts.filter(a =>
       a.state === State.ACTIVE && (a.remainingQuota === null || a.remainingQuota > 0)
@@ -192,7 +213,7 @@ export class AccountPool {
       const exhausted = this.accounts.filter(a => a.state === State.EXHAUSTED);
       for (const acc of exhausted) {
         const quota = await getQuota(acc.cookies);
-        if (quota && quota.remaining > 0) {
+        if (quota && typeof quota.remaining === 'number' && quota.remaining > 0) {
           acc.remainingQuota = quota.remaining;
           acc.state = State.ACTIVE;
           acc.lastCheckedAt = Date.now();
@@ -213,15 +234,23 @@ export class AccountPool {
       throw new Error('No available account in pool');
     }
 
-    // Round-Robin: 选最久未用的账号 (兼顾额度)
-    // 将 exhausted (quota=0) 的排到最后，其余按 lastUsedAt 升序
+    // 负载均衡: Round-Robin + 额度加权
+    // 1. quota=0 或 EXHAUSTED 的排最后
+    // 2. 额度未知 (null) 的正常参与轮询
+    // 3. 在 lastUsedAt 相近的账号中，优先选剩余额度多的
     candidates.sort((a, b) => {
-      // quota=0 的排最后
-      const aExhausted = a.remainingQuota !== null && a.remainingQuota <= 0;
-      const bExhausted = b.remainingQuota !== null && b.remainingQuota <= 0;
-      if (aExhausted !== bExhausted) return aExhausted ? 1 : -1;
-      // 最久未用的排前面 (Round-Robin 效果)
-      return a.lastUsedAt - b.lastUsedAt;
+      const aLow = typeof a.remainingQuota === 'number' && a.remainingQuota <= 0;
+      const bLow = typeof b.remainingQuota === 'number' && b.remainingQuota <= 0;
+      if (aLow !== bLow) return aLow ? 1 : -1;
+
+      // 时间窗口分组: 同一秒内视为"同时"，比较额度
+      const timeDiff = Math.floor(a.lastUsedAt / 1000) - Math.floor(b.lastUsedAt / 1000);
+      if (timeDiff !== 0) return timeDiff;
+
+      // 同一时间窗口内，额度多的优先 (null 视为中等额度 50)
+      const aQuota = typeof a.remainingQuota === 'number' ? a.remainingQuota : 50;
+      const bQuota = typeof b.remainingQuota === 'number' ? b.remainingQuota : 50;
+      return bQuota - aQuota;  // 降序，额度多的排前面
     });
 
     const account = candidates[0];
@@ -232,8 +261,10 @@ export class AccountPool {
 
   /**
    * 归还账号
+   * @param {object} opts
+   * @param {number} [opts.cost] - 本次请求消耗的 requests 数 (由 adapter 传入)
    */
-  release(account, { success = true, quotaExhausted = false, sessionExpired = false } = {}) {
+  release(account, { success = true, quotaExhausted = false, sessionExpired = false, cost = 0 } = {}) {
     if (sessionExpired) {
       account.state = State.NEEDS_LOGIN;
       account.errorCount++;
@@ -243,8 +274,14 @@ export class AccountPool {
     } else if (success) {
       account.state = State.ACTIVE;
       account.errorCount = 0;
-      // 异步查询真实剩余 requests 数量 (不同模型扣费不同)
-      this._refreshQuota(account);
+      // 本地扣减额度 (仅当有已知的 remaining 时)
+      if (cost > 0 && typeof account.remainingQuota === 'number') {
+        account.remainingQuota = Math.max(0, account.remainingQuota - cost);
+        if (account.remainingQuota <= 0) {
+          account.state = State.EXHAUSTED;
+          this._addLog(`额度耗尽 (本地扣减): ${account.email} (0 remaining)`, 'warn');
+        }
+      }
     } else {
       account.errorCount++;
       account.state = account.errorCount >= 5 ? State.DEAD : State.ACTIVE;
@@ -442,23 +479,6 @@ export class AccountPool {
     }
   }
 
-  /**
-   * 异步刷新账号的真实剩余 requests
-   */
-  async _refreshQuota(account) {
-    try {
-      const quota = await getQuota(account.cookies);
-      if (quota) {
-        account.remainingQuota = quota.remaining;
-        account.lastCheckedAt = Date.now();
-        if (quota.remaining <= 0) {
-          account.state = State.EXHAUSTED;
-          this._addLog(`额度耗尽: ${account.email} (${quota.remaining} requests)`, 'warn');
-        }
-      }
-    } catch {}
-  }
-
   /**
    * 将 dead 账号归档 (DB: 标记 state=dead / 文件: 移到 dead/ 子目录)
    */
@@ -515,7 +535,7 @@ export class AccountPool {
       for (const acc of this.accounts.filter(a => a.state === State.ACTIVE || a.state === State.EXHAUSTED)) {
         if (Date.now() - acc.lastCheckedAt < interval) continue;
         const quota = await getQuota(acc.cookies);
-        if (quota) {
+        if (quota && typeof quota.remaining === 'number') {
           acc.remainingQuota = quota.remaining;
           acc.lastCheckedAt = Date.now();
           if (acc.state === State.EXHAUSTED && quota.remaining > 0) {

server.js

@@ -23,10 +23,19 @@ await pool.init();
 
 // ==================== HTTP 服务器 ====================
 
-function parseBody(req) {
+function parseBody(req, maxSize = 2 * 1024 * 1024) {  // 默认 2MB
   return new Promise((resolve, reject) => {
     let data = '';
-    req.on('data', chunk => data += chunk);
+    let size = 0;
+    req.on('data', chunk => {
+      size += chunk.length;
+      if (size > maxSize) {
+        req.destroy();
+        reject(new Error('Request body too large'));
+        return;
+      }
+      data += chunk;
+    });
     req.on('end', () => {
       try { resolve(JSON.parse(data)); }
       catch { reject(new Error('Invalid JSON body')); }

stream-transform.js

@@ -85,34 +85,44 @@ function createRepeatDetector(minRepeatLen = 150) {
  * @returns {Promise<ReadableStream>}
  * @throws {Error} 如果流的第一个实质性对象就是 streamingError
  */
-export function probeStream(upstreamStream) {
+export function probeStream(upstreamStream, timeoutMs = 30000) {
   return new Promise((resolve, reject) => {
     let resolved = false;
-    let rawChunks = [];  // 缓存预检阶段读取的原始数据
+    let rawChunks = [];
+
+    function cleanup() {
+      clearTimeout(timer);
+      upstreamStream.removeListener('data', onData);
+      upstreamStream.removeListener('end', onEnd);
+      upstreamStream.removeListener('error', onError);
+    }
+
+    // 超时保护: 防止上游永不响应导致请求永久挂起
+    const timer = setTimeout(() => {
+      if (resolved) return;
+      resolved = true;
+      cleanup();
+      upstreamStream.destroy();
+      reject(new Error('probeStream timeout'));
+    }, timeoutMs);
 
     const parser = createJsonStreamParser((obj) => {
       if (resolved) return;
 
       if (obj.type === 'streamingError') {
         resolved = true;
-        upstreamStream.removeListener('data', onData);
-        upstreamStream.removeListener('end', onEnd);
-        upstreamStream.removeListener('error', onError);
+        cleanup();
         upstreamStream.resume();
-        const err = Object.assign(
+        reject(Object.assign(
           new Error(obj.data || 'Streaming error'),
           { statusCode: 429 }
-        );
-        reject(err);
+        ));
         return;
       }
 
-      // 收到第一个正常内容就确认流可用
       if (obj.type === 'chunk' || obj.type === 'reasoningContent') {
         resolved = true;
-        upstreamStream.removeListener('data', onData);
-        upstreamStream.removeListener('end', onEnd);
-        upstreamStream.removeListener('error', onError);
+        cleanup();
         upstreamStream.pause();
 
         const wrapped = new PassThrough();
@@ -133,6 +143,7 @@ export function probeStream(upstreamStream) {
       parser.flush();
       if (!resolved) {
         resolved = true;
+        cleanup();
         const wrapped = new PassThrough();
         for (const chunk of rawChunks) wrapped.write(chunk);
         wrapped.end();
@@ -142,6 +153,7 @@ export function probeStream(upstreamStream) {
     function onError(err) {
       if (resolved) return;
       resolved = true;
+      cleanup();
       reject(err);
     }