opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-27 02:36:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2

Browse Source
This commit is contained in:
jyong 2025-06-11 16:36:36 +08:00
parent a6f7560d2f
commit 66fa68fa18
3 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/app/apps/common/workflow_response_converter.py
View File

@ -43,10 +43,12 @@ from core.app.entities.task_entities import (
WorkflowStartStreamResponse, WorkflowStartStreamResponse,
) )
from core.file import FILE_MODEL_IDENTITY, File from core.file import FILE_MODEL_IDENTITY, File
from core.plugin.impl.datasource import PluginDatasourceManager
from core.tools.tool_manager import ToolManager from core.tools.tool_manager import ToolManager
from core.workflow.entities.workflow_execution import WorkflowExecution from core.workflow.entities.workflow_execution import WorkflowExecution
from core.workflow.entities.workflow_node_execution import WorkflowNodeExecution, WorkflowNodeExecutionStatus from core.workflow.entities.workflow_node_execution import WorkflowNodeExecution, WorkflowNodeExecutionStatus
from core.workflow.nodes import NodeType from core.workflow.nodes import NodeType
from core.workflow.nodes.datasource.entities import DatasourceNodeData
from core.workflow.nodes.tool.entities import ToolNodeData from core.workflow.nodes.tool.entities import ToolNodeData
from models import ( from models import (
Account, Account,
@ -181,6 +183,11 @@ class WorkflowResponseConverter:
provider_type=node_data.provider_type, provider_type=node_data.provider_type,
provider_id=node_data.provider_id, provider_id=node_data.provider_id,
) )
elif event.node_type == NodeType.DATASOURCE:
node_data = cast(DatasourceNodeData, event.node_data)
manager = PluginDatasourceManager()
provider_entity = manager.fetch_datasource_provider(self._application_generate_entity.app_config.tenant_id, f"{node_data.plugin_id}/{node_data.provider_name}")
response.data.extras["icon"] = provider_entity.declaration.identity.icon
return response return response

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

@ -66,6 +66,9 @@ class CustomizedPipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
) )
if not pipeline_template: if not pipeline_template:
return None return None
dsl_data = yaml.safe_load(pipeline_template.yaml_content)
graph_data = dsl_data.get("workflow", {}).get("graph", {})
return { return {
"id": pipeline_template.id, "id": pipeline_template.id,
@ -74,5 +77,6 @@ class CustomizedPipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
"description": pipeline_template.description, "description": pipeline_template.description,
"chunk_structure": pipeline_template.chunk_structure, "chunk_structure": pipeline_template.chunk_structure,
"export_data": pipeline_template.yaml_content, "export_data": pipeline_template.yaml_content,
"graph": graph_data,
"created_by": pipeline_template.created_user_name, "created_by": pipeline_template.created_user_name,
} }

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

@ -66,13 +66,15 @@ class DatabasePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
if not pipeline_template: if not pipeline_template:
return None return None
dsl_data = yaml.safe_load(pipeline_template.yaml_content)
graph_data = dsl_data.get("workflow", {}).get("graph", {})
return { return {
"id": pipeline_template.id, "id": pipeline_template.id,
"name": pipeline_template.name, "name": pipeline_template.name,
"icon_info": pipeline_template.icon, "icon_info": pipeline_template.icon,
"description": pipeline_template.description,
"chunk_structure": pipeline_template.chunk_structure, "chunk_structure": pipeline_template.chunk_structure,
"export_data": pipeline_template.yaml_content, "export_data": pipeline_template.yaml_content,
"graph": graph_data,
"created_by": pipeline_template.created_user_name, "created_by": pipeline_template.created_user_name,
"description": pipeline_template.description,
} }