diff --git a/src/langchain_google_spanner/vector_store.py b/src/langchain_google_spanner/vector_store.py index b5084e80..4be49a05 100644 --- a/src/langchain_google_spanner/vector_store.py +++ b/src/langchain_google_spanner/vector_store.py @@ -672,6 +672,13 @@ def add_texts( embeds = self._embedding_service.embed_documents(texts_list) + if len(embeds) != number_of_records: + raise ValueError( + "Number of embeddings should equal the number of documents." + "Try reducing the batch size when adding the documents." + f" Expected: {number_of_records}, but found {len(embeds)}" + ) + if metadatas is None: metadatas = [{} for _ in texts]