feat: workflow add knowledge-retrieval node
This commit is contained in:
@@ -15,6 +15,7 @@ interface RadioCardProps extends Omit<RadioGroupProps, 'onChange'> {
|
|||||||
onValueChange?: (value: string | null | undefined, option?: RadioCardOption) => void;
|
onValueChange?: (value: string | null | undefined, option?: RadioCardOption) => void;
|
||||||
onChange?: (value: string | null | undefined, option?: RadioCardOption) => void;
|
onChange?: (value: string | null | undefined, option?: RadioCardOption) => void;
|
||||||
itemRender?: (option: RadioCardOption) => ReactNode;
|
itemRender?: (option: RadioCardOption) => ReactNode;
|
||||||
|
allowClear?: boolean;
|
||||||
}
|
}
|
||||||
|
|
||||||
const RadioGroupCard: FC<RadioCardProps> = ({
|
const RadioGroupCard: FC<RadioCardProps> = ({
|
||||||
@@ -22,7 +23,8 @@ const RadioGroupCard: FC<RadioCardProps> = ({
|
|||||||
value,
|
value,
|
||||||
onValueChange,
|
onValueChange,
|
||||||
onChange,
|
onChange,
|
||||||
itemRender
|
itemRender,
|
||||||
|
allowClear = true
|
||||||
}) => {
|
}) => {
|
||||||
// 监听value变化
|
// 监听value变化
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
@@ -34,23 +36,27 @@ const RadioGroupCard: FC<RadioCardProps> = ({
|
|||||||
const handleChange = (option: RadioCardOption) => {
|
const handleChange = (option: RadioCardOption) => {
|
||||||
if (option.disabled) return
|
if (option.disabled) return
|
||||||
if (onChange) {
|
if (onChange) {
|
||||||
onChange(value === option.value ? null : String(option.value), value === option.value ? undefined : option);
|
if (allowClear && value === option.value) {
|
||||||
|
onChange(null, undefined);
|
||||||
|
} else {
|
||||||
|
onChange(String(option.value), option);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className={`rb:grid rb:grid-cols-${options.length} rb:gap-[12px]`}>
|
<div className={`rb:grid rb:grid-cols-${options.length} rb:gap-3`}>
|
||||||
{options.map(option => (
|
{options.map(option => (
|
||||||
<div key={String(option.value)} className={clsx("rb:border rb:rounded-[8px] rb:w-full rb:p-[20px_12px] rb:text-center rb:cursor-pointer", {
|
<div key={String(option.value)} className={clsx("rb:border rb:rounded-lg rb:w-full rb:p-[20px_12px] rb:text-center rb:cursor-pointer", {
|
||||||
'rb:bg-[rgba(21,94,239,0.06)] rb:border-[#155EEF]': option.value === value,
|
'rb:bg-[rgba(21,94,239,0.06)] rb:border-[#155EEF]': option.value === value,
|
||||||
'rb:border-[#EBEBEB] rb:bg-[#ffffff]': option.value !== value,
|
'rb:border-[#EBEBEB] rb:bg-[#ffffff]': option.value !== value,
|
||||||
'rb:opacity-[0.75]': option.disabled
|
'rb:opacity-[0.75]': option.disabled
|
||||||
})} onClick={() => handleChange(option)}>
|
})} onClick={() => handleChange(option)}>
|
||||||
{itemRender ? itemRender(option) : (
|
{itemRender ? itemRender(option) : (
|
||||||
<>
|
<>
|
||||||
{option.icon && <img src={option.icon} className="rb:w-[40px] rb:h-[40px] rb:mb-[12px] rb:m-[0_auto]" />}
|
{option.icon && <img src={option.icon} className="rb:w-10 rb:h-10 rb:mb-3 rb:m-[0_auto]" />}
|
||||||
<div className="rb:text-[14px] rb:font-medium">{option.label}</div>
|
<div className="rb:text-[14px] rb:font-medium">{option.label}</div>
|
||||||
<div className="rb:mt-[6px] rb:text-[#5B6167] rb:text-[12px] rb:font-regular">{option.labelDesc}</div>
|
<div className="rb:mt-1.5 rb:text-[#5B6167] rb:text-[12px] rb:font-regular">{option.labelDesc}</div>
|
||||||
</>
|
</>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -393,12 +393,6 @@ export const zh = {
|
|||||||
clusterName: '集群名称',
|
clusterName: '集群名称',
|
||||||
clusterDescription: '集群描述',
|
clusterDescription: '集群描述',
|
||||||
clusterDescriptionPlaceholder: '这是一个专门处理核心业务的Agent集群,能够协作完成复杂的业务处理任务。',
|
clusterDescriptionPlaceholder: '这是一个专门处理核心业务的Agent集群,能够协作完成复杂的业务处理任务。',
|
||||||
toolCalling: '工具调用',
|
|
||||||
toolCallingDesc: '主控代理将子代理作为工具调用',
|
|
||||||
toolCallingFeature: '集中控制,适合结构化工作流',
|
|
||||||
handoffs: '交接',
|
|
||||||
handoffsDesc: '代理之间动态转移控制权',
|
|
||||||
handoffsFeature: '去中心化控制,适合复杂对话场景',
|
|
||||||
recommend: '推荐',
|
recommend: '推荐',
|
||||||
advanced: '高级',
|
advanced: '高级',
|
||||||
multiAgentArchitecture: '多代理架构模式',
|
multiAgentArchitecture: '多代理架构模式',
|
||||||
@@ -425,7 +419,7 @@ export const zh = {
|
|||||||
stateSharingStrategy: '状态共享策略',
|
stateSharingStrategy: '状态共享策略',
|
||||||
intermediateResultProcessing: '中间结果处理',
|
intermediateResultProcessing: '中间结果处理',
|
||||||
metadataTransfer: '元数据传输',
|
metadataTransfer: '元数据传输',
|
||||||
|
knowledgeConfig: '知识库配置',
|
||||||
temperature: '温度',
|
temperature: '温度',
|
||||||
temperature_desc: '温度参数,控制输出的随机性',
|
temperature_desc: '温度参数,控制输出的随机性',
|
||||||
max_tokens: '最大令牌数',
|
max_tokens: '最大令牌数',
|
||||||
@@ -453,9 +447,9 @@ export const zh = {
|
|||||||
releasePreview: '发布预览',
|
releasePreview: '发布预览',
|
||||||
globalConfig: '全局配置',
|
globalConfig: '全局配置',
|
||||||
globalConfigDesc: '全局配置将应用于所有关联的知识库作为默认配置。单个知识库的配置将覆盖全局配置。',
|
globalConfigDesc: '全局配置将应用于所有关联的知识库作为默认配置。单个知识库的配置将覆盖全局配置。',
|
||||||
rerankModel: '重排序模型',
|
rerankModel: 'Reranker 模型',
|
||||||
rerankModelDesc: '激活后,搜索结果将被重新排序以提高相关性',
|
rerankModelDesc: '激活后,搜索结果将被重新排序以提高相关性',
|
||||||
rearrangementModel: '重排序模型',
|
rearrangementModel: 'Reranker 模型',
|
||||||
rearrangementModelDesc: '选择用于重新排序的模型',
|
rearrangementModelDesc: '选择用于重新排序的模型',
|
||||||
reranker_top_k: 'Top K',
|
reranker_top_k: 'Top K',
|
||||||
reranker_top_k_desc: '返回的文档片段数量,范围从1到20',
|
reranker_top_k_desc: '返回的文档片段数量,范围从1到20',
|
||||||
@@ -514,6 +508,21 @@ export const zh = {
|
|||||||
promptChatPlaceholder: '描述你需要的提示词,例如:我需要一个客服助手',
|
promptChatPlaceholder: '描述你需要的提示词,例如:我需要一个客服助手',
|
||||||
promptChatEmpty: '目前没有对话内容',
|
promptChatEmpty: '目前没有对话内容',
|
||||||
promptEmpty: '在左侧描述您的用例,编排预览将在此处显示。',
|
promptEmpty: '在左侧描述您的用例,编排预览将在此处显示。',
|
||||||
|
|
||||||
|
master: '主管模式',
|
||||||
|
master_agent: '主管模式',
|
||||||
|
masterDesc: '由主 Agent 统一调度和管理,子 Agent 按照主管分配的任务执行,适合需要集中控制的场景。',
|
||||||
|
handoffs: '协作模式',
|
||||||
|
handoffsDesc: '多个 Agent 平等协作,根据任务需求自主协调配合,适合需要灵活互动的复杂场景。',
|
||||||
|
masterConfig: '主管配置',
|
||||||
|
orchestrationMode: '任务分配策略',
|
||||||
|
conditional: '智能分配',
|
||||||
|
sequential: '顺序分配',
|
||||||
|
parallel: '并行分配',
|
||||||
|
aggregationStrategy: '结果汇总方式',
|
||||||
|
merge: '完整汇总',
|
||||||
|
vote: '关键信息提取',
|
||||||
|
priority: '结构化整合',
|
||||||
},
|
},
|
||||||
// 角色管理相关翻译
|
// 角色管理相关翻译
|
||||||
role: {
|
role: {
|
||||||
@@ -1629,7 +1638,7 @@ export const zh = {
|
|||||||
llm: '大语言模型 (LLM)',
|
llm: '大语言模型 (LLM)',
|
||||||
model_selection: '多模型选择',
|
model_selection: '多模型选择',
|
||||||
model_voting: '多模型投票',
|
model_voting: '多模型投票',
|
||||||
rag: '知识检索 (RAG)',
|
'knowledge-retrieval': '知识检索 (RAG)',
|
||||||
classification: '智能分类',
|
classification: '智能分类',
|
||||||
parameter_extraction: '参数提取',
|
parameter_extraction: '参数提取',
|
||||||
flowControl: '流程控制',
|
flowControl: '流程控制',
|
||||||
@@ -1694,6 +1703,11 @@ export const zh = {
|
|||||||
},
|
},
|
||||||
end: {
|
end: {
|
||||||
output: '回复'
|
output: '回复'
|
||||||
|
},
|
||||||
|
'knowledge-retrieval': {
|
||||||
|
query: '查询变量',
|
||||||
|
knowledge_retrieval: '知识库',
|
||||||
|
recallConfig: '召回测试',
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,155 @@
|
|||||||
|
import { type FC, useRef, useState, useEffect } from 'react'
|
||||||
|
import { useTranslation } from 'react-i18next'
|
||||||
|
import { Space, Button, List } from 'antd'
|
||||||
|
import knowledgeEmpty from '@/assets/images/application/knowledgeEmpty.svg'
|
||||||
|
import type {
|
||||||
|
KnowledgeConfigForm,
|
||||||
|
KnowledgeConfig,
|
||||||
|
RerankerConfig,
|
||||||
|
KnowledgeBase,
|
||||||
|
KnowledgeModalRef,
|
||||||
|
KnowledgeConfigModalRef,
|
||||||
|
KnowledgeGlobalConfigModalRef,
|
||||||
|
} from './types'
|
||||||
|
import Empty from '@/components/Empty'
|
||||||
|
import KnowledgeListModal from './KnowledgeListModal'
|
||||||
|
import KnowledgeConfigModal from './KnowledgeConfigModal'
|
||||||
|
import KnowledgeGlobalConfigModal from './KnowledgeGlobalConfigModal'
|
||||||
|
import Tag from '@/components/Tag'
|
||||||
|
|
||||||
|
const Knowledge: FC<{value?: KnowledgeConfig; onChange?: (config: KnowledgeConfig) => void}> = ({value = {knowledge_bases: []}, onChange}) => {
|
||||||
|
const { t } = useTranslation()
|
||||||
|
const knowledgeModalRef = useRef<KnowledgeModalRef>(null)
|
||||||
|
const knowledgeConfigModalRef = useRef<KnowledgeConfigModalRef>(null)
|
||||||
|
const knowledgeGlobalConfigModalRef = useRef<KnowledgeGlobalConfigModalRef>(null)
|
||||||
|
const [knowledgeList, setKnowledgeList] = useState<KnowledgeBase[]>([])
|
||||||
|
const [editConfig, setEditConfig] = useState<KnowledgeConfig>({} as KnowledgeConfig)
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (value && JSON.stringify(value) !== JSON.stringify(editConfig)) {
|
||||||
|
setEditConfig({ ...(value || {}) })
|
||||||
|
const knowledge_bases = [...(value.knowledge_bases || [])]
|
||||||
|
setKnowledgeList(knowledge_bases)
|
||||||
|
}
|
||||||
|
}, [value])
|
||||||
|
|
||||||
|
const handleKnowledgeConfig = () => {
|
||||||
|
knowledgeGlobalConfigModalRef.current?.handleOpen()
|
||||||
|
}
|
||||||
|
const handleAddKnowledge = () => {
|
||||||
|
knowledgeModalRef.current?.handleOpen()
|
||||||
|
}
|
||||||
|
const handleDeleteKnowledge = (id: string) => {
|
||||||
|
const list = knowledgeList.filter(item => item.id !== id)
|
||||||
|
setKnowledgeList([...list])
|
||||||
|
onChange && onChange({
|
||||||
|
...editConfig,
|
||||||
|
knowledge_bases: [...list],
|
||||||
|
})
|
||||||
|
}
|
||||||
|
const handleEditKnowledge = (item: KnowledgeBase) => {
|
||||||
|
knowledgeConfigModalRef.current?.handleOpen(item)
|
||||||
|
}
|
||||||
|
const refresh = (values: KnowledgeBase[] | KnowledgeConfigForm | RerankerConfig, type: 'knowledge' | 'knowledgeConfig' | 'rerankerConfig') => {
|
||||||
|
if (type === 'knowledge') {
|
||||||
|
let list = [...knowledgeList]
|
||||||
|
if (list.length > 0) {
|
||||||
|
(Array.isArray(values) ? values : [values]).forEach(vo => {
|
||||||
|
const index = list.findIndex(item => item.id === (vo as KnowledgeBase).id)
|
||||||
|
if (index === -1) {
|
||||||
|
list.push(vo as KnowledgeBase)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
list = [...values as KnowledgeBase[]]
|
||||||
|
}
|
||||||
|
setKnowledgeList([...list])
|
||||||
|
onChange && onChange({
|
||||||
|
...editConfig,
|
||||||
|
knowledge_bases: [...list],
|
||||||
|
})
|
||||||
|
} else if (type === 'knowledgeConfig') {
|
||||||
|
const index = knowledgeList.findIndex(item => item.id === (values as KnowledgeBase).kb_id)
|
||||||
|
const list = [...knowledgeList]
|
||||||
|
list[index] = {
|
||||||
|
...list[index],
|
||||||
|
config: {...values as KnowledgeConfigForm}
|
||||||
|
}
|
||||||
|
setKnowledgeList([...list])
|
||||||
|
onChange && onChange({
|
||||||
|
...editConfig,
|
||||||
|
knowledge_bases: [...list],
|
||||||
|
})
|
||||||
|
} else if (type === 'rerankerConfig') {
|
||||||
|
const rerankerValues = values as RerankerConfig
|
||||||
|
setEditConfig(prev => ({ ...prev, ...rerankerValues }))
|
||||||
|
onChange && onChange({
|
||||||
|
...editConfig,
|
||||||
|
...rerankerValues,
|
||||||
|
reranker_id: rerankerValues.rerank_model ? rerankerValues.reranker_id : undefined,
|
||||||
|
reranker_top_k: rerankerValues.rerank_model ? rerankerValues.reranker_top_k : undefined,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return (
|
||||||
|
<div>
|
||||||
|
<div className="rb:flex rb:justify-between rb:items-center">
|
||||||
|
<div>{t('application.knowledgeBaseAssociation')}</div>
|
||||||
|
|
||||||
|
<Space>
|
||||||
|
<Button style={{ padding: '0 8px', height: '24px' }} onClick={handleKnowledgeConfig}>{t('workflow.config.knowledge-retrieval.recallConfig')}</Button>
|
||||||
|
<Button style={{ padding: '0 8px', height: '24px' }} onClick={handleAddKnowledge}>+</Button>
|
||||||
|
</Space>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{knowledgeList.length === 0
|
||||||
|
? <Empty url={knowledgeEmpty} size={88} subTitle={t('application.knowledgeEmpty')} />
|
||||||
|
:
|
||||||
|
<List
|
||||||
|
grid={{ gutter: 12, column: 1 }}
|
||||||
|
dataSource={knowledgeList}
|
||||||
|
renderItem={(item) => (
|
||||||
|
<List.Item>
|
||||||
|
<div key={item.id} className="rb:flex rb:items-center rb:justify-between rb:p-[12px_16px] rb:bg-[#FBFDFF] rb:border rb:border-[#DFE4ED] rb:rounded-lg">
|
||||||
|
<div className="rb:font-medium rb:leading-4">
|
||||||
|
{item.name}
|
||||||
|
<Tag color={item.status === 1 ? 'success' : item.status === 0 ? 'default' : 'error'} className="rb:ml-2">
|
||||||
|
{item.status === 1 ? t('common.enable') : item.status === 0 ? t('common.disabled') : t('common.deleted')}
|
||||||
|
</Tag>
|
||||||
|
<div className="rb:mt-1 rb:text-[12px] rb:text-[#5B6167] rb:font-regular rb:leading-5">{t('application.contains', {include_count: item.doc_num})}</div>
|
||||||
|
</div>
|
||||||
|
<Space size={12}>
|
||||||
|
<div
|
||||||
|
className="rb:w-6 rb:h-6 rb:cursor-pointer rb:bg-[url('@/assets/images/editBorder.svg')] rb:hover:bg-[url('@/assets/images/editBg.svg')]"
|
||||||
|
onClick={() => handleEditKnowledge(item)}
|
||||||
|
></div>
|
||||||
|
<div
|
||||||
|
className="rb:w-6 rb:h-6 rb:cursor-pointer rb:bg-[url('@/assets/images/deleteBorder.svg')] rb:hover:bg-[url('@/assets/images/deleteBg.svg')]"
|
||||||
|
onClick={() => handleDeleteKnowledge(item.id)}
|
||||||
|
></div>
|
||||||
|
</Space>
|
||||||
|
</div>
|
||||||
|
</List.Item>
|
||||||
|
)}
|
||||||
|
/>
|
||||||
|
}
|
||||||
|
{/* 全局设置 */}
|
||||||
|
<KnowledgeGlobalConfigModal
|
||||||
|
data={editConfig}
|
||||||
|
ref={knowledgeGlobalConfigModalRef}
|
||||||
|
refresh={refresh}
|
||||||
|
/>
|
||||||
|
{/* 知识库列表 */}
|
||||||
|
<KnowledgeListModal
|
||||||
|
ref={knowledgeModalRef}
|
||||||
|
selectedList={knowledgeList}
|
||||||
|
refresh={refresh}
|
||||||
|
/>
|
||||||
|
<KnowledgeConfigModal
|
||||||
|
ref={knowledgeConfigModalRef}
|
||||||
|
refresh={refresh}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
export default Knowledge
|
||||||
@@ -0,0 +1,186 @@
|
|||||||
|
import { forwardRef, useEffect, useImperativeHandle, useState } from 'react';
|
||||||
|
import { Form, Select, InputNumber } from 'antd';
|
||||||
|
import { useTranslation } from 'react-i18next';
|
||||||
|
|
||||||
|
import type { KnowledgeConfigModalRef, KnowledgeBase, KnowledgeConfigForm, RetrieveType } from './types'
|
||||||
|
import RbModal from '@/components/RbModal'
|
||||||
|
import RbSlider from '@/components/RbSlider'
|
||||||
|
import { formatDateTime } from '@/utils/format';
|
||||||
|
|
||||||
|
const FormItem = Form.Item;
|
||||||
|
|
||||||
|
interface KnowledgeConfigModalProps {
|
||||||
|
refresh: (values: KnowledgeConfigForm, type: 'knowledgeConfig') => void;
|
||||||
|
}
|
||||||
|
const retrieveTypes: RetrieveType[] = ['participle', 'semantic', 'hybrid']
|
||||||
|
|
||||||
|
const KnowledgeConfigModal = forwardRef<KnowledgeConfigModalRef, KnowledgeConfigModalProps>(({
|
||||||
|
refresh,
|
||||||
|
}, ref) => {
|
||||||
|
const { t } = useTranslation();
|
||||||
|
const [visible, setVisible] = useState(false);
|
||||||
|
const [form] = Form.useForm<KnowledgeConfigForm>();
|
||||||
|
const [data, setData] = useState<KnowledgeBase | null>(null);
|
||||||
|
|
||||||
|
const values = Form.useWatch<KnowledgeConfigForm>([], form);
|
||||||
|
|
||||||
|
// 封装取消方法,添加关闭弹窗逻辑
|
||||||
|
const handleClose = () => {
|
||||||
|
setVisible(false);
|
||||||
|
form.resetFields();
|
||||||
|
setData(null)
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleOpen = (data: KnowledgeBase) => {
|
||||||
|
form.setFieldsValue({
|
||||||
|
retrieve_type: data?.config?.retrieve_type || retrieveTypes[0],
|
||||||
|
kb_id: data.id,
|
||||||
|
top_k: data?.config?.top_k || 5,
|
||||||
|
similarity_threshold: data?.config?.similarity_threshold || 0.5,
|
||||||
|
vector_similarity_weight: data?.config?.vector_similarity_weight || 0.5,
|
||||||
|
...(data || {}),
|
||||||
|
...(data?.config || {}),
|
||||||
|
})
|
||||||
|
setData({...data})
|
||||||
|
setVisible(true);
|
||||||
|
};
|
||||||
|
// 封装保存方法,添加提交逻辑
|
||||||
|
const handleSave = () => {
|
||||||
|
form
|
||||||
|
.validateFields()
|
||||||
|
.then(() => {
|
||||||
|
refresh(values, 'knowledgeConfig')
|
||||||
|
handleClose()
|
||||||
|
})
|
||||||
|
.catch((err) => {
|
||||||
|
console.log('err', err)
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 暴露给父组件的方法
|
||||||
|
useImperativeHandle(ref, () => ({
|
||||||
|
handleOpen,
|
||||||
|
handleClose
|
||||||
|
}));
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (values?.retrieve_type) {
|
||||||
|
const fieldsToReset = Object.keys(values).filter(key =>
|
||||||
|
key !== 'kb_id' && key !== 'retrieve_type'
|
||||||
|
) as (keyof KnowledgeConfigForm)[];
|
||||||
|
form.resetFields(fieldsToReset);
|
||||||
|
}
|
||||||
|
}, [values?.retrieve_type])
|
||||||
|
|
||||||
|
return (
|
||||||
|
<RbModal
|
||||||
|
title={t('application.knowledgeConfig')}
|
||||||
|
open={visible}
|
||||||
|
onCancel={handleClose}
|
||||||
|
okText={t('common.save')}
|
||||||
|
onOk={handleSave}
|
||||||
|
>
|
||||||
|
<Form
|
||||||
|
form={form}
|
||||||
|
layout="vertical"
|
||||||
|
>
|
||||||
|
{data && (
|
||||||
|
<div className="rb:mb-6 rb:flex rb:items-center rb:justify-between rb:border rb:rounded-lg rb:p-[17px_16px] rb:cursor-pointer rb:bg-[#F0F3F8] rb:border-[#DFE4ED] rb:text-[#212332]">
|
||||||
|
<div className="rb:text-[16px] rb:leading-5.5">
|
||||||
|
{data.name}
|
||||||
|
<div className="rb:text-[12px] rb:leading-4 rb:text-[#5B6167] rb:mt-2">{t('application.contains', {include_count: data.doc_num})}</div>
|
||||||
|
</div>
|
||||||
|
<div className="rb:text-[12px] rb:leading-4 rb:text-[#5B6167]">{formatDateTime(data.updated_at, 'YYYY-MM-DD HH:mm:ss')}</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
<FormItem name="kb_id" hidden />
|
||||||
|
{/* 检索模式 */}
|
||||||
|
<FormItem
|
||||||
|
name="retrieve_type"
|
||||||
|
label={t('application.retrieve_type')}
|
||||||
|
extra={t('application.retrieve_type_desc')}
|
||||||
|
rules={[{ required: true, message: t('common.pleaseSelect') }]}
|
||||||
|
>
|
||||||
|
|
||||||
|
<Select
|
||||||
|
options={retrieveTypes.map(key => ({
|
||||||
|
label: t(`application.${key}`),
|
||||||
|
value: key,
|
||||||
|
}))}
|
||||||
|
/>
|
||||||
|
</FormItem>
|
||||||
|
{/* Top K */}
|
||||||
|
<FormItem
|
||||||
|
name="top_k"
|
||||||
|
label={t('application.top_k')}
|
||||||
|
rules={[{ required: true, message: t('common.pleaseEnter') }]}
|
||||||
|
extra={t('application.top_k_desc')}
|
||||||
|
initialValue={5}
|
||||||
|
>
|
||||||
|
<InputNumber style={{ width: '100%' }} min={1} max={20} />
|
||||||
|
</FormItem>
|
||||||
|
{/* 语义相似度阈值 similarity_threshold */}
|
||||||
|
{values?.retrieve_type === 'semantic' && (
|
||||||
|
<FormItem
|
||||||
|
name="similarity_threshold"
|
||||||
|
label={t('application.similarity_threshold')}
|
||||||
|
extra={t('application.similarity_threshold_desc')}
|
||||||
|
initialValue={0.5}
|
||||||
|
>
|
||||||
|
<RbSlider
|
||||||
|
max={1.0}
|
||||||
|
step={0.1}
|
||||||
|
min={0.0}
|
||||||
|
/>
|
||||||
|
</FormItem>
|
||||||
|
)}
|
||||||
|
{/* 分词匹配度阈值 vector_similarity_weight */}
|
||||||
|
{values?.retrieve_type === 'participle' && (
|
||||||
|
<FormItem
|
||||||
|
name="vector_similarity_weight"
|
||||||
|
label={t('application.vector_similarity_weight')}
|
||||||
|
extra={t('application.vector_similarity_weight_desc')}
|
||||||
|
initialValue={0.5}
|
||||||
|
>
|
||||||
|
<RbSlider
|
||||||
|
max={1.0}
|
||||||
|
step={0.1}
|
||||||
|
min={0.0}
|
||||||
|
/>
|
||||||
|
</FormItem>
|
||||||
|
)}
|
||||||
|
{/* 混合检索权重 */}
|
||||||
|
{values?.retrieve_type === 'hybrid' && (
|
||||||
|
<>
|
||||||
|
<FormItem
|
||||||
|
name="similarity_threshold"
|
||||||
|
label={t('application.similarity_threshold')}
|
||||||
|
extra={t('application.similarity_threshold_desc1')}
|
||||||
|
initialValue={0.5}
|
||||||
|
>
|
||||||
|
<RbSlider
|
||||||
|
max={1.0}
|
||||||
|
step={0.1}
|
||||||
|
min={0.0}
|
||||||
|
/>
|
||||||
|
</FormItem>
|
||||||
|
<FormItem
|
||||||
|
name="vector_similarity_weight"
|
||||||
|
label={t('application.vector_similarity_weight')}
|
||||||
|
extra={t('application.vector_similarity_weight_desc1')}
|
||||||
|
initialValue={0.5}
|
||||||
|
>
|
||||||
|
<RbSlider
|
||||||
|
max={1.0}
|
||||||
|
step={0.1}
|
||||||
|
min={0.0}
|
||||||
|
/>
|
||||||
|
</FormItem>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</Form>
|
||||||
|
</RbModal>
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
export default KnowledgeConfigModal;
|
||||||
@@ -0,0 +1,121 @@
|
|||||||
|
import { forwardRef, useImperativeHandle, useState, useEffect } from 'react';
|
||||||
|
import { Form, InputNumber, Switch } from 'antd';
|
||||||
|
import { useTranslation } from 'react-i18next';
|
||||||
|
|
||||||
|
import type { RerankerConfig, KnowledgeGlobalConfigModalRef } from './types'
|
||||||
|
import RbModal from '@/components/RbModal'
|
||||||
|
import CustomSelect from '@/components/CustomSelect'
|
||||||
|
import { getModelListUrl } from '@/api/models'
|
||||||
|
|
||||||
|
const FormItem = Form.Item;
|
||||||
|
|
||||||
|
interface KnowledgeGlobalConfigModalProps {
|
||||||
|
data: RerankerConfig;
|
||||||
|
refresh: (values: RerankerConfig, type: 'rerankerConfig') => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
const KnowledgeGlobalConfigModal = forwardRef<KnowledgeGlobalConfigModalRef, KnowledgeGlobalConfigModalProps>(({
|
||||||
|
refresh,
|
||||||
|
data,
|
||||||
|
}, ref) => {
|
||||||
|
const { t } = useTranslation();
|
||||||
|
const [visible, setVisible] = useState(false);
|
||||||
|
const [form] = Form.useForm<RerankerConfig>();
|
||||||
|
const values = Form.useWatch<RerankerConfig>([], form);
|
||||||
|
|
||||||
|
// 封装取消方法,添加关闭弹窗逻辑
|
||||||
|
const handleClose = () => {
|
||||||
|
setVisible(false);
|
||||||
|
form.resetFields();
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleOpen = () => {
|
||||||
|
form.setFieldsValue({ ...data, rerank_model: !!data?.reranker_id })
|
||||||
|
setVisible(true);
|
||||||
|
};
|
||||||
|
// 封装保存方法,添加提交逻辑
|
||||||
|
const handleSave = () => {
|
||||||
|
form
|
||||||
|
.validateFields()
|
||||||
|
.then(() => {
|
||||||
|
refresh(values, 'rerankerConfig')
|
||||||
|
handleClose()
|
||||||
|
})
|
||||||
|
.catch((err) => {
|
||||||
|
console.log('err', err)
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (values?.rerank_model) {
|
||||||
|
form.setFieldsValue({ ...data })
|
||||||
|
} else {
|
||||||
|
form.setFieldsValue({ reranker_id: undefined, reranker_top_k: undefined })
|
||||||
|
}
|
||||||
|
}, [values?.rerank_model])
|
||||||
|
|
||||||
|
// 暴露给父组件的方法
|
||||||
|
useImperativeHandle(ref, () => ({
|
||||||
|
handleOpen,
|
||||||
|
}));
|
||||||
|
|
||||||
|
return (
|
||||||
|
<RbModal
|
||||||
|
title={t('application.globalConfig')}
|
||||||
|
open={visible}
|
||||||
|
onCancel={handleClose}
|
||||||
|
okText={t('common.save')}
|
||||||
|
onOk={handleSave}
|
||||||
|
>
|
||||||
|
<Form
|
||||||
|
form={form}
|
||||||
|
layout="vertical"
|
||||||
|
>
|
||||||
|
<div className="rb:text-[#5B6167] rb:mb-6">{t('application.globalConfigDesc')}</div>
|
||||||
|
|
||||||
|
{/* 结果重排 */}
|
||||||
|
<div className="rb:flex rb:items-center rb:justify-between rb:my-6">
|
||||||
|
<div className="rb:text-[14px] rb:font-medium rb:leading-5">
|
||||||
|
{t('application.rerankModel')}
|
||||||
|
<div className="rb:mt-1 rb:text-[12px] rb:text-[#5B6167] rb:font-regular rb:leading-4">{t('application.rerankModelDesc')}</div>
|
||||||
|
</div>
|
||||||
|
<FormItem
|
||||||
|
name="rerank_model"
|
||||||
|
valuePropName="checked"
|
||||||
|
className="rb:mb-0!"
|
||||||
|
>
|
||||||
|
<Switch />
|
||||||
|
</FormItem>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{values?.rerank_model && <>
|
||||||
|
<FormItem
|
||||||
|
name="reranker_id"
|
||||||
|
label={t('application.rearrangementModel')}
|
||||||
|
rules={[{ required: true, message: t('common.pleaseSelect') }]}
|
||||||
|
extra={t('application.rearrangementModelDesc')}
|
||||||
|
>
|
||||||
|
<CustomSelect
|
||||||
|
url={getModelListUrl}
|
||||||
|
params={{ type: 'rerank', pagesize: 100 }}
|
||||||
|
valueKey="id"
|
||||||
|
labelKey="name"
|
||||||
|
hasAll={false}
|
||||||
|
/>
|
||||||
|
</FormItem>
|
||||||
|
{/* Top K */}
|
||||||
|
<FormItem
|
||||||
|
name="reranker_top_k"
|
||||||
|
label={t('application.reranker_top_k')}
|
||||||
|
rules={[{ required: true, message: t('common.pleaseEnter') }]}
|
||||||
|
extra={t('application.reranker_top_k_desc')}
|
||||||
|
>
|
||||||
|
<InputNumber style={{ width: '100%' }} min={1} max={20} />
|
||||||
|
</FormItem>
|
||||||
|
</>}
|
||||||
|
</Form>
|
||||||
|
</RbModal>
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
export default KnowledgeGlobalConfigModal;
|
||||||
@@ -0,0 +1,148 @@
|
|||||||
|
import { forwardRef, useEffect, useImperativeHandle, useState } from 'react';
|
||||||
|
import { Space, List } from 'antd';
|
||||||
|
import { useTranslation } from 'react-i18next';
|
||||||
|
import clsx from 'clsx'
|
||||||
|
import type { KnowledgeModalRef, KnowledgeBase } from './types'
|
||||||
|
import type { KnowledgeBaseListItem } from '@/views/KnowledgeBase/types'
|
||||||
|
import RbModal from '@/components/RbModal'
|
||||||
|
import { getKnowledgeBaseList } from '@/api/knowledgeBase'
|
||||||
|
import SearchInput from '@/components/SearchInput'
|
||||||
|
import Empty from '@/components/Empty'
|
||||||
|
import { formatDateTime } from '@/utils/format';
|
||||||
|
interface KnowledgeModalProps {
|
||||||
|
refresh: (rows: KnowledgeBase[], type: 'knowledge') => void;
|
||||||
|
selectedList: KnowledgeBase[];
|
||||||
|
}
|
||||||
|
|
||||||
|
const KnowledgeListModal = forwardRef<KnowledgeModalRef, KnowledgeModalProps>(({
|
||||||
|
refresh,
|
||||||
|
selectedList
|
||||||
|
}, ref) => {
|
||||||
|
const { t } = useTranslation();
|
||||||
|
const [visible, setVisible] = useState(false);
|
||||||
|
const [list, setList] = useState<KnowledgeBaseListItem[]>([])
|
||||||
|
const [filterList, setFilterList] = useState<KnowledgeBaseListItem[]>([])
|
||||||
|
const [query, setQuery] = useState<{keywords?: string}>({})
|
||||||
|
const [selectedIds, setSelectedIds] = useState<string[]>([])
|
||||||
|
const [selectedRows, setSelectedRows] = useState<KnowledgeBase[]>([])
|
||||||
|
|
||||||
|
// 封装取消方法,添加关闭弹窗逻辑
|
||||||
|
const handleClose = () => {
|
||||||
|
setVisible(false);
|
||||||
|
setQuery({})
|
||||||
|
setSelectedIds([])
|
||||||
|
setSelectedRows([])
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleOpen = () => {
|
||||||
|
setVisible(true);
|
||||||
|
setQuery({})
|
||||||
|
setSelectedIds([])
|
||||||
|
setSelectedRows([])
|
||||||
|
};
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (visible) {
|
||||||
|
getList()
|
||||||
|
}
|
||||||
|
}, [query.keywords, visible])
|
||||||
|
const getList = () => {
|
||||||
|
getKnowledgeBaseList(undefined, {
|
||||||
|
...query,
|
||||||
|
pagesize: 100,
|
||||||
|
orderby:'created_at',
|
||||||
|
desc:true,
|
||||||
|
})
|
||||||
|
.then(res => {
|
||||||
|
const response = res as { items: KnowledgeBaseListItem[] }
|
||||||
|
setList(response.items || [])
|
||||||
|
})
|
||||||
|
}
|
||||||
|
// 封装保存方法,添加提交逻辑
|
||||||
|
const handleSave = () => {
|
||||||
|
refresh(selectedRows.map(item => ({
|
||||||
|
...item,
|
||||||
|
config: {
|
||||||
|
similarity_threshold: 0.7,
|
||||||
|
strategy: "hybrid",
|
||||||
|
top_k: 3,
|
||||||
|
weight: 1,
|
||||||
|
}
|
||||||
|
})), 'knowledge')
|
||||||
|
setVisible(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 暴露给父组件的方法
|
||||||
|
useImperativeHandle(ref, () => ({
|
||||||
|
handleOpen,
|
||||||
|
handleClose
|
||||||
|
}));
|
||||||
|
const handleSearch = (value?: string) => {
|
||||||
|
setQuery({keywords: value})
|
||||||
|
setSelectedIds([])
|
||||||
|
setSelectedRows([])
|
||||||
|
}
|
||||||
|
const handleSelect = (item: KnowledgeBase) => {
|
||||||
|
const index = selectedIds.indexOf(item.id)
|
||||||
|
if (index === -1) {
|
||||||
|
setSelectedIds([...selectedIds, item.id])
|
||||||
|
setSelectedRows([...selectedRows, item])
|
||||||
|
} else {
|
||||||
|
setSelectedIds(selectedIds.filter(id => id !== item.id))
|
||||||
|
setSelectedRows(selectedRows.filter(row => row.id !== item.id))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (list.length && selectedList.length) {
|
||||||
|
const unSelectedList = list.filter(item => selectedList.findIndex(vo => vo.id === item.id) < 0)
|
||||||
|
setFilterList([...unSelectedList])
|
||||||
|
} else if (list.length) {
|
||||||
|
setFilterList([...list])
|
||||||
|
}
|
||||||
|
}, [list, selectedList])
|
||||||
|
|
||||||
|
return (
|
||||||
|
<>
|
||||||
|
<RbModal
|
||||||
|
title={t('application.chooseKnowledge')}
|
||||||
|
open={visible}
|
||||||
|
onCancel={handleClose}
|
||||||
|
okText={t('common.save')}
|
||||||
|
onOk={handleSave}
|
||||||
|
width={1000}
|
||||||
|
>
|
||||||
|
<Space size={24} direction="vertical" className="rb:w-full">
|
||||||
|
<SearchInput
|
||||||
|
placeholder={t('knowledgeBase.searchPlaceholder')}
|
||||||
|
onSearch={handleSearch}
|
||||||
|
style={{ width: '100%' }}
|
||||||
|
/>
|
||||||
|
{filterList.length === 0
|
||||||
|
? <Empty />
|
||||||
|
: <List
|
||||||
|
grid={{ gutter: 16, column: 2 }}
|
||||||
|
dataSource={filterList}
|
||||||
|
renderItem={(item: KnowledgeBase) => (
|
||||||
|
<List.Item>
|
||||||
|
<div key={item.id} className={clsx("rb:flex rb:items-center rb:justify-between rb:border rb:rounded-lg rb:p-[17px_16px] rb:cursor-pointer rb:hover:bg-[#F0F3F8]", {
|
||||||
|
"rb:bg-[rgba(21,94,239,0.06)] rb:border-[#155EEF] rb:text-[#155EEF]": selectedIds.includes(item.id),
|
||||||
|
"rb:border-[#DFE4ED] rb:text-[#212332]": !selectedIds.includes(item.id),
|
||||||
|
})} onClick={() => handleSelect(item)}>
|
||||||
|
<div className="rb:text-[16px] rb:leading-5.5">
|
||||||
|
{item.name}
|
||||||
|
<div className="rb:text-[12px] rb:leading-4 rb:text-[#5B6167] rb:mt-2">{t('application.contains', {include_count: item.doc_num})}</div>
|
||||||
|
</div>
|
||||||
|
<div className="rb:text-[12px] rb:leading-4 rb:text-[#5B6167]">{formatDateTime(item.created_at, 'YYYY-MM-DD HH:mm:ss')}</div>
|
||||||
|
</div>
|
||||||
|
</List.Item>
|
||||||
|
)}
|
||||||
|
/>
|
||||||
|
}
|
||||||
|
</Space>
|
||||||
|
</RbModal>
|
||||||
|
</>
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
export default KnowledgeListModal;
|
||||||
@@ -0,0 +1,32 @@
|
|||||||
|
import type { KnowledgeBaseListItem } from '@/views/KnowledgeBase/types'
|
||||||
|
|
||||||
|
export interface RerankerConfig {
|
||||||
|
rerank_model?: boolean | undefined;
|
||||||
|
reranker_id?: string | undefined;
|
||||||
|
reranker_top_k?: number | undefined;
|
||||||
|
}
|
||||||
|
export type RetrieveType = 'participle' | 'semantic' | 'hybrid'
|
||||||
|
export interface KnowledgeConfigForm {
|
||||||
|
kb_id?: string;
|
||||||
|
similarity_threshold?: number;
|
||||||
|
vector_similarity_weight?: number;
|
||||||
|
top_k?: number;
|
||||||
|
retrieve_type?: RetrieveType;
|
||||||
|
}
|
||||||
|
export interface KnowledgeBase extends KnowledgeBaseListItem, KnowledgeConfigForm {
|
||||||
|
config?: KnowledgeConfigForm
|
||||||
|
}
|
||||||
|
export interface KnowledgeConfig extends RerankerConfig {
|
||||||
|
knowledge_bases: KnowledgeBase[];
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface KnowledgeConfigModalRef {
|
||||||
|
handleOpen: (data: KnowledgeBase) => void;
|
||||||
|
}
|
||||||
|
export interface KnowledgeGlobalConfigModalRef {
|
||||||
|
handleOpen: () => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface KnowledgeModalRef {
|
||||||
|
handleOpen: (config?: KnowledgeConfig[]) => void;
|
||||||
|
}
|
||||||
@@ -0,0 +1,82 @@
|
|||||||
|
import { type FC } from 'react'
|
||||||
|
import { Select, type SelectProps } from 'antd'
|
||||||
|
import type { Suggestion } from '../Editor/plugin/AutocompletePlugin'
|
||||||
|
type LabelRender = SelectProps['labelRender'];
|
||||||
|
|
||||||
|
interface VariableSelectProps extends SelectProps {
|
||||||
|
options: Suggestion[];
|
||||||
|
value?: string;
|
||||||
|
onChange?: (value: string) => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
const VariableSelect: FC<VariableSelectProps> = ({
|
||||||
|
placeholder,
|
||||||
|
options,
|
||||||
|
value,
|
||||||
|
onChange,
|
||||||
|
}) => {
|
||||||
|
|
||||||
|
const handleChange = (value: string) => {
|
||||||
|
onChange?.(value);
|
||||||
|
}
|
||||||
|
const labelRender: LabelRender = (props) => {
|
||||||
|
const { value } = props
|
||||||
|
const filterOption = options.find(vo => vo.value === value)
|
||||||
|
|
||||||
|
if (filterOption) {
|
||||||
|
return (
|
||||||
|
<span
|
||||||
|
className="rb:border rb:border-[#DFE4ED] rb:rounded-md rb:bg-white rb:leading-5.5! rb:text-[12px] rb:inline-flex rb:items-center rb:px-1.5 rb:cursor-pointer"
|
||||||
|
contentEditable={false}
|
||||||
|
>
|
||||||
|
<img
|
||||||
|
src={filterOption.nodeData?.icon}
|
||||||
|
style={{ width: '12px', height: '12px', marginRight: '4px' }}
|
||||||
|
alt=""
|
||||||
|
/>
|
||||||
|
{filterOption.nodeData?.name}
|
||||||
|
<span className="rb:text-[#DFE4ED] rb:mx-0.5">/</span>
|
||||||
|
<span className="rb:text-[#155EEF]">{filterOption.label}</span>
|
||||||
|
</span>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
const groupedSuggestions = options.reduce((groups: Record<string, any[]>, suggestion) => {
|
||||||
|
const { nodeData } = suggestion
|
||||||
|
const nodeId = nodeData.id as string;
|
||||||
|
if (!groups[nodeId]) {
|
||||||
|
groups[nodeId] = [];
|
||||||
|
}
|
||||||
|
groups[nodeId].push(suggestion);
|
||||||
|
return groups;
|
||||||
|
}, {});
|
||||||
|
|
||||||
|
const groupedOptions = Object.entries(groupedSuggestions).map(([nodeId, suggestions]) => ({
|
||||||
|
label: suggestions[0].nodeData.name,
|
||||||
|
options: suggestions.map(s => ({ label: s.label, value: s.value }))
|
||||||
|
}));
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Select
|
||||||
|
placeholder={placeholder}
|
||||||
|
value={value}
|
||||||
|
style={{ width: '100%' }}
|
||||||
|
options={groupedOptions}
|
||||||
|
labelRender={labelRender}
|
||||||
|
onChange={handleChange}
|
||||||
|
showSearch
|
||||||
|
filterOption={(input, option) => {
|
||||||
|
if (option?.options) {
|
||||||
|
return option.label?.toLowerCase().includes(input.toLowerCase()) ||
|
||||||
|
option.options.some((opt: any) =>
|
||||||
|
opt.value.toLowerCase().includes(input.toLowerCase())
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return option?.label?.toLowerCase().includes(input.toLowerCase()) ?? false;
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default VariableSelect
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { type FC, useEffect, useState, useRef } from "react";
|
import { type FC, useEffect, useState, useRef, useMemo } from "react";
|
||||||
import { useTranslation } from 'react-i18next'
|
import { useTranslation } from 'react-i18next'
|
||||||
import { Graph, Node } from '@antv/x6';
|
import { Graph, Node } from '@antv/x6';
|
||||||
import { Form, Input, Button, Select, InputNumber, Slider, Space, Divider, App } from 'antd'
|
import { Form, Input, Button, Select, InputNumber, Slider, Space, Divider, App } from 'antd'
|
||||||
@@ -9,6 +9,9 @@ import emptyIcon from '@/assets/images/workflow/empty.png'
|
|||||||
import CustomSelect from "@/components/CustomSelect";
|
import CustomSelect from "@/components/CustomSelect";
|
||||||
import VariableEditModal from './VariableEditModal';
|
import VariableEditModal from './VariableEditModal';
|
||||||
import MessageEditor from './MessageEditor'
|
import MessageEditor from './MessageEditor'
|
||||||
|
import Knowledge from './Knowledge/Knowledge';
|
||||||
|
import type { Suggestion } from '../Editor/plugin/AutocompletePlugin'
|
||||||
|
import VariableSelect from './VariableSelect';
|
||||||
|
|
||||||
interface PropertiesProps {
|
interface PropertiesProps {
|
||||||
selectedNode?: Node | null;
|
selectedNode?: Node | null;
|
||||||
@@ -67,7 +70,16 @@ const Properties: FC<PropertiesProps> = ({
|
|||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (values && selectedNode) {
|
if (values && selectedNode) {
|
||||||
const { id, ...rest } = values
|
const { id, knowledge_retrieval, ...rest } = values
|
||||||
|
const { knowledge_bases, ...restKnowledgeConfig } = knowledge_retrieval || {}
|
||||||
|
let allRest = {
|
||||||
|
...rest,
|
||||||
|
...restKnowledgeConfig,
|
||||||
|
knowledge_bases: knowledge_bases?.map(vo => ({
|
||||||
|
id: vo.id,
|
||||||
|
...vo.config
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
|
||||||
Object.keys(values).forEach(key => {
|
Object.keys(values).forEach(key => {
|
||||||
if (selectedNode.data?.config[key]) {
|
if (selectedNode.data?.config[key]) {
|
||||||
@@ -75,7 +87,10 @@ const Properties: FC<PropertiesProps> = ({
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
selectedNode?.setData({ ...selectedNode.data, ...rest })
|
selectedNode?.setData({
|
||||||
|
...selectedNode.data,
|
||||||
|
...allRest,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}, [values, selectedNode])
|
}, [values, selectedNode])
|
||||||
|
|
||||||
@@ -119,6 +134,83 @@ const Properties: FC<PropertiesProps> = ({
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
const variableList = useMemo(() => {
|
||||||
|
if (!selectedNode || !graphRef?.current) return [];
|
||||||
|
|
||||||
|
const variableList: Suggestion[] = [];
|
||||||
|
const graph = graphRef.current;
|
||||||
|
const edges = graph.getEdges();
|
||||||
|
const nodes = graph.getNodes();
|
||||||
|
|
||||||
|
// Find all connected previous nodes (recursive)
|
||||||
|
const getAllPreviousNodes = (nodeId: string, visited = new Set<string>()): string[] => {
|
||||||
|
if (visited.has(nodeId)) return [];
|
||||||
|
visited.add(nodeId);
|
||||||
|
|
||||||
|
const directPrevious = edges
|
||||||
|
.filter(edge => edge.getTargetCellId() === nodeId)
|
||||||
|
.map(edge => edge.getSourceCellId());
|
||||||
|
|
||||||
|
const allPrevious = [...directPrevious];
|
||||||
|
directPrevious.forEach(prevNodeId => {
|
||||||
|
allPrevious.push(...getAllPreviousNodes(prevNodeId, visited));
|
||||||
|
});
|
||||||
|
|
||||||
|
return allPrevious;
|
||||||
|
};
|
||||||
|
|
||||||
|
const allPreviousNodeIds = getAllPreviousNodes(selectedNode.id);
|
||||||
|
|
||||||
|
allPreviousNodeIds.forEach(nodeId => {
|
||||||
|
const node = nodes.find(n => n.id === nodeId);
|
||||||
|
if (!node) return;
|
||||||
|
|
||||||
|
const nodeData = node.getData();
|
||||||
|
|
||||||
|
switch(nodeData.type) {
|
||||||
|
case 'start':
|
||||||
|
const list = [
|
||||||
|
...(nodeData.config?.variables?.defaultValue ?? []),
|
||||||
|
...(nodeData.config?.variables?.value ?? [])
|
||||||
|
]
|
||||||
|
list.forEach((variable: any) => {
|
||||||
|
variableList.push({
|
||||||
|
key: `${nodeId}_${variable.name}`,
|
||||||
|
label: variable.name,
|
||||||
|
type: 'variable',
|
||||||
|
dataType: variable.type,
|
||||||
|
value: `{{${nodeId}.${variable.name}}}`,
|
||||||
|
nodeData: nodeData,
|
||||||
|
});
|
||||||
|
});
|
||||||
|
nodeData.config?.variables?.sys.forEach((variable: any) => {
|
||||||
|
variableList.push({
|
||||||
|
key: `${nodeId}_${variable.name}`,
|
||||||
|
label: `sys.${variable.name}`,
|
||||||
|
type: 'variable',
|
||||||
|
dataType: variable.type,
|
||||||
|
value: `{{sys.${variable.name}}}`,
|
||||||
|
nodeData: nodeData,
|
||||||
|
});
|
||||||
|
});
|
||||||
|
break
|
||||||
|
case 'llm':
|
||||||
|
variableList.push({
|
||||||
|
key: `${nodeId}_output`,
|
||||||
|
label: 'output',
|
||||||
|
type: 'variable',
|
||||||
|
dataType: 'String',
|
||||||
|
value: `${nodeId}.output`,
|
||||||
|
nodeData: nodeData,
|
||||||
|
});
|
||||||
|
break
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return variableList;
|
||||||
|
}, [selectedNode, graphRef]);
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="rb:w-75 rb:fixed rb:right-0 rb:top-16 rb:bottom-0 rb:p-3">
|
<div className="rb:w-75 rb:fixed rb:right-0 rb:top-16 rb:bottom-0 rb:p-3">
|
||||||
<div className="rb:font-medium rb:leading-5 rb:mb-3">{t('workflow.nodeProperties')}</div>
|
<div className="rb:font-medium rb:leading-5 rb:mb-3">{t('workflow.nodeProperties')}</div>
|
||||||
@@ -203,6 +295,17 @@ const Properties: FC<PropertiesProps> = ({
|
|||||||
return null
|
return null
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (config.type === 'knowledge') {
|
||||||
|
return (
|
||||||
|
<Form.Item
|
||||||
|
key={key}
|
||||||
|
name={key}
|
||||||
|
>
|
||||||
|
<Knowledge />
|
||||||
|
</Form.Item>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<Form.Item
|
<Form.Item
|
||||||
key={key}
|
key={key}
|
||||||
@@ -231,6 +334,11 @@ const Properties: FC<PropertiesProps> = ({
|
|||||||
valueKey={config.valueKey}
|
valueKey={config.valueKey}
|
||||||
labelKey={config.labelKey}
|
labelKey={config.labelKey}
|
||||||
/>
|
/>
|
||||||
|
: config.type === 'variableList'
|
||||||
|
? <VariableSelect
|
||||||
|
placeholder={t('common.pleaseSelect')}
|
||||||
|
options={variableList}
|
||||||
|
/>
|
||||||
: null
|
: null
|
||||||
}
|
}
|
||||||
</Form.Item>
|
</Form.Item>
|
||||||
|
|||||||
@@ -131,7 +131,16 @@ export const nodeLibrary: NodeLibrary[] = [
|
|||||||
},
|
},
|
||||||
// { type: "model_selection", icon: modelSelectionIcon },
|
// { type: "model_selection", icon: modelSelectionIcon },
|
||||||
// { type: "model_voting", icon: modelVotingIcon },
|
// { type: "model_voting", icon: modelVotingIcon },
|
||||||
// { type: "rag", icon: ragIcon },
|
{ type: "knowledge-retrieval", icon: ragIcon,
|
||||||
|
config: {
|
||||||
|
query: {
|
||||||
|
type: 'variableList',
|
||||||
|
},
|
||||||
|
knowledge_retrieval: {
|
||||||
|
type: 'knowledge'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
// { type: "classification", icon: classificationIcon },
|
// { type: "classification", icon: classificationIcon },
|
||||||
// { type: "parameter_extraction", icon: parameterExtractionIcon }
|
// { type: "parameter_extraction", icon: parameterExtractionIcon }
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -32,7 +32,7 @@ export interface UseWorkflowGraphReturn {
|
|||||||
deleteEvent: () => boolean | void;
|
deleteEvent: () => boolean | void;
|
||||||
copyEvent: () => boolean | void;
|
copyEvent: () => boolean | void;
|
||||||
parseEvent: () => boolean | void;
|
parseEvent: () => boolean | void;
|
||||||
handleSave: (flag?: boolean) => Promise<any>;
|
handleSave: (flag?: boolean) => Promise<unknown>;
|
||||||
}
|
}
|
||||||
|
|
||||||
const edge_color = '#155EEF';
|
const edge_color = '#155EEF';
|
||||||
@@ -83,7 +83,13 @@ export const useWorkflowGraph = ({
|
|||||||
|
|
||||||
if (nodeLibraryConfig?.config) {
|
if (nodeLibraryConfig?.config) {
|
||||||
Object.keys(nodeLibraryConfig.config).forEach(key => {
|
Object.keys(nodeLibraryConfig.config).forEach(key => {
|
||||||
if (nodeLibraryConfig.config && nodeLibraryConfig.config[key] && config[key]) {
|
if (key === 'knowledge_retrieval' && nodeLibraryConfig.config && nodeLibraryConfig.config[key]) {
|
||||||
|
const { query, ...rest } = config
|
||||||
|
nodeLibraryConfig.config[key].defaultValue = {
|
||||||
|
...rest
|
||||||
|
}
|
||||||
|
console.log(type, config, nodeLibraryConfig)
|
||||||
|
} else if (nodeLibraryConfig.config && nodeLibraryConfig.config[key] && config[key]) {
|
||||||
nodeLibraryConfig.config[key].defaultValue = config[key]
|
nodeLibraryConfig.config[key].defaultValue = config[key]
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@@ -686,12 +692,17 @@ export const useWorkflowGraph = ({
|
|||||||
nodes: nodes.map((node: Node) => {
|
nodes: nodes.map((node: Node) => {
|
||||||
const data = node.getData();
|
const data = node.getData();
|
||||||
const position = node.getPosition();
|
const position = node.getPosition();
|
||||||
const config: Record<string, any> = {}
|
let itemConfig: Record<string, any> = {}
|
||||||
|
|
||||||
if (data.config) {
|
if (data.config) {
|
||||||
Object.keys(data.config).forEach(key => {
|
Object.keys(data.config).forEach(key => {
|
||||||
if (data.config[key] && 'defaultValue' in data.config[key]) {
|
if (data.config[key] && 'defaultValue' in data.config[key] && key !== 'knowledge_retrieval') {
|
||||||
config[key] = data.config[key].defaultValue
|
itemConfig[key] = data.config[key].defaultValue
|
||||||
|
} else if (key === 'knowledge_retrieval' && data.config[key] && 'defaultValue' in data.config[key]) {
|
||||||
|
itemConfig = {
|
||||||
|
...itemConfig,
|
||||||
|
...data.config[key].defaultValue
|
||||||
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@@ -704,7 +715,7 @@ export const useWorkflowGraph = ({
|
|||||||
x: position.x,
|
x: position.x,
|
||||||
y: position.y,
|
y: position.y,
|
||||||
},
|
},
|
||||||
config: config
|
config: itemConfig
|
||||||
};
|
};
|
||||||
}),
|
}),
|
||||||
edges: edges.map((edge: Edge) => {
|
edges: edges.map((edge: Edge) => {
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
|
|
||||||
import { Graph } from '@antv/x6';
|
import { Graph } from '@antv/x6';
|
||||||
export interface NodeConfig {
|
export interface NodeConfig {
|
||||||
type: 'input' | 'textarea' | 'select' | 'inputNumber' | 'slider' | 'customSelect' | 'define';
|
type: 'input' | 'textarea' | 'select' | 'inputNumber' | 'slider' | 'customSelect' | 'define' | 'knowledge' | 'variableList';
|
||||||
options?: { label: string; value: string }[];
|
options?: { label: string; value: string }[];
|
||||||
|
|
||||||
max?: number;
|
max?: number;
|
||||||
|
|||||||
Reference in New Issue
Block a user