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

feat/datasource
twwu 11 months ago
commit 346d066128

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

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

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

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

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

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

Loading…
Cancel
Save