MaxKB/ui/src/workflow/nodes/intent-classify-node/index.ts
2025-09-26 13:42:48 +08:00

77 lines
2.1 KiB
TypeScript

import IntentNodeVue from './index.vue'
import { AppNode, AppNodeModel } from '@/workflow/common/app-node'
class IntentNode extends AppNode {
constructor(props: any) {
super(props, IntentNodeVue)
}
}
const get_up_index_height = (branch_list: Array<any>, index: number) => {
return branch_list
.filter((item, i) => i < index)
.map((item) => item.height + 8)
.reduce((x,y) => x+y, 0)
}
class IntentModel extends AppNodeModel {
refreshBranch() {
// 更新节点连接边的path
this.incoming.edges.forEach((edge: any) => {
// 调用自定义的更新方案
edge.updatePathByAnchor()
})
this.outgoing.edges.forEach((edge: any) => {
edge.updatePathByAnchor()
})
}
getDefaultAnchor() {
const {
id,
x,
y,
width,
height,
properties: { branch_condition_list }
} = this
if (this.height === undefined) {
this.height = 200
}
const showNode = this.properties.showNode === undefined ? true : this.properties.showNode
const anchors: any = []
anchors.push({
x: x - width / 2 + 10,
y: showNode ? y : y - 15,
id: `${id}_left`,
edgeAddable: false,
type: 'left'
})
if (branch_condition_list) {
const FORM_ITEMS_HEIGHT = 397 // 上方表单占用高度
for (let index = 0; index < branch_condition_list.length; index++) {
const element = branch_condition_list[index]
anchors.push({
x: x + width / 2 - 10,
y: showNode
? y - height / 2 + FORM_ITEMS_HEIGHT + index *41.36
: y - 15,
id: `${id}_${element.id}_right`,
type: 'right'
})
console.log(y - height / 2 + FORM_ITEMS_HEIGHT + 100/ 2)
}
}
return anchors
}
}
export default {
type: 'intent-node',
model: IntentModel,
view: IntentNode
}