fix(get_supported_openai_params.py): fix linting error

This commit is contained in:
Krrish Dholakia 2025-02-11 19:28:37 -08:00
parent 04d07b090e
commit e5ed711d55

View file

@ -122,7 +122,9 @@ def get_supported_openai_params( # noqa: PLR0915
elif custom_llm_provider == "vertex_ai" or custom_llm_provider == "vertex_ai_beta":
if request_type == "chat_completion":
if model.startswith("meta/"):
return litellm.VertexAILlama3Config().get_supported_openai_params()
return litellm.VertexAILlama3Config().get_supported_openai_params(
model=model
)
if model.startswith("mistral"):
return litellm.MistralConfig().get_supported_openai_params(model=model)
if model.startswith("codestral"):