From 245df6d3a8bf02467a0ec64fda40561a9fdee042 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 15 Aug 2024 08:59:24 -0300 Subject: [PATCH] fix: Update config to enable caching in vector store RAG test. --- .../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 2d11bc4c337f..a2f8043af66f 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(config={"output": True}): + for result in graph.start(config={"output": {"cache": True}}): results.append(result) assert len(results) == len_results