aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-18 14:34:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Implements handling of OpenAI rate limit error.

Browse Source
This commit is contained in:
Torantulino
2023-04-01 01:30:13 +01:00
parent 4100961745
commit ffeb1aca73
1 changed files with 38 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/chat.py
View File

@@ -1,3 +1,4 @@
import time
import openai import openai
import keys import keys
@@ -18,6 +19,8 @@ def create_chat_message(role, content):
return {"role": role, "content": content} return {"role": role, "content": content}
def chat_with_ai(prompt, user_input, full_message_history, permanent_memory, token_limit, debug = False): def chat_with_ai(prompt, user_input, full_message_history, permanent_memory, token_limit, debug = False):
while True:
try:
""" """
Interact with the OpenAI API, sending the prompt, user input, message history, and permanent memory. Interact with the OpenAI API, sending the prompt, user input, message history, and permanent memory.
@@ -57,3 +60,6 @@ def chat_with_ai(prompt, user_input, full_message_history, permanent_memory, tok
full_message_history.append(create_chat_message("assistant", assistant_reply)) full_message_history.append(create_chat_message("assistant", assistant_reply))
return assistant_reply return assistant_reply
except openai.RateLimitError:
print("Error: ", "API Rate Limit Reached. Waiting 60 seconds...")
time.sleep(60)