mirror of
https://github.com/langgenius/dify.git
synced 2025-12-20 06:32:45 +00:00
feat: update RAG recommended plugins hook to accept type parameter (#29735)
This commit is contained in:
@@ -37,7 +37,7 @@ const useRefreshPluginList = () => {
|
||||
if ((manifest && PluginCategoryEnum.tool.includes(manifest.category)) || refreshAllType) {
|
||||
invalidateAllToolProviders()
|
||||
invalidateAllBuiltInTools()
|
||||
invalidateRAGRecommendedPlugins()
|
||||
invalidateRAGRecommendedPlugins('tool')
|
||||
// TODO: update suggested tools. It's a function in hook useMarketplacePlugins,handleUpdatePlugins
|
||||
}
|
||||
|
||||
|
||||
@@ -52,7 +52,7 @@ const RAGToolRecommendations = ({
|
||||
data: ragRecommendedPlugins,
|
||||
isLoading: isLoadingRAGRecommendedPlugins,
|
||||
isFetching: isFetchingRAGRecommendedPlugins,
|
||||
} = useRAGRecommendedPlugins()
|
||||
} = useRAGRecommendedPlugins('tool')
|
||||
|
||||
const recommendedPlugins = useMemo(() => {
|
||||
if (ragRecommendedPlugins)
|
||||
|
||||
@@ -330,15 +330,24 @@ export const useRemoveProviderCredentials = ({
|
||||
|
||||
const useRAGRecommendedPluginListKey = [NAME_SPACE, 'rag-recommended-plugins']
|
||||
|
||||
export const useRAGRecommendedPlugins = () => {
|
||||
export const useRAGRecommendedPlugins = (type: 'tool' | 'datasource' | 'all' = 'all') => {
|
||||
return useQuery<RAGRecommendedPlugins>({
|
||||
queryKey: useRAGRecommendedPluginListKey,
|
||||
queryFn: () => get<RAGRecommendedPlugins>('/rag/pipelines/recommended-plugins'),
|
||||
queryKey: [...useRAGRecommendedPluginListKey, type],
|
||||
queryFn: () => get<RAGRecommendedPlugins>('/rag/pipelines/recommended-plugins', {
|
||||
params: {
|
||||
type,
|
||||
},
|
||||
}),
|
||||
})
|
||||
}
|
||||
|
||||
export const useInvalidateRAGRecommendedPlugins = () => {
|
||||
return useInvalid(useRAGRecommendedPluginListKey)
|
||||
const queryClient = useQueryClient()
|
||||
return (type: 'tool' | 'datasource' | 'all' = 'all') => {
|
||||
queryClient.invalidateQueries({
|
||||
queryKey: [...useRAGRecommendedPluginListKey, type],
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// App Triggers API hooks
|
||||
|
||||
Reference in New Issue
Block a user