mirror of
https://github.com/BerriAI/litellm.git
synced 2026-08-28 05:25:59 +00:00
test(vertex_ai): parametrize placeholder scoping across gemini-3 model variants
This commit is contained in:
parent
d5af42717e
commit
db50e123d5
1 changed files with 36 additions and 0 deletions
|
|
@ -1,5 +1,7 @@
|
|||
import base64
|
||||
|
||||
import pytest
|
||||
|
||||
from litellm.litellm_core_utils.prompt_templates.factory import (
|
||||
convert_to_gemini_tool_call_result,
|
||||
)
|
||||
|
|
@ -1052,6 +1054,40 @@ def test_parallel_tool_call_history_replayed_through_full_message_conversion():
|
|||
assert "thoughtSignature" not in model_parts[2]
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"model",
|
||||
[
|
||||
"gemini-3-pro-preview",
|
||||
"gemini-3-flash-preview",
|
||||
"gemini-3.1-pro-preview",
|
||||
"gemini-3.6-flash",
|
||||
"gemini-3.7-flash",
|
||||
"vertex_ai/gemini-3.7-flash",
|
||||
"gemini/gemini-3.7-flash",
|
||||
],
|
||||
)
|
||||
def test_placeholder_scoped_to_first_call_across_gemini_3_variants(model):
|
||||
"""Every gemini-3 family member, bare or provider-prefixed, gets one placeholder at most."""
|
||||
from litellm.litellm_core_utils.prompt_templates.factory import (
|
||||
_get_dummy_thought_signature,
|
||||
convert_to_gemini_tool_call_invoke,
|
||||
)
|
||||
|
||||
gemini_parts = convert_to_gemini_tool_call_invoke(
|
||||
{
|
||||
"role": "assistant",
|
||||
"content": None,
|
||||
"tool_calls": _parallel_tool_calls(None, None, None),
|
||||
},
|
||||
model=model,
|
||||
)
|
||||
|
||||
assert len(gemini_parts) == 3
|
||||
assert gemini_parts[0]["thoughtSignature"] == _get_dummy_thought_signature()
|
||||
assert "thoughtSignature" not in gemini_parts[1]
|
||||
assert "thoughtSignature" not in gemini_parts[2]
|
||||
|
||||
|
||||
def test_signed_text_part_survives_alongside_unsigned_parallel_tool_calls():
|
||||
"""gemini-2.5 history with a signed text part and parallel unsigned calls: the text keeps its real
|
||||
signature, only the first call gets the placeholder, and the siblings stay bare."""
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue