opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 11:56:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2 transform

Browse Source
This commit is contained in:
jyong 2025-07-21 14:51:40 +08:00
parent 51d7a9b6be
commit 9c96f1db6c
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
api/services/rag_pipeline/rag_pipeline_transform_service.py
View File

@ -32,9 +32,11 @@ class RagPipelineTransformService:
indexing_technique = dataset.indexing_technique indexing_technique = dataset.indexing_technique
if not datasource_type and not indexing_technique: if not datasource_type and not indexing_technique:
self._transfrom_to_empty_pipeline(dataset)
return return
doc_form = dataset.doc_form doc_form = dataset.doc_form
if not doc_form: if not doc_form:
self._transfrom_to_empty_pipeline(dataset)
return return
retrieval_model = dataset.retrieval_model retrieval_model = dataset.retrieval_model
pipeline_yaml = self._get_transform_yaml(doc_form, datasource_type, indexing_technique) pipeline_yaml = self._get_transform_yaml(doc_form, datasource_type, indexing_technique)
@ -262,3 +264,21 @@ class RagPipelineTransformService:
for identifier in need_install_plugin_unique_identifiers for identifier in need_install_plugin_unique_identifiers
], ],
) )
def _transfrom_to_empty_pipeline(self, dataset: Dataset):
pipeline = Pipeline(
tenant_id=dataset.tenant_id,
name=dataset.name,
description=dataset.description,
created_by=current_user.id,
)
db.session.add(pipeline)
db.session.flush()
dataset.pipeline_id = pipeline.id
dataset.runtime_mode = "rag_pipeline"
dataset.updated_by = current_user.id
dataset.updated_at = datetime.now(UTC).replace(tzinfo=None)
db.session.add(dataset)
db.session.commit()