mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-09 03:18:44 +00:00
test: fix test
This commit is contained in:
parent
f93d577967
commit
b18e958dda
1 changed files with 2 additions and 1 deletions
|
|
@ -143,7 +143,8 @@ def test_all_model_configs():
|
|||
)
|
||||
|
||||
assert (
|
||||
"max_completion_tokens" in VertexAILlama3Config().get_supported_openai_params()
|
||||
"max_completion_tokens"
|
||||
in VertexAILlama3Config().get_supported_openai_params(model="llama3")
|
||||
)
|
||||
assert VertexAILlama3Config().map_openai_params(
|
||||
{"max_completion_tokens": 10}, {}, "llama3", drop_params=False
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue