From e648b8e8ecbf6c1ada7a66f140301099c667ca4d Mon Sep 17 00:00:00 2001
From: Logan <logan.markewich@live.com>
Date: Mon, 20 May 2024 16:03:15 -0600
Subject: [PATCH] update how retries get triggered for openai (#13608)
---
.../llama_index/embeddings/openai/utils.py | 3 +--
.../embeddings/llama-index-embeddings-openai/pyproject.toml | 2 +-
.../llama_index/embeddings/upstage/utils.py | 5 ++---
.../embeddings/llama-index-embeddings-upstage/pyproject.toml | 2 +-
.../llama-index-llms-openai/llama_index/llms/openai/utils.py | 3 +--
.../llms/llama-index-llms-openai/pyproject.toml | 2 +-
6 files changed, 7 insertions(+), 10 deletions(-)
diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-openai/llama_index/embeddings/openai/utils.py b/llama-index-integrations/embeddings/llama-index-embeddings-openai/llama_index/embeddings/openai/utils.py
index 6fc3d9ade3..a9ceedc7eb 100644
--- a/llama-index-integrations/embeddings/llama-index-embeddings-openai/llama_index/embeddings/openai/utils.py
+++ b/llama-index-integrations/embeddings/llama-index-embeddings-openai/llama_index/embeddings/openai/utils.py
@@ -57,9 +57,8 @@ def create_retry_decorator(
retry=(
retry_if_exception_type(
(
- openai.APITimeoutError,
- openai.APIError,
openai.APIConnectionError,
+ openai.APITimeoutError,
openai.RateLimitError,
openai.InternalServerError,
)
diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-openai/pyproject.toml b/llama-index-integrations/embeddings/llama-index-embeddings-openai/pyproject.toml
index 8d303813b0..417a1fbe20 100644
--- a/llama-index-integrations/embeddings/llama-index-embeddings-openai/pyproject.toml
+++ b/llama-index-integrations/embeddings/llama-index-embeddings-openai/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
license = "MIT"
name = "llama-index-embeddings-openai"
readme = "README.md"
-version = "0.1.9"
+version = "0.1.10"
[tool.poetry.dependencies]
python = ">=3.8.1,<4.0"
diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-upstage/llama_index/embeddings/upstage/utils.py b/llama-index-integrations/embeddings/llama-index-embeddings-upstage/llama_index/embeddings/upstage/utils.py
index 543aab2d98..01cb73e239 100644
--- a/llama-index-integrations/embeddings/llama-index-embeddings-upstage/llama_index/embeddings/upstage/utils.py
+++ b/llama-index-integrations/embeddings/llama-index-embeddings-upstage/llama_index/embeddings/upstage/utils.py
@@ -62,11 +62,10 @@ def create_retry_decorator(
retry=(
retry_if_exception_type(
(
- openai.APITimeoutError,
- openai.APIError,
openai.APIConnectionError,
+ openai.APITimeoutError,
openai.RateLimitError,
- openai.APIStatusError,
+ openai.InternalServerError,
)
)
),
diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-upstage/pyproject.toml b/llama-index-integrations/embeddings/llama-index-embeddings-upstage/pyproject.toml
index 348618780b..20c749d37a 100644
--- a/llama-index-integrations/embeddings/llama-index-embeddings-upstage/pyproject.toml
+++ b/llama-index-integrations/embeddings/llama-index-embeddings-upstage/pyproject.toml
@@ -30,7 +30,7 @@ license = "MIT"
name = "llama-index-embeddings-upstage"
packages = [{include = "llama_index/"}]
readme = "README.md"
-version = "0.1.0"
+version = "0.1.1"
[tool.poetry.dependencies]
python = ">=3.8.1,<4.0"
diff --git a/llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/utils.py b/llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/utils.py
index b4fdfbb489..c2d3d85de2 100644
--- a/llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/utils.py
+++ b/llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/utils.py
@@ -162,9 +162,8 @@ def create_retry_decorator(
retry=(
retry_if_exception_type(
(
- openai.APITimeoutError,
- openai.APIError,
openai.APIConnectionError,
+ openai.APITimeoutError,
openai.RateLimitError,
openai.InternalServerError,
)
diff --git a/llama-index-integrations/llms/llama-index-llms-openai/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-openai/pyproject.toml
index 97046212d1..dac2287b73 100644
--- a/llama-index-integrations/llms/llama-index-llms-openai/pyproject.toml
+++ b/llama-index-integrations/llms/llama-index-llms-openai/pyproject.toml
@@ -29,7 +29,7 @@ exclude = ["**/BUILD"]
license = "MIT"
name = "llama-index-llms-openai"
readme = "README.md"
-version = "0.1.19"
+version = "0.1.20"
[tool.poetry.dependencies]
python = ">=3.8.1,<4.0"
--
GitLab