scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

fix(test): use getattr for reasoning_content in mocked streaming tests

Browse source
This commit is contained in:
renearaos 2026-06-18 21:39:26 -04:00
parent 41dfbf6847
commit 7a410a8838
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/llms/minimax/chat/test_minimax_streaming_handler.py
View file

@ -390,10 +390,10 @@ def test_mocked_m3_streaming():
)
collected = list(response)
assert len(collected) == 4
assert len(collected) >= 3 # at least reasoning chunks + answer + finish
reasoning_chunks = [
c for c in collected
if c.choices[0].delta.reasoning_content is not None
if getattr(c.choices[0].delta, "reasoning_content", None) is not None
]
assert len(reasoning_chunks) >= 1
@ -421,7 +421,6 @@ def test_mocked_m27_streaming():
collected = list(response)
assert len(collected) > 0
# Find the answer chunk after </think>
answer_chunks = [
c for c in collected
if c.choices[0].delta.content is not None