xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-06 05:40:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

update litellm

Browse Source
This commit is contained in:
mrT23
2023-08-23 09:19:15 +03:00
parent 6599cbc7f2
commit 16150e9c84
2 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pr_agent/algo/ai_handler.py
View File

@ -48,8 +48,6 @@ class AiHandler:
litellm.replicate_key = get_settings().replicate.key litellm.replicate_key = get_settings().replicate.key
if get_settings().get("HUGGINGFACE.KEY", None): if get_settings().get("HUGGINGFACE.KEY", None):
litellm.huggingface_key = get_settings().huggingface.key 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: except AttributeError as e:
raise ValueError("OpenAI key is required") from e raise ValueError("OpenAI key is required") from e

2
pyproject.toml
View File

@ -42,7 +42,7 @@ dependencies = [
"atlassian-python-api==3.39.0", "atlassian-python-api==3.39.0",
"GitPython~=3.1.32", "GitPython~=3.1.32",
"starlette-context==0.3.6", "starlette-context==0.3.6",
"litellm~=0.1.351", "litellm~=0.1.445",
"PyYAML==6.0", "PyYAML==6.0",
"boto3~=1.28.25" "boto3~=1.28.25"
] ]