feat(web): add question classifier node
This commit is contained in:
@@ -1553,6 +1553,7 @@ Memory Bear: After the rebellion, regional warlordism intensified for several re
|
|||||||
'parameter-extractor': 'Parameter Extraction',
|
'parameter-extractor': 'Parameter Extraction',
|
||||||
flowControl: 'Flow Control',
|
flowControl: 'Flow Control',
|
||||||
'if-else': 'Conditional Branch',
|
'if-else': 'Conditional Branch',
|
||||||
|
'question-classifier': 'Question Classifier',
|
||||||
iteration: 'Iteration',
|
iteration: 'Iteration',
|
||||||
loop: 'Loop',
|
loop: 'Loop',
|
||||||
parallel: 'Parallel Execution',
|
parallel: 'Parallel Execution',
|
||||||
@@ -1664,7 +1665,18 @@ Memory Bear: After the rebellion, regional warlordism intensified for several re
|
|||||||
"gt": '>',
|
"gt": '>',
|
||||||
"ge": '>=',
|
"ge": '>=',
|
||||||
else_desc: 'Used to define the logic that should be executed when the if condition is not met.'
|
else_desc: 'Used to define the logic that should be executed when the if condition is not met.'
|
||||||
}
|
},
|
||||||
|
'question-classifier': {
|
||||||
|
model_id: 'Model',
|
||||||
|
input_variable: 'Input Variable',
|
||||||
|
categories: 'Categories',
|
||||||
|
user_supplement_prompt: 'Instruction',
|
||||||
|
class_name: 'Category',
|
||||||
|
addClassName: 'Add Category'
|
||||||
|
},
|
||||||
|
name: 'Key',
|
||||||
|
type: 'Type',
|
||||||
|
value: 'Value',
|
||||||
},
|
},
|
||||||
|
|
||||||
clear: 'Clear',
|
clear: 'Clear',
|
||||||
|
|||||||
@@ -1654,6 +1654,7 @@ export const zh = {
|
|||||||
'parameter-extractor': '参数提取',
|
'parameter-extractor': '参数提取',
|
||||||
flowControl: '流程控制',
|
flowControl: '流程控制',
|
||||||
'if-else': '条件分支',
|
'if-else': '条件分支',
|
||||||
|
'question-classifier': '问题分类器',
|
||||||
iteration: '迭代 (Iteration)',
|
iteration: '迭代 (Iteration)',
|
||||||
loop: '循环 (Loop)',
|
loop: '循环 (Loop)',
|
||||||
parallel: '并行执行',
|
parallel: '并行执行',
|
||||||
@@ -1793,6 +1794,14 @@ export const zh = {
|
|||||||
template: '代码',
|
template: '代码',
|
||||||
mapping: '输入变量'
|
mapping: '输入变量'
|
||||||
},
|
},
|
||||||
|
'question-classifier': {
|
||||||
|
model_id: '模型',
|
||||||
|
input_variable: '输入变量',
|
||||||
|
categories: '分类',
|
||||||
|
user_supplement_prompt: '指令',
|
||||||
|
class_name: '分类',
|
||||||
|
addClassName: '添加分类'
|
||||||
|
},
|
||||||
name: '键',
|
name: '键',
|
||||||
type: '类型',
|
type: '类型',
|
||||||
value: '值',
|
value: '值',
|
||||||
|
|||||||
@@ -0,0 +1,64 @@
|
|||||||
|
import { type FC } from 'react';
|
||||||
|
import { useTranslation } from 'react-i18next';
|
||||||
|
import { Input, Button, Form, Space } from 'antd';
|
||||||
|
import { PlusOutlined, CopyOutlined, DeleteOutlined, ExpandOutlined } from '@ant-design/icons';
|
||||||
|
|
||||||
|
interface CategoryListProps {
|
||||||
|
parentName: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
const CategoryList: FC<CategoryListProps> = ({ parentName }) => {
|
||||||
|
const { t } = useTranslation();
|
||||||
|
const form = Form.useFormInstance();
|
||||||
|
const formValues = Form.useWatch([parentName], form);
|
||||||
|
|
||||||
|
console.log('formValues', formValues)
|
||||||
|
return (
|
||||||
|
<Form.List name={parentName}>
|
||||||
|
{(fields, { add, remove }) => (
|
||||||
|
<Space direction="vertical" size={12} className="rb:w-full">
|
||||||
|
{fields.map(({ key, name, ...restField }, index) => {
|
||||||
|
const currentItem = formValues?.[key] || {};
|
||||||
|
const contentLength = (currentItem.class_name || '').length;
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div key={key} className="rb:border rb:border-[#DFE4ED] rb:rounded-md rb:p-3 rb:bg-[#F8F9FB]">
|
||||||
|
<div className="rb:flex rb:items-center rb:justify-between rb:mb-2">
|
||||||
|
<div>{t('workflow.config.question-classifier.class_name')} {index + 1}</div>
|
||||||
|
<div className="rb:flex rb:items-center rb:gap-1">
|
||||||
|
<span className="rb:text-xs rb:text-gray-500">{contentLength}</span>
|
||||||
|
<Button
|
||||||
|
type="text"
|
||||||
|
size="small"
|
||||||
|
icon={<DeleteOutlined />}
|
||||||
|
onClick={() => remove(name)}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<Form.Item
|
||||||
|
{...restField}
|
||||||
|
name={[name, 'class_name']}
|
||||||
|
noStyle
|
||||||
|
>
|
||||||
|
<Input.TextArea
|
||||||
|
placeholder={t('common.pleaseEnter')}
|
||||||
|
rows={2}
|
||||||
|
/>
|
||||||
|
</Form.Item>
|
||||||
|
</div>
|
||||||
|
)})}
|
||||||
|
|
||||||
|
<Button
|
||||||
|
type="dashed"
|
||||||
|
onClick={() => add({})}
|
||||||
|
className="rb:w-full"
|
||||||
|
>
|
||||||
|
+ {t('workflow.config.question-classifier.addClassName')}
|
||||||
|
</Button>
|
||||||
|
</Space>
|
||||||
|
)}
|
||||||
|
</Form.List>
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
export default CategoryList;
|
||||||
@@ -17,6 +17,7 @@ import GroupVariableList from './GroupVariableList'
|
|||||||
import CaseList from './CaseList'
|
import CaseList from './CaseList'
|
||||||
import HttpRequest from './HttpRequest';
|
import HttpRequest from './HttpRequest';
|
||||||
import MappingList from './MappingList'
|
import MappingList from './MappingList'
|
||||||
|
import CategoryList from './CategoryList'
|
||||||
|
|
||||||
interface PropertiesProps {
|
interface PropertiesProps {
|
||||||
selectedNode?: Node | null;
|
selectedNode?: Node | null;
|
||||||
@@ -472,10 +473,15 @@ const Properties: FC<PropertiesProps> = ({
|
|||||||
: config.type === 'variableList'
|
: config.type === 'variableList'
|
||||||
? <VariableSelect
|
? <VariableSelect
|
||||||
placeholder={t('common.pleaseSelect')}
|
placeholder={t('common.pleaseSelect')}
|
||||||
options={variableList}
|
options={variableList.map(vo => ({
|
||||||
|
...vo,
|
||||||
|
value: `{{${vo.value}}}`
|
||||||
|
}))}
|
||||||
/>
|
/>
|
||||||
: config.type === 'switch'
|
: config.type === 'switch'
|
||||||
? <Switch />
|
? <Switch />
|
||||||
|
: config.type === 'categoryList'
|
||||||
|
? <CategoryList parentName={key} />
|
||||||
: null
|
: null
|
||||||
}
|
}
|
||||||
</Form.Item>
|
</Form.Item>
|
||||||
|
|||||||
@@ -37,6 +37,7 @@ import sensitiveDetectionIcon from '@/assets/images/workflow/sensitive_detection
|
|||||||
import outputAuditIcon from '@/assets/images/workflow/output_audit.png';
|
import outputAuditIcon from '@/assets/images/workflow/output_audit.png';
|
||||||
import selfOptimizationIcon from '@/assets/images/workflow/self_optimization.png';
|
import selfOptimizationIcon from '@/assets/images/workflow/self_optimization.png';
|
||||||
import processEvolutionIcon from '@/assets/images/workflow/process_evolution.png';
|
import processEvolutionIcon from '@/assets/images/workflow/process_evolution.png';
|
||||||
|
import questionClassifierIcon from '@/assets/images/workflow/question-classifier.png'
|
||||||
|
|
||||||
import { getModelListUrl } from '@/api/models'
|
import { getModelListUrl } from '@/api/models'
|
||||||
import type { NodeLibrary } from './types'
|
import type { NodeLibrary } from './types'
|
||||||
@@ -201,6 +202,28 @@ export const nodeLibrary: NodeLibrary[] = [
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
type: "question-classifier", icon: questionClassifierIcon,
|
||||||
|
config: {
|
||||||
|
model_id: {
|
||||||
|
type: 'customSelect',
|
||||||
|
url: getModelListUrl,
|
||||||
|
params: { type: 'llm,chat' }, // llm/chat
|
||||||
|
valueKey: 'id',
|
||||||
|
labelKey: 'name',
|
||||||
|
},
|
||||||
|
input_variable: {
|
||||||
|
type: 'variableList',
|
||||||
|
},
|
||||||
|
categories: {
|
||||||
|
type: 'categoryList'
|
||||||
|
},
|
||||||
|
user_supplement_prompt: {
|
||||||
|
type: 'messageEditor',
|
||||||
|
isArray: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
// { type: "iteration", icon: iterationIcon },
|
// { type: "iteration", icon: iterationIcon },
|
||||||
// { type: "loop", icon: loopIcon },
|
// { type: "loop", icon: loopIcon },
|
||||||
// { type: "parallel", icon: parallelIcon },
|
// { type: "parallel", icon: parallelIcon },
|
||||||
|
|||||||
@@ -771,7 +771,7 @@ export const useWorkflowGraph = ({
|
|||||||
itemConfig = {
|
itemConfig = {
|
||||||
...itemConfig,
|
...itemConfig,
|
||||||
...data.config[key].defaultValue,
|
...data.config[key].defaultValue,
|
||||||
knowledge_bases: knowledge_bases.map((vo: any) => ({ kb_id: vo.id, ...vo.config }))
|
knowledge_bases: knowledge_bases?.map((vo: any) => ({ kb_id: vo.id, ...vo.config }))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|||||||
Reference in New Issue
Block a user