From 55a5c8b546c0d17be8900d8dc85d81670549a22c Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 6 Jan 2026 21:36:34 +0000 Subject: [PATCH] Fix integration tests --- tests/integration/test_document_rag_streaming_integration.py | 2 +- tests/integration/test_graph_rag_streaming_integration.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/test_document_rag_streaming_integration.py b/tests/integration/test_document_rag_streaming_integration.py index e1584aeb..84061add 100644 --- a/tests/integration/test_document_rag_streaming_integration.py +++ b/tests/integration/test_document_rag_streaming_integration.py @@ -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( diff --git a/tests/integration/test_graph_rag_streaming_integration.py b/tests/integration/test_graph_rag_streaming_integration.py index f7e306ea..47dd84b6 100644 --- a/tests/integration/test_graph_rag_streaming_integration.py +++ b/tests/integration/test_graph_rag_streaming_integration.py @@ -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(