add retry for openai chatbot

ElmiraGhorbani · Jul 25, 2023 · fee7cde · fee7cde
1 parent 0509350
commit fee7cde
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chatgpt_long_term_memory/openai_engine/openai_chatbot.py b/chatgpt_long_term_memory/openai_engine/openai_chatbot.py
@@ -53,7 +53,7 @@ def __init__(self, openai_chatbot_config: OpenAIChatConfig, **kw):
             'gpt-3.5-turbo-16k': 16384
         }
 
-    # @retry_on_openai_errors(max_retry=3)
+    @retry_on_openai_errors(max_retry=3)
     def chat(self, user_input, chat_history=[]):
         """
         This function serves as a wrapper for the OpenAI API, allowing the chatbot to generate a response to a user's query.
@@ -73,7 +73,7 @@ def chat(self, user_input, chat_history=[]):
         7. The API returns a response that contains the generated message from the language model.
         8. The chatbot extracts the assistant's response from the API response and returns it as the final output.
         """
-        
+
         if chat_history:
             history = "\n".join(chat_history)
         else: