diff --git a/libs/partners/chroma/langchain_chroma/vectorstores.py b/libs/partners/chroma/langchain_chroma/vectorstores.py index 945057f4adb22..0dfbd8bab1944 100644 --- a/libs/partners/chroma/langchain_chroma/vectorstores.py +++ b/libs/partners/chroma/langchain_chroma/vectorstores.py @@ -997,7 +997,7 @@ def update_documents(self, ids: List[str], documents: List[Document]) -> None: embeddings = self._embedding_function.embed_documents(text) if hasattr( - self._collection._client, "max_batch_size" + self._collection._client, "get_max_batch_size" ): # for Chroma 0.4.10 and above from chromadb.utils.batch_utils import create_batches @@ -1070,7 +1070,7 @@ def from_texts( if ids is None: ids = [str(uuid.uuid4()) for _ in texts] if hasattr( - chroma_collection._client, "max_batch_size" + chroma_collection._client, "get_max_batch_size" ): # for Chroma 0.4.10 and above from chromadb.utils.batch_utils import create_batches