aicodeswitch 1.6.2 → 1.7.1

package/dist/server/transformers/streaming.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ClaudeToOpenAIResponsesEventTransform = exports.OpenAIResponsesToClaudeEventTransform = exports.OpenAIToClaudeEventTransform = exports.rewriteStream = exports.SSESerializerTransform = exports.SSEParserTransform = void 0;
+exports.ClaudeToOpenAIChatEventTransform = exports.OpenAIToClaudeEventTransform = exports.rewriteStream = exports.SSESerializerTransform = exports.SSEParserTransform = void 0;
 const stream_1 = require("stream");
 const crypto_1 = __importDefault(require("crypto"));
 const claude_openai_1 = require("./claude-openai");
@@ -406,45 +406,33 @@ class OpenAIToClaudeEventTransform extends stream_1.Transform {
     }
 }
 exports.OpenAIToClaudeEventTransform = OpenAIToClaudeEventTransform;
-class OpenAIResponsesToClaudeEventTransform extends stream_1.Transform {
+class ClaudeToOpenAIChatEventTransform extends stream_1.Transform {
     constructor(options) {
         var _a;
         super({ objectMode: true });
-        Object.defineProperty(this, "contentIndex", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-        Object.defineProperty(this, "textBlockIndex", {
+        Object.defineProperty(this, "pendingToolCallId", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: null
         });
-        Object.defineProperty(this, "toolCalls", {
+        Object.defineProperty(this, "pendingToolName", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: new Map()
-        });
-        Object.defineProperty(this, "toolCallKeyToBlockIndex", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: new Map()
+            value: null
         });
-        Object.defineProperty(this, "hasMessageStart", {
+        Object.defineProperty(this, "pendingToolArgs", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: false
+            value: ''
         });
-        Object.defineProperty(this, "stopReason", {
+        Object.defineProperty(this, "toolCallIndex", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: 'end_turn'
+            value: 0
         });
         Object.defineProperty(this, "usage", {
             enumerable: true,
@@ -452,437 +440,104 @@ class OpenAIResponsesToClaudeEventTransform extends stream_1.Transform {
             writable: true,
             value: null
         });
-        Object.defineProperty(this, "messageId", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: null
-        });
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: null
-        });
-        Object.defineProperty(this, "finalized", {
+        Object.defineProperty(this, "finished", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: false
         });
-        this.model = (_a = options === null || options === void 0 ? void 0 : options.model) !== null && _a !== void 0 ? _a : null;
-    }
-    getUsage() {
-        if (!this.usage)
-            return undefined;
-        return Object.assign({}, this.usage);
-    }
-    _transform(event, _encoding, callback) {
-        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4;
-        if (this.finalized) {
-            callback();
-            return;
-        }
-        const eventType = event.event || ((_a = event.data) === null || _a === void 0 ? void 0 : _a.type) || '';
-        if (eventType.includes('response.created')) {
-            const response = ((_b = event.data) === null || _b === void 0 ? void 0 : _b.response) || event.data;
-            if (response === null || response === void 0 ? void 0 : response.id)
-                this.messageId = response.id;
-            if (response === null || response === void 0 ? void 0 : response.model)
-                this.model = response.model;
-            this.ensureMessageStart();
-            callback();
-            return;
-        }
-        if (eventType.includes('output_text')) {
-            const deltaText = (_d = (_c = event.data) === null || _c === void 0 ? void 0 : _c.delta) !== null && _d !== void 0 ? _d : (_e = event.data) === null || _e === void 0 ? void 0 : _e.text;
-            if (typeof deltaText === 'string' && deltaText.length > 0) {
-                this.handleTextDelta(deltaText);
-            }
-            if (eventType.includes('done')) {
-                this.closeTextBlock();
-            }
-            callback();
-            return;
-        }
-        if (eventType.includes('tool_call')) {
-            const toolId = ((_f = event.data) === null || _f === void 0 ? void 0 : _f.tool_call_id) || ((_g = event.data) === null || _g === void 0 ? void 0 : _g.id) || ((_j = (_h = event.data) === null || _h === void 0 ? void 0 : _h.tool_call) === null || _j === void 0 ? void 0 : _j.id) || `tool_${this.toolCalls.size + 1}`;
-            const toolName = ((_k = event.data) === null || _k === void 0 ? void 0 : _k.name) || ((_m = (_l = event.data) === null || _l === void 0 ? void 0 : _l.tool_call) === null || _m === void 0 ? void 0 : _m.name) || 'tool';
-            const delta = (_p = (_o = event.data) === null || _o === void 0 ? void 0 : _o.delta) !== null && _p !== void 0 ? _p : (_q = event.data) === null || _q === void 0 ? void 0 : _q.arguments;
-            if (typeof delta === 'string') {
-                this.handleToolDelta(toolId, toolName, delta);
-            }
-            if (eventType.includes('done')) {
-                const key = toolId || toolName;
-                this.closeToolBlock(key);
-            }
-            callback();
-            return;
-        }
-        if (eventType.includes('response.completed')) {
-            const response = ((_r = event.data) === null || _r === void 0 ? void 0 : _r.response) || event.data;
-            if (response === null || response === void 0 ? void 0 : response.usage) {
-                const inputTokens = (_v = (_t = (_s = response.usage) === null || _s === void 0 ? void 0 : _s.input_tokens) !== null && _t !== void 0 ? _t : (_u = response.usage) === null || _u === void 0 ? void 0 : _u.prompt_tokens) !== null && _v !== void 0 ? _v : 0;
-                const outputTokens = (_z = (_x = (_w = response.usage) === null || _w === void 0 ? void 0 : _w.output_tokens) !== null && _x !== void 0 ? _x : (_y = response.usage) === null || _y === void 0 ? void 0 : _y.completion_tokens) !== null && _z !== void 0 ? _z : 0;
-                const cacheRead = (_4 = (_1 = (_0 = response.usage) === null || _0 === void 0 ? void 0 : _0.cache_read_input_tokens) !== null && _1 !== void 0 ? _1 : (_3 = (_2 = response.usage) === null || _2 === void 0 ? void 0 : _2.prompt_tokens_details) === null || _3 === void 0 ? void 0 : _3.cached_tokens) !== null && _4 !== void 0 ? _4 : 0;
-                this.usage = {
-                    input_tokens: inputTokens,
-                    output_tokens: outputTokens,
-                    cache_read_input_tokens: cacheRead,
-                };
-            }
-            this.finalize();
-            callback();
-            return;
-        }
-        callback();
-    }
-    _flush(callback) {
-        this.finalize();
-        callback();
-    }
-    assignContentBlockIndex() {
-        const index = this.contentIndex;
-        this.contentIndex += 1;
-        return index;
-    }
-    pushEvent(type, data) {
-        this.push({ event: type, data });
-    }
-    ensureMessageStart() {
-        if (this.hasMessageStart)
-            return;
-        const message = {
-            id: this.messageId || `msg_${crypto_1.default.randomUUID()}`,
-            type: 'message',
-            role: 'assistant',
-            content: [],
-            model: this.model || 'unknown',
-            stop_reason: null,
-            stop_sequence: null,
-            usage: {
-                input_tokens: 0,
-                output_tokens: 0,
-            },
-        };
-        this.pushEvent('message_start', { type: 'message_start', message });
-        this.hasMessageStart = true;
-    }
-    handleTextDelta(text) {
-        this.ensureMessageStart();
-        if (this.textBlockIndex === null) {
-            this.textBlockIndex = this.assignContentBlockIndex();
-            this.pushEvent('content_block_start', {
-                type: 'content_block_start',
-                index: this.textBlockIndex,
-                content_block: { type: 'text' },
-            });
-        }
-        this.pushEvent('content_block_delta', {
-            type: 'content_block_delta',
-            index: this.textBlockIndex,
-            delta: { type: 'text_delta', text },
-        });
-    }
-    closeTextBlock() {
-        if (this.textBlockIndex === null)
-            return;
-        this.pushEvent('content_block_stop', {
-            type: 'content_block_stop',
-            index: this.textBlockIndex,
-        });
-        this.textBlockIndex = null;
-    }
-    handleToolDelta(toolId, toolName, delta) {
-        this.ensureMessageStart();
-        const key = toolId || toolName;
-        if (!this.toolCalls.has(key)) {
-            const toolBlockIndex = this.assignContentBlockIndex();
-            this.toolCalls.set(key, { id: toolId, name: toolName, arguments: '' });
-            this.toolCallKeyToBlockIndex.set(key, toolBlockIndex);
-            this.pushEvent('content_block_start', {
-                type: 'content_block_start',
-                index: toolBlockIndex,
-                content_block: { type: 'tool_use', id: toolId, name: toolName },
-            });
-        }
-        const toolEntry = this.toolCalls.get(key);
-        if (toolEntry) {
-            toolEntry.arguments += delta;
-        }
-        const blockIndex = this.toolCallKeyToBlockIndex.get(key);
-        if (blockIndex !== undefined) {
-            this.pushEvent('content_block_delta', {
-                type: 'content_block_delta',
-                index: blockIndex,
-                delta: { type: 'input_json_delta', partial_json: delta },
-            });
-        }
-    }
-    closeToolBlock(key) {
-        const blockIndex = this.toolCallKeyToBlockIndex.get(key);
-        if (blockIndex === undefined)
-            return;
-        this.pushEvent('content_block_stop', {
-            type: 'content_block_stop',
-            index: blockIndex,
-        });
-        this.toolCallKeyToBlockIndex.delete(key);
-        this.toolCalls.delete(key);
-    }
-    finalize() {
-        if (this.finalized)
-            return;
-        this.ensureMessageStart();
-        this.closeTextBlock();
-        for (const blockIndex of this.toolCallKeyToBlockIndex.values()) {
-            this.pushEvent('content_block_stop', {
-                type: 'content_block_stop',
-                index: blockIndex,
-            });
-        }
-        this.toolCallKeyToBlockIndex.clear();
-        this.toolCalls.clear();
-        const usage = this.usage || {
-            input_tokens: 0,
-            output_tokens: 0,
-            cache_read_input_tokens: 0,
-        };
-        this.pushEvent('message_delta', {
-            type: 'message_delta',
-            delta: { stop_reason: this.stopReason, stop_sequence: null },
-            usage,
-        });
-        this.pushEvent('message_stop', { type: 'message_stop' });
-        this.finalized = true;
-    }
-}
-exports.OpenAIResponsesToClaudeEventTransform = OpenAIResponsesToClaudeEventTransform;
-class ClaudeToOpenAIResponsesEventTransform extends stream_1.Transform {
-    constructor(options) {
-        var _a;
-        super({ objectMode: true });
-        Object.defineProperty(this, "responseId", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: null
-        });
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: null
         });
-        Object.defineProperty(this, "createdAt", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: Date.now()
-        });
-        Object.defineProperty(this, "outputText", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ''
-        });
-        Object.defineProperty(this, "textBlockIndex", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: null
-        });
-        Object.defineProperty(this, "toolCalls", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: new Map()
-        });
-        Object.defineProperty(this, "completedToolCalls", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: []
-        });
-        Object.defineProperty(this, "usage", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: null
-        });
-        Object.defineProperty(this, "hasCreated", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
         this.model = (_a = options === null || options === void 0 ? void 0 : options.model) !== null && _a !== void 0 ? _a : null;
     }
     getUsage() {
-        if (!this.usage)
-            return undefined;
-        return Object.assign({}, this.usage);
+        return this.usage;
     }
     _transform(event, _encoding, callback) {
-        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
-        const eventType = event.event || '';
-        if (eventType === 'message_start') {
-            const message = (_a = event.data) === null || _a === void 0 ? void 0 : _a.message;
-            if (message === null || message === void 0 ? void 0 : message.id)
-                this.responseId = message.id;
-            if (message === null || message === void 0 ? void 0 : message.model)
-                this.model = message.model;
-            this.ensureResponseCreated();
+        if (this.finished) {
             callback();
             return;
         }
-        if (eventType === 'content_block_start') {
-            const block = (_b = event.data) === null || _b === void 0 ? void 0 : _b.content_block;
-            const index = (_c = event.data) === null || _c === void 0 ? void 0 : _c.index;
-            if ((block === null || block === void 0 ? void 0 : block.type) === 'text') {
-                this.textBlockIndex = index;
-            }
-            if ((block === null || block === void 0 ? void 0 : block.type) === 'tool_use' && typeof index === 'number') {
-                this.toolCalls.set(index, {
-                    id: block.id || `tool_${index}`,
-                    name: block.name || 'tool',
-                    arguments: '',
-                });
-            }
+        const type = event.event;
+        const data = event.data;
+        if (type === 'message_start' && (data === null || data === void 0 ? void 0 : data.message)) {
+            this.model = data.message.model || this.model;
+            this.push({ event: null, data: { id: data.message.id, model: this.model, choices: [{ index: 0, delta: { role: 'assistant' }, finish_reason: null }] } });
             callback();
             return;
         }
-        if (eventType === 'content_block_delta') {
-            const delta = (_d = event.data) === null || _d === void 0 ? void 0 : _d.delta;
-            const index = (_e = event.data) === null || _e === void 0 ? void 0 : _e.index;
-            if ((delta === null || delta === void 0 ? void 0 : delta.type) === 'text_delta' && typeof delta.text === 'string') {
-                this.ensureResponseCreated();
-                this.outputText += delta.text;
-                this.pushEvent('response.output_text.delta', {
-                    type: 'response.output_text.delta',
-                    delta: delta.text,
-                    output_index: 0,
-                    content_index: 0,
-                });
+        if (type === 'content_block_start' && (data === null || data === void 0 ? void 0 : data.content_block)) {
+            if (data.content_block.type === 'text') {
+                // 文本块开始
             }
-            if ((delta === null || delta === void 0 ? void 0 : delta.type) === 'input_json_delta' && typeof delta.partial_json === 'string' && typeof index === 'number') {
-                const tool = this.toolCalls.get(index);
-                if (tool) {
-                    tool.arguments += delta.partial_json;
-                    this.ensureResponseCreated();
-                    this.pushEvent('response.output_tool_call.delta', {
-                        type: 'response.output_tool_call.delta',
-                        delta: delta.partial_json,
-                        output_index: index,
-                        tool_call_id: tool.id,
-                        name: tool.name,
-                    });
-                }
+            else if (data.content_block.type === 'tool_use') {
+                this.pendingToolCallId = data.content_block.id;
+                this.pendingToolName = data.content_block.name;
+                this.pendingToolArgs = '';
             }
             callback();
             return;
         }
-        if (eventType === 'content_block_stop') {
-            const index = (_f = event.data) === null || _f === void 0 ? void 0 : _f.index;
-            if (typeof index === 'number') {
-                if (this.textBlockIndex === index) {
-                    this.pushEvent('response.output_text.done', {
-                        type: 'response.output_text.done',
-                        text: this.outputText,
-                        output_index: 0,
-                        content_index: 0,
-                    });
-                    this.textBlockIndex = null;
-                }
-                const tool = this.toolCalls.get(index);
-                if (tool) {
-                    this.completedToolCalls.push(tool);
-                    this.pushEvent('response.output_tool_call.done', {
-                        type: 'response.output_tool_call.done',
-                        output_index: index,
-                        tool_call: {
-                            id: tool.id,
-                            name: tool.name,
-                            arguments: tool.arguments,
-                        },
-                    });
-                    this.toolCalls.delete(index);
+        if (type === 'content_block_delta' && (data === null || data === void 0 ? void 0 : data.delta)) {
+            if (data.delta.type === 'text_delta') {
+                const text = data.delta.text || '';
+                if (text) {
+                    this.push({ event: null, data: { id: '', model: this.model, choices: [{ index: 0, delta: { content: text } }] } });
                 }
             }
+            else if (data.delta.type === 'input_json_delta') {
+                this.pendingToolArgs += data.delta.partial_json || '';
+            }
             callback();
             return;
         }
-        if (eventType === 'message_delta') {
-            if ((_g = event.data) === null || _g === void 0 ? void 0 : _g.usage) {
-                this.usage = {
-                    input_tokens: (_h = event.data.usage.input_tokens) !== null && _h !== void 0 ? _h : 0,
-                    output_tokens: (_j = event.data.usage.output_tokens) !== null && _j !== void 0 ? _j : 0,
-                    cache_read_input_tokens: (_k = event.data.usage.cache_read_input_tokens) !== null && _k !== void 0 ? _k : 0,
+        if (type === 'content_block_stop') {
+            if (this.pendingToolCallId && this.pendingToolName !== null) {
+                const toolCall = {
+                    index: this.toolCallIndex,
+                    id: this.pendingToolCallId,
+                    type: 'function',
+                    function: {
+                        name: this.pendingToolName,
+                        arguments: this.pendingToolArgs,
+                    },
                 };
+                this.push({ event: null, data: { id: '', model: this.model, choices: [{ index: 0, delta: { tool_calls: [toolCall] } }] } });
+                this.toolCallIndex++;
+                this.pendingToolCallId = null;
+                this.pendingToolName = null;
+                this.pendingToolArgs = '';
             }
             callback();
             return;
         }
-        if (eventType === 'message_stop') {
-            this.pushCompletedResponse();
+        if (type === 'message_stop' || (data === null || data === void 0 ? void 0 : data.type) === 'message_stop') {
+            this.finished = true;
+            this.push({ event: null, data: { id: '', model: this.model, choices: [{ index: 0, delta: {}, finish_reason: 'stop' }] } });
+            this.push({ event: 'done', data: { type: 'done' } });
             callback();
             return;
         }
+        if (data === null || data === void 0 ? void 0 : data.usage) {
+            this.usage = {
+                prompt_tokens: data.usage.input_tokens || 0,
+                completion_tokens: data.usage.output_tokens || 0,
+                total_tokens: (data.usage.input_tokens || 0) + (data.usage.output_tokens || 0),
+            };
+        }
         callback();
     }
-    ensureResponseCreated() {
-        if (this.hasCreated)
-            return;
-        const response = {
-            id: this.responseId || `resp_${crypto_1.default.randomUUID()}`,
-            object: 'response',
-            model: this.model || 'unknown',
-            output: [],
-            created_at: this.createdAt,
-        };
-        this.pushEvent('response.created', { type: 'response.created', response });
-        this.hasCreated = true;
-    }
-    pushEvent(event, data) {
-        this.push({ event, data });
-    }
-    pushCompletedResponse() {
-        var _a, _b, _c, _d, _e, _f;
-        this.ensureResponseCreated();
-        const output = [];
-        if (this.outputText) {
-            output.push({
-                type: 'message',
-                role: 'assistant',
-                content: [{ type: 'output_text', text: this.outputText }],
-            });
-        }
-        for (const tool of this.completedToolCalls) {
-            output.push({
-                type: 'tool_call',
-                id: tool.id,
-                name: tool.name,
-                arguments: tool.arguments,
-            });
+    _flush(callback) {
+        if (!this.finished) {
+            this.finished = true;
+            this.push({ event: null, data: { id: '', model: this.model, choices: [{ index: 0, delta: {}, finish_reason: 'stop' }] } });
+            this.push({ event: 'done', data: { type: 'done' } });
         }
-        const inputTokens = (_b = (_a = this.usage) === null || _a === void 0 ? void 0 : _a.input_tokens) !== null && _b !== void 0 ? _b : 0;
-        const cacheRead = (_d = (_c = this.usage) === null || _c === void 0 ? void 0 : _c.cache_read_input_tokens) !== null && _d !== void 0 ? _d : 0;
-        const outputTokens = (_f = (_e = this.usage) === null || _e === void 0 ? void 0 : _e.output_tokens) !== null && _f !== void 0 ? _f : 0;
-        const response = {
-            id: this.responseId || `resp_${crypto_1.default.randomUUID()}`,
-            object: 'response',
-            model: this.model || 'unknown',
-            output,
-            output_text: this.outputText,
-            status: 'completed',
-            created_at: this.createdAt,
-            usage: {
-                input_tokens: inputTokens + cacheRead,
-                output_tokens: outputTokens,
-                total_tokens: inputTokens + cacheRead + outputTokens,
-            },
-        };
-        this.pushEvent('response.completed', { type: 'response.completed', response });
+        callback();
     }
 }
-exports.ClaudeToOpenAIResponsesEventTransform = ClaudeToOpenAIResponsesEventTransform;
+exports.ClaudeToOpenAIChatEventTransform = ClaudeToOpenAIChatEventTransform;