test: update tests usage -> gneration.usage

markstur · markstur · commit 5b737abefc4a · 2026-04-30T17:27:59.000-07:00
More new tests need fixing after rebase.

Signed-off-by: Mark Sturdevant &lt;mark.sturdevant@ibm.com&gt;
diff --git a/test/cli/test_serve_integration.py b/test/cli/test_serve_integration.py
@@ -196,7 +196,7 @@ def test_streaming_content_chunks(self, client, mock_module):
     def test_streaming_with_usage_field(self, client, mock_module):
         """Test streaming response includes usage when stream_options.include_usage=True."""
         mock_output = ModelOutputThunk("Response")
-        mock_output.usage = {
+        mock_output.generation.usage = {
             "prompt_tokens": 10,
             "completion_tokens": 5,
             "total_tokens": 15,
@@ -346,7 +346,7 @@ def test_tool_calls_with_usage_info(self, client, mock_module):
                 name="get_weather", func=mock_tool, args={"location": "Paris"}
             )
         }
-        mock_output.usage = {
+        mock_output.generation.usage = {
             "prompt_tokens": 50,
             "completion_tokens": 20,
             "total_tokens": 70,
@@ -481,7 +481,7 @@ def test_streaming_tool_calls_with_usage(self, client, mock_module):
                 name="get_weather", func=mock_tool, args={"location": "Paris"}
             )
         }
-        mock_output.usage = {
+        mock_output.generation.usage = {
             "prompt_tokens": 30,
             "completion_tokens": 15,
             "total_tokens": 45,
diff --git a/test/cli/test_serve_tool_calling.py b/test/cli/test_serve_tool_calling.py
@@ -275,7 +275,7 @@ async def test_tool_calls_with_usage_info(self, mock_module, sample_tool_request
                 name="get_weather", func=mock_tool, args={"location": "Paris"}
             )
         }
-        mock_output.usage = {
+        mock_output.generation.usage = {
             "prompt_tokens": 50,
             "completion_tokens": 20,
             "total_tokens": 70,

Original file line number	Diff line number	Diff line change
`@@ -275,7 +275,7 @@ async def test_tool_calls_with_usage_info(self, mock_module, sample_tool_request`
`275`	`275`	`name="get_weather", func=mock_tool, args={"location": "Paris"}`
`276`	`276`	`)`
`277`	`277`	`}`
`278`		`- mock_output.usage = {`
	`278`	`+ mock_output.generation.usage = {`
`279`	`279`	`"prompt_tokens": 50,`
`280`	`280`	`"completion_tokens": 20,`
`281`	`281`	`"total_tokens": 70,`