From 9cf7dcb5782995ad405cd205750a4993fc8dba7b Mon Sep 17 00:00:00 2001 From: cybermaggedon Date: Thu, 2 Jul 2026 11:14:54 +0100 Subject: [PATCH] fix: wire variant into remaining streaming integration test mocks (#1013) Three more streaming tests were missing _wire_variant after the async for change in create_completion_stream. --- .../integration/test_text_completion_streaming_integration.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/integration/test_text_completion_streaming_integration.py b/tests/integration/test_text_completion_streaming_integration.py index caa3ec9c..7d514522 100644 --- a/tests/integration/test_text_completion_streaming_integration.py +++ b/tests/integration/test_text_completion_streaming_integration.py @@ -270,6 +270,7 @@ class TestTextCompletionStreaming: processor.generate_content_stream = Processor.generate_content_stream.__get__( processor, Processor ) + _wire_variant(processor) # Act & Assert with pytest.raises(Exception) as exc_info: @@ -307,6 +308,7 @@ class TestTextCompletionStreaming: processor.generate_content_stream = Processor.generate_content_stream.__get__( processor, Processor ) + _wire_variant(processor) # Act chunks = [] @@ -330,6 +332,7 @@ class TestTextCompletionStreaming: processor.generate_content_stream = Processor.generate_content_stream.__get__( processor, Processor ) + _wire_variant(processor) system_prompt = "You are an expert." user_prompt = "Explain quantum physics."