From a0f9022028235a3609f2377c60c66762e2b70a1c Mon Sep 17 00:00:00 2001 From: dishaprakash <57954147+dishaprakash@users.noreply.github.com> Date: Tue, 27 Jan 2026 07:52:32 +0530 Subject: [PATCH 1/2] Add batching to document insertion in vector store Implement batching for adding documents to vector store. --- samples/langchain_on_vertexai/create_embeddings.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/samples/langchain_on_vertexai/create_embeddings.py b/samples/langchain_on_vertexai/create_embeddings.py index 370d8262..fdb3b50e 100644 --- a/samples/langchain_on_vertexai/create_embeddings.py +++ b/samples/langchain_on_vertexai/create_embeddings.py @@ -114,7 +114,12 @@ async def grant_select(engine): ) ids = [str(uuid.uuid4()) for i in range(len(docs))] - await vector_store.aadd_documents(docs, ids=ids) + batch_size = 100 + for i in range(0, len(docs), batch_size): + batch_docs = docs[i : i + batch_size] + batch_ids = ids[i : i + batch_size] + print(f"Adding batch {i // batch_size + 1} ({len(batch_docs)} documents)...") + await vector_store.aadd_documents(batch_docs, ids=batch_ids) async def main(): From 785c01fec3d9dc8eeb9c55cdbd8c8c7549c1bcdb Mon Sep 17 00:00:00 2001 From: dishaprakash <57954147+dishaprakash@users.noreply.github.com> Date: Tue, 27 Jan 2026 07:54:27 +0530 Subject: [PATCH 2/2] Update package versions in requirements.txt Updated version of google-cloud-aiplatform and langchain packages. --- samples/langchain_on_vertexai/requirements.txt | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/samples/langchain_on_vertexai/requirements.txt b/samples/langchain_on_vertexai/requirements.txt index 064bf76a..5f476db6 100644 --- a/samples/langchain_on_vertexai/requirements.txt +++ b/samples/langchain_on_vertexai/requirements.txt @@ -1,5 +1,5 @@ -google-cloud-aiplatform[reasoningengine,langchain]==1.121.0 +google-cloud-aiplatform[reasoningengine]==1.130.0 google-cloud-resource-manager==1.14.2 -langchain-community==0.3.31 -langchain-google-cloud-sql-pg==0.14.1 -langchain-google-vertexai==2.1.2 +langchain-community==0.4.1 +langchain-google-cloud-sql-pg==0.15.0 +langchain-google-vertexai==3.2.1