Add tests in CI

Author: cbornet

langchain/tests/test_retrievers.py

@@ -1,6 +1,7 @@
 import json
 import logging
 import os
+import time
 from collections.abc import Iterator
 from pathlib import Path
 from typing import Literal
@@ -170,8 +171,15 @@ def test_retrieve_init_args(
         pipeline_id=pipeline_id,
         num_results=2,
     )
-    docs = retriever.invoke(input="What are you?")
-    assert len(docs) == 2
+    start = time.time()
+    while True:
+        docs = retriever.invoke(input="What are you?")
+        if len(docs) == 2:
+            break
+        if time.time() - start > 180:
+            msg = "Docs not retrieved in time"
+            raise RuntimeError(msg)
+        time.sleep(1)
 
 
 def test_retrieve_invoke_args(
@@ -181,10 +189,17 @@ def test_retrieve_invoke_args(
     pipeline_id: str,
 ) -> None:
     retriever = VectorizeRetriever(environment=environment, api_token=api_token)
-    docs = retriever.invoke(
-        input="What are you?",
-        organization=org_id,
-        pipeline_id=pipeline_id,
-        num_results=2,
-    )
-    assert len(docs) == 2
+    start = time.time()
+    while True:
+        docs = retriever.invoke(
+            input="What are you?",
+            organization=org_id,
+            pipeline_id=pipeline_id,
+            num_results=2,
+        )
+        if len(docs) == 2:
+            break
+        if time.time() - start > 180:
+            msg = "Docs not retrieved in time"
+            raise RuntimeError(msg)
+        time.sleep(1)