datasource icon

12 months ago · cef77a3717
parent 28726b6cf3
commit cef77a3717
1 changed files with 1 additions and 1 deletions
--- 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'