encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-03-30 10:57:20 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: Update expected UsageMetadataChunk in LiteLLM tests

Browse Source 
Close #4680

Co-authored-by: George Weale <gweale@google.com>
PiperOrigin-RevId: 877646178
This commit is contained in:
Drew Afromsky
2026-03-02 17:10:12 -08:00
committed by Copybara-Service
parent 5770cd3776
commit dd0851ac74
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/models/test_litellm.py
+8 -4
View File
@@ -2819,7 +2819,7 @@ def test_to_litellm_role():
"content": "this is a test",
}
}
]
],
),
[TextChunk(text="this is a test")],
UsageMetadataChunk(
@@ -2877,7 +2877,9 @@ def test_to_litellm_role():
(None, "stop"),
),
(
ModelResponse(choices=[{"finish_reason": "tool_calls"}]),
ModelResponse(
choices=[{"finish_reason": "tool_calls"}],
),
[None],
UsageMetadataChunk(
prompt_tokens=0, completion_tokens=0, total_tokens=0
@@ -2962,7 +2964,8 @@ def test_to_litellm_role():
finish_reason=None,
delta=Delta(role="assistant", content="Hello"),
)
]
],
usage=None,
),
[TextChunk(text="Hello")],
None,
@@ -2977,7 +2980,8 @@ def test_to_litellm_role():
role="assistant", reasoning_content="thinking..."
),
)
]
],
usage=None,
),
[
ReasoningChunk(