xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-03 04:10:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
1 changed files with 3 additions and 0 deletions
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:
logging.error("Rate limit error during OpenAI inference: ", e)
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:
raise TryAgain
resp = response.choices[0]['message']['content']