Bläddra i källkod

refactor: update dataset handling to use runtime_mode instead of pipeline_id

tags/2.0.0-beta.1
twwu 4 månader sedan
förälder
incheckning
6f17200dec

+ 3
- 3
web/app/(commonLayout)/datasets/(datasetDetailLayout)/[datasetId]/layout-main.tsx Visa fil

@@ -125,7 +125,7 @@ const DatasetDetailLayout: FC<IAppDetailLayoutProps> = (props) => {
return true
if (datasetRes.provider === 'external')
return false
if (!datasetRes.pipeline_id)
if (datasetRes.runtime_mode === 'general')
return false
return !datasetRes.is_published
}, [datasetRes])
@@ -149,7 +149,7 @@ const DatasetDetailLayout: FC<IAppDetailLayoutProps> = (props) => {
]

if (datasetRes?.provider !== 'external') {
if (datasetRes?.pipeline_id) {
if (datasetRes?.runtime_mode === 'rag_pipeline') {
baseNavigation.unshift({
name: t('common.datasetMenus.pipeline'),
href: `/datasets/${datasetId}/pipeline`,
@@ -168,7 +168,7 @@ const DatasetDetailLayout: FC<IAppDetailLayoutProps> = (props) => {
}

return baseNavigation
}, [t, datasetId, isButtonDisabledWithPipeline, datasetRes?.provider, datasetRes?.pipeline_id])
}, [t, datasetId, isButtonDisabledWithPipeline, datasetRes?.provider, datasetRes?.runtime_mode])

useDocumentTitle(datasetRes?.name || t('common.menus.datasets'))


+ 3
- 2
web/app/components/datasets/documents/detail/settings/index.tsx Visa fil

@@ -13,9 +13,10 @@ const Settings = ({
datasetId,
documentId,
}: SettingsProps) => {
const pipelineId = useDatasetDetailContextWithSelector(s => s.dataset?.pipeline_id)
const runtimeMode = useDatasetDetailContextWithSelector(s => s.dataset?.runtime_mode)
const isGeneralDataset = runtimeMode === 'general'

if (!pipelineId) {
if (isGeneralDataset) {
return (
<DocumentSettings
datasetId={datasetId}

+ 1
- 1
web/app/components/datasets/documents/index.tsx Visa fil

@@ -173,7 +173,7 @@ const Documents: FC<IDocumentsProps> = ({ datasetId }) => {

const routeToDocCreate = () => {
// if dataset is create from pipeline, redirect to create from pipeline page
if (dataset?.pipeline_id) {
if (dataset?.runtime_mode === 'rag_pipeline') {
router.push(`/datasets/${datasetId}/documents/create-from-pipeline`)
return
}

+ 2
- 2
web/app/components/datasets/list/dataset-card/index.tsx Visa fil

@@ -50,8 +50,8 @@ const DatasetCard = ({
return dataset.provider === EXTERNAL_PROVIDER
}, [dataset.provider])
const isPipelineUnpublished = useMemo(() => {
return !!dataset.pipeline_id && !dataset.is_published
}, [dataset.pipeline_id, dataset.is_published])
return dataset.runtime_mode === 'rag_pipeline' && !dataset.is_published
}, [dataset.runtime_mode, dataset.is_published])
const chunkingModeIcon = dataset.doc_form ? DOC_FORM_ICON_WITH_BG[dataset.doc_form] : React.Fragment
const Icon = isExternalProvider ? DOC_FORM_ICON_WITH_BG.external : chunkingModeIcon
const iconInfo = dataset.icon_info || {

+ 1
- 1
web/app/components/rag-pipeline/components/panel/test-run/data-source/online-documents/online-document-selector.tsx Visa fil

@@ -50,7 +50,7 @@ const OnlineDocumentSelector = ({
datasource_type: DatasourceType.onlineDocument,
}, {
onSuccess(documentsData) {
setDocumentsData(documentsData as DataSourceNotionWorkspace[])
setDocumentsData(documentsData.result as DataSourceNotionWorkspace[])
},
})
}

+ 1
- 0
web/models/datasets.ts Visa fil

@@ -83,6 +83,7 @@ export type DataSet = {
keyword_number?: number
pipeline_id?: string
is_published?: boolean // Indicates if the pipeline is published
runtime_mode: 'rag_pipeline' | 'general'
}

export type ExternalAPIItem = {

+ 1
- 1
web/models/pipeline.ts Visa fil

@@ -167,7 +167,7 @@ export type PipelineDatasourceNodeRunRequest = {
export type PipelineDatasourceNodeRunResponse = {
job_id?: string
status: 'processing' | 'completed'
result: Record<string, any>
result: any
provider_type: DatasourceType
}


Laddar…
Avbryt
Spara