From f3287a9f673c6962b5b28627b8ae80d702a8be3e Mon Sep 17 00:00:00 2001 From: mrT23 Date: Fri, 8 Aug 2025 09:34:31 +0300 Subject: [PATCH] fix: update model prefix in litellm_ai_handler and adjust dependencies in requirements.txt --- pr_agent/algo/ai_handlers/litellm_ai_handler.py | 2 +- requirements.txt | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index 78d12ffe..31e3d12b 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -300,7 +300,7 @@ class LiteLLMAIHandler(BaseAiHandler): "reasoning_effort": 'minimal', "allowed_openai_params": ["reasoning_effort"], } - model = model.replace('_thinking', '') # remove _thinking suffix + model = 'openai/'+model.replace('_thinking', '') # remove _thinking suffix # Currently, some models do not support a separate system and user prompts diff --git a/requirements.txt b/requirements.txt index 9c138ad9..ec9f3451 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,16 +1,19 @@ +aiohttp==3.10.2 anthropic>=0.52.0 #anthropic[vertex]==0.47.1 atlassian-python-api==3.41.4 azure-devops==7.1.0b3 azure-identity==1.15.0 +boto3==1.33.6 certifi==2024.8.30 dynaconf==3.2.4 +fastapi==0.115.6 GitPython==3.1.41 google-cloud-aiplatform==1.38.0 google-generativeai==0.8.3 google-cloud-storage==2.10.0 Jinja2==3.1.2 -litellm[proxy]==1.75.2 +litellm==1.73.6 loguru==0.7.2 msrest==0.7.1 openai>=1.55.3 @@ -22,7 +25,9 @@ retry==0.9.2 starlette-context==0.3.6 tiktoken==0.8.0 ujson==5.8.0 +uvicorn==0.22.0 tenacity==8.2.3 +gunicorn==23.0.0 pytest-cov==5.0.0 pydantic==2.8.2 html2text==2024.2.26