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

add baseClasses for outputs

This commit is contained in:
Henry 2023-04-20 00:46:33 +01:00
parent df7a3995cd
commit 27e026d943
3 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
packages/components/nodes/chains/LLMChain/LLMChain.ts
View File

@ -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']
}
]
}

2
packages/components/src/Interface.ts
View File

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

13
packages/ui/src/utils/genericHelper.js
View File

@ -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 = {