scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(vertex_ai): move stray Gemma streaming tests to tests/unit so CI coverage passes (#43422)

Browse source 
Co-authored-by: Krrish Dholakia <krrishdholakia@berri.ai>
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
devin-ai-integration[bot] 2026-09-27 05:02:44 +00:00 • committed by GitHub
parent 9a0ff249d5
commit c1f761eba5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/unit/llms/vertex_ai/vertex_gemma_models/test_vertex_gemma_transformation.py
View file

@ -1332,7 +1332,7 @@ def test_sync_gemma_stream(_gemma_cached_access_token):
def handle(request):
captured["body"] = json.loads(request.content)
return httpx.Response(200, json=_make_gemma_vertex_response(content="READY", total_tokens=15))
return httpx.Response(200, json=_make_gemma_vertex_response(content="READY"))
stream = litellm.completion(
model="vertex_ai/gemma/test-model",
@ -1362,7 +1362,7 @@ async def test_async_gemma_responses_stream(_gemma_cached_access_token):
def handle(request):
captured["body"] = json.loads(request.content)
return httpx.Response(200, json=_make_gemma_vertex_response(content="READY", total_tokens=15))
return httpx.Response(200, json=_make_gemma_vertex_response(content="READY"))
response = await litellm.aresponses(
model="vertex_ai/gemma/test-model",
@ -1380,4 +1380,4 @@ async def test_async_gemma_responses_stream(_gemma_cached_access_token):
assert "READY" in "".join(event.delta for event in events if isinstance(event, OutputTextDeltaEvent))
assert isinstance(events[-1], ResponseCompletedEvent)
assert events[-1].response.usage is not None
assert events[-1].response.usage.total_tokens == 15
assert events[-1].response.usage.total_tokens == 114