feat: workflow add knowledge-retrieval node

This commit is contained in:
zhaoying
2025-12-29 14:39:57 +08:00
parent 06e5f4f8ff
commit 0fce86f76b
12 changed files with 899 additions and 27 deletions

View File

@@ -15,6 +15,7 @@ interface RadioCardProps extends Omit<RadioGroupProps, 'onChange'> {
onValueChange?: (value: string | null | undefined, option?: RadioCardOption) => void;
onChange?: (value: string | null | undefined, option?: RadioCardOption) => void;
itemRender?: (option: RadioCardOption) => ReactNode;
allowClear?: boolean;
}
const RadioGroupCard: FC<RadioCardProps> = ({
@@ -22,7 +23,8 @@ const RadioGroupCard: FC<RadioCardProps> = ({
value,
onValueChange,
onChange,
itemRender
itemRender,
allowClear = true
}) => {
// 监听value变化
useEffect(() => {
@@ -34,23 +36,27 @@ const RadioGroupCard: FC<RadioCardProps> = ({
const handleChange = (option: RadioCardOption) => {
if (option.disabled) return
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 (
<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 => (
<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:border-[#EBEBEB] rb:bg-[#ffffff]': option.value !== value,
'rb:opacity-[0.75]': option.disabled
})} onClick={() => handleChange(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: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>

View File

@@ -393,12 +393,6 @@ export const zh = {
clusterName: '集群名称',
clusterDescription: '集群描述',
clusterDescriptionPlaceholder: '这是一个专门处理核心业务的Agent集群能够协作完成复杂的业务处理任务。',
toolCalling: '工具调用',
toolCallingDesc: '主控代理将子代理作为工具调用',
toolCallingFeature: '集中控制,适合结构化工作流',
handoffs: '交接',
handoffsDesc: '代理之间动态转移控制权',
handoffsFeature: '去中心化控制,适合复杂对话场景',
recommend: '推荐',
advanced: '高级',
multiAgentArchitecture: '多代理架构模式',
@@ -425,7 +419,7 @@ export const zh = {
stateSharingStrategy: '状态共享策略',
intermediateResultProcessing: '中间结果处理',
metadataTransfer: '元数据传输',
knowledgeConfig: '知识库配置',
temperature: '温度',
temperature_desc: '温度参数,控制输出的随机性',
max_tokens: '最大令牌数',
@@ -453,9 +447,9 @@ export const zh = {
releasePreview: '发布预览',
globalConfig: '全局配置',
globalConfigDesc: '全局配置将应用于所有关联的知识库作为默认配置。单个知识库的配置将覆盖全局配置。',
rerankModel: '重排序模型',
rerankModel: 'Reranker 模型',
rerankModelDesc: '激活后,搜索结果将被重新排序以提高相关性',
rearrangementModel: '重排序模型',
rearrangementModel: 'Reranker 模型',
rearrangementModelDesc: '选择用于重新排序的模型',
reranker_top_k: 'Top K',
reranker_top_k_desc: '返回的文档片段数量范围从1到20',
@@ -514,6 +508,21 @@ export const zh = {
promptChatPlaceholder: '描述你需要的提示词,例如:我需要一个客服助手',
promptChatEmpty: '目前没有对话内容',
promptEmpty: '在左侧描述您的用例,编排预览将在此处显示。',
master: '主管模式',
master_agent: '主管模式',
masterDesc: '由主 Agent 统一调度和管理,子 Agent 按照主管分配的任务执行,适合需要集中控制的场景。',
handoffs: '协作模式',
handoffsDesc: '多个 Agent 平等协作,根据任务需求自主协调配合,适合需要灵活互动的复杂场景。',
masterConfig: '主管配置',
orchestrationMode: '任务分配策略',
conditional: '智能分配',
sequential: '顺序分配',
parallel: '并行分配',
aggregationStrategy: '结果汇总方式',
merge: '完整汇总',
vote: '关键信息提取',
priority: '结构化整合',
},
// 角色管理相关翻译
role: {
@@ -1629,7 +1638,7 @@ export const zh = {
llm: '大语言模型 (LLM)',
model_selection: '多模型选择',
model_voting: '多模型投票',
rag: '知识检索 (RAG)',
'knowledge-retrieval': '知识检索 (RAG)',
classification: '智能分类',
parameter_extraction: '参数提取',
flowControl: '流程控制',
@@ -1694,6 +1703,11 @@ export const zh = {
},
end: {
output: '回复'
},
'knowledge-retrieval': {
query: '查询变量',
knowledge_retrieval: '知识库',
recallConfig: '召回测试',
}
},

View File

@@ -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

View File

@@ -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;

View File

@@ -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;

View File

@@ -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;

View File

@@ -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;
}

View File

@@ -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

View File

@@ -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 { Graph, Node } from '@antv/x6';
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 VariableEditModal from './VariableEditModal';
import MessageEditor from './MessageEditor'
import Knowledge from './Knowledge/Knowledge';
import type { Suggestion } from '../Editor/plugin/AutocompletePlugin'
import VariableSelect from './VariableSelect';
interface PropertiesProps {
selectedNode?: Node | null;
@@ -67,7 +70,16 @@ const Properties: FC<PropertiesProps> = ({
useEffect(() => {
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 => {
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])
@@ -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 (
<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>
@@ -203,6 +295,17 @@ const Properties: FC<PropertiesProps> = ({
return null
}
if (config.type === 'knowledge') {
return (
<Form.Item
key={key}
name={key}
>
<Knowledge />
</Form.Item>
)
}
return (
<Form.Item
key={key}
@@ -231,6 +334,11 @@ const Properties: FC<PropertiesProps> = ({
valueKey={config.valueKey}
labelKey={config.labelKey}
/>
: config.type === 'variableList'
? <VariableSelect
placeholder={t('common.pleaseSelect')}
options={variableList}
/>
: null
}
</Form.Item>

View File

@@ -131,7 +131,16 @@ export const nodeLibrary: NodeLibrary[] = [
},
// { type: "model_selection", icon: modelSelectionIcon },
// { 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: "parameter_extraction", icon: parameterExtractionIcon }
]

View File

@@ -32,7 +32,7 @@ export interface UseWorkflowGraphReturn {
deleteEvent: () => boolean | void;
copyEvent: () => boolean | void;
parseEvent: () => boolean | void;
handleSave: (flag?: boolean) => Promise<any>;
handleSave: (flag?: boolean) => Promise<unknown>;
}
const edge_color = '#155EEF';
@@ -83,7 +83,13 @@ export const useWorkflowGraph = ({
if (nodeLibraryConfig?.config) {
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]
}
})
@@ -686,12 +692,17 @@ export const useWorkflowGraph = ({
nodes: nodes.map((node: Node) => {
const data = node.getData();
const position = node.getPosition();
const config: Record<string, any> = {}
let itemConfig: Record<string, any> = {}
if (data.config) {
Object.keys(data.config).forEach(key => {
if (data.config[key] && 'defaultValue' in data.config[key]) {
config[key] = data.config[key].defaultValue
if (data.config[key] && 'defaultValue' in data.config[key] && key !== 'knowledge_retrieval') {
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,
y: position.y,
},
config: config
config: itemConfig
};
}),
edges: edges.map((edge: Edge) => {

View File

@@ -1,7 +1,7 @@
import { Graph } from '@antv/x6';
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 }[];
max?: number;