diff --git a/web/app/components/workflow/block-selector/all-tools.tsx b/web/app/components/workflow/block-selector/all-tools.tsx index 1bb5bfd04f..910cf684c4 100644 --- a/web/app/components/workflow/block-selector/all-tools.tsx +++ b/web/app/components/workflow/block-selector/all-tools.tsx @@ -35,6 +35,7 @@ type AllToolsProps = { canNotSelectMultiple?: boolean onSelectMultiple?: (type: BlockEnum, tools: ToolDefaultValue[]) => void selectedTools?: ToolValue[] + isHideMCPTools?: boolean } const DEFAULT_TAGS: AllToolsProps['tags'] = [] @@ -52,9 +53,10 @@ const AllTools = ({ customTools, mcpTools = [], selectedTools, + isHideMCPTools, }: AllToolsProps) => { const language = useGetLanguage() - const tabs = useToolTabs() + const tabs = useToolTabs(isHideMCPTools) const [activeTab, setActiveTab] = useState(ToolTypeEnum.All) const [activeView, setActiveView] = useState(ViewType.flat) const hasFilter = searchText || tags.length > 0 diff --git a/web/app/components/workflow/block-selector/hooks.ts b/web/app/components/workflow/block-selector/hooks.ts index 791eb7f73f..d00815584d 100644 --- a/web/app/components/workflow/block-selector/hooks.ts +++ b/web/app/components/workflow/block-selector/hooks.ts @@ -31,10 +31,9 @@ export const useTabs = () => { ] } -export const useToolTabs = () => { +export const useToolTabs = (isHideMCPTools?: boolean) => { const { t } = useTranslation() - - return [ + const tabs = [ { key: ToolTypeEnum.All, name: t('workflow.tabs.allTool'), @@ -51,9 +50,13 @@ export const useToolTabs = () => { key: ToolTypeEnum.Workflow, name: t('workflow.tabs.workflowTool'), }, - { + ] + if(!isHideMCPTools) { + tabs.push({ key: ToolTypeEnum.MCP, name: 'MCP', - }, - ] + }) + } + + return tabs } diff --git a/web/app/components/workflow/block-selector/tabs.tsx b/web/app/components/workflow/block-selector/tabs.tsx index f32ab89692..7e2b768afc 100644 --- a/web/app/components/workflow/block-selector/tabs.tsx +++ b/web/app/components/workflow/block-selector/tabs.tsx @@ -83,6 +83,7 @@ const Tabs: FC = ({ customTools={customTools || []} workflowTools={workflowTools || []} mcpTools={mcpTools || []} + isHideMCPTools /> ) }