Skip to content

Commit

Permalink
fix: Update config to enable caching in vector store RAG test.
Browse files Browse the repository at this point in the history
  • Loading branch information
ogabrielluiz committed Aug 15, 2024
1 parent 68aebef commit 245df6d
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 245df6d

Please sign in to comment.