Update Chroma_Existing.ts - first OSS PR that works!!

This commit is contained in:
Pooria Arab 2023-09-07 08:00:09 -07:00 committed by GitHub
parent 86e09d726a
commit bf459bf5ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 5 deletions

View File

@ -93,18 +93,20 @@ class Chroma_Existing_VectorStores implements INode {
const credentialData = await getCredentialData(nodeData.credential ?? '', options)
const chromaApiKey = getCredentialParam('chromaApiKey', credentialData, nodeData)
const chromaMetadataFilter = nodeData.inputs?.chromaMetadataFilter;
const metadataFilter = chromaMetadataFilter ? JSON.parse(chromaMetadataFilter) : {};
const chromaMetadataFilter = nodeData.inputs?.chromaMetadataFilter
const metadataFilter = chromaMetadataFilter ? JSON.parse(chromaMetadataFilter) : {}
const obj: {
collectionName: string
url?: string
chromaApiKey?: string
metadataFilter?: any
filter?: object | undefined
} = { collectionName }
if (chromaURL) obj.url = chromaURL
if (chromaApiKey) obj.chromaApiKey = chromaApiKey
if (chromaMetadataFilter) obj.metadataFilter = metadataFilter
if (chromaMetadataFilter) {
const metadatafilter = typeof chromaMetadataFilter === 'object' ? chromaMetadataFilter : JSON.parse(chromaMetadataFilter)
obj.filter = metadatafilter
}
const vectorStore = await ChromaExtended.fromExistingCollection(embeddings, obj)