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 = [ this.outputs = [
{ {
label: this.label, label: 'LLM Chain',
name: this.name, name: 'llmChain',
type: this.type baseClasses: [this.type, ...getBaseClasses(LLMChain)]
}, },
{ {
label: 'Output Prediction', label: 'Output Prediction',
name: 'outputPrediction', name: 'outputPrediction',
type: 'string' baseClasses: ['string']
} }
] ]
} }

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

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

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

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