opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-07-25 05:28:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(workflow): hide Agent V2 in chatflow (#39439)

Browse Source
This commit is contained in:
yyh 2026-07-23 11:37:13 +08:00 committed by GitHub
parent 45711eb6dd
commit 625ed2a0e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
web/app/components/workflow-app/hooks/__tests__/use-available-nodes-meta-data.spec.ts
View File

@ -40,6 +40,18 @@ describe('useAvailableNodesMetaData', () => {
)
})
it('should expose legacy Agent instead of Agent v2 in chat mode when Agent v2 is enabled', () => {
mockUseIsChatMode.mockReturnValue(true)
const { result } = renderHook(() => useAvailableNodesMetaData())
const nodeTypes = result.current.nodes.map((node) => node.metaData.type)
expect(nodeTypes).toContain(BlockEnum.Agent)
expect(nodeTypes).not.toContain(BlockEnum.AgentV2)
expect(result.current.nodesMap?.[BlockEnum.Agent]).toBeDefined()
expect(result.current.nodesMap?.[BlockEnum.AgentV2]).toBeUndefined()
})
it('should include workflow-specific trigger and end nodes outside chat mode', () => {
mockUseIsChatMode.mockReturnValue(false)

5
web/app/components/workflow-app/hooks/use-available-nodes-meta-data.ts
View File

@ -31,6 +31,7 @@ export const useAvailableNodesMetaData = () => {
const isChatMode = useIsChatMode()
const docLink = useDocLink()
const agentV2Enabled = isAgentV2Enabled()
const shouldUseAgentV2 = agentV2Enabled && !isChatMode
const startNodeMetaData = useMemo(
() => ({
@ -45,7 +46,7 @@ export const useAvailableNodesMetaData = () => {
const mergedNodesMetaData = useMemo(() => {
const commonNodes = WORKFLOW_COMMON_NODES.filter((node) =>
agentV2Enabled
shouldUseAgentV2
? node.metaData.type !== BlockEnum.Agent
: node.metaData.type !== BlockEnum.AgentV2,
)
@ -63,7 +64,7 @@ export const useAvailableNodesMetaData = () => {
TriggerPluginDefault,
]),
]
}, [agentV2Enabled, isChatMode, startNodeMetaData])
}, [isChatMode, shouldUseAgentV2, startNodeMetaData])
const availableNodesMetaData = useMemo(
() =>