aicodeswitch 3.0.20 → 3.6.1

package/README.md

@@ -29,6 +29,7 @@ AI Code Switch 是帮助你在本地管理 AI 编程工具接入大模型的工
 * 导入和导出：一键备份数据，在多太电脑间共享aicodeswitch配置
 * 自定义API Key，支持B/S架构，让aicodeswitch成为在线服务，提供给团队使用
 * 数据完全本地，自主可控
+* 特殊语法：在发送的提示词最前面添加!!来直接切换为高智商模型服务，简单快捷
 
 ## 桌面客户端
 

package/dist/server/fs-database.js

@@ -28,6 +28,7 @@ const path_1 = __importDefault(require("path"));
 const promises_1 = __importDefault(require("fs/promises"));
 const crypto_1 = __importDefault(require("crypto"));
 const crypto_js_1 = __importDefault(require("crypto-js"));
+const type_migration_1 = require("./type-migration");
 /**
  * 基于文件系统的数据库管理器
  * 使用 JSON 文件存储数据，无需编译依赖
@@ -206,6 +207,8 @@ class FileSystemDatabaseManager {
             yield promises_1.default.mkdir(this.dataPath, { recursive: true });
             // 加载所有数据
             yield this.loadAllData();
+            // 执行数据源类型迁移（在加载数据之后）
+            yield this.migrateSourceTypes();
             // 确保默认配置
             yield this.ensureDefaultConfig();
         });
@@ -293,6 +296,77 @@ class FileSystemDatabaseManager {
             }
         });
     }
+    /**
+     * 迁移数据源类型（在初始化时执行）
+     * 处理 vendors.json 中的 services[].sourceType
+     * 将旧类型 ('claude-code', 'openai-responses') 迁移为新类型 ('claude', 'openai')
+     */
+    migrateSourceTypes() {
+        return __awaiter(this, void 0, void 0, function* () {
+            console.log('[TypeMigration] Checking for source type migration...');
+            let needsMigration = false;
+            // 检查是否需要迁移
+            for (const vendor of this.vendors) {
+                if (vendor.services) {
+                    for (const service of vendor.services) {
+                        if (service.sourceType && (0, type_migration_1.isLegacySourceType)(service.sourceType)) {
+                            needsMigration = true;
+                            break;
+                        }
+                    }
+                }
+                if (needsMigration)
+                    break;
+            }
+            if (!needsMigration) {
+                console.log('[TypeMigration] No migration needed');
+                return;
+            }
+            console.log('[TypeMigration] Starting source type migration...');
+            // 备份当前数据到 ~/.aicodeswitch/backup/YYYY-MM-DD-HH-MM/
+            const now = new Date();
+            const year = now.getFullYear();
+            const month = String(now.getMonth() + 1).padStart(2, '0');
+            const day = String(now.getDate()).padStart(2, '0');
+            const hours = String(now.getHours()).padStart(2, '0');
+            const minutes = String(now.getMinutes()).padStart(2, '0');
+            const dateStr = `${year}-${month}-${day}-${hours}-${minutes}`;
+            const appDir = path_1.default.dirname(this.dataPath); // ~/.aicodeswitch/
+            const backupBaseDir = path_1.default.join(appDir, 'backup');
+            const backupDir = path_1.default.join(backupBaseDir, dateStr);
+            yield promises_1.default.mkdir(backupDir, { recursive: true });
+            const vendorsBackupPath = path_1.default.join(backupDir, 'vendors.json');
+            yield promises_1.default.writeFile(vendorsBackupPath, JSON.stringify(this.vendors, null, 2));
+            console.log(`[TypeMigration] Backup created: ${vendorsBackupPath}`);
+            // 执行迁移
+            let migratedCount = 0;
+            for (const vendor of this.vendors) {
+                if (vendor.services) {
+                    for (const service of vendor.services) {
+                        if (service.sourceType && (0, type_migration_1.isLegacySourceType)(service.sourceType)) {
+                            const oldType = service.sourceType;
+                            service.sourceType = (0, type_migration_1.migrateSourceType)(service.sourceType);
+                            console.log(`[TypeMigration] Migrated service "${service.name}": ${oldType} -> ${service.sourceType}`);
+                            migratedCount++;
+                        }
+                    }
+                }
+            }
+            // 保存迁移后的数据
+            yield this.saveVendors();
+            console.log(`[TypeMigration] Migration completed. Migrated ${migratedCount} services.`);
+        });
+    }
+    /**
+     * 迁移导入数据中的类型
+     * 用于导入功能，自动将旧类型转换为新类型
+     */
+    migrateVendorsOnImport(vendors) {
+        return vendors.map(vendor => {
+            var _a;
+            return (Object.assign(Object.assign({}, vendor), { services: (_a = vendor.services) === null || _a === void 0 ? void 0 : _a.map(service => (Object.assign(Object.assign({}, service), { sourceType: service.sourceType ? (0, type_migration_1.normalizeSourceType)(service.sourceType) : undefined }))) }));
+        });
+    }
     saveVendors() {
         return __awaiter(this, void 0, void 0, function* () {
             // 确保每个供应商都有 services 数组
@@ -1507,7 +1581,7 @@ class FileSystemDatabaseManager {
             const existing = this.blacklist.get(key);
             if (existing) {
                 existing.blacklistedAt = now;
-                existing.expiresAt = now + 10 * 60 * 1000;
+                existing.expiresAt = now + 2 * 60 * 1000; // 2分钟黑名单（从10分钟缩短）
                 existing.errorCount++;
                 existing.lastError = errorMessage;
                 existing.lastStatusCode = statusCode;
@@ -1519,7 +1593,7 @@ class FileSystemDatabaseManager {
                     routeId,
                     contentType,
                     blacklistedAt: now,
-                    expiresAt: now + 10 * 60 * 1000,
+                    expiresAt: now + 2 * 60 * 1000, // 2分钟黑名单（从10分钟缩短）
                     errorCount: 1,
                     lastError: errorMessage,
                     lastStatusCode: statusCode,
@@ -1807,6 +1881,10 @@ class FileSystemDatabaseManager {
                 if (!validation.valid) {
                     return { success: false, message: '导入失败', details: `数据验证失败：${validation.error}` };
                 }
+                // 自动迁移导入数据中的旧类型
+                if (importData.vendors) {
+                    importData.vendors = this.migrateVendorsOnImport(importData.vendors);
+                }
                 // 导入数据（更新 updatedAt）
                 const now = Date.now();
                 this.vendors = importData.vendors.map((v) => (Object.assign(Object.assign({}, v), { updatedAt: now })));

package/dist/server/main.js

@@ -28,6 +28,7 @@ const utils_1 = require("./utils");
 const tools_service_1 = require("./tools-service");
 const websocket_service_1 = require("./websocket-service");
 const rules_status_service_1 = require("./rules-status-service");
+const type_migration_1 = require("./type-migration");
 const config_metadata_1 = require("./config-metadata");
 const config_1 = require("./config");
 const appDir = path_1.default.join(os_1.default.homedir(), '.aicodeswitch');
@@ -75,6 +76,28 @@ const app = (0, express_1.default)();
 app.use((0, cors_1.default)());
 app.use(express_1.default.json({ limit: 'Infinity' }));
 app.use(express_1.default.urlencoded({ extended: true, limit: 'Infinity' }));
+// 类型转换中间件：自动将旧的数据源类型转换为新类型
+app.use((req, _res, next) => {
+    if (req.body && typeof req.body === 'object') {
+        // 转换 sourceType
+        if (req.body.sourceType && typeof req.body.sourceType === 'string') {
+            if ((0, type_migration_1.isLegacySourceType)(req.body.sourceType)) {
+                console.log(`[API] Converting legacy sourceType: ${req.body.sourceType} -> ${(0, type_migration_1.normalizeSourceType)(req.body.sourceType)}`);
+                req.body.sourceType = (0, type_migration_1.normalizeSourceType)(req.body.sourceType);
+            }
+        }
+        // 转换数组中的 sourceType（如 vendors 的 services）
+        if (Array.isArray(req.body.services)) {
+            req.body.services = req.body.services.map((service) => {
+                if (service.sourceType && (0, type_migration_1.isLegacySourceType)(service.sourceType)) {
+                    return Object.assign(Object.assign({}, service), { sourceType: (0, type_migration_1.normalizeSourceType)(service.sourceType) });
+                }
+                return service;
+            });
+        }
+    }
+    next();
+});
 const asyncHandler = (handler) => (req, res, next) => {
     Promise.resolve(handler(req, res, next)).catch((err) => {
         console.error('[asyncHandler] Caught error:', err);

package/dist/server/proxy-server.js

@@ -158,6 +158,8 @@ class ProxyServer {
                 }
                 // 尝试每个规则,直到成功或全部失败
                 let lastError = null;
+                let lastFailedRule = null;
+                let lastFailedService = null;
                 for (let index = 0; index < allRules.length; index++) {
                     const rule = allRules[index];
                     const service = this.getServiceById(rule.targetServiceId);
@@ -181,6 +183,8 @@ class ProxyServer {
                     catch (error) {
                         console.error(`Service ${service.name} failed:`, error.message);
                         lastError = error;
+                        lastFailedRule = rule;
+                        lastFailedService = service;
                         // 检测是否是 timeout 错误
                         const isTimeout = error.code === 'ECONNABORTED' ||
                             ((_b = error.message) === null || _b === void 0 ? void 0 : _b.toLowerCase().includes('timeout')) ||
@@ -206,6 +210,21 @@ class ProxyServer {
                 }
                 // 所有服务都失败了
                 console.error('All services failed');
+                // 如果有失败的服务但都在黑名单中，尝试使用最后一个失败的服务（作为 fallback）
+                if (lastFailedRule && lastFailedService) {
+                    console.log(`All services in blacklist, attempting fallback to last failed service: ${lastFailedService.name}`);
+                    try {
+                        yield this.proxyRequest(req, res, route, lastFailedRule, lastFailedService, {
+                            failoverEnabled: false, // Fallback 模式不启用故障切换
+                            forwardedToServiceName: undefined,
+                        });
+                        return;
+                    }
+                    catch (fallbackError) {
+                        console.error(`Fallback to service ${lastFailedService.name} also failed:`, fallbackError.message);
+                        lastError = fallbackError;
+                    }
+                }
                 // 记录日志
                 if (((_d = this.config) === null || _d === void 0 ? void 0 : _d.enableLogging) !== false && SUPPORTED_TARGETS.some(target => req.path.startsWith(`/${target}/`))) {
                     yield this.dbManager.addLog({
@@ -344,6 +363,8 @@ class ProxyServer {
                 }
                 // 尝试每个规则,直到成功或全部失败
                 let lastError = null;
+                let lastFailedRule = null;
+                let lastFailedService = null;
                 for (let index = 0; index < allRules.length; index++) {
                     const rule = allRules[index];
                     const service = this.getServiceById(rule.targetServiceId);
@@ -367,6 +388,8 @@ class ProxyServer {
                     catch (error) {
                         console.error(`Service ${service.name} failed:`, error.message);
                         lastError = error;
+                        lastFailedRule = rule;
+                        lastFailedService = service;
                         // 检测是否是 timeout 错误
                         const isTimeout = error.code === 'ECONNABORTED' ||
                             ((_b = error.message) === null || _b === void 0 ? void 0 : _b.toLowerCase().includes('timeout')) ||
@@ -392,6 +415,21 @@ class ProxyServer {
                 }
                 // 所有服务都失败了
                 console.error('All services failed');
+                // 如果有失败的服务但都在黑名单中，尝试使用最后一个失败的服务（作为 fallback）
+                if (lastFailedRule && lastFailedService) {
+                    console.log(`All services in blacklist, attempting fallback to last failed service: ${lastFailedService.name}`);
+                    try {
+                        yield this.proxyRequest(req, res, route, lastFailedRule, lastFailedService, {
+                            failoverEnabled: false, // Fallback 模式不启用故障切换
+                            forwardedToServiceName: undefined,
+                        });
+                        return;
+                    }
+                    catch (fallbackError) {
+                        console.error(`Fallback to service ${lastFailedService.name} also failed:`, fallbackError.message);
+                        lastError = fallbackError;
+                    }
+                }
                 // 记录日志
                 if (((_d = this.config) === null || _d === void 0 ? void 0 : _d.enableLogging) !== false && SUPPORTED_TARGETS.some(target => req.path.startsWith(`/${target}/`))) {
                     yield this.dbManager.addLog({
@@ -985,6 +1023,10 @@ class ProxyServer {
                 type: 'thinking',
                 match: (_req, body) => this.hasThinkingSignal(body),
             },
+            {
+                type: 'high-iq',
+                match: (_req, body) => this.hasHighIqSignal(body),
+            },
             {
                 type: 'long-context',
                 match: (_req, body) => this.hasLongContextSignal(body),
@@ -1049,6 +1091,10 @@ class ProxyServer {
             bg: 'background',
             thinking: 'thinking',
             reasoning: 'thinking',
+            'high-iq': 'high-iq',
+            high_iq: 'high-iq',
+            highiq: 'high-iq',
+            smart: 'high-iq',
             'long-context': 'long-context',
             long_context: 'long-context',
             long: 'long-context',
@@ -1090,6 +1136,65 @@ class ProxyServer {
             ((_a = body === null || body === void 0 ? void 0 : body.reasoning) === null || _a === void 0 ? void 0 : _a.effort) ||
             ((_b = body === null || body === void 0 ? void 0 : body.reasoning) === null || _b === void 0 ? void 0 : _b.enabled));
     }
+    hasHighIqSignal(body) {
+        const messages = body === null || body === void 0 ? void 0 : body.messages;
+        if (!Array.isArray(messages)) {
+            return false;
+        }
+        for (const message of messages) {
+            if ((message === null || message === void 0 ? void 0 : message.role) !== 'user')
+                continue;
+            const content = message === null || message === void 0 ? void 0 : message.content;
+            // 处理字符串类型的 content
+            if (typeof content === 'string') {
+                if (content.trim().startsWith('!!')) {
+                    return true;
+                }
+            }
+            // 处理数组类型的 content
+            else if (Array.isArray(content)) {
+                for (const block of content) {
+                    if ((block === null || block === void 0 ? void 0 : block.type) === 'text' && typeof block.text === 'string') {
+                        if (block.text.trim().startsWith('!!')) {
+                            return true;
+                        }
+                    }
+                }
+            }
+        }
+        return false;
+    }
+    removeHighIqPrefix(body) {
+        if (!(body === null || body === void 0 ? void 0 : body.messages) || !Array.isArray(body.messages)) {
+            return body;
+        }
+        // 深拷贝 body 以避免修改原始对象
+        const processedBody = JSON.parse(JSON.stringify(body));
+        for (const message of processedBody.messages) {
+            if ((message === null || message === void 0 ? void 0 : message.role) !== 'user')
+                continue;
+            const content = message === null || message === void 0 ? void 0 : message.content;
+            // 处理字符串类型的 content
+            if (typeof content === 'string') {
+                if (content.trim().startsWith('!!')) {
+                    // 移除 !! 前缀并执行 trim
+                    message.content = content.replace(/^!!\s*/, '').trim();
+                }
+            }
+            // 处理数组类型的 content
+            else if (Array.isArray(content)) {
+                for (const block of content) {
+                    if ((block === null || block === void 0 ? void 0 : block.type) === 'text' && typeof block.text === 'string') {
+                        if (block.text.trim().startsWith('!!')) {
+                            // 移除 !! 前缀并执行 trim
+                            block.text = block.text.replace(/^!!\s*/, '').trim();
+                        }
+                    }
+                }
+            }
+        }
+        return processedBody;
+    }
     hasBackgroundSignal(body) {
         var _a, _b, _c;
         // 检测 count tokens 请求：messages 只有一条，role 为 "user"，content 为 "count"
@@ -1211,15 +1316,19 @@ class ProxyServer {
     }
     /** 判断是否为 Claude 相关类型（使用 x-api-key 认证） */
     isClaudeSource(sourceType) {
-        return sourceType === 'claude-chat' || sourceType === 'claude-code';
+        return sourceType === 'claude-chat' || sourceType === 'claude';
     }
     isOpenAIChatSource(sourceType) {
-        return sourceType === 'openai-chat' || sourceType === 'openai-responses' || sourceType === 'deepseek-reasoning-chat';
+        return sourceType === 'openai-chat' || sourceType === 'openai' || sourceType === 'deepseek-reasoning-chat';
     }
     /** 判断是否为 Gemini 类型 */
     isGeminiSource(sourceType) {
         return sourceType === 'gemini';
     }
+    /** 判断是否为 Gemini Chat 类型 */
+    isGeminiChatSource(sourceType) {
+        return sourceType === 'gemini-chat';
+    }
     isChatType(sourceType) {
         return sourceType.endsWith('-chat') || sourceType === 'gemini';
     }
@@ -1341,8 +1450,8 @@ class ProxyServer {
         // 向下兼容：检测旧数据的 'auto' 值
         // TODO: 删除
         const isAuto = authType === 'auto';
-        // 使用 x-goog-api-key 认证（适用于 Google Gemini API）
-        if (authType === types_1.AuthType.G_API_KEY || (isAuto && this.isGeminiSource(sourceType))) {
+        // 使用 x-goog-api-key 认证（适用于 Google Gemini API 和 Gemini Chat）
+        if (authType === types_1.AuthType.G_API_KEY || (isAuto && (this.isGeminiSource(sourceType) || this.isGeminiChatSource(sourceType)))) {
             headers['x-goog-api-key'] = service.apiKey;
         }
         // 使用 x-api-key 认证（适用于 claude-chat, claude-code 及某些需要 x-api-key 的 openai-chat 兼容 API）
@@ -1728,6 +1837,11 @@ class ProxyServer {
                     useMCPProcessing = false;
                 }
             }
+            // 高智商请求处理：移除 !! 前缀
+            if (rule.contentType === 'high-iq' && requestBody.messages) {
+                requestBody = this.removeHighIqPrefix(requestBody);
+                console.log('[HIGH-IQ] Removed !! prefix from user messages');
+            }
             // 用于收集响应数据的变量
             let responseHeadersForLog;
             let responseBodyForLog;
@@ -1907,7 +2021,7 @@ class ProxyServer {
                     else if (this.isOpenAIChatSource(sourceType)) {
                         requestBody = (0, claude_openai_1.transformClaudeRequestToOpenAIChat)(requestBody, rule.targetModel);
                     }
-                    else if (this.isGeminiSource(sourceType)) {
+                    else if (this.isGeminiSource(sourceType) || this.isGeminiChatSource(sourceType)) {
                         requestBody = (0, gemini_1.transformClaudeRequestToGemini)(requestBody);
                     }
                     else {
@@ -1923,7 +2037,7 @@ class ProxyServer {
                     else if (this.isClaudeSource(sourceType)) {
                         requestBody = (0, claude_openai_1.transformClaudeRequestToOpenAIChat)(requestBody, rule.targetModel);
                     }
-                    else if (this.isGeminiSource(sourceType)) {
+                    else if (this.isGeminiSource(sourceType) || this.isGeminiChatSource(sourceType)) {
                         requestBody = (0, gemini_1.transformOpenAIChatRequestToGemini)(requestBody);
                     }
                     else {
@@ -1952,7 +2066,8 @@ class ProxyServer {
                     const model = requestBody.model || rule.targetModel || 'gemini-pro';
                     upstreamUrl = this.buildGeminiUrl(service.apiUrl, model, streamRequested);
                 }
-                else if (this.isChatType(sourceType)) {
+                else if (this.isChatType(sourceType) || this.isGeminiChatSource(sourceType)) {
+                    // Chat 类型（包括 gemini-chat）直接使用用户配置的完整 URL
                     upstreamUrl = service.apiUrl;
                 }
                 else {
@@ -2235,8 +2350,8 @@ class ProxyServer {
                         }));
                         return;
                     }
-                    // Gemini -> Claude Code 流式转换
-                    if (targetType === 'claude-code' && this.isGeminiSource(sourceType)) {
+                    // Gemini / Gemini Chat -> Claude Code 流式转换
+                    if (targetType === 'claude-code' && (this.isGeminiSource(sourceType) || this.isGeminiChatSource(sourceType))) {
                         res.setHeader('Content-Type', 'text/event-stream');
                         res.setHeader('Cache-Control', 'no-cache');
                         res.setHeader('Connection', 'keep-alive');
@@ -2331,8 +2446,8 @@ class ProxyServer {
                         }));
                         return;
                     }
-                    // Gemini -> Codex 流式转换
-                    if (targetType === 'codex' && this.isGeminiSource(sourceType)) {
+                    // Gemini / Gemini Chat -> Codex 流式转换
+                    if (targetType === 'codex' && (this.isGeminiSource(sourceType) || this.isGeminiChatSource(sourceType))) {
                         res.setHeader('Content-Type', 'text/event-stream');
                         res.setHeader('Cache-Control', 'no-cache');
                         res.setHeader('Connection', 'keep-alive');
@@ -2499,7 +2614,7 @@ class ProxyServer {
                     responseBodyForLog = JSON.stringify(converted);
                     res.status(response.status).json(converted);
                 }
-                else if (targetType === 'claude-code' && this.isGeminiSource(sourceType)) {
+                else if (targetType === 'claude-code' && (this.isGeminiSource(sourceType) || this.isGeminiChatSource(sourceType))) {
                     const converted = (0, gemini_1.transformGeminiResponseToClaude)(responseData, rule.targetModel);
                     usageForLog = (0, gemini_1.extractTokenUsageFromGeminiUsage)(responseData === null || responseData === void 0 ? void 0 : responseData.usageMetadata);
                     responseBodyForLog = JSON.stringify(converted);
@@ -2511,7 +2626,7 @@ class ProxyServer {
                     responseBodyForLog = JSON.stringify(converted);
                     res.status(response.status).json(converted);
                 }
-                else if (targetType === 'codex' && this.isGeminiSource(sourceType)) {
+                else if (targetType === 'codex' && (this.isGeminiSource(sourceType) || this.isGeminiChatSource(sourceType))) {
                     const converted = (0, gemini_1.transformGeminiResponseToOpenAIChat)(responseData, rule.targetModel);
                     usageForLog = (0, gemini_1.extractTokenUsageFromGeminiUsage)(responseData === null || responseData === void 0 ? void 0 : responseData.usageMetadata);
                     responseBodyForLog = JSON.stringify(converted);

package/dist/server/type-migration.js

@@ -0,0 +1,68 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.migrateSourceType = migrateSourceType;
+exports.downgradeSourceType = downgradeSourceType;
+exports.isLegacySourceType = isLegacySourceType;
+exports.normalizeSourceType = normalizeSourceType;
+/**
+ * 旧类型 → 新类型映射表
+ */
+const SOURCE_TYPE_MIGRATION_MAP = {
+    'openai-chat': 'openai-chat',
+    'openai-responses': 'openai', // 重命名
+    'claude-chat': 'claude-chat',
+    'claude-code': 'claude', // 重命名
+    'deepseek-reasoning-chat': 'deepseek-reasoning-chat',
+    'gemini': 'gemini',
+    'gemini-chat': 'gemini-chat',
+};
+/**
+ * 新类型 → 旧类型映射表
+ * 用于向下兼容导出
+ */
+const SOURCE_TYPE_REVERSE_MAP = {
+    'openai-chat': 'openai-chat',
+    'openai': 'openai-responses',
+    'claude-chat': 'claude-chat',
+    'claude': 'claude-code',
+    'deepseek-reasoning-chat': 'deepseek-reasoning-chat',
+    'gemini': 'gemini',
+    'gemini-chat': 'gemini-chat',
+};
+/**
+ * 将旧类型转换为新类型
+ * @param legacyType 旧的数据源类型
+ * @returns 新的数据源类型
+ */
+function migrateSourceType(legacyType) {
+    return SOURCE_TYPE_MIGRATION_MAP[legacyType];
+}
+/**
+ * 将新类型转换为旧类型
+ * 用于向下兼容导出
+ * @param newType 新的数据源类型
+ * @returns 旧的数据源类型
+ */
+function downgradeSourceType(newType) {
+    return SOURCE_TYPE_REVERSE_MAP[newType];
+}
+/**
+ * 检查是否为旧类型
+ * @param type 类型字符串
+ * @returns 是否为旧类型
+ */
+function isLegacySourceType(type) {
+    return type === 'openai-responses' || type === 'claude-code';
+}
+/**
+ * 标准化类型
+ * 自动处理新旧类型，将旧类型转换为新类型，新类型保持不变
+ * @param type 类型字符串（可能是旧类型或新类型）
+ * @returns 标准化后的新类型
+ */
+function normalizeSourceType(type) {
+    if (isLegacySourceType(type)) {
+        return migrateSourceType(type);
+    }
+    return type;
+}