scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-03 07:23:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 166

test(gemini): assert vertex lists on content chunks (#49734)

Browse source 
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Mateo Wang 2026-11-02 23:40:32 +00:00 • committed by GitHub
parent a2d27881e8
commit 4e268c2c55
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/integration/providers/test_gemini_long_decode_wire.py
View file

@ -380,7 +380,9 @@ def test_cut_chat_stream_hands_its_token_on_a_chunk_with_empty_vertex_ai_lists_a
chunks: Final = payloads(post(gateway, "chat", request_body("chat", model, True)).text, True)
carriers: Final = [chunk for chunk in chunks if "continuation_token" in json.dumps(chunk)]
assert [_vertex_lists(chunk) for chunk in carriers] == [dict(_EMPTY_VERTEX_LISTS)], chunks
assert [_vertex_lists(chunk) for chunk in chunks if chunk not in carriers] == [{}] * (len(chunks) - 1), chunks
assert [_vertex_lists(chunk) for chunk in chunks] == [
dict(_EMPTY_VERTEX_LISTS) if at(chunk, "choices", 0, "delta", "content") else {} for chunk in chunks
], chunks
metered: Final = [chunk for chunk in chunks if isinstance(chunk.get("usage"), dict)]
assert [at(chunk, "usage", "completion_tokens_details") for chunk in metered] == [
{"text_tokens": CUT.output_tokens}