farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 21:00:58 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix mongodb vector database where no documents are returned

Browse Source
This commit is contained in:
Henry
2023-11-17 13:52:41 +00:00
parent 261e45d74a
commit f8c4ec28b4
4 changed files with 16 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/vectorstores/MongoDB/MongoDB_Existing.ts
+5 -7
View File
@@ -1,11 +1,10 @@
import { ICommonObject, INode, INodeData } from '../../../src/Interface'
import { Collection } from 'mongodb'
import { MongoDBAtlasVectorSearch } from 'langchain/vectorstores/mongodb_atlas'
import { Embeddings } from 'langchain/embeddings/base'
import { VectorStore } from 'langchain/vectorstores/base'
import { Document } from 'langchain/document'
import { MongoDBSearchBase } from './MongoDBSearchBase'
import { Collection } from 'mongodb'
import { MongoDBAtlasVectorSearch } from 'langchain/vectorstores/mongodb_atlas'
import { ICommonObject, INode, INodeData } from '../../../src/Interface'
class MongoDBExisting_VectorStores extends MongoDBSearchBase implements INode {
constructor() {
@@ -20,7 +19,7 @@ class MongoDBExisting_VectorStores extends MongoDBSearchBase implements INode {
return super.init(nodeData, _, options, undefined)
}
constructVectorStore(
async constructVectorStore(
embeddings: Embeddings,
collection: Collection,
indexName: string,
@@ -28,13 +27,12 @@ class MongoDBExisting_VectorStores extends MongoDBSearchBase implements INode {
embeddingKey: string,
_: Document<Record<string, any>>[] | undefined
): Promise<VectorStore> {
const mongoDBAtlasVectorSearch = new MongoDBAtlasVectorSearch(embeddings, {
return new MongoDBAtlasVectorSearch(embeddings, {
collection: collection,
indexName: indexName,
textKey: textKey,
embeddingKey: embeddingKey
})
return Promise.resolve(mongoDBAtlasVectorSearch)
}
}
packages/components/nodes/vectorstores/MongoDB/MongoDB_Upsert.ts
+9 -8
View File
@@ -1,12 +1,11 @@
import { ICommonObject, INode, INodeData } from '../../../src/Interface'
import { flatten } from 'lodash'
import { Collection } from 'mongodb'
import { Embeddings } from 'langchain/embeddings/base'
import { Document } from 'langchain/document'
import { flatten } from 'lodash'
import { VectorStore } from 'langchain/vectorstores/base'
import { MongoDBSearchBase } from './MongoDBSearchBase'
import { Collection } from 'mongodb'
import { MongoDBAtlasVectorSearch } from 'langchain/vectorstores/mongodb_atlas'
import { ICommonObject, INode, INodeData } from '../../../src/Interface'
import { MongoDBSearchBase } from './MongoDBSearchBase'
class MongoDBUpsert_VectorStores extends MongoDBSearchBase implements INode {
constructor() {
@@ -23,7 +22,7 @@ class MongoDBUpsert_VectorStores extends MongoDBSearchBase implements INode {
})
}
constructVectorStore(
async constructVectorStore(
embeddings: Embeddings,
collection: Collection,
indexName: string,
@@ -31,12 +30,14 @@ class MongoDBUpsert_VectorStores extends MongoDBSearchBase implements INode {
embeddingKey: string,
docs: Document<Record<string, any>>[]
): Promise<VectorStore> {
return MongoDBAtlasVectorSearch.fromDocuments(docs, embeddings, {
const mongoDBAtlasVectorSearch = new MongoDBAtlasVectorSearch(embeddings, {
collection: collection,
indexName: indexName,
textKey: textKey,
embeddingKey: embeddingKey
})
await mongoDBAtlasVectorSearch.addDocuments(docs)
return mongoDBAtlasVectorSearch
}
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
@@ -51,7 +52,7 @@ class MongoDBUpsert_VectorStores extends MongoDBSearchBase implements INode {
}
}
return super.init(nodeData, _, options, flattenDocs)
return super.init(nodeData, _, options, finalDocs)
}
}