From 1150f51884cd136aeeb9f810471845aa56b74d52 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 14 Aug 2024 15:39:47 -0300 Subject: [PATCH] fix: Add config parameter to graph start function in test_vector_store_rag. --- .../initial_setup/starter_projects/test_vector_store_rag.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/tests/unit/initial_setup/starter_projects/test_vector_store_rag.py b/src/backend/tests/unit/initial_setup/starter_projects/test_vector_store_rag.py index 4fbed60066cd..d6f9241574b7 100644 --- a/src/backend/tests/unit/initial_setup/starter_projects/test_vector_store_rag.py +++ b/src/backend/tests/unit/initial_setup/starter_projects/test_vector_store_rag.py @@ -112,7 +112,7 @@ def test_vector_store_rag(ingestion_graph, rag_graph): ] for ids, graph, len_results in zip([ingestion_ids, rag_ids], [ingestion_graph, rag_graph], [5, 8]): results = [] - for result in graph.start(): + for result in graph.start(config={"output": True}): results.append(result) assert len(results) == len_results