From f8f57419c40ad2d777200a936d541bf20c868ee1 Mon Sep 17 00:00:00 2001 From: tmokmss Date: Tue, 28 Nov 2023 23:07:46 +0900 Subject: [PATCH] Update ai_handler.py --- pr_agent/algo/ai_handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pr_agent/algo/ai_handler.py b/pr_agent/algo/ai_handler.py index 7f071ef3..5b6a05f4 100644 --- a/pr_agent/algo/ai_handler.py +++ b/pr_agent/algo/ai_handler.py @@ -1,5 +1,6 @@ import os +import boto3 import litellm import openai from litellm import acompletion @@ -7,7 +8,6 @@ from openai.error import APIError, RateLimitError, Timeout, TryAgain from retry import retry from pr_agent.config_loader import get_settings from pr_agent.log import get_logger -import boto3 OPENAI_RETRIES = 5