Skip to content

Commit

Permalink
fix: Add config parameter to graph start function in test_vector_stor…
Browse files Browse the repository at this point in the history
…e_rag.
  • Loading branch information
ogabrielluiz committed Aug 14, 2024
1 parent f1cdaf6 commit 1150f51
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():
for result in graph.start(config={"output": True}):
results.append(result)

assert len(results) == len_results
Expand Down

0 comments on commit 1150f51

Please sign in to comment.