From 578d7c69f8c6e84db7070e6d439b30142a78888a Mon Sep 17 00:00:00 2001 From: woung717 Date: Thu, 29 Aug 2024 21:45:48 +0900 Subject: [PATCH] fix: change deprecated timeout parameter for litellm --- pr_agent/algo/ai_handlers/litellm_ai_handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index f577bead..282f662b 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -190,7 +190,7 @@ class LiteLLMAIHandler(BaseAiHandler): "deployment_id": deployment_id, "messages": messages, "temperature": temperature, - "force_timeout": get_settings().config.ai_timeout, + "timeout": get_settings().config.ai_timeout, "api_base": self.api_base, }