@mmmbuto/zai-codex-bridge 0.3.0 → 0.3.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mmmbuto/zai-codex-bridge",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "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

@@ -12,6 +12,8 @@
 
 const http = require('http');
 const crypto = require('crypto');
+const { createGunzip } = require('zlib');
+const { pipeline } = require('stream');
 
 // Configuration from environment
 const PORT = parseInt(process.env.PORT || '31415', 10);
@@ -269,7 +271,8 @@ async function makeUpstreamRequest(path, body, headers) {
   const auth = pickAuth(headers);
   const upstreamHeaders = {
     'Content-Type': 'application/json',
-    'Authorization': auth
+    'Authorization': auth,
+    'Accept-Encoding': 'identity'  // Disable compression to avoid gzip issues
   };
 
   log('info', 'Upstream request:', {
@@ -456,107 +459,124 @@ async function streamChatToResponses(stream, res, responseId, messageItemId) {
 
   try {
     for await (const chunk of stream) {
-      buffer += chunk.toString('utf8');
-
-      // Z.ai stream: SSE lines "data: {...}\n\n"
-      let idx;
-      while ((idx = buffer.indexOf('\n\n')) !== -1) {
-        const raw = buffer.slice(0, idx);
-        buffer = buffer.slice(idx + 2);
-
-        const lines = raw.split('\n');
-        for (const line of lines) {
-          if (!line.startsWith('data:')) continue;
-          const payload = line.slice(5).trim();
-          if (!payload || payload === '[DONE]') continue;
-
-          let json;
-          try { json = JSON.parse(payload); } catch { continue; }
-
-          const choice = json?.choices?.[0];
-          const delta = choice?.delta ?? {};
-
-          // 1) reasoning
-          if (typeof delta.reasoning_content === 'string' && delta.reasoning_content.length) {
-            reasoningText += delta.reasoning_content;
-            send({
-              type: 'response.reasoning_text.delta',
-              sequence_number: seq++,
-              item_id: messageItemId,
-              output_index: 0,
-              content_index: 0,
-              delta: delta.reasoning_content,
-            });
-            log('debug', `Reasoning delta: ${delta.reasoning_content.substring(0, 30)}...`);
+      const chunkStr = Buffer.from(chunk).toString('utf8');
+      buffer += chunkStr;
+
+      // Z.ai stream: SSE lines "data: {...}\n"
+      // Split by newline and process each complete line
+      const lines = buffer.split('\n');
+      // Keep the last line if it's incomplete (doesn't end with data pattern)
+      buffer = lines.pop() || '';
+
+      for (const line of lines) {
+        if (!line.trim() || !line.startsWith('data:')) {
+          // Skip empty lines and comments (starting with :)
+          if (line.trim() && !line.startsWith(':')) {
+            log('debug', 'Non-data line:', line.substring(0, 50));
           }
+          continue;
+        }
 
-          // 2) normal output
-          if (typeof delta.content === 'string' && delta.content.length) {
-            outputText += delta.content;
-            send({
-              type: 'response.output_text.delta',
-              sequence_number: seq++,
-              item_id: messageItemId,
-              output_index: 0,
-              content_index: reasoningText ? 1 : 0,
-              delta: delta.content,
-            });
-            log('debug', `Output delta: ${delta.content.substring(0, 30)}...`);
-          }
+        const payload = line.slice(5).trim();
+        if (payload === '[DONE]') {
+          log('info', 'Stream received [DONE]');
+          await finalizeAndClose();
+          return;
+        }
+
+        if (!payload) continue;
+
+        let json;
+        try {
+          json = JSON.parse(payload);
+        } catch (e) {
+          log('warn', 'Failed to parse SSE payload:', e.message, 'payload:', payload.substring(0, 100));
+          continue;
+        }
 
-          // 3) tool calls (OpenAI-style in chat.completions delta.tool_calls)
-          if (Array.isArray(delta.tool_calls)) {
-            for (const tc of delta.tool_calls) {
-              // tc: {id, type:"function", function:{name, arguments}}
-              const callId = tc.id || `call_${tc.index ?? 0}`;
-              const name = tc.function?.name || 'unknown';
-              const argsDelta = tc.function?.arguments || '';
-
-              let st = toolCalls.get(callId);
-              if (!st) {
-                st = {
-                  itemId: `fc_${crypto.randomUUID().replace(/-/g, '')}`,
-                  outputIndex: nextOutputIndex++,
-                  name,
-                  args: '',
-                };
-                toolCalls.set(callId, st);
-
-                send({
-                  type: 'response.output_item.added',
-                  sequence_number: seq++,
-                  output_index: st.outputIndex,
-                  item: {
-                    type: 'function_call',
-                    id: st.itemId,
-                    call_id: callId,
-                    name: st.name,
-                    arguments: '',
-                  },
-                });
-                log('debug', `Tool call added: ${name} (${callId})`);
-              }
-
-              if (argsDelta) {
-                st.args += argsDelta;
-                send({
-                  type: 'response.function_call_arguments.delta',
-                  sequence_number: seq++,
-                  item_id: st.itemId,
-                  output_index: st.outputIndex,
-                  delta: argsDelta,
-                });
-              }
+        const choice = json?.choices?.[0];
+        const delta = choice?.delta ?? {};
+
+        // 1) reasoning
+        if (typeof delta.reasoning_content === 'string' && delta.reasoning_content.length) {
+          reasoningText += delta.reasoning_content;
+          send({
+            type: 'response.reasoning_text.delta',
+            sequence_number: seq++,
+            item_id: messageItemId,
+            output_index: 0,
+            content_index: 0,
+            delta: delta.reasoning_content,
+          });
+          log('debug', `Reasoning delta: ${delta.reasoning_content.substring(0, 30)}...`);
+        }
+
+        // 2) normal output
+        if (typeof delta.content === 'string' && delta.content.length) {
+          outputText += delta.content;
+          send({
+            type: 'response.output_text.delta',
+            sequence_number: seq++,
+            item_id: messageItemId,
+            output_index: 0,
+            content_index: reasoningText ? 1 : 0,
+            delta: delta.content,
+          });
+          log('debug', `Output delta: ${delta.content.substring(0, 30)}...`);
+        }
+
+        // 3) tool calls (OpenAI-style in chat.completions delta.tool_calls)
+        if (Array.isArray(delta.tool_calls)) {
+          for (const tc of delta.tool_calls) {
+            // tc: {id, type:"function", function:{name, arguments}}
+            const callId = tc.id || `call_${tc.index ?? 0}`;
+            const name = tc.function?.name || 'unknown';
+            const argsDelta = tc.function?.arguments || '';
+
+            let st = toolCalls.get(callId);
+            if (!st) {
+              st = {
+                itemId: `fc_${crypto.randomUUID().replace(/-/g, '')}`,
+                outputIndex: nextOutputIndex++,
+                name,
+                args: '',
+              };
+              toolCalls.set(callId, st);
+
+              send({
+                type: 'response.output_item.added',
+                sequence_number: seq++,
+                output_index: st.outputIndex,
+                item: {
+                  type: 'function_call',
+                  id: st.itemId,
+                  call_id: callId,
+                  name: st.name,
+                  arguments: '',
+                },
+              });
+              log('debug', `Tool call added: ${name} (${callId})`);
             }
-          }
 
-          // 4) finish
-          if (choice?.finish_reason) {
-            log('info', `Stream finish_reason: ${choice.finish_reason}`);
-            await finalizeAndClose();
-            return;
+            if (argsDelta) {
+              st.args += argsDelta;
+              send({
+                type: 'response.function_call_arguments.delta',
+                sequence_number: seq++,
+                item_id: st.itemId,
+                output_index: st.outputIndex,
+                delta: argsDelta,
+              });
+            }
           }
         }
+
+        // 4) finish
+        if (choice?.finish_reason) {
+          log('info', `Stream finish_reason: ${choice.finish_reason}`);
+          await finalizeAndClose();
+          return;
+        }
       }
     }
   } catch (e) {