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

Update ai_handler.py

Browse Source
This commit is contained in:
tmokmss
2023-11-28 23:07:46 +09:00
parent 917f4b6a01
commit f8f57419c4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pr_agent/algo/ai_handler.py
View File

@ -1,5 +1,6 @@
import os import os
import boto3
import litellm import litellm
import openai import openai
from litellm import acompletion from litellm import acompletion
@ -7,7 +8,6 @@ from openai.error import APIError, RateLimitError, Timeout, TryAgain
from retry import retry from retry import retry
from pr_agent.config_loader import get_settings from pr_agent.config_loader import get_settings
from pr_agent.log import get_logger from pr_agent.log import get_logger
import boto3
OPENAI_RETRIES = 5 OPENAI_RETRIES = 5