aicodeswitch 3.0.18 → 3.0.19

package/README.md

@@ -30,6 +30,10 @@ AI Code Switch 是帮助你在本地管理 AI 编程工具接入大模型的工
 * 自定义API Key，支持B/S架构，让aicodeswitch成为在线服务，提供给团队使用
 * 数据完全本地，自主可控
 
+## 桌面客户端
+
+[进入下载](https://github.com/tangshuang/aicodeswitch/releases)
+
 ## 命令行工具
 
 ### 安装

package/dist/server/proxy-server.js

@@ -111,6 +111,14 @@ class ProxyServer {
             writable: true,
             value: 60000
         }); // 去重缓存1分钟过期
+        // 频率限制跟踪：用于跟踪每个规则在当前时间窗口内的请求数
+        // key: ruleId, value: { count: number, windowStart: number }
+        Object.defineProperty(this, "frequencyLimitTracker", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new Map()
+        });
         this.dbManager = dbManager;
         this.config = dbManager.getConfig();
         this.app = app;
@@ -466,6 +474,9 @@ class ProxyServer {
         const routeRules = this.dbManager.getRules(routeId);
         return routeRules.sort((a, b) => (b.sortOrder || 0) - (a.sortOrder || 0));
     }
+    getRuleById(ruleId) {
+        return this.dbManager.getRule(ruleId);
+    }
     findMatchingRoute(req) {
         // 根据请求路径确定目标类型
         let targetType;
@@ -564,6 +575,32 @@ class ProxyServer {
             }
         }
     }
+    /**
+     * 清理过期的频率限制跟踪数据
+     */
+    cleanExpiredFrequencyTrackers() {
+        const now = Date.now();
+        const rules = this.dbManager.getRules();
+        const activeRuleIds = new Set(rules.map((r) => r.id));
+        for (const ruleId of this.frequencyLimitTracker.keys()) {
+            // 清理不再存在的规则的跟踪数据
+            if (!activeRuleIds.has(ruleId)) {
+                this.frequencyLimitTracker.delete(ruleId);
+                continue;
+            }
+            // 清理超时的跟踪数据
+            const tracker = this.frequencyLimitTracker.get(ruleId);
+            if (tracker) {
+                const rule = this.dbManager.getRule(ruleId);
+                if (rule && rule.frequencyWindow) {
+                    const windowMs = rule.frequencyWindow * 1000;
+                    if (now - tracker.windowStart > windowMs * 2) {
+                        this.frequencyLimitTracker.delete(ruleId);
+                    }
+                }
+            }
+        }
+    }
     /**
      * 根据GLM计费逻辑判断请求是否应该计费
      * 核心规则：
@@ -678,6 +715,10 @@ class ProxyServer {
                     if (rule.requestCountLimit && rule.totalRequestsUsed !== undefined && rule.totalRequestsUsed >= rule.requestCountLimit) {
                         continue; // 跳过超限规则
                     }
+                    // 检查频率限制
+                    if (this.isFrequencyLimitExceeded(rule)) {
+                        continue; // 跳过达到频率限制的规则
+                    }
                     return rule;
                 }
             }
@@ -701,6 +742,10 @@ class ProxyServer {
                 if (rule.requestCountLimit && rule.totalRequestsUsed !== undefined && rule.totalRequestsUsed >= rule.requestCountLimit) {
                     continue; // 跳过超限规则
                 }
+                // 检查频率限制
+                if (this.isFrequencyLimitExceeded(rule)) {
+                    continue; // 跳过达到频率限制的规则
+                }
                 return rule;
             }
             // 3. 最后返回 default 规则
@@ -722,6 +767,10 @@ class ProxyServer {
                 if (rule.requestCountLimit && rule.totalRequestsUsed !== undefined && rule.totalRequestsUsed >= rule.requestCountLimit) {
                     continue; // 跳过超限规则
                 }
+                // 检查频率限制
+                if (this.isFrequencyLimitExceeded(rule)) {
+                    continue; // 跳过达到频率限制的规则
+                }
                 return rule;
             }
             return undefined;
@@ -772,6 +821,10 @@ class ProxyServer {
                         return false;
                     }
                 }
+                // 检查频率限制
+                if (this.isFrequencyLimitExceeded(rule)) {
+                    return false;
+                }
                 return true; // 没有设置限制的规则总是可用
             });
             // 如果过滤后还有规则，使用过滤后的结果
@@ -781,6 +834,88 @@ class ProxyServer {
         }
         return candidates;
     }
+    /**
+     * 检查规则是否达到频率限制
+     * 如果设置了频率限制(frequencyLimit)和时间窗口(frequencyWindow)，
+     * 则跟踪当前时间窗口内的请求数，超过限制则返回true
+     *
+     * frequencyWindow = 0 表示"同一时刻"，计数器不会按时间窗口重置，
+     * 持续累积直到达到 frequencyLimit
+     */
+    isFrequencyLimitExceeded(rule) {
+        if (!rule.frequencyLimit || rule.frequencyLimit <= 0) {
+            return false; // 没有设置频率限制，不超过限制
+        }
+        // frequencyWindow 为 0 表示"同一时刻"（不按时间窗口重置）
+        const isZeroWindow = rule.frequencyWindow === 0;
+        if (!rule.frequencyWindow && !isZeroWindow) {
+            return false; // 没有设置时间窗口且不是0，不启用频率限制
+        }
+        const now = Date.now();
+        const existing = this.frequencyLimitTracker.get(rule.id);
+        if (!existing) {
+            // 首次请求，创建新记录
+            this.frequencyLimitTracker.set(rule.id, { count: 1, windowStart: now });
+            return false;
+        }
+        // 如果是零窗口（同一时刻），不按时间重置，持续累积
+        if (!isZeroWindow && rule.frequencyWindow) {
+            const windowMs = rule.frequencyWindow * 1000;
+            // 检查是否在当前时间窗口内
+            if (now - existing.windowStart >= windowMs) {
+                // 时间窗口已过，重置计数器
+                this.frequencyLimitTracker.set(rule.id, { count: 1, windowStart: now });
+                return false;
+            }
+        }
+        // 检查是否超过限制
+        if (existing.count >= rule.frequencyLimit) {
+            return true; // 超过频率限制
+        }
+        // 增加计数
+        existing.count++;
+        this.frequencyLimitTracker.set(rule.id, existing);
+        return false;
+    }
+    /**
+     * 记录请求（增加频率计数）
+     * 在请求成功处理后调用
+     * frequencyWindow = 0 表示"同一时刻"，计数器不会按时间窗口重置
+     */
+    recordRequest(ruleId) {
+        const rule = this.getRuleById(ruleId);
+        if (!rule || !rule.frequencyLimit || rule.frequencyLimit <= 0) {
+            return;
+        }
+        // frequencyWindow 为 0 表示"同一时刻"
+        const isZeroWindow = rule.frequencyWindow === 0;
+        // 如果 frequencyWindow 既不是 0 也不是正数，则不记录
+        if (!isZeroWindow && !rule.frequencyWindow) {
+            return;
+        }
+        const now = Date.now();
+        const existing = this.frequencyLimitTracker.get(ruleId);
+        if (!existing) {
+            this.frequencyLimitTracker.set(ruleId, { count: 1, windowStart: now });
+        }
+        else if (isZeroWindow) {
+            // 零窗口：持续累积，不按时间重置
+            existing.count++;
+            this.frequencyLimitTracker.set(ruleId, existing);
+        }
+        else if (rule.frequencyWindow) {
+            const windowMs = rule.frequencyWindow * 1000;
+            if (now - existing.windowStart < windowMs) {
+                // 在时间窗口内，增加计数
+                existing.count++;
+                this.frequencyLimitTracker.set(ruleId, existing);
+            }
+            else {
+                // 时间窗口已过，重置
+                this.frequencyLimitTracker.set(ruleId, { count: 1, windowStart: now });
+            }
+        }
+    }
     determineContentType(req) {
         const body = req.body;
         if (!body)
@@ -1426,7 +1561,7 @@ class ProxyServer {
     }
     proxyRequest(req, res, route, rule, service) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d;
+            var _a, _b, _c, _d, _e, _f;
             res.locals.skipLog = true;
             const startTime = Date.now();
             const sourceType = (service.sourceType || 'openai-chat');
@@ -1595,7 +1730,7 @@ class ProxyServer {
                     targetType,
                     targetServiceId: service.id,
                     targetServiceName: service.name,
-                    targetModel: rule.targetModel,
+                    targetModel: rule.targetModel || requestModel,
                     vendorId: service.vendorId,
                     vendorName: vendor === null || vendor === void 0 ? void 0 : vendor.name,
                     requestModel,
@@ -1643,6 +1778,8 @@ class ProxyServer {
                     // 检查是否是重复请求（如网络重试）
                     if (!this.isRequestProcessed(requestHash)) {
                         this.dbManager.incrementRuleRequestCount(rule.id, 1);
+                        // 更新频率限制跟踪
+                        this.recordRequest(rule.id);
                         // 获取更新后的规则数据并广播
                         const updatedRule = this.dbManager.getRule(rule.id);
                         if (updatedRule) {
@@ -1654,6 +1791,10 @@ class ProxyServer {
                         this.cleanExpiredDedupeCache();
                     }
                 }
+                // 定期清理过期的频率限制跟踪数据
+                if (Math.random() < 0.01) { // 1%概率清理
+                    this.cleanExpiredFrequencyTrackers();
+                }
                 // 清理 MCP 临时图片文件
                 if (tempImageFiles.length > 0) {
                     (0, mcp_image_handler_1.cleanupTempImages)(tempImageFiles);
@@ -1832,7 +1973,7 @@ class ProxyServer {
                             console.error('[Proxy] Response stream error:', err);
                         });
                         (0, stream_1.pipeline)(response.data, parser, eventCollector, converter, serializer, res, (error) => __awaiter(this, void 0, void 0, function* () {
-                            var _a;
+                            var _a, _b;
                             if (error) {
                                 console.error('[Proxy] Pipeline error for claude-code:', error);
                                 // 记录到错误日志 - 包含请求详情和实际转发信息
@@ -1855,10 +1996,10 @@ class ProxyServer {
                                         targetType,
                                         targetServiceId: service.id,
                                         targetServiceName: service.name,
-                                        targetModel: rule.targetModel,
+                                        targetModel: rule.targetModel || ((_a = req.body) === null || _a === void 0 ? void 0 : _a.model),
                                         vendorId: service.vendorId,
                                         vendorName: vendor === null || vendor === void 0 ? void 0 : vendor.name,
-                                        requestModel: (_a = req.body) === null || _a === void 0 ? void 0 : _a.model,
+                                        requestModel: (_b = req.body) === null || _b === void 0 ? void 0 : _b.model,
                                         responseTime: Date.now() - startTime,
                                     });
                                 }
@@ -1934,7 +2075,7 @@ class ProxyServer {
                             console.error('[Proxy] Response stream error:', err);
                         });
                         (0, stream_1.pipeline)(response.data, parser, eventCollector, converter, serializer, res, (error) => __awaiter(this, void 0, void 0, function* () {
-                            var _a;
+                            var _a, _b;
                             if (error) {
                                 console.error('[Proxy] Pipeline error for codex:', error);
                                 // 记录到错误日志 - 包含请求详情和实际转发信息
@@ -1957,10 +2098,10 @@ class ProxyServer {
                                         targetType,
                                         targetServiceId: service.id,
                                         targetServiceName: service.name,
-                                        targetModel: rule.targetModel,
+                                        targetModel: rule.targetModel || ((_a = req.body) === null || _a === void 0 ? void 0 : _a.model),
                                         vendorId: service.vendorId,
                                         vendorName: vendor === null || vendor === void 0 ? void 0 : vendor.name,
-                                        requestModel: (_a = req.body) === null || _a === void 0 ? void 0 : _a.model,
+                                        requestModel: (_b = req.body) === null || _b === void 0 ? void 0 : _b.model,
                                         responseTime: Date.now() - startTime,
                                     });
                                 }
@@ -2030,7 +2171,7 @@ class ProxyServer {
                             console.error('[Proxy] Response stream error:', err);
                         });
                         (0, stream_1.pipeline)(response.data, parser, eventCollector, converter, serializer, res, (error) => __awaiter(this, void 0, void 0, function* () {
-                            var _a;
+                            var _a, _b;
                             if (error) {
                                 console.error('[Proxy] Pipeline error for gemini->claude-code:', error);
                                 try {
@@ -2050,10 +2191,10 @@ class ProxyServer {
                                         targetType,
                                         targetServiceId: service.id,
                                         targetServiceName: service.name,
-                                        targetModel: rule.targetModel,
+                                        targetModel: rule.targetModel || ((_a = req.body) === null || _a === void 0 ? void 0 : _a.model),
                                         vendorId: service.vendorId,
                                         vendorName: vendor === null || vendor === void 0 ? void 0 : vendor.name,
-                                        requestModel: (_a = req.body) === null || _a === void 0 ? void 0 : _a.model,
+                                        requestModel: (_b = req.body) === null || _b === void 0 ? void 0 : _b.model,
                                         responseTime: Date.now() - startTime,
                                     });
                                 }
@@ -2126,7 +2267,7 @@ class ProxyServer {
                             console.error('[Proxy] Response stream error:', err);
                         });
                         (0, stream_1.pipeline)(response.data, parser, eventCollector, converter, serializer, res, (error) => __awaiter(this, void 0, void 0, function* () {
-                            var _a;
+                            var _a, _b;
                             if (error) {
                                 console.error('[Proxy] Pipeline error for gemini->codex:', error);
                                 try {
@@ -2146,10 +2287,10 @@ class ProxyServer {
                                         targetType,
                                         targetServiceId: service.id,
                                         targetServiceName: service.name,
-                                        targetModel: rule.targetModel,
+                                        targetModel: rule.targetModel || ((_a = req.body) === null || _a === void 0 ? void 0 : _a.model),
                                         vendorId: service.vendorId,
                                         vendorName: vendor === null || vendor === void 0 ? void 0 : vendor.name,
-                                        requestModel: (_a = req.body) === null || _a === void 0 ? void 0 : _a.model,
+                                        requestModel: (_b = req.body) === null || _b === void 0 ? void 0 : _b.model,
                                         responseTime: Date.now() - startTime,
                                     });
                                 }
@@ -2280,10 +2421,10 @@ class ProxyServer {
                         targetType,
                         targetServiceId: service.id,
                         targetServiceName: service.name,
-                        targetModel: rule.targetModel,
+                        targetModel: rule.targetModel || ((_b = req.body) === null || _b === void 0 ? void 0 : _b.model),
                         vendorId: service.vendorId,
                         vendorName: vendor === null || vendor === void 0 ? void 0 : vendor.name,
-                        requestModel: (_b = req.body) === null || _b === void 0 ? void 0 : _b.model,
+                        requestModel: (_c = req.body) === null || _c === void 0 ? void 0 : _c.model,
                         upstreamRequest: upstreamRequestForLog,
                         responseTime: Date.now() - startTime,
                     });
@@ -2341,7 +2482,7 @@ class ProxyServer {
                 console.error('Proxy error:', error);
                 // 检测是否是 timeout 错误
                 const isTimeout = error.code === 'ECONNABORTED' ||
-                    ((_c = error.message) === null || _c === void 0 ? void 0 : _c.toLowerCase().includes('timeout')) ||
+                    ((_d = error.message) === null || _d === void 0 ? void 0 : _d.toLowerCase().includes('timeout')) ||
                     (error.errno && error.errno === 'ETIMEDOUT');
                 const errorMessage = isTimeout
                     ? 'Request timeout - the upstream API took too long to respond'
@@ -2364,10 +2505,10 @@ class ProxyServer {
                     targetType,
                     targetServiceId: service.id,
                     targetServiceName: service.name,
-                    targetModel: rule.targetModel,
+                    targetModel: rule.targetModel || ((_e = req.body) === null || _e === void 0 ? void 0 : _e.model),
                     vendorId: service.vendorId,
                     vendorName: vendor === null || vendor === void 0 ? void 0 : vendor.name,
-                    requestModel: (_d = req.body) === null || _d === void 0 ? void 0 : _d.model,
+                    requestModel: (_f = req.body) === null || _f === void 0 ? void 0 : _f.model,
                     upstreamRequest: upstreamRequestForLog,
                     responseTime: Date.now() - startTime,
                 });