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

Fix duplicate assignment of replicate_key in LiteLLMAIHandler

Browse Source
This commit is contained in:
Randy, Huang
2024-04-21 14:47:25 +09:00
parent e429c5d012
commit 7a9e73702d
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View File

@ -54,8 +54,6 @@ class LiteLLMAIHandler(BaseAiHandler):
litellm.cohere_key = get_settings().cohere.key litellm.cohere_key = get_settings().cohere.key
if get_settings().get("REPLICATE.KEY", None): if get_settings().get("REPLICATE.KEY", None):
litellm.replicate_key = get_settings().replicate.key litellm.replicate_key = get_settings().replicate.key
if get_settings().get("REPLICATE.KEY", None):
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.API_BASE", None) and 'huggingface' in get_settings().config.model: if get_settings().get("HUGGINGFACE.API_BASE", None) and 'huggingface' in get_settings().config.model: