Add error handling to bailout from failure cases

SuDa2103 · May 30, 2023 · 1bc49c5 · 1bc49c5
1 parent 17ae6ce
commit 1bc49c5
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/utils/OpenAIStream.ts b/utils/OpenAIStream.ts
@@ -45,6 +45,19 @@ export async function OpenAIStream(payload: OpenAIStreamPayload) {
           controller.enqueue(encoder.encode(data));
         }
       }
+
+      // optimistic error handling
+      if (res.status !== 200) {
+        const data = {
+          status: res.status,
+          statusText: res.statusText,
+          body: await res.text(),
+        }
+        console.log(`Error: recieved non-200 status code, ${JSON.stringify(data)}`);
+        controller.close();
+        return
+      }
+
       // stream response (SSE) from OpenAI may be fragmented into multiple chunks
       // this ensures we properly read chunks and invoke an event for each SSE event stream
       const parser = createParser(onParse);