From 0ed4ec5cd0f7483b4e6bfc3e977bfb8268ff5d05 Mon Sep 17 00:00:00 2001
From: AkaraChen <akarachen@outlook.com>
Date: Thu, 2 Jan 2025 14:48:40 +0800
Subject: [PATCH] fix: agent strategy selector show workflow

---
 web/app/components/workflow/block-selector/index-bar.tsx     | 5 ++++-
 .../block-selector/tool/tool-list-tree-view/list.tsx         | 5 ++++-
 web/app/components/workflow/block-selector/tools.tsx         | 1 -
 web/i18n/en-US/workflow.ts                                   | 1 +
 web/i18n/zh-Hans/workflow.ts                                 | 1 +
 5 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/web/app/components/workflow/block-selector/index-bar.tsx b/web/app/components/workflow/block-selector/index-bar.tsx
index 3c5bf8d6e2..8d4b3de10e 100644
--- a/web/app/components/workflow/block-selector/index-bar.tsx
+++ b/web/app/components/workflow/block-selector/index-bar.tsx
@@ -6,6 +6,7 @@ import classNames from '@/utils/classnames'
 
 export const CUSTOM_GROUP_NAME = '@@@custom@@@'
 export const WORKFLOW_GROUP_NAME = '@@@workflow@@@'
+export const AGENT_GROUP_NAME = '@@@agent@@@'
 /*
 {
   A: {
@@ -46,8 +47,10 @@ export const groupItems = (items: ToolWithProvider[], getFirstChar: (item: ToolW
       groupName = item.author
     else if (item.type === CollectionType.custom)
       groupName = CUSTOM_GROUP_NAME
-    else
+    else if (item.type === CollectionType.workflow)
       groupName = WORKFLOW_GROUP_NAME
+    else
+      groupName = AGENT_GROUP_NAME
 
     if (!acc[letter][groupName])
       acc[letter][groupName] = []
diff --git a/web/app/components/workflow/block-selector/tool/tool-list-tree-view/list.tsx b/web/app/components/workflow/block-selector/tool/tool-list-tree-view/list.tsx
index 8bf5095833..a8fd34b98a 100644
--- a/web/app/components/workflow/block-selector/tool/tool-list-tree-view/list.tsx
+++ b/web/app/components/workflow/block-selector/tool/tool-list-tree-view/list.tsx
@@ -6,7 +6,7 @@ import type { BlockEnum } from '../../../types'
 import type { ToolDefaultValue } from '../../types'
 import Item from './item'
 import { useTranslation } from 'react-i18next'
-import { CUSTOM_GROUP_NAME, WORKFLOW_GROUP_NAME } from '../../index-bar'
+import { AGENT_GROUP_NAME, CUSTOM_GROUP_NAME, WORKFLOW_GROUP_NAME } from '../../index-bar'
 
 type Props = {
   payload: Record<string, ToolWithProvider[]>
@@ -27,6 +27,9 @@ const ToolListTreeView: FC<Props> = ({
     if (name === WORKFLOW_GROUP_NAME)
       return t('workflow.tabs.workflowTool')
 
+    if (name === AGENT_GROUP_NAME)
+      return t('workflow.tabs.agent')
+
     return name
   }, [t])
 
diff --git a/web/app/components/workflow/block-selector/tools.tsx b/web/app/components/workflow/block-selector/tools.tsx
index 060f6dfa2c..5b5d1da20b 100644
--- a/web/app/components/workflow/block-selector/tools.tsx
+++ b/web/app/components/workflow/block-selector/tools.tsx
@@ -60,7 +60,6 @@ const Blocks = ({
       Object.keys(withLetterAndGroupViewToolsData[letter]).forEach((groupName) => {
         if (!result[groupName])
           result[groupName] = []
-
         result[groupName].push(...withLetterAndGroupViewToolsData[letter][groupName])
       })
     })
diff --git a/web/i18n/en-US/workflow.ts b/web/i18n/en-US/workflow.ts
index 4b8d2e1452..c2f9685036 100644
--- a/web/i18n/en-US/workflow.ts
+++ b/web/i18n/en-US/workflow.ts
@@ -218,6 +218,7 @@ const translation = {
     'transform': 'Transform',
     'utilities': 'Utilities',
     'noResult': 'No match found',
+    'agent': 'Agent Strategy',
   },
   blocks: {
     'start': 'Start',
diff --git a/web/i18n/zh-Hans/workflow.ts b/web/i18n/zh-Hans/workflow.ts
index 635d9cf3ce..c72e97b588 100644
--- a/web/i18n/zh-Hans/workflow.ts
+++ b/web/i18n/zh-Hans/workflow.ts
@@ -218,6 +218,7 @@ const translation = {
     'transform': '转换',
     'utilities': '工具',
     'noResult': '未找到匹配项',
+    'agent': 'Agent 策略',
   },
   blocks: {
     'start': '开始',