From 83c24f4d50a58a2ffba96730158daeddbdf58ed3 Mon Sep 17 00:00:00 2001
From: justinmann <justin@dispo.fun>
Date: Mon, 3 Jun 2024 09:08:33 -0700
Subject: [PATCH] cannot pass embedModel to MongoDBAtlasVectorStore (#887)
---
.../core/src/storage/vectorStore/MongoDBAtlasVectorStore.ts | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/packages/core/src/storage/vectorStore/MongoDBAtlasVectorStore.ts b/packages/core/src/storage/vectorStore/MongoDBAtlasVectorStore.ts
index 7d0fbafca..3ba30595c 100644
--- a/packages/core/src/storage/vectorStore/MongoDBAtlasVectorStore.ts
+++ b/packages/core/src/storage/vectorStore/MongoDBAtlasVectorStore.ts
@@ -3,6 +3,7 @@ import type { BulkWriteOptions, Collection } from "mongodb";
import { MongoClient } from "mongodb";
import type { BaseNode } from "../../Node.js";
import { MetadataMode } from "../../Node.js";
+import { BaseEmbedding } from "../../embeddings/types.js";
import {
VectorStoreBase,
type MetadataFilters,
@@ -44,9 +45,10 @@ export class MongoDBAtlasVectorSearch
init: Partial<MongoDBAtlasVectorSearch> & {
dbName: string;
collectionName: string;
+ embedModel?: BaseEmbedding;
},
) {
- super();
+ super(init.embedModel);
if (init.mongodbClient) {
this.mongodbClient = init.mongodbClient;
} else {
@@ -100,7 +102,7 @@ export class MongoDBAtlasVectorSearch
}
async delete(refDocId: string, deleteOptions?: any): Promise<void> {
- await this.collection.deleteOne(
+ await this.collection.deleteMany(
{
[`${this.metadataKey}.ref_doc_id`]: refDocId,
},
--
GitLab