scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 136

fix(__init__.py): correctly return azure_text models in models_by_provider dictionary

Browse source
This commit is contained in:
Krrish Dholakia 2024-12-21 14:56:07 -08:00
parent 7b61ffde5e
commit f6db785088
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
litellm/__init__.py
View file

@ -469,6 +469,7 @@ friendliai_models: List = []
palm_models: List = []
groq_models: List = []
azure_models: List = []
azure_text_models: List = []
anyscale_models: List = []
cerebras_models: List = []
galadriel_models: List = []
@ -481,6 +482,8 @@ def add_known_models():
open_ai_chat_completion_models.append(key)
elif value.get("litellm_provider") == "text-completion-openai":
open_ai_text_completion_models.append(key)
elif value.get("litellm_provider") == "azure_text":
azure_text_models.append(key)
elif value.get("litellm_provider") == "cohere":
cohere_models.append(key)
elif value.get("litellm_provider") == "cohere_chat":
@ -847,6 +850,7 @@ model_list = (
+ cerebras_models
+ galadriel_models
+ sambanova_models
+ azure_text_models
)
@ -893,7 +897,8 @@ models_by_provider: dict = {
"friendliai": friendliai_models,
"palm": palm_models,
"groq": groq_models,
"azure": azure_models,
"azure": azure_models + azure_text_models,
"azure_text": azure_text_models,
"anyscale": anyscale_models,
"cerebras": cerebras_models,
"galadriel": galadriel_models,