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