Fix duplicate assignment of replicate_key in LiteLLMAIHandler

This commit is contained in:
Randy, Huang
2024-04-21 14:47:25 +09:00
parent e429c5d012
commit 7a9e73702d

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: