diff --git a/web/app/components/plugins/install-plugin/hooks/use-refresh-plugin-list.tsx b/web/app/components/plugins/install-plugin/hooks/use-refresh-plugin-list.tsx index ac26971359..6294887356 100644 --- a/web/app/components/plugins/install-plugin/hooks/use-refresh-plugin-list.tsx +++ b/web/app/components/plugins/install-plugin/hooks/use-refresh-plugin-list.tsx @@ -6,6 +6,7 @@ import { useInvalidateAllBuiltInTools, useInvalidateAllToolProviders } from '@/s import { useInvalidateStrategyProviders } from '@/service/use-strategy' import type { Plugin, PluginDeclaration, PluginManifestInMarket } from '../../types' import { PluginType } from '../../types' +import { useInvalidDataSourceList } from '@/service/use-pipeline' const useRefreshPluginList = () => { const invalidateInstalledPluginList = useInvalidateInstalledPluginList() @@ -16,6 +17,7 @@ const useRefreshPluginList = () => { const invalidateAllToolProviders = useInvalidateAllToolProviders() const invalidateAllBuiltInTools = useInvalidateAllBuiltInTools() + const invalidateAllDataSources = useInvalidDataSourceList() const invalidateStrategyProviders = useInvalidateStrategyProviders() return { @@ -30,6 +32,9 @@ const useRefreshPluginList = () => { // TODO: update suggested tools. It's a function in hook useMarketplacePlugins,handleUpdatePlugins } + if ((manifest && PluginType.datasource.includes(manifest.category)) || refreshAllType) + invalidateAllDataSources() + // model select if ((manifest && PluginType.model.includes(manifest.category)) || refreshAllType) { refreshModelProviders()