@mmmbuto/zai-codex-bridge 0.1.8 → 0.1.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mmmbuto/zai-codex-bridge",
-  "version": "0.1.8",
+  "version": "0.1.10",
   "description": "Local proxy that translates OpenAI Responses API format to Z.AI Chat Completions format for Codex",
   "main": "src/server.js",
   "bin": {

package/src/server.js

@@ -260,6 +260,7 @@ async function streamChatToResponses(stream, res) {
   let buffer = '';
   let chunkCount = 0;
   let deltaCount = 0;
+  let lastParsed = null; // Keep track of last parsed SSE for ID extraction
 
   log('debug', 'Starting to process stream');
 
@@ -284,14 +285,32 @@ async function streamChatToResponses(stream, res) {
       // Check for stream end
       if (data === '[DONE]') {
         log('info', `Stream end received - wrote ${deltaCount} deltas total`);
-        res.write(`event: completed\n`);
-        res.write(`data: ${JSON.stringify({ status: 'completed' })}\n\n`);
-        log('info', 'Sent completed event');
+
+        // Send response.completed event (required by Codex Responses API)
+        // Map Z.AI usage format to Responses API format
+        const zaiUsage = lastParsed?.usage;
+        const completedEvent = {
+          id: lastParsed?.id || 'msg_' + Date.now(),
+          usage: zaiUsage ? {
+            input_tokens: zaiUsage.prompt_tokens || 0,
+            output_tokens: zaiUsage.completion_tokens || 0,
+            total_tokens: zaiUsage.total_tokens || 0
+          } : {
+            input_tokens: 0,
+            output_tokens: 0,
+            total_tokens: 0
+          }
+        };
+
+        res.write(`event: response.completed\n`);
+        res.write(`data: ${JSON.stringify(completedEvent)}\n\n`);
+        log('info', 'Sent response.completed event');
         return;
       }
 
       try {
         const parsed = JSON.parse(data);
+        lastParsed = parsed; // Save for later use in completed event
         log('debug', 'Parsed SSE:', JSON.stringify(parsed).substring(0, 150));
 
         const delta = parsed.choices?.[0]?.delta;