Update app.js

app.js

@@ -148,33 +148,54 @@ async function handleRequest(req, res) {
     res.setHeader('Access-Control-Allow-Methods', 'POST, GET, OPTIONS');
     res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization, x-api-key, anthropic-version, model');
 
+    const contentType = response.headers.get('content-type');
     const contentEncoding = response.headers.get('content-encoding');
+    console.log('Content-Type:', contentType);
     console.log('Content-Encoding:', contentEncoding);
 
-    // 读取整个响应体
-    const buffer = await response.buffer();
-    console.log('Response body length:', buffer.length);
+    if (contentType && contentType.includes('text/event-stream')) {
+      // 处理 SSE
+      res.setHeader('Content-Type', 'text/event-stream');
+      res.setHeader('Cache-Control', 'no-cache');
+      res.setHeader('Connection', 'keep-alive');
 
-    let data;
-    if (contentEncoding === 'gzip') {
-      // 检查是否已经是JSON格式
-      if (buffer.toString().trim().startsWith('{')) {
-        console.log('Response appears to be JSON despite gzip header. Using as-is.');
-        data = buffer;
-      } else {
-        try {
-          data = await gunzip(buffer);
-        } catch (error) {
-          console.error('Gunzip error:', error);
-          console.log('First 100 bytes of response:', buffer.slice(0, 100).toString('hex'));
-          throw new Error('Failed to decompress the response');
-        }
+      const reader = response.body.getReader();
+      const decoder = new TextDecoder();
+
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        const chunk = decoder.decode(value, { stream: true });
+        res.write(chunk);
       }
+
+      res.end();
     } else {
-      data = buffer;
+      // 非流式响应的处理
+      const buffer = await response.buffer();
+      console.log('Response body length:', buffer.length);
+
+      let data;
+      if (contentEncoding === 'gzip') {
+        if (buffer.toString().trim().startsWith('{') || buffer.toString().trim().startsWith('event:')) {
+          console.log('Response appears to be JSON or SSE despite gzip header. Using as-is.');
+          data = buffer;
+        } else {
+          try {
+            data = await gunzip(buffer);
+          } catch (error) {
+            console.error('Gunzip error:', error);
+            console.log('First 100 bytes of response:', buffer.slice(0, 100).toString('hex'));
+            throw new Error('Failed to decompress the response');
+          }
+        }
+      } else {
+        data = buffer;
+      }
+
+      res.send(data);
     }
 
-    res.send(data);
     logRequest(req, response.status, `Request forwarded successfully for model: ${model}`);
   } catch (error) {
     console.error('Request error:', error);