From cef77a371799fbd3b1f4b76ab3148f9d13a86f56 Mon Sep 17 00:00:00 2001
From: zxhlyh <jasonapring2015@outlook.com>
Date: Mon, 26 May 2025 16:41:50 +0800
Subject: [PATCH] datasource icon

---
 web/app/components/workflow/block-icon.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/web/app/components/workflow/block-icon.tsx b/web/app/components/workflow/block-icon.tsx
index 4ec78a49bc..234e20c7ff 100644
--- a/web/app/components/workflow/block-icon.tsx
+++ b/web/app/components/workflow/block-icon.tsx
@@ -109,7 +109,7 @@ const BlockIcon: FC<BlockIconProps> = ({
         )
       }
       {
-        type === BlockEnum.Tool && toolIcon && (
+        (type === BlockEnum.Tool || type === BlockEnum.DataSource) && toolIcon && (
           <>
             {
               typeof toolIcon === 'string'