feat(add_texts): ensure index readiness

langchain-ai · Nov 11, 2024 · 6cd571f · 6cd571f
1 parent f9e0f24
commit 6cd571f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libs/partners/pinecone/langchain_pinecone/vectorstores.py b/libs/partners/pinecone/langchain_pinecone/vectorstores.py
@@ -294,6 +294,8 @@ def add_texts(
         for metadata, text in zip(metadatas, texts):
             metadata[self._text_key] = text
 
+        initial_vector_count = self._get_vector_count()
+
         # For loops to avoid memory issues and optimize when using HTTP based embeddings
         # The first loop runs the embeddings, it benefits when using OpenAI embeddings
         # The second loops runs the pinecone upsert asynchronously.
@@ -323,6 +325,8 @@ def add_texts(
                     **kwargs,
                 )
 
+        asyncio.run(self._wait_on_index(len(texts) + initial_vector_count))
+
         return ids
 
     def similarity_search_with_score(