fix: handle rate_limit_exceeded in openai assistants (#210)

Co-authored-by: Saikat Mitra <saikatmitra91@gmail.com>
empirical-run · May 6, 2024 · 0b2d1c0 · 0b2d1c0
1 parent 973c9b2
commit 0b2d1c0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.changeset/wild-owls-agree.md b/.changeset/wild-owls-agree.md
@@ -0,0 +1,5 @@
+---
+"@empiricalrun/ai": patch
+---
+
+fix: handle rate_limit_exceeded in openai assistants
diff --git a/packages/ai/src/providers/openai/index.ts b/packages/ai/src/providers/openai/index.ts
@@ -156,7 +156,11 @@ const runAssistant: ICreateAndRunAssistantThread = async (body) => {
           }
           return asstRunResp;
         })().catch((err: any) => {
-          if ((err.message as string).includes("server_error")) {
+          const serverError = (err.message as string).includes("server_error");
+          const rateLimitError = (err.message as string).includes(
+            "rate_limit_exceeded",
+          );
+          if (serverError || rateLimitError) {
             console.warn(
               `Retrying request due to server error (attempt ${attempt})`,
             );