From 4921893bd6128c7376338a5ae8f67feca15cce7b Mon Sep 17 00:00:00 2001 From: James Braza Date: Fri, 13 Sep 2024 08:58:02 -0700 Subject: [PATCH] Also adjusted test_get_directory_index_w_manifest --- tests/test_agents.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_agents.py b/tests/test_agents.py index dd75d58a5..c94af42f4 100644 --- a/tests/test_agents.py +++ b/tests/test_agents.py @@ -61,8 +61,8 @@ async def test_get_directory_index_w_manifest( "title", "year", ], "Incorrect fields in index" - # paper.pdf + flag_day.html + bates.txt + obama.txt - assert len(await index.index_files) == 4, "Incorrect number of index files" + # paper.pdf + empty.txt + flag_day.html + bates.txt + obama.txt + assert len(await index.index_files) == 5, "Incorrect number of index files" results = await index.query(query="who is Frederick Bates?") top_result = next(iter(results[0].docs.values())) paper_dir = cast(Path, agent_test_settings.paper_directory)