refactor Chroma configuration handling

Author: HenryHengZJ

packages/components/nodes/vectorstores/Chroma/Chroma.ts

@@ -100,6 +100,43 @@ class Chroma_VectorStores implements INode {
         ]
     }
 
+    private _buildChromaConfig(
+        collectionName: string,
+        chromaURL: string | undefined,
+        chromaApiKey: string | undefined,
+        chromaTenant: string | undefined,
+        chromaDatabase: string | undefined
+    ): ICommonObject {
+        const obj: {
+            collectionName: string
+            url?: string
+            chromaCloudAPIKey?: string
+            clientParams?: {
+                host?: string
+                port?: number
+                ssl?: boolean
+                tenant?: string
+                database?: string
+            }
+        } = { collectionName }
+
+        if (chromaURL) obj.url = chromaURL
+        if (chromaApiKey) obj.chromaCloudAPIKey = chromaApiKey
+
+        if (chromaTenant || chromaDatabase) {
+            obj.clientParams = {}
+            if (chromaTenant) obj.clientParams.tenant = chromaTenant
+            if (chromaDatabase) obj.clientParams.database = chromaDatabase
+            if (chromaApiKey) {
+                obj.clientParams.host = 'api.trychroma.com'
+                obj.clientParams.port = 8000
+                obj.clientParams.ssl = true
+            }
+        }
+
+        return obj
+    }
+
     //@ts-ignore
     vectorStoreMethods = {
         async upsert(nodeData: INodeData, options: ICommonObject): Promise<Partial<IndexingResult>> {
@@ -122,32 +159,7 @@ class Chroma_VectorStores implements INode {
                 }
             }
 
-            const obj: {
-                collectionName: string
-                url?: string
-                chromaCloudAPIKey?: string
-                clientParams?: {
-                    host?: string
-                    port?: number
-                    ssl?: boolean
-                    tenant?: string
-                    database?: string
-                }
-            } = { collectionName }
-
-            if (chromaURL) obj.url = chromaURL
-            if (chromaApiKey) obj.chromaCloudAPIKey = chromaApiKey
-
-            if (chromaTenant || chromaDatabase) {
-                obj.clientParams = {}
-                if (chromaTenant) obj.clientParams.tenant = chromaTenant
-                if (chromaDatabase) obj.clientParams.database = chromaDatabase
-                if (chromaApiKey) {
-                    obj.clientParams.host = 'api.trychroma.com'
-                    obj.clientParams.port = 8000
-                    obj.clientParams.ssl = true
-                }
-            }
+            const obj = this._buildChromaConfig(collectionName, chromaURL, chromaApiKey, chromaTenant, chromaDatabase)
 
             try {
                 if (recordManager) {
@@ -183,32 +195,7 @@ class Chroma_VectorStores implements INode {
             const chromaTenant = getCredentialParam('chromaTenant', credentialData, nodeData)
             const chromaDatabase = getCredentialParam('chromaDatabase', credentialData, nodeData)
 
-            const obj: {
-                collectionName: string
-                url?: string
-                chromaCloudAPIKey?: string
-                clientParams?: {
-                    host?: string
-                    port?: number
-                    ssl?: boolean
-                    tenant?: string
-                    database?: string
-                }
-            } = { collectionName }
-
-            if (chromaURL) obj.url = chromaURL
-            if (chromaApiKey) obj.chromaCloudAPIKey = chromaApiKey
-
-            if (chromaTenant || chromaDatabase) {
-                obj.clientParams = {}
-                if (chromaTenant) obj.clientParams.tenant = chromaTenant
-                if (chromaDatabase) obj.clientParams.database = chromaDatabase
-                if (chromaApiKey) {
-                    obj.clientParams.host = 'api.trychroma.com'
-                    obj.clientParams.port = 8000
-                    obj.clientParams.ssl = true
-                }
-            }
+            const obj = this._buildChromaConfig(collectionName, chromaURL, chromaApiKey, chromaTenant, chromaDatabase)
 
             try {
                 if (recordManager) {
@@ -249,33 +236,7 @@ class Chroma_VectorStores implements INode {
         const chromaDatabase = getCredentialParam('chromaDatabase', credentialData, nodeData)
         const chromaMetadataFilter = nodeData.inputs?.chromaMetadataFilter
 
-        const obj: {
-            collectionName: string
-            url?: string
-            chromaCloudAPIKey?: string
-            clientParams?: {
-                host?: string
-                port?: number
-                ssl?: boolean
-                tenant?: string
-                database?: string
-            }
-            filter?: object | undefined
-        } = { collectionName }
-
-        if (chromaURL) obj.url = chromaURL
-        if (chromaApiKey) obj.chromaCloudAPIKey = chromaApiKey
-
-        if (chromaTenant || chromaDatabase) {
-            obj.clientParams = {}
-            if (chromaTenant) obj.clientParams.tenant = chromaTenant
-            if (chromaDatabase) obj.clientParams.database = chromaDatabase
-            if (chromaApiKey) {
-                obj.clientParams.host = 'api.trychroma.com'
-                obj.clientParams.port = 8000
-                obj.clientParams.ssl = true
-            }
-        }
+        const obj: ICommonObject = this._buildChromaConfig(collectionName, chromaURL, chromaApiKey, chromaTenant, chromaDatabase)
 
         if (chromaMetadataFilter) {
             const metadatafilter = typeof chromaMetadataFilter === 'object' ? chromaMetadataFilter : parseJsonBody(chromaMetadataFilter)