fix

Author: cbornet

langchain/tests/conftest.py

@@ -1,14 +1,15 @@
 import json
 import logging
 import os
+import time
 from collections.abc import Iterator
 from pathlib import Path
 from typing import Literal
 
 import pytest
 import urllib3
 import vectorize_client as v
-from vectorize_client import ApiClient
+from vectorize_client import ApiClient, RetrieveDocumentsRequest
 
 
 @pytest.fixture(scope="session")
@@ -147,6 +148,24 @@ def pipeline_id(api_client: v.ApiClient, org_id: str) -> Iterator[str]:
     pipeline_id = pipeline_response.data.id
     logging.info("Created pipeline %s", pipeline_id)
 
+    # Wait for the pipeline to be created
+    request = RetrieveDocumentsRequest(
+        question="query",
+        num_results=2,
+    )
+    start = time.time()
+    while True:
+        response = pipelines.retrieve_documents(
+            org_id, pipeline_id, request
+        )
+        docs = response.documents
+        if len(docs) == 2:
+            break
+        if time.time() - start > 180:
+            msg = "Docs not retrieved in time"
+            raise RuntimeError(msg)
+        time.sleep(1)
+
     yield pipeline_id
 
     try:

langchain/tests/test_langchain_integration_tests.py

@@ -1,3 +1,4 @@
+import time
 from typing import Literal
 
 import pytest
@@ -20,6 +21,7 @@ def setup(
         cls.api_token = api_token
         cls.org_id = org_id
         cls.pipeline_id = pipeline_id
+        time.sleep()
 
     @property
     def retriever_constructor(self) -> type[VectorizeRetriever]: