fix: add delay to langchain vs tests

GitOrigin-RevId: 2e5de5db1780fe7eeafcbc974e750280a12f6835
pathwaycom · Mar 27, 2024 · 8c2581e · 8c2581e
1 parent cb06e04
commit 8c2581e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pathway/xpacks/llm/tests/test_vector_store.py b/python/pathway/xpacks/llm/tests/test_vector_store.py
@@ -158,7 +158,7 @@ def pathway_server(tmp_path, host, port):
     thread.join()
 
 
-@pytest.mark.flaky(reruns=4)
+@pytest.mark.flaky(reruns=4, reruns_delay=20)
 @xfail_on_multiple_threads
 def test_vector_store_with_langchain(tmp_path: pathlib.Path) -> None:
     host = "0.0.0.0"
@@ -171,7 +171,7 @@ def test_vector_store_with_langchain(tmp_path: pathlib.Path) -> None:
     p.start()
     time.sleep(5)
     client = VectorStoreClient(host=host, port=port)
-    MAX_ATTEMPTS = 7
+    MAX_ATTEMPTS = 8
     attempts = 0
     output = []
     while attempts < MAX_ATTEMPTS: