Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Enable Filters in MongoDB Vector Search Retriever (#3333) 

* added: support to metadata filter using mongodb vector search retriever

* fix: format lint
This commit is contained in:
Anthony Bryan Gavilan Vinces 2024-10-16 10:35:37 -05:00 committed by GitHub
parent 6c3541915b
commit 235fcfeb40
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
packages/components/nodes/vectorstores/MongoDBAtlas/MongoDBAtlas.ts
View File

@ -86,6 +86,13 @@ class MongoDBAtlas_VectorStores implements INode {
additionalParams: true,
optional: true
},
{
label: 'Mongodb Metadata Filter',
name: 'mongoMetadataFilter',
type: 'json',
optional: true,
additionalParams: true
},
{
label: 'Top K',
name: 'topK',
@ -164,9 +171,12 @@ class MongoDBAtlas_VectorStores implements INode {
let textKey = nodeData.inputs?.textKey as string
let embeddingKey = nodeData.inputs?.embeddingKey as string
const embeddings = nodeData.inputs?.embeddings as Embeddings
const mongoMetadataFilter = nodeData.inputs?.mongoMetadataFilter as object
let mongoDBConnectUrl = getCredentialParam('mongoDBConnectUrl', credentialData, nodeData)
const filter: MongoDBAtlasVectorSearch['FilterType'] = {}
const mongoClient = await getMongoClient(mongoDBConnectUrl)
try {
const collection = mongoClient.db(databaseName).collection(collectionName)
@ -181,7 +191,20 @@ class MongoDBAtlas_VectorStores implements INode {
embeddingKey
}) as unknown as VectorStore
return resolveVectorStoreOrRetriever(nodeData, vectorStore)
if (mongoMetadataFilter) {
const metadataFilter = typeof mongoMetadataFilter === 'object' ? mongoMetadataFilter : JSON.parse(mongoMetadataFilter)
for (const key in metadataFilter) {
filter.preFilter = {
...filter.preFilter,
[key]: {
$eq: metadataFilter[key]
}
}
}
}
return resolveVectorStoreOrRetriever(nodeData, vectorStore, filter)
} catch (e) {
throw new Error(e)
}