youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 09:58:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: question classify output

Browse Source
This commit is contained in:
Joel
2024-03-13 15:43:03 +08:00
parent 1f41521c21
commit 25a11c5bb7
3 changed files with 115 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
web/app/components/workflow/nodes/_base/components/variable/utils.ts
View File

@ -2,7 +2,14 @@ import type { CodeNodeType } from '../../../code/types'
import { BlockEnum, InputVarType, VarType } from '@/app/components/workflow/types'
import type { StartNodeType } from '@/app/components/workflow/nodes/start/types'
import type { NodeOutPutVar } from '@/app/components/workflow/types'
import { KNOWLEDGE_RETRIEVAL_OUTPUT_STRUCT, LLM_OUTPUT_STRUCT, SUPPORT_OUTPUT_VARS_NODE, TEMPLATE_TRANSFORM_OUTPUT_STRUCT } from '@/app/components/workflow/constants'
import {
CHAT_QUESTION_CLASSIFIER_OUTPUT_STRUCT,
COMPLETION_QUESTION_CLASSIFIER_OUTPUT_STRUCT,
KNOWLEDGE_RETRIEVAL_OUTPUT_STRUCT,
LLM_OUTPUT_STRUCT,
SUPPORT_OUTPUT_VARS_NODE,
TEMPLATE_TRANSFORM_OUTPUT_STRUCT,
} from '@/app/components/workflow/constants'
const inputVarTypeToVarType = (type: InputVarType): VarType => {
if (type === InputVarType.number)
@ -11,7 +18,7 @@ const inputVarTypeToVarType = (type: InputVarType): VarType => {
return VarType.string
}
const formatItem = (item: any): NodeOutPutVar => {
const formatItem = (item: any, isChatMode: boolean): NodeOutPutVar => {
const { id, data } = item
const res: NodeOutPutVar = {
nodeId: id,
@ -59,10 +66,15 @@ const formatItem = (item: any): NodeOutPutVar => {
res.vars = TEMPLATE_TRANSFORM_OUTPUT_STRUCT
break
}
case BlockEnum.QuestionClassifier: {
res.vars = isChatMode ? CHAT_QUESTION_CLASSIFIER_OUTPUT_STRUCT : COMPLETION_QUESTION_CLASSIFIER_OUTPUT_STRUCT
break
}
}
return res
}
export const toNodeOutputVars = (nodes: any[]): NodeOutPutVar[] => {
return nodes.filter(node => SUPPORT_OUTPUT_VARS_NODE.includes(node.data.type)).map(formatItem)
export const toNodeOutputVars = (nodes: any[], isChatMode: boolean): NodeOutPutVar[] => {
return nodes.filter(node => SUPPORT_OUTPUT_VARS_NODE.includes(node.data.type)).map(node => formatItem(node, isChatMode))
}

9
web/app/components/workflow/nodes/_base/components/variable/var-reference-picker.tsx
View File

@ -14,7 +14,10 @@ import {
PortalToFollowElemContent,
PortalToFollowElemTrigger,
} from '@/app/components/base/portal-to-follow-elem'
import { useWorkflow } from '@/app/components/workflow/hooks'
import {
useIsChatMode,
useWorkflow,
} from '@/app/components/workflow/hooks'
type Props = {
className?: string
@ -37,9 +40,11 @@ const VarReferencePicker: FC<Props> = ({
value,
onChange,
}) => {
const isChatMode = useIsChatMode()
const { getTreeLeafNodes, getBeforeNodesInSameBranch } = useWorkflow()
const availableNodes = getBeforeNodesInSameBranch(nodeId)
const outputVars = toNodeOutputVars(availableNodes)
const outputVars = toNodeOutputVars(availableNodes, isChatMode)
const [open, setOpen] = useState(false)
const hasValue = value.length > 0
const outputVarNodeId = hasValue ? value[0] : ''