feat: Add Ollama as a provider in the proxy UI

This commit is contained in:
sha-ahammed 2024-06-05 16:48:38 +05:30
parent 499933f943
commit faa4dfe03e
3 changed files with 21 additions and 1 deletions

View file

@ -7344,6 +7344,10 @@ def get_provider_fields(custom_llm_provider: str) -> List[ProviderField]:
if custom_llm_provider == "databricks":
return litellm.DatabricksConfig().get_required_params()
elif custom_llm_provider == "ollama":
return litellm.OllamaConfig().get_required_params()
else:
return []