youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 00:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2

Browse Source
This commit is contained in:
jyong
2025-06-11 13:12:18 +08:00
parent 2972a06f16
commit 874e1bc41d
5 changed files with 70 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/services/rag_pipeline/pipeline_template/customized/customized_retrieval.py
View File

@ -70,6 +70,9 @@ class CustomizedPipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
return {
"id": pipeline_template.id,
"name": pipeline_template.name,
"icon": pipeline_template.icon,
"icon_info": pipeline_template.icon,
"description": pipeline_template.description,
"chunk_structure": pipeline_template.chunk_structure,
"export_data": yaml.safe_load(pipeline_template.yaml_content),
"created_by": pipeline_template.created_user_name,
}

4
api/services/rag_pipeline/pipeline_template/database/database_retrieval.py
View File

@ -70,7 +70,9 @@ class DatabasePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
return {
"id": pipeline_template.id,
"name": pipeline_template.name,
"icon": pipeline_template.icon,
"icon_info": pipeline_template.icon,
"chunk_structure": pipeline_template.chunk_structure,
"export_data": yaml.safe_load(pipeline_template.yaml_content),
"created_by": pipeline_template.created_user_name,
"description": pipeline_template.description,
}

3
api/services/rag_pipeline/rag_pipeline.py
View File

@ -116,6 +116,7 @@ class RagPipelineService:
customized_template.name = template_info.name
customized_template.description = template_info.description
customized_template.icon = template_info.icon_info.model_dump()
customized_template.updated_by = current_user.id
db.session.commit()
return customized_template
@ -694,7 +695,6 @@ class RagPipelineService:
db.session.add(document)
db.session.commit()
return workflow_node_execution
def update_workflow(
@ -969,6 +969,7 @@ class RagPipelineService:
position=max_position + 1 if max_position else 1,
chunk_structure=dataset.chunk_structure,
language="en-US",
created_by=current_user.id,
)
db.session.add(pipeline_customized_template)
db.session.commit()