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

Fix integration tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-01-06 21:36:34 +00:00
parent 35ee0141f7
commit 55a5c8b546
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/test_document_rag_streaming_integration.py
View file

@ -127,7 +127,7 @@ class TestDocumentRagStreaming:
# Act - Streaming
streaming_chunks = []
async def collect(chunk):
async def collect(chunk, end_of_stream):
streaming_chunks.append(chunk)
streaming_result = await document_rag_streaming.query(

2
tests/integration/test_graph_rag_streaming_integration.py
View file

@ -138,7 +138,7 @@ class TestGraphRagStreaming:
# Act - Streaming
streaming_chunks = []
async def collect(chunk):
async def collect(chunk, end_of_stream):
streaming_chunks.append(chunk)
streaming_result = await graph_rag_streaming.query(