scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-05 07:37:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 360

test(databricks): assert the full usage payload on tool-only responses

Browse source
This commit is contained in:
Prasanna Jain 2026-10-28 10:27:01 -04:00
parent eca8b429f5
commit c7fa3584ce
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/unit/llms/databricks/chat/test_databricks_chat_transformation.py
View file

@ -2464,7 +2464,8 @@ def test_transform_response_keeps_a_tool_only_message_without_content() -> None:
completion: Final = ChatCompletion.model_validate(result.model_dump())
assert completion.choices[0].finish_reason == "tool_calls"
assert completion.choices[0].message.model_dump(exclude_none=True) == {"role": "assistant", "tool_calls": [tool_call]}
assert completion.usage is not None and completion.usage.total_tokens == 8
assert completion.usage is not None
assert (completion.usage.prompt_tokens, completion.usage.completion_tokens, completion.usage.total_tokens) == (5, 3, 8)
_DATABRICKS_TOOLS_WITH_OPENAI_TOOL_SEARCH: Final = [