fix: handle badrequesterror

r2e-project · Jan 17, 2025 · a594dca · a594dca
1 parent b145991
commit a594dca
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/r2e/llms/openai_runner.py b/r2e/llms/openai_runner.py
@@ -46,6 +46,11 @@ def _run_single(self, payload: list[dict[str, str]]) -> list[str]:
                 messages=payload,  # type: ignore
                 **self.client_kwargs,
             )
+        except openai.BadRequestError as e:
+            return [
+                f"OpenAI BadRequestError: {e}"
+                for _ in range(self.client_kwargs.get("n", 1))
+            ]
         except (
             openai.APIError,
             openai.RateLimitError,