From b40076ffe1860845674375df2c7f42b6e624383a Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Sat, 7 Mar 2026 23:26:27 +0000 Subject: [PATCH] Fixing tests --- tests/integration/test_rag_streaming_protocol.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/integration/test_rag_streaming_protocol.py b/tests/integration/test_rag_streaming_protocol.py index bebfaafa..ba492687 100644 --- a/tests/integration/test_rag_streaming_protocol.py +++ b/tests/integration/test_rag_streaming_protocol.py @@ -234,12 +234,13 @@ class TestDocumentRagStreamingProtocol: @pytest.fixture def document_rag(self, mock_embeddings_client, mock_doc_embeddings_client, - mock_streaming_prompt_client): + mock_streaming_prompt_client, mock_fetch_chunk): """Create DocumentRag instance with mocked dependencies""" return DocumentRag( embeddings_client=mock_embeddings_client, doc_embeddings_client=mock_doc_embeddings_client, prompt_client=mock_streaming_prompt_client, + fetch_chunk=mock_fetch_chunk, verbose=False )