diff --git a/web/src/pages/agent/canvas/node/tool-node.tsx b/web/src/pages/agent/canvas/node/tool-node.tsx index ccdf7be83..42c7bf3b5 100644 --- a/web/src/pages/agent/canvas/node/tool-node.tsx +++ b/web/src/pages/agent/canvas/node/tool-node.tsx @@ -5,6 +5,7 @@ import { MouseEventHandler, memo, useCallback } from 'react'; import { NodeHandleId, Operator } from '../../constant'; import { ToolCard } from '../../form/agent-form/agent-tools'; import { useFindMcpById } from '../../hooks/use-find-mcp-by-id'; +import OperatorIcon from '../../operator-icon'; import useGraphStore from '../../store'; import { NodeWrapper } from './node-wrapper'; @@ -57,7 +58,10 @@ function InnerToolNode({ className="cursor-pointer" data-tool={x.component_name} > - {x.component_name} +
+ + {x.component_name} +
))} diff --git a/web/src/pages/agent/constant.tsx b/web/src/pages/agent/constant.tsx index 8950b658b..fe3255d17 100644 --- a/web/src/pages/agent/constant.tsx +++ b/web/src/pages/agent/constant.tsx @@ -788,18 +788,8 @@ export const CategorizeAnchorPointPositions = [ // no connection lines are allowed between key and value export const RestrictedUpstreamMap = { [Operator.Begin]: [Operator.Relevant], - [Operator.Categorize]: [ - Operator.Begin, - Operator.Categorize, - Operator.Answer, - Operator.Relevant, - ], - [Operator.Answer]: [ - Operator.Begin, - Operator.Answer, - Operator.Message, - Operator.Relevant, - ], + [Operator.Categorize]: [Operator.Begin, Operator.Categorize, Operator.Answer], + [Operator.Answer]: [Operator.Begin, Operator.Answer, Operator.Message], [Operator.Retrieval]: [Operator.Begin, Operator.Retrieval], [Operator.Generate]: [Operator.Begin, Operator.Relevant], [Operator.Message]: [ @@ -809,9 +799,8 @@ export const RestrictedUpstreamMap = { Operator.Retrieval, Operator.RewriteQuestion, Operator.Categorize, - Operator.Relevant, ], - [Operator.Relevant]: [Operator.Begin, Operator.Answer, Operator.Relevant], + [Operator.Relevant]: [Operator.Begin, Operator.Answer], [Operator.RewriteQuestion]: [ Operator.Begin, Operator.Message, diff --git a/web/src/pages/agent/form/agent-form/agent-tools.tsx b/web/src/pages/agent/form/agent-form/agent-tools.tsx index 60ae326b8..079297e0b 100644 --- a/web/src/pages/agent/form/agent-form/agent-tools.tsx +++ b/web/src/pages/agent/form/agent-form/agent-tools.tsx @@ -18,6 +18,7 @@ import { Operator } from '../../constant'; import { AgentInstanceContext } from '../../context'; import { useFindMcpById } from '../../hooks/use-find-mcp-by-id'; import { INextOperatorForm } from '../../interface'; +import OperatorIcon from '../../operator-icon'; import useGraphStore from '../../store'; import { filterDownstreamAgentNodeIds } from '../../utils/filter-downstream-nodes'; import { ToolPopover } from './tool-popover'; @@ -109,7 +110,10 @@ export function AgentTools() {