Fixing flaky test test_cli_can_build_and_search_index by retrying (#816)

Author: jamesbraza

tests/test_cli.py

@@ -1,8 +1,10 @@
 import os
 import sys
+import zlib
 from pathlib import Path
 
 import pytest
+from tenacity import Retrying, retry_if_exception_type, stop_after_attempt
 
 from paperqa import Docs
 from paperqa.settings import Settings
@@ -76,7 +78,13 @@ def test_cli_can_build_and_search_index(
     settings.agent.index.paper_directory = rel_path_home_to_stub_data
     settings.agent.index.index_directory = agent_index_dir
     index_name = "test"
-    build_index(index_name, stub_data_dir, settings)
+    for attempt in Retrying(
+        stop=stop_after_attempt(3),
+        # zlib.error: Error -5 while decompressing data: incomplete or truncated stream
+        retry=retry_if_exception_type(zlib.error),
+    ):
+        with attempt:
+            build_index(index_name, stub_data_dir, settings)
     result = search_query("XAI", index_name, settings)
     assert len(result) == 1
     assert isinstance(result[0][0], Docs)