mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-10 22:41:41 +00:00
Add test for gpt-5 completion verbosity parameter
This commit is contained in:
parent
190a26a4da
commit
7bb4b08162
1 changed files with 16 additions and 0 deletions
|
|
@ -1501,3 +1501,19 @@ class TestCallTypesOCR:
|
|||
|
||||
call_type = CallTypes("aocr")
|
||||
assert call_type == CallTypes.aocr
|
||||
|
||||
|
||||
def test_gpt5_completion_verbosity_parameter():
|
||||
"""Test that verbosity parameter is passed to optional_params for GPT-5 models"""
|
||||
with patch("litellm.main.openai_chat_completions.completion") as mock_completion:
|
||||
mock_completion.return_value = MagicMock()
|
||||
|
||||
litellm.completion(
|
||||
model="gpt-5",
|
||||
messages=[{"role": "user", "content": "Hi"}],
|
||||
verbosity="low",
|
||||
)
|
||||
|
||||
mock_completion.assert_called_once()
|
||||
optional_params = mock_completion.call_args.kwargs["optional_params"]
|
||||
assert optional_params.get("verbosity") == "low"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue