fixes from copilot review

Signed-off-by: Mark Kurtz <[email protected]>

Author: markurtz

src/guidellm/mock_server/handlers/chat_completions.py

@@ -251,7 +251,7 @@ async def generate_stream(stream_response):
             await stream_response.write(f"data: {json.dumps(final_chunk)}\n\n")
 
             # Send usage if requested
-            if req.stream_options and req.stream_options.get("include_usage"):
+            if req.stream_options and req.stream_options.include_usage:
                 usage_chunk = {
                     "id": completion_id,
                     "object": "chat.completion.chunk",

src/guidellm/mock_server/handlers/completions.py

@@ -251,7 +251,7 @@ async def generate_stream(stream_response):
             await stream_response.write(f"data: {json.dumps(final_chunk)}\n\n")
 
             # Send usage if requested
-            if req.stream_options and req.stream_options.get("include_usage"):
+            if req.stream_options and req.stream_options.include_usage:
                 usage_chunk = {
                     "id": completion_id,
                     "object": "text_completion",

tests/unit/mock_server/test_server.py

@@ -378,11 +378,11 @@ async def test_streaming_completions(self, mock_server_instance):
                     except json.JSONDecodeError:
                         continue
 
-                assert len(chunks) > 0
-                # Verify chunk structure
-                for chunk in chunks:
-                    assert "choices" in chunk
-                    assert len(chunk["choices"]) > 0
+            assert len(chunks) > 0
+            # Verify chunk structure
+            for chunk in chunks:
+                assert "choices" in chunk
+                assert len(chunk["choices"]) > 0
 
     @pytest.mark.smoke
     @pytest.mark.parametrize(