ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 02:45:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Retry on rate limit error on OpenAI calls

Browse source
This commit is contained in:
Ori Kotek 2023-07-20 15:02:34 +03:00
parent 2f9546e144
commit 2e246869d0
No known key found for this signature in database
GPG key ID: 5AAFD0757D1EAC4C
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pr_agent/algo/ai_handler.py
View file

@ -72,6 +72,9 @@ class AiHandler:
except (RateLimitError) as e: except (RateLimitError) as e:
logging.error("Rate limit error during OpenAI inference: ", e) logging.error("Rate limit error during OpenAI inference: ", e)
raise raise
except (Exception) as e:
logging.error("Unknown error during OpenAI inference: ", e)
raise TryAgain from e
if response is None or len(response.choices) == 0: if response is None or len(response.choices) == 0:
raise TryAgain raise TryAgain
resp = response.choices[0]['message']['content'] resp = response.choices[0]['message']['content']