aicodeswitch 1.3.9 → 1.4.1

package/dist/server/proxy-server.js

@@ -143,8 +143,8 @@ class ProxyServer {
         this.app.use('/codex', this.createFixedRouteHandler('codex'));
         // Dynamic proxy middleware
         this.app.use((req, res, next) => __awaiter(this, void 0, void 0, function* () {
-            var _a;
-            // 根路径 / 不应该被代理中间件处理，应该传递给静态文件服务
+            var _a, _b, _c, _d;
+            // 根路径 / 不应该被代理中间件处理,应该传递给静态文件服务
             if (req.path === '/') {
                 return next();
             }
@@ -153,19 +153,88 @@ class ProxyServer {
                 if (!route) {
                     return res.status(404).json({ error: 'No matching route found' });
                 }
-                const rule = this.findMatchingRule(route.id, req);
-                if (!rule) {
+                // 检查是否启用故障切换
+                const enableFailover = ((_a = this.config) === null || _a === void 0 ? void 0 : _a.enableFailover) !== false; // 默认为 true
+                if (!enableFailover) {
+                    // 故障切换已禁用,使用传统的单一规则匹配
+                    const rule = yield this.findMatchingRule(route.id, req);
+                    if (!rule) {
+                        return res.status(404).json({ error: 'No matching rule found' });
+                    }
+                    const service = this.services.get(rule.targetServiceId);
+                    if (!service) {
+                        return res.status(500).json({ error: 'Target service not configured' });
+                    }
+                    yield this.proxyRequest(req, res, route, rule, service);
+                    return;
+                }
+                // 启用故障切换:获取所有候选规则
+                const allRules = this.getAllMatchingRules(route.id, req);
+                if (allRules.length === 0) {
                     return res.status(404).json({ error: 'No matching rule found' });
                 }
-                const service = this.services.get(rule.targetServiceId);
-                if (!service) {
-                    return res.status(500).json({ error: 'Target service not configured' });
+                // 尝试每个规则,直到成功或全部失败
+                let lastError = null;
+                for (const rule of allRules) {
+                    const service = this.services.get(rule.targetServiceId);
+                    if (!service)
+                        continue;
+                    // 检查黑名单
+                    const isBlacklisted = yield this.dbManager.isServiceBlacklisted(service.id, route.id, rule.contentType);
+                    if (isBlacklisted) {
+                        console.log(`Service ${service.name} is blacklisted, skipping...`);
+                        continue;
+                    }
+                    try {
+                        // 尝试代理请求
+                        yield this.proxyRequest(req, res, route, rule, service);
+                        return; // 成功,直接返回
+                    }
+                    catch (error) {
+                        console.error(`Service ${service.name} failed:`, error.message);
+                        lastError = error;
+                        // 判断是否应该加入黑名单 (4xx + 5xx)
+                        const statusCode = ((_b = error.response) === null || _b === void 0 ? void 0 : _b.status) || 500;
+                        if (statusCode >= 400) {
+                            yield this.dbManager.addToBlacklist(service.id, route.id, rule.contentType, error.message, statusCode);
+                            console.log(`Service ${service.name} added to blacklist (${route.id}:${rule.contentType}:${service.id})`);
+                        }
+                        // 继续尝试下一个服务
+                        continue;
+                    }
+                }
+                // 所有服务都失败了
+                console.error('All services failed');
+                // 记录日志
+                if (((_c = this.config) === null || _c === void 0 ? void 0 : _c.enableLogging) && SUPPORTED_TARGETS.some(target => req.path.startsWith(`/${target}/`))) {
+                    yield this.dbManager.addLog({
+                        timestamp: Date.now(),
+                        method: req.method,
+                        path: req.path,
+                        headers: this.normalizeHeaders(req.headers),
+                        body: req.body ? JSON.stringify(req.body) : undefined,
+                        error: (lastError === null || lastError === void 0 ? void 0 : lastError.message) || 'All services failed',
+                    });
                 }
-                yield this.proxyRequest(req, res, route, rule, service);
+                // 记录错误日志
+                yield this.dbManager.addErrorLog({
+                    timestamp: Date.now(),
+                    method: req.method,
+                    path: req.path,
+                    statusCode: 503,
+                    errorMessage: 'All services failed',
+                    errorStack: lastError === null || lastError === void 0 ? void 0 : lastError.stack,
+                    requestHeaders: this.normalizeHeaders(req.headers),
+                    requestBody: req.body ? JSON.stringify(req.body) : undefined,
+                });
+                res.status(503).json({
+                    error: 'All services failed',
+                    details: lastError === null || lastError === void 0 ? void 0 : lastError.message
+                });
             }
             catch (error) {
                 console.error('Proxy error:', error);
-                if (((_a = this.config) === null || _a === void 0 ? void 0 : _a.enableLogging) && SUPPORTED_TARGETS.some(target => req.path.startsWith(`/${target}/`))) {
+                if (((_d = this.config) === null || _d === void 0 ? void 0 : _d.enableLogging) && SUPPORTED_TARGETS.some(target => req.path.startsWith(`/${target}/`))) {
                     yield this.dbManager.addLog({
                         timestamp: Date.now(),
                         method: req.method,
@@ -192,7 +261,7 @@ class ProxyServer {
     }
     createFixedRouteHandler(targetType) {
         return (req, res) => __awaiter(this, void 0, void 0, function* () {
-            var _a;
+            var _a, _b, _c, _d;
             try {
                 // 检查API Key验证
                 if (this.config.apiKey) {
@@ -206,19 +275,88 @@ class ProxyServer {
                 if (!route) {
                     return res.status(404).json({ error: `No active route found for target type: ${targetType}` });
                 }
-                const rule = this.findMatchingRule(route.id, req);
-                if (!rule) {
+                // 检查是否启用故障切换
+                const enableFailover = ((_a = this.config) === null || _a === void 0 ? void 0 : _a.enableFailover) !== false; // 默认为 true
+                if (!enableFailover) {
+                    // 故障切换已禁用,使用传统的单一规则匹配
+                    const rule = yield this.findMatchingRule(route.id, req);
+                    if (!rule) {
+                        return res.status(404).json({ error: 'No matching rule found' });
+                    }
+                    const service = this.services.get(rule.targetServiceId);
+                    if (!service) {
+                        return res.status(500).json({ error: 'Target service not configured' });
+                    }
+                    yield this.proxyRequest(req, res, route, rule, service);
+                    return;
+                }
+                // 启用故障切换:获取所有候选规则
+                const allRules = this.getAllMatchingRules(route.id, req);
+                if (allRules.length === 0) {
                     return res.status(404).json({ error: 'No matching rule found' });
                 }
-                const service = this.services.get(rule.targetServiceId);
-                if (!service) {
-                    return res.status(500).json({ error: 'Target service not configured' });
+                // 尝试每个规则,直到成功或全部失败
+                let lastError = null;
+                for (const rule of allRules) {
+                    const service = this.services.get(rule.targetServiceId);
+                    if (!service)
+                        continue;
+                    // 检查黑名单
+                    const isBlacklisted = yield this.dbManager.isServiceBlacklisted(service.id, route.id, rule.contentType);
+                    if (isBlacklisted) {
+                        console.log(`Service ${service.name} is blacklisted, skipping...`);
+                        continue;
+                    }
+                    try {
+                        // 尝试代理请求
+                        yield this.proxyRequest(req, res, route, rule, service);
+                        return; // 成功,直接返回
+                    }
+                    catch (error) {
+                        console.error(`Service ${service.name} failed:`, error.message);
+                        lastError = error;
+                        // 判断是否应该加入黑名单 (4xx + 5xx)
+                        const statusCode = ((_b = error.response) === null || _b === void 0 ? void 0 : _b.status) || 500;
+                        if (statusCode >= 400) {
+                            yield this.dbManager.addToBlacklist(service.id, route.id, rule.contentType, error.message, statusCode);
+                            console.log(`Service ${service.name} added to blacklist (${route.id}:${rule.contentType}:${service.id})`);
+                        }
+                        // 继续尝试下一个服务
+                        continue;
+                    }
                 }
-                yield this.proxyRequest(req, res, route, rule, service);
+                // 所有服务都失败了
+                console.error('All services failed');
+                // 记录日志
+                if (((_c = this.config) === null || _c === void 0 ? void 0 : _c.enableLogging) && SUPPORTED_TARGETS.some(target => req.path.startsWith(`/${target}/`))) {
+                    yield this.dbManager.addLog({
+                        timestamp: Date.now(),
+                        method: req.method,
+                        path: req.path,
+                        headers: this.normalizeHeaders(req.headers),
+                        body: req.body ? JSON.stringify(req.body) : undefined,
+                        error: (lastError === null || lastError === void 0 ? void 0 : lastError.message) || 'All services failed',
+                    });
+                }
+                // 记录错误日志
+                yield this.dbManager.addErrorLog({
+                    timestamp: Date.now(),
+                    method: req.method,
+                    path: req.path,
+                    statusCode: 503,
+                    errorMessage: 'All services failed',
+                    errorStack: lastError === null || lastError === void 0 ? void 0 : lastError.stack,
+                    requestHeaders: this.normalizeHeaders(req.headers),
+                    requestBody: req.body ? JSON.stringify(req.body) : undefined,
+                });
+                res.status(503).json({
+                    error: 'All services failed',
+                    details: lastError === null || lastError === void 0 ? void 0 : lastError.message
+                });
             }
             catch (error) {
                 console.error(`Fixed route error for ${targetType}:`, error);
-                if (((_a = this.config) === null || _a === void 0 ? void 0 : _a.enableLogging) && SUPPORTED_TARGETS.some(target => req.path.startsWith(`/${target}/`))) {
+                if (((_d = this.config) === null || _d === void 0 ? void 0 : _d.enableLogging) && SUPPORTED_TARGETS.some(target => req.path.startsWith(`/${target}/`))) {
                     yield this.dbManager.addLog({
                         timestamp: Date.now(),
                         method: req.method,
@@ -252,32 +390,69 @@ class ProxyServer {
         return this.routes.find(route => route.targetType === targetType && route.isActive);
     }
     findMatchingRule(routeId, req) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const rules = this.rules.get(routeId);
+            if (!rules)
+                return undefined;
+            const body = req.body;
+            const requestModel = body === null || body === void 0 ? void 0 : body.model;
+            // 1. 首先查找 model-mapping 类型的规则，按 sortOrder 降序匹配
+            if (requestModel) {
+                const modelMappingRules = rules.filter(rule => rule.contentType === 'model-mapping' &&
+                    rule.replacedModel &&
+                    requestModel.includes(rule.replacedModel));
+                // 过滤黑名单
+                for (const rule of modelMappingRules) {
+                    const isBlacklisted = yield this.dbManager.isServiceBlacklisted(rule.targetServiceId, routeId, rule.contentType);
+                    if (!isBlacklisted) {
+                        return rule;
+                    }
+                }
+            }
+            // 2. 查找其他内容类型的规则
+            const contentType = this.determineContentType(req);
+            const contentTypeRules = rules.filter(rule => rule.contentType === contentType);
+            // 过滤黑名单
+            for (const rule of contentTypeRules) {
+                const isBlacklisted = yield this.dbManager.isServiceBlacklisted(rule.targetServiceId, routeId, contentType);
+                if (!isBlacklisted) {
+                    return rule;
+                }
+            }
+            // 3. 最后返回 default 规则
+            const defaultRules = rules.filter(rule => rule.contentType === 'default');
+            // 过滤黑名单
+            for (const rule of defaultRules) {
+                const isBlacklisted = yield this.dbManager.isServiceBlacklisted(rule.targetServiceId, routeId, 'default');
+                if (!isBlacklisted) {
+                    return rule;
+                }
+            }
+            return undefined;
+        });
+    }
+    getAllMatchingRules(routeId, req) {
         const rules = this.rules.get(routeId);
         if (!rules)
-            return undefined;
+            return [];
         const body = req.body;
         const requestModel = body === null || body === void 0 ? void 0 : body.model;
-        // 1. 首先查找 model-mapping 类型的规则，按 sortOrder 降序匹配
+        const candidates = [];
+        // 1. Model mapping rules
         if (requestModel) {
             const modelMappingRules = rules.filter(rule => rule.contentType === 'model-mapping' &&
                 rule.replacedModel &&
                 requestModel.includes(rule.replacedModel));
-            if (modelMappingRules.length > 0) {
-                return modelMappingRules[0]; // 已按 sortOrder 降序排序
-            }
+            candidates.push(...modelMappingRules);
         }
-        // 2. 查找其他内容类型的规则
+        // 2. Content type specific rules
         const contentType = this.determineContentType(req);
         const contentTypeRules = rules.filter(rule => rule.contentType === contentType);
-        if (contentTypeRules.length > 0) {
-            return contentTypeRules[0]; // 已按 sortOrder 降序排序
-        }
-        // 3. 最后返回 default 规则
+        candidates.push(...contentTypeRules);
+        // 3. Default rules
         const defaultRules = rules.filter(rule => rule.contentType === 'default');
-        if (defaultRules.length > 0) {
-            return defaultRules[0]; // 已按 sortOrder 降序排序
-        }
-        return undefined;
+        candidates.push(...defaultRules);
+        return candidates;
     }
     determineContentType(req) {
         const body = req.body;
@@ -720,9 +895,17 @@ class ProxyServer {
                     }
                 }
                 const streamRequested = this.isStreamRequested(req, requestBody);
+                // Build the full URL by appending the request path to the service API URL
+                let pathToAppend = req.path;
+                if (route.targetType === 'claude-code' && req.path.startsWith('/claude-code')) {
+                    pathToAppend = req.path.slice('/claude-code'.length);
+                }
+                else if (route.targetType === 'codex' && req.path.startsWith('/codex')) {
+                    pathToAppend = req.path.slice('/codex'.length);
+                }
                 const config = {
                     method: req.method,
-                    url: service.apiUrl,
+                    url: `${service.apiUrl}${pathToAppend}`,
                     headers: this.buildUpstreamHeaders(req, service, sourceType, streamRequested),
                     timeout: service.timeout || 30000,
                     validateStatus: () => true,
@@ -745,7 +928,7 @@ class ProxyServer {
                         res.setHeader('Cache-Control', 'no-cache');
                         res.setHeader('Connection', 'keep-alive');
                         const parser = new streaming_1.SSEParserTransform();
-                        const chunkCollector = new chunk_collector_1.ChunkCollectorTransform();
+                        const eventCollector = new chunk_collector_1.SSEEventCollectorTransform();
                         const converter = new streaming_1.OpenAIToClaudeEventTransform({ model: requestBody === null || requestBody === void 0 ? void 0 : requestBody.model });
                         const serializer = new streaming_1.SSESerializerTransform();
                         // 收集响应头
@@ -755,11 +938,18 @@ class ProxyServer {
                             if (usage) {
                                 usageForLog = (0, claude_openai_1.extractTokenUsageFromClaudeUsage)(usage);
                             }
-                            // 收集stream chunks
-                            streamChunksForLog = chunkCollector.getChunks();
+                            else {
+                                // 尝试从event collector中提取usage
+                                const extractedUsage = eventCollector.extractUsage();
+                                if (extractedUsage) {
+                                    usageForLog = this.extractTokenUsage(extractedUsage);
+                                }
+                            }
+                            // 收集stream chunks（每个chunk是一个完整的SSE事件）
+                            streamChunksForLog = eventCollector.getChunks();
                             void finalizeLog(res.statusCode);
                         });
-                        (0, stream_1.pipeline)(response.data, parser, chunkCollector, converter, serializer, res, (error) => {
+                        (0, stream_1.pipeline)(response.data, parser, eventCollector, converter, serializer, res, (error) => {
                             if (error) {
                                 void finalizeLog(500, error.message);
                             }
@@ -771,7 +961,7 @@ class ProxyServer {
                         res.setHeader('Cache-Control', 'no-cache');
                         res.setHeader('Connection', 'keep-alive');
                         const parser = new streaming_1.SSEParserTransform();
-                        const chunkCollector = new chunk_collector_1.ChunkCollectorTransform();
+                        const eventCollector = new chunk_collector_1.SSEEventCollectorTransform();
                         const converter = new streaming_1.OpenAIResponsesToClaudeEventTransform({ model: requestBody === null || requestBody === void 0 ? void 0 : requestBody.model });
                         const serializer = new streaming_1.SSESerializerTransform();
                         responseHeadersForLog = this.normalizeResponseHeaders(responseHeaders);
@@ -780,10 +970,17 @@ class ProxyServer {
                             if (usage) {
                                 usageForLog = (0, claude_openai_1.extractTokenUsageFromClaudeUsage)(usage);
                             }
-                            streamChunksForLog = chunkCollector.getChunks();
+                            else {
+                                // 尝试从event collector中提取usage
+                                const extractedUsage = eventCollector.extractUsage();
+                                if (extractedUsage) {
+                                    usageForLog = this.extractTokenUsage(extractedUsage);
+                                }
+                            }
+                            streamChunksForLog = eventCollector.getChunks();
                             void finalizeLog(res.statusCode);
                         });
-                        (0, stream_1.pipeline)(response.data, parser, chunkCollector, converter, serializer, res, (error) => {
+                        (0, stream_1.pipeline)(response.data, parser, eventCollector, converter, serializer, res, (error) => {
                             if (error) {
                                 void finalizeLog(500, error.message);
                             }
@@ -795,7 +992,7 @@ class ProxyServer {
                         res.setHeader('Cache-Control', 'no-cache');
                         res.setHeader('Connection', 'keep-alive');
                         const parser = new streaming_1.SSEParserTransform();
-                        const chunkCollector = new chunk_collector_1.ChunkCollectorTransform();
+                        const eventCollector = new chunk_collector_1.SSEEventCollectorTransform();
                         const converter = new streaming_1.ClaudeToOpenAIResponsesEventTransform({ model: requestBody === null || requestBody === void 0 ? void 0 : requestBody.model });
                         const serializer = new streaming_1.SSESerializerTransform();
                         responseHeadersForLog = this.normalizeResponseHeaders(responseHeaders);
@@ -804,10 +1001,17 @@ class ProxyServer {
                             if (usage) {
                                 usageForLog = (0, claude_openai_1.extractTokenUsageFromClaudeUsage)(usage);
                             }
-                            streamChunksForLog = chunkCollector.getChunks();
+                            else {
+                                // 尝试从event collector中提取usage
+                                const extractedUsage = eventCollector.extractUsage();
+                                if (extractedUsage) {
+                                    usageForLog = this.extractTokenUsage(extractedUsage);
+                                }
+                            }
+                            streamChunksForLog = eventCollector.getChunks();
                             void finalizeLog(res.statusCode);
                         });
-                        (0, stream_1.pipeline)(response.data, parser, chunkCollector, converter, serializer, res, (error) => {
+                        (0, stream_1.pipeline)(response.data, parser, eventCollector, converter, serializer, res, (error) => {
                             if (error) {
                                 void finalizeLog(500, error.message);
                             }
@@ -819,7 +1023,7 @@ class ProxyServer {
                         res.setHeader('Cache-Control', 'no-cache');
                         res.setHeader('Connection', 'keep-alive');
                         const parser = new streaming_1.SSEParserTransform();
-                        const chunkCollector = new chunk_collector_1.ChunkCollectorTransform();
+                        const eventCollector = new chunk_collector_1.SSEEventCollectorTransform();
                         const toClaude = new streaming_1.OpenAIToClaudeEventTransform({ model: requestBody === null || requestBody === void 0 ? void 0 : requestBody.model });
                         const toResponses = new streaming_1.ClaudeToOpenAIResponsesEventTransform({ model: requestBody === null || requestBody === void 0 ? void 0 : requestBody.model });
                         const serializer = new streaming_1.SSESerializerTransform();
@@ -829,10 +1033,17 @@ class ProxyServer {
                             if (usage) {
                                 usageForLog = (0, claude_openai_1.extractTokenUsageFromClaudeUsage)(usage);
                             }
-                            streamChunksForLog = chunkCollector.getChunks();
+                            else {
+                                // 尝试从event collector中提取usage
+                                const extractedUsage = eventCollector.extractUsage();
+                                if (extractedUsage) {
+                                    usageForLog = this.extractTokenUsage(extractedUsage);
+                                }
+                            }
+                            streamChunksForLog = eventCollector.getChunks();
                             void finalizeLog(res.statusCode);
                         });
-                        (0, stream_1.pipeline)(response.data, parser, chunkCollector, toClaude, toResponses, serializer, res, (error) => {
+                        (0, stream_1.pipeline)(response.data, parser, eventCollector, toClaude, toResponses, serializer, res, (error) => {
                             if (error) {
                                 void finalizeLog(500, error.message);
                             }
@@ -840,34 +1051,19 @@ class ProxyServer {
                         return;
                     }
                     // 默认stream处理(无转换)
-                    const chunkCollector = new chunk_collector_1.ChunkCollectorTransform();
+                    const eventCollector = new chunk_collector_1.SSEEventCollectorTransform();
                     responseHeadersForLog = this.normalizeResponseHeaders(responseHeaders);
                     this.copyResponseHeaders(responseHeaders, res);
                     res.on('finish', () => {
-                        streamChunksForLog = chunkCollector.getChunks();
-                        // 尝试从stream chunks中解析usage信息
-                        if (streamChunksForLog && streamChunksForLog.length > 0) {
-                            // 合并所有chunks并尝试解析usage
-                            const allChunks = streamChunksForLog.join('');
-                            // 查找包含usage信息的部分
-                            const usageMatch = allChunks.match(/usage[\s\S]*?\{[\s\S]*?\}/);
-                            if (usageMatch) {
-                                try {
-                                    // 尝试解析usage信息
-                                    const usageStr = usageMatch[0];
-                                    const jsonStart = usageStr.indexOf('{');
-                                    const jsonEnd = usageStr.lastIndexOf('}') + 1;
-                                    const usageJson = JSON.parse(usageStr.slice(jsonStart, jsonEnd));
-                                    usageForLog = this.extractTokenUsage(usageJson);
-                                }
-                                catch (e) {
-                                    console.error('Failed to parse usage from stream chunks:', e);
-                                }
-                            }
+                        streamChunksForLog = eventCollector.getChunks();
+                        // 尝试从event collector中提取usage信息
+                        const extractedUsage = eventCollector.extractUsage();
+                        if (extractedUsage) {
+                            usageForLog = this.extractTokenUsage(extractedUsage);
                         }
                         void finalizeLog(res.statusCode);
                     });
-                    (0, stream_1.pipeline)(response.data, chunkCollector, res, (error) => {
+                    (0, stream_1.pipeline)(response.data, eventCollector, res, (error) => {
                         if (error) {
                             void finalizeLog(500, error.message);
                         }

package/dist/server/transformers/chunk-collector.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChunkCollectorTransform = void 0;
+exports.SSEEventCollectorTransform = exports.ChunkCollectorTransform = void 0;
 const stream_1 = require("stream");
 /**
  * ChunkCollectorTransform - 收集stream chunks用于日志记录
@@ -37,3 +37,152 @@ class ChunkCollectorTransform extends stream_1.Transform {
     }
 }
 exports.ChunkCollectorTransform = ChunkCollectorTransform;
+/**
+ * SSEEventCollectorTransform - 智能收集完整的SSE事件
+ * 这个Transform会解析SSE流并将每个完整的事件存储为一个单独的entry
+ * 确保每个chunk代表一条完整的消息,而不是随机的buffer片段
+ */
+class SSEEventCollectorTransform extends stream_1.Transform {
+    constructor() {
+        super();
+        Object.defineProperty(this, "buffer", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ''
+        });
+        Object.defineProperty(this, "currentEvent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: {
+                dataLines: [],
+                rawLines: []
+            }
+        });
+        Object.defineProperty(this, "events", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+    }
+    _transform(chunk, _encoding, callback) {
+        this.buffer += chunk.toString('utf8');
+        this.processBuffer();
+        // 将chunk传递给下一个stream
+        this.push(chunk);
+        callback();
+    }
+    _flush(callback) {
+        // 处理剩余的buffer
+        if (this.buffer.trim()) {
+            this.processBuffer();
+        }
+        // 刷新最后一个事件
+        this.flushEvent();
+        callback();
+    }
+    processBuffer() {
+        const lines = this.buffer.split('\n');
+        // 保留最后一行(可能不完整)
+        this.buffer = lines.pop() || '';
+        for (const line of lines) {
+            this.processLine(line);
+        }
+    }
+    processLine(line) {
+        // 记录原始行
+        this.currentEvent.rawLines.push(line);
+        // 空行表示一个事件结束
+        if (!line.trim()) {
+            this.flushEvent();
+            return;
+        }
+        if (line.startsWith('event:')) {
+            this.currentEvent.event = line.slice(6).trim();
+            return;
+        }
+        if (line.startsWith('id:')) {
+            this.currentEvent.id = line.slice(3).trim();
+            return;
+        }
+        if (line.startsWith('data:')) {
+            this.currentEvent.dataLines.push(line.slice(5).trim());
+            return;
+        }
+    }
+    flushEvent() {
+        // 只有当有内容时才创建事件
+        if (!this.currentEvent.event && this.currentEvent.dataLines.length === 0 && !this.currentEvent.id) {
+            this.currentEvent = { dataLines: [], rawLines: [] };
+            return;
+        }
+        const raw = this.currentEvent.rawLines.join('\n');
+        const event = {
+            event: this.currentEvent.event,
+            id: this.currentEvent.id,
+            data: this.currentEvent.dataLines.length > 0 ? this.currentEvent.dataLines.join('\n') : undefined,
+            raw
+        };
+        this.events.push(event);
+        this.currentEvent = { dataLines: [], rawLines: [] };
+    }
+    /**
+     * 获取收集的所有SSE事件
+     * 每个事件都是一个完整的SSE消息
+     */
+    getEvents() {
+        return this.events;
+    }
+    /**
+     * 获取原始chunks(兼容旧接口)
+     */
+    getChunks() {
+        return this.events.map(e => e.raw);
+    }
+    /**
+     * 清空已收集的事件
+     */
+    clearEvents() {
+        this.events = [];
+    }
+    /**
+     * 从events中提取usage信息
+     */
+    extractUsage() {
+        for (const event of this.events) {
+            if (!event.data)
+                continue;
+            try {
+                const data = JSON.parse(event.data);
+                // 尝试从不同的位置提取usage
+                // 1. message_delta事件中的usage
+                if (event.event === 'message_delta' && data.usage) {
+                    return data.usage;
+                }
+                // 2. 直接在data中的usage
+                if (data.usage) {
+                    return data.usage;
+                }
+                // 3. OpenAI格式: choices数组中最后一个元素的usage
+                if (Array.isArray(data.choices) && data.choices.length > 0) {
+                    const lastChoice = data.choices[data.choices.length - 1];
+                    if (lastChoice === null || lastChoice === void 0 ? void 0 : lastChoice.usage) {
+                        return lastChoice.usage;
+                    }
+                }
+                // 4. 直接在顶级的usage字段
+                if (data.input_tokens !== undefined || data.output_tokens !== undefined ||
+                    data.prompt_tokens !== undefined || data.completion_tokens !== undefined) {
+                    return data;
+                }
+            }
+            catch (_a) {
+                // JSON解析失败,跳过
+            }
+        }
+        return null;
+    }
+}
+exports.SSEEventCollectorTransform = SSEEventCollectorTransform;