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

fix

Browse Source
This commit is contained in:
jyong
2025-08-25 13:55:42 +08:00
parent b7ed2cade1
commit cc1f0d4d8d
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
api/services/rag_pipeline/rag_pipeline_dsl_service.py
View File

@ -37,6 +37,7 @@ from models import Account
from models.dataset import Dataset, DatasetCollectionBinding, Pipeline from models.dataset import Dataset, DatasetCollectionBinding, Pipeline
from models.workflow import Workflow, WorkflowType from models.workflow import Workflow, WorkflowType
from services.entities.knowledge_entities.rag_pipeline_entities import ( from services.entities.knowledge_entities.rag_pipeline_entities import (
IconInfo,
KnowledgeConfiguration, KnowledgeConfiguration,
RagPipelineDatasetCreateEntity, RagPipelineDatasetCreateEntity,
) )
@ -126,6 +127,7 @@ class RagPipelineDslService:
pipeline_id: Optional[str] = None, pipeline_id: Optional[str] = None,
dataset: Optional[Dataset] = None, dataset: Optional[Dataset] = None,
dataset_name: Optional[str] = None, dataset_name: Optional[str] = None,
icon_info: Optional[IconInfo] = None,
) -> RagPipelineImportInfo: ) -> RagPipelineImportInfo:
"""Import an app from YAML content or URL.""" """Import an app from YAML content or URL."""
import_id = str(uuid.uuid4()) import_id = str(uuid.uuid4())
@ -274,10 +276,16 @@ class RagPipelineDslService:
# create dataset # create dataset
name = dataset_name or pipeline.name name = dataset_name or pipeline.name
description = pipeline.description description = pipeline.description
icon_type = data.get("rag_pipeline", {}).get("icon_type") if icon_info:
icon = data.get("rag_pipeline", {}).get("icon") icon_type = icon_info.icon_type
icon_background = data.get("rag_pipeline", {}).get("icon_background") icon = icon_info.icon
icon_url = data.get("rag_pipeline", {}).get("icon_url") icon_background = icon_info.icon_background
icon_url = icon_info.icon_url
else:
icon_type = data.get("rag_pipeline", {}).get("icon_type")
icon = data.get("rag_pipeline", {}).get("icon")
icon_background = data.get("rag_pipeline", {}).get("icon_background")
icon_url = data.get("rag_pipeline", {}).get("icon_url")
workflow = data.get("workflow", {}) workflow = data.get("workflow", {})
graph = workflow.get("graph", {}) graph = workflow.get("graph", {})
nodes = graph.get("nodes", []) nodes = graph.get("nodes", [])
@ -925,6 +933,7 @@ class RagPipelineDslService:
yaml_content=rag_pipeline_dataset_create_entity.yaml_content, yaml_content=rag_pipeline_dataset_create_entity.yaml_content,
dataset=None, dataset=None,
dataset_name=rag_pipeline_dataset_create_entity.name, dataset_name=rag_pipeline_dataset_create_entity.name,
icon_info=rag_pipeline_dataset_create_entity.icon_info,
) )
return { return {
"id": rag_pipeline_import_info.id, "id": rag_pipeline_import_info.id,