Merge branch 'feat/rag-pipeline' into deploy/rag-dev

This commit is contained in:
twwu 2025-06-11 18:12:17 +08:00
commit 346d066128
6 changed files with 21 additions and 17 deletions

View File

@ -149,22 +149,26 @@ const DatasetDetailLayout: FC<IAppDetailLayoutProps> = (props) => {
]
if (datasetRes?.provider !== 'external') {
baseNavigation.unshift(...[{
if (datasetRes?.pipeline_id) {
baseNavigation.unshift({
name: t('common.datasetMenus.pipeline'),
href: `/datasets/${datasetId}/pipeline`,
icon: PipelineLine as RemixiconComponentType,
selectedIcon: PipelineFill as RemixiconComponentType,
disabled: false,
})
}
baseNavigation.unshift({
name: t('common.datasetMenus.documents'),
href: `/datasets/${datasetId}/documents`,
icon: RiFileTextLine,
selectedIcon: RiFileTextFill,
disabled: isButtonDisabledWithPipeline,
}, {
name: t('common.datasetMenus.pipeline'),
href: `/datasets/${datasetId}/pipeline`,
icon: PipelineLine as RemixiconComponentType,
selectedIcon: PipelineFill as RemixiconComponentType,
disabled: false,
}])
})
}
return baseNavigation
}, [t, datasetId, isButtonDisabledWithPipeline, datasetRes?.provider])
}, [t, datasetId, isButtonDisabledWithPipeline, datasetRes?.provider, datasetRes?.pipeline_id])
useDocumentTitle(datasetRes?.name || t('common.menus.datasets'))

View File

@ -108,7 +108,7 @@ const TemplateCard = ({
const blob = new Blob([res.data], { type: 'application/yaml' })
downloadFile({
data: blob,
fileName: `${pipeline.name}.dsl`,
fileName: `${pipeline.name}.yml`,
})
Toast.notify({
type: 'success',

View File

@ -73,7 +73,7 @@ const DatasetCard = ({
if (availableDocCount === dataset.document_count)
return t('dataset.docAllEnabled', { count: availableDocCount })
if (availableDocCount < dataset.document_count)
return t('dataset.docAllEnabled', { count: dataset.document_count, num: availableDocCount })
return t('dataset.partialEnabled', { count: dataset.document_count, num: availableDocCount })
}, [t, dataset.document_count, dataset.total_available_documents])
const language = useGetLanguage()

View File

@ -25,7 +25,7 @@ const translation = {
externalKnowledgeBase: 'External Knowledge Base',
localDocs: 'Local Docs',
documentCount: ' docs',
docAllEnabled_one: 'All {{count}} document enabled',
docAllEnabled_one: '{{count}} document enabled',
docAllEnabled_other: 'All {{count}} documents enabled',
partialEnabled_one: 'Total of {{count}} document, {{num}} available',
partialEnabled_other: 'Total of {{count}} documents, {{num}} available',

View File

@ -25,7 +25,7 @@ const translation = {
externalKnowledgeBase: '外部知识库',
localDocs: '本地文档',
documentCount: ' 文档',
docAllEnabled_one: '所有 {{count}} 个文档可用',
docAllEnabled_one: '{{count}} 个文档可用',
docAllEnabled_other: '所有 {{count}} 个文档均可用',
partialEnabled_one: '共计 {{count}} 个文档, {{num}} 可用',
partialEnabled_other: '共计 {{count}} 个文档, {{num}} 可用',

View File

@ -33,7 +33,7 @@ import type { IconInfo } from '@/models/datasets'
const NAME_SPACE = 'pipeline'
export const PipelineTemplateListQueryKeyPrefix = [NAME_SPACE, 'template', 'list']
export const PipelineTemplateListQueryKeyPrefix = [NAME_SPACE, 'template-list']
export const usePipelineTemplateList = (params: PipelineTemplateListParams) => {
return useQuery<PipelineTemplateListResponse>({
queryKey: [...PipelineTemplateListQueryKeyPrefix, params.type],
@ -62,7 +62,7 @@ export const useUpdateTemplateInfo = (
mutationOptions: MutationOptions<UpdateTemplateInfoResponse, Error, UpdateTemplateInfoRequest> = {},
) => {
return useMutation({
mutationKey: [NAME_SPACE, 'template', 'update'],
mutationKey: [NAME_SPACE, 'template-update'],
mutationFn: (request: UpdateTemplateInfoRequest) => {
const { template_id, ...rest } = request
return patch<UpdateTemplateInfoResponse>(`/rag/pipeline/customized/templates/${template_id}`, {
@ -77,7 +77,7 @@ export const useDeleteTemplate = (
mutationOptions: MutationOptions<DeleteTemplateResponse, Error, string> = {},
) => {
return useMutation({
mutationKey: [NAME_SPACE, 'template', 'delete'],
mutationKey: [NAME_SPACE, 'template-delete'],
mutationFn: (templateId: string) => {
return del<DeleteTemplateResponse>(`/rag/pipeline/customized/templates/${templateId}`)
},
@ -89,7 +89,7 @@ export const useExportTemplateDSL = (
mutationOptions: MutationOptions<ExportTemplateDSLResponse, Error, string> = {},
) => {
return useMutation({
mutationKey: [NAME_SPACE, 'dsl-export'],
mutationKey: [NAME_SPACE, 'template-dsl-export'],
mutationFn: (templateId: string) => {
return post<ExportTemplateDSLResponse>(`/rag/pipeline/customized/templates/${templateId}`)
},