add baseClasses for outputs

packages/components/nodes/chains/LLMChain/LLMChain.ts

@@ -43,14 +43,14 @@ class LLMChain_Chains implements INode {
         ]
         this.outputs = [
             {
-                label: this.label,
-                name: this.name,
-                type: this.type
+                label: 'LLM Chain',
+                name: 'llmChain',
+                baseClasses: [this.type, ...getBaseClasses(LLMChain)]
             },
             {
                 label: 'Output Prediction',
                 name: 'outputPrediction',
-                type: 'string'
+                baseClasses: ['string']
             }
         ]
     }

packages/components/src/Interface.ts

@@ -32,7 +32,7 @@ export interface INodeOptionsValue {
 export interface INodeOutputsValue {
     label: string
     name: string
-    type: string
+    baseClasses: string[]
     description?: string
 }
 

packages/ui/src/utils/genericHelper.js

@@ -61,12 +61,13 @@ export const initNode = (nodeData, newNodeId) => {
                 let baseClasses = ''
                 let type = ''
 
-                if (whitelistTypes.includes(nodeData.outputs[j].type)) {
-                    baseClasses = nodeData.outputs[j].type
-                    type = nodeData.outputs[j].type
-                } else {
-                    baseClasses = nodeData.baseClasses.join('|')
-                    type = nodeData.baseClasses.join(' | ')
+                const outputBaseClasses = nodeData.outputs[j].baseClasses ?? []
+                if (outputBaseClasses.length > 1) {
+                    baseClasses = outputBaseClasses.join('|')
+                    type = outputBaseClasses.join(' | ')
+                } else if (outputBaseClasses.length === 1) {
+                    baseClasses = outputBaseClasses[0]
+                    type = outputBaseClasses[0]
                 }
 
                 const newOutputOption = {