Merge branch 'main' into litellm_cleanup_traceback

This commit is contained in:
Krish Dholakia 2024-06-06 16:32:08 -07:00 committed by GitHub
commit 677e0255c8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
12 changed files with 1105 additions and 560 deletions

View file

@ -7355,6 +7355,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 []