apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-03 15:01:00 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

fix: wire variant into remaining streaming integration test mocks (#1013)

Browse source 
Three more streaming tests were missing _wire_variant after the
async for change in create_completion_stream.
This commit is contained in:
cybermaggedon 2026-07-02 11:14:54 +01:00 • committed by GitHub
parent 6c9a545a06
commit 9cf7dcb578
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/integration/test_text_completion_streaming_integration.py
View file

@ -270,6 +270,7 @@ class TestTextCompletionStreaming:
processor.generate_content_stream = Processor.generate_content_stream.__get__( processor.generate_content_stream = Processor.generate_content_stream.__get__(
processor, Processor processor, Processor
) )
_wire_variant(processor)
# Act & Assert # Act & Assert
with pytest.raises(Exception) as exc_info: with pytest.raises(Exception) as exc_info:
@ -307,6 +308,7 @@ class TestTextCompletionStreaming:
processor.generate_content_stream = Processor.generate_content_stream.__get__( processor.generate_content_stream = Processor.generate_content_stream.__get__(
processor, Processor processor, Processor
) )
_wire_variant(processor)
# Act # Act
chunks = [] chunks = []
@ -330,6 +332,7 @@ class TestTextCompletionStreaming:
processor.generate_content_stream = Processor.generate_content_stream.__get__( processor.generate_content_stream = Processor.generate_content_stream.__get__(
processor, Processor processor, Processor
) )
_wire_variant(processor)
system_prompt = "You are an expert." system_prompt = "You are an expert."
user_prompt = "Explain quantum physics." user_prompt = "Explain quantum physics."