Merge pull request #876 from randy-tsukemen/fix-duplication

Fix duplicate assignment of replicate_key in LiteLLMAIHandler
This commit is contained in:
Tal
2024-04-21 10:55:01 +03:00
committed by GitHub

View File

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