Raise exception if no response after all retries

ls1intum · Oct 11, 2024 · b213250 · b213250
1 parent 2438c5e
commit b213250
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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):