From b2132507d3ea6a45def105a712ee31551e84c7c6 Mon Sep 17 00:00:00 2001 From: Michael Dyer Date: Fri, 11 Oct 2024 15:54:24 +0200 Subject: [PATCH] Raise exception if no response after all retries --- app/llm/external/openai_chat.py | 2 +- app/llm/external/openai_embeddings.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/app/llm/external/openai_chat.py b/app/llm/external/openai_chat.py index 0204ff7..c6a8969 100644 --- a/app/llm/external/openai_chat.py +++ b/app/llm/external/openai_chat.py @@ -131,7 +131,7 @@ def chat( traceback.print_exc() logging.info(f"Retrying in {wait_time} seconds...") time.sleep(wait_time) - logging.error(f"Failed to get response from OpenAI after {retries} retries") + raise Exception(f"Failed to get response from OpenAI after {retries} retries") class DirectOpenAIChatModel(OpenAIChatModel): diff --git a/app/llm/external/openai_embeddings.py b/app/llm/external/openai_embeddings.py index b5d00e5..6e0d8ef 100644 --- a/app/llm/external/openai_embeddings.py +++ b/app/llm/external/openai_embeddings.py @@ -42,8 +42,7 @@ def embed(self, text: str) -> list[float]: logging.warning(f"OpenAI error on attempt {attempt + 1}: {e}") logging.info(f"Retrying in {wait_time} seconds...") time.sleep(wait_time) - logging.error(f"Failed to get embedding after {retries} retries.") - return [] + raise Exception(f"Failed to get embedding from OpenAI after {retries} retries.") class DirectOpenAIEmbeddingModel(OpenAIEmbeddingModel):