From 5f08a9314ca0920577a70ca9161528bc55d4d10c Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Wed, 11 Jun 2025 17:10:20 +0800
Subject: [PATCH] r2

---
 api/core/plugin/impl/datasource.py        | 2 +-
 api/services/rag_pipeline/rag_pipeline.py | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/api/core/plugin/impl/datasource.py b/api/core/plugin/impl/datasource.py
index 775f36b506..98ee0bb11e 100644
--- a/api/core/plugin/impl/datasource.py
+++ b/api/core/plugin/impl/datasource.py
@@ -242,7 +242,7 @@ class PluginDatasourceManager(BasePluginClient):
                     "author": "langgenius",
                     "name": "file",
                     "label": {"zh_Hans": "File", "en_US": "File", "pt_BR": "File", "ja_JP": "File"},
-                    "icon": "https://cloud.dify.ai/console/api/workspaces/current/plugin/icon?tenant_id=945b4365-9d99-48c1-8c47-90593fe8b9c9&filename=13d9312f6b1352d3939b90a5257de58ff3cd619d5be4f5b266ff0298935ac328.svg",
+                    "icon": "https://assets.dify.ai/images/File%20Upload.svg",
                     "description": {"zh_Hans": "File", "en_US": "File", "pt_BR": "File", "ja_JP": "File"},
                 },
                 "credentials_schema": [],
diff --git a/api/services/rag_pipeline/rag_pipeline.py b/api/services/rag_pipeline/rag_pipeline.py
index 429c53d829..2ed44c4a36 100644
--- a/api/services/rag_pipeline/rag_pipeline.py
+++ b/api/services/rag_pipeline/rag_pipeline.py
@@ -126,7 +126,7 @@ class RagPipelineService:
         Delete customized pipeline template.
         """
         customized_template: PipelineCustomizedTemplate | None = (
-            db.query(PipelineCustomizedTemplate)
+            db.session.query(PipelineCustomizedTemplate)
             .filter(
                 PipelineCustomizedTemplate.id == template_id,
                 PipelineCustomizedTemplate.tenant_id == current_user.current_tenant_id,
@@ -135,8 +135,8 @@ class RagPipelineService:
         )
         if not customized_template:
             raise ValueError("Customized pipeline template not found.")
-        db.delete(customized_template)
-        db.commit()
+        db.session.delete(customized_template)
+        db.session.commit()
 
     def get_draft_workflow(self, pipeline: Pipeline) -> Optional[Workflow]:
         """