Fix error propagation on rate limit.

norpadon · Nov 21, 2024 · 3d51e2a · 3d51e2a
1 parent 7a2a79e
commit 3d51e2a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "wanga"
-version = "0.2.2"
+version = "0.2.3"
 description = "A library for interacting with Large Language Models."
 authors = [{ name = "Artur Chakhvadze", email = "norpadon@gmail.com" }]
 license = { text = "MIT" }

diff --git a/wanga/models/openai.py b/wanga/models/openai.py
@@ -189,7 +189,6 @@ def _create_retry_decorator(self):
             stop=stop_after_attempt(self._num_retries + 1),
             wait=wait_exponential(multiplier=1, min=4, max=10),
             retry=retry_conditions,
-            retry_error_callback=lambda retry_state: retry_state.outcome.exception() if retry_state.outcome else None,
             reraise=True,
         )