phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

change friendli_ai -> friendliai

Browse source
This commit is contained in:
wslee 2024-06-11 16:17:30 +09:00
parent b9d261ce21
commit 18cc703aa2
5 changed files with 17 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/utils.py
View file

@ -6607,9 +6607,9 @@ def get_llm_provider(
or get_secret("TOGETHERAI_API_KEY")
or get_secret("TOGETHER_AI_TOKEN")
)
elif custom_llm_provider == "friendli_ai":
elif custom_llm_provider == "friendliai":
api_base = "https://inference.friendli.ai/v1"
dynamic_api_key = get_secret("FRIENDLI_AI_API_KEY") or get_secret(
dynamic_api_key = get_secret("FRIENDLIAI_API_KEY") or get_secret(
"FRIENDLI_TOKEN"
)
if api_base is not None and not isinstance(api_base, str):
@ -6660,9 +6660,9 @@ def get_llm_provider(
custom_llm_provider = "deepseek"
dynamic_api_key = get_secret("DEEPSEEK_API_KEY")
elif endpoint == "inference.friendli.ai/v1":
custom_llm_provider = "friendli_ai"
custom_llm_provider = "friendliai"
dynamic_api_key = get_secret(
"FRIENDLI_AI_API_KEY"
"FRIENDLIAI_API_KEY"
) or get_secret("FRIENDLI_TOKEN")
if api_base is not None and not isinstance(api_base, str):