mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 22:34:59 +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
10
coder.py
10
coder.py
|
@ -239,12 +239,22 @@ class Coder:
|
||||||
if not model:
|
if not model:
|
||||||
model = self.main_model
|
model = self.main_model
|
||||||
|
|
||||||
|
import time
|
||||||
|
from openai.error import RateLimitError
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
completion = openai.ChatCompletion.create(
|
completion = openai.ChatCompletion.create(
|
||||||
model=model,
|
model=model,
|
||||||
messages=messages,
|
messages=messages,
|
||||||
temperature=0,
|
temperature=0,
|
||||||
stream=True,
|
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
|
interrupted = False
|
||||||
try:
|
try:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue