Update chatStream.js

server/chatStream.js

@@ -192,9 +192,24 @@ export async function websocketChatStream(body, headers, onToken, abortSignal) {
   let finished = false;
 
   return new Promise((resolve, reject) => {
+    // Timeout fallback - resolve after 120 seconds if no finish_reason
+    const timeoutId = setTimeout(() => {
+      if (!finished) {
+        finished = true;
+        cleanup();
+        const toolCalls = [...toolCallBuffer.values()].map((t) => ({
+          id: t.id || `call_${crypto.randomUUID()}`,
+          type: "function",
+          function: { name: t.name, arguments: t.arguments },
+        }));
+        resolve({ assistantText, toolCalls });
+      }
+    }, 120000);
+
     const cleanup = () => {
       activeStreamHandlers.delete(currentRequestId);
       errorHandlers.delete(currentRequestId);
+      clearTimeout(timeoutId);
       if (abortSignal) {
         abortSignal.removeEventListener("abort", abortHandler);
       }
@@ -220,12 +235,12 @@ export async function websocketChatStream(body, headers, onToken, abortSignal) {
         return;
       }
 
-      // Log server-side errors
-      if (payload.error) {
+      // Only treat as fatal error if it's a structured error response (not delta data with error field)
+      if (payload.error && !payload.choices) {
         console.error("[WS PAYLOAD ERROR]", payload.error);
         if (onToken) onToken(`[ERROR] ${payload.error}`);
         
-        // Mark as finished to end the stream
+        // Mark as finished to end the stream on fatal errors only
         if (!finished) {
           finished = true;
           cleanup();