Skip to content

Commit 77e9b44

Browse files
aclark4lifeCopilot
andauthored
Update libs/langchain-mongodb/tests/utils.py
Co-authored-by: Copilot <[email protected]>
1 parent 57cf687 commit 77e9b44

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

libs/langchain-mongodb/tests/utils.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,12 @@ def bulk_embed_and_insert_texts(
6565
) -> List:
6666
"""Patched insert_texts that waits for data to be indexed before returning"""
6767
ids_inserted = bulk_embed_and_insert_texts(
68-
self.embedding, self.collection, self._embedding_field_config, texts, metadatas, ids
68+
embedding_func=self._embedding.embed_documents,
69+
collection=self.collection,
70+
embedding_field_config=self._embedding_field_config,
71+
texts=texts,
72+
metadatas=metadatas,
73+
ids=ids,
6974
)
7075
n_docs = self.collection.count_documents({})
7176
start = monotonic()

0 commit comments

Comments
 (0)