From 16150e9c84dfe1482fc58e39a72122f52649cc2e Mon Sep 17 00:00:00 2001 From: mrT23 Date: Wed, 23 Aug 2023 09:19:15 +0300 Subject: [PATCH] update litellm --- pr_agent/algo/ai_handler.py | 2 -- pyproject.toml | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/pr_agent/algo/ai_handler.py b/pr_agent/algo/ai_handler.py index 4e1f86e3..1a12564b 100644 --- a/pr_agent/algo/ai_handler.py +++ b/pr_agent/algo/ai_handler.py @@ -48,8 +48,6 @@ class AiHandler: litellm.replicate_key = get_settings().replicate.key if get_settings().get("HUGGINGFACE.KEY", None): litellm.huggingface_key = get_settings().huggingface.key - if get_settings().get("HUGGINGFACE.KEY", None): - litellm.huggingface_key = get_settings().huggingface.key except AttributeError as e: raise ValueError("OpenAI key is required") from e diff --git a/pyproject.toml b/pyproject.toml index 802cd0d8..9a945dca 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -42,7 +42,7 @@ dependencies = [ "atlassian-python-api==3.39.0", "GitPython~=3.1.32", "starlette-context==0.3.6", - "litellm~=0.1.351", + "litellm~=0.1.445", "PyYAML==6.0", "boto3~=1.28.25" ]