-
+ {icon ||
}
diff --git a/web/app/components/workflow/block-icon.tsx b/web/app/components/workflow/block-icon.tsx
index 3656c42b3f..7f7aeca092 100644
--- a/web/app/components/workflow/block-icon.tsx
+++ b/web/app/components/workflow/block-icon.tsx
@@ -2,6 +2,7 @@ import type { FC } from 'react'
import { memo } from 'react'
import { BlockEnum } from './types'
import {
+ Agent,
Answer,
Assigner,
Code,
@@ -53,8 +54,7 @@ const getIcon = (type: BlockEnum, className: string) => {
[BlockEnum.ParameterExtractor]:
,
[BlockEnum.DocExtractor]:
,
[BlockEnum.ListFilter]:
,
- // TODO: add icon for Agent
- [BlockEnum.Agent]:
,
+ [BlockEnum.Agent]:
,
}[type]
}
const ICON_CONTAINER_BG_COLOR_MAP: Record
= {
diff --git a/web/app/components/workflow/nodes/_base/components/agent-strategy.tsx b/web/app/components/workflow/nodes/_base/components/agent-strategy.tsx
index 636baaa08b..50bf260311 100644
--- a/web/app/components/workflow/nodes/_base/components/agent-strategy.tsx
+++ b/web/app/components/workflow/nodes/_base/components/agent-strategy.tsx
@@ -5,6 +5,7 @@ import { AgentStrategySelector } from './agent-strategy-selector'
import Link from 'next/link'
import { useTranslation } from 'react-i18next'
import Form from '@/app/components/header/account-setting/model-provider-page/model-modal/Form'
+import { Agent } from '@/app/components/base/icons/src/vender/workflow'
export type Strategy = {
agent_strategy_provider_name: string
@@ -39,8 +40,8 @@ export const AgentStrategy = (props: AgentStrategyProps) => {
fieldLabelClassName='uppercase'
/>
}
title={t('workflow.nodes.agent.strategy.configureTip')}
description={