dify

dify/web/service/use-strategy.ts

@@ -0,0 +1,36 @@
+import type {
+  StrategyPluginDetail,
+} from '@/app/components/plugins/types'
+import { useInvalid } from './use-base'
+import type { QueryOptions } from '@tanstack/react-query'
+import {
+  useQuery,
+} from '@tanstack/react-query'
+import { fetchStrategyDetail, fetchStrategyList } from './strategy'
+
+const NAME_SPACE = 'agent_strategy'
+
+const useStrategyListKey = [NAME_SPACE, 'strategyList']
+export const useStrategyProviders = () => {
+  return useQuery<StrategyPluginDetail[]>({
+    queryKey: useStrategyListKey,
+    queryFn: fetchStrategyList,
+  })
+}
+
+export const useInvalidateStrategyProviders = () => {
+  return useInvalid(useStrategyListKey)
+}
+
+export const useStrategyProviderDetail = (agentProvider: string, options?: QueryOptions<StrategyPluginDetail>) => {
+  return useQuery<StrategyPluginDetail>({
+    ...options,
+    queryKey: [NAME_SPACE, 'detail', agentProvider],
+    queryFn: () => fetchStrategyDetail(agentProvider),
+    enabled: !!agentProvider,
+  })
+}
+
+export const useInvalidateStrategyProviderDetail = (agentProvider: string) => {
+  return useInvalid([NAME_SPACE, 'detail', agentProvider])
+}