mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 06:15:00 +00:00
Handle openai.error.RateLimitError
and retry requests with a delay.
This commit is contained in:
parent
a85cc59120
commit
06351d3205
1 changed files with 16 additions and 6 deletions
22
coder.py
22
coder.py
|
@ -239,12 +239,22 @@ class Coder:
|
|||
if not model:
|
||||
model = self.main_model
|
||||
|
||||
completion = openai.ChatCompletion.create(
|
||||
model=model,
|
||||
messages=messages,
|
||||
temperature=0,
|
||||
stream=True,
|
||||
)
|
||||
import time
|
||||
from openai.error import RateLimitError
|
||||
|
||||
while True:
|
||||
try:
|
||||
completion = openai.ChatCompletion.create(
|
||||
model=model,
|
||||
messages=messages,
|
||||
temperature=0,
|
||||
stream=True,
|
||||
)
|
||||
break
|
||||
except RateLimitError as e:
|
||||
retry_after = e.retry_after
|
||||
print(f"Rate limit exceeded. Retrying in {retry_after} seconds.")
|
||||
time.sleep(retry_after)
|
||||
|
||||
interrupted = False
|
||||
try:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue