fix: utils.py

fix supported openai params
This commit is contained in:
Krrish Dholakia 2024-07-27 22:03:40 -07:00
parent b854d2100c
commit 1ac42d8464
2 changed files with 62 additions and 1 deletions

View file

@ -4216,7 +4216,7 @@ def get_supported_openai_params(
if model.startswith("meta/"):
return litellm.VertexAILlama3Config().get_supported_openai_params()
return litellm.VertexAIConfig().get_supported_openai_params()
return litellm.VertexGeminiConfig().get_supported_openai_params()
elif request_type == "embeddings":
return litellm.VertexAITextEmbeddingConfig().get_supported_openai_params()
elif custom_llm_provider == "vertex_ai_beta":