Merge branch 'BerriAI:main' into main

This commit is contained in:
Raymond1415926 2024-06-06 10:12:20 -07:00 committed by GitHub
commit 38b44c301a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
19 changed files with 237 additions and 141 deletions

View file

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