Add kwargs to bulk_embed_and_insert_texts

Author: aclark4life

libs/langchain-mongodb/langchain_mongodb/vectorstores.py

@@ -364,11 +364,24 @@ def add_texts(
                 if (j + 1) % batch_size == 0 or size >= 47_000_000:
                     if ids:
                         batch_res = bulk_embed_and_insert_texts(
-                            texts_batch, metadatas_batch, ids[i : j + 1]
+                            embedding_model=self._embedding,
+                            collection=self._collection,
+                            text_key=self._text_key,
+                            embedding_key=self._embedding_key,
+                            metadata_key=self._metadata_key,
+                            texts=texts_batch,
+                            metadatas=metadatas_batch,
+                            ids=ids[i : j + 1],
                         )
                     else:
                         batch_res = bulk_embed_and_insert_texts(
-                            texts_batch, metadatas_batch
+                            embedding_model=self._embedding,
+                            collection=self._collection,
+                            text_key=self._text_key,
+                            embedding_key=self._embedding_key,
+                            metadata_key=self._metadata_key,
+                            texts=texts_batch,
+                            metadatas=metadatas_batch,
                         )
                     result_ids.extend(batch_res)
                     texts_batch = []
@@ -378,10 +391,25 @@ def add_texts(
         if texts_batch:
             if ids:
                 batch_res = bulk_embed_and_insert_texts(
-                    texts_batch, metadatas_batch, ids[i : j + 1]
+                    embedding_model=self._embedding,
+                    collection=self._collection,
+                    text_key=self._text_key,
+                    embedding_key=self._embedding_key,
+                    metadata_key=self._metadata_key,
+                    texts=texts_batch,
+                    metadatas=metadatas_batch,
+                    ids=ids[i : j + 1],
                 )
             else:
-                batch_res = bulk_embed_and_insert_texts(texts_batch, metadatas_batch)
+                batch_res = bulk_embed_and_insert_texts(
+                    embedding_model=self._embedding,
+                    collection=self._collection,
+                    text_key=self._text_key,
+                    embedding_key=self._embedding_key,
+                    metadata_key=self._metadata_key,
+                    texts=texts_batch,
+                    metadatas=metadatas_batch,
+                )
             result_ids.extend(batch_res)
         return result_ids