feat - add azure ai studio models on litellm ui

This commit is contained in:
Ishaan Jaff 2024-06-12 20:28:16 -07:00
parent 2a32817963
commit e128dc4e1f
3 changed files with 24 additions and 0 deletions

View file

@ -7221,6 +7221,9 @@ def get_provider_fields(custom_llm_provider: str) -> List[ProviderField]:
elif custom_llm_provider == "ollama":
return litellm.OllamaConfig().get_required_params()
elif custom_llm_provider == "azure_ai":
return litellm.AzureAIStudioConfig().get_required_params()
else:
return []