dify/web/service/use-strategy.ts

import type { QueryOptions } from '@tanstack/react-query'
import type { StrategyPluginDetail } from '@/app/components/plugins/types'
import { useQuery } from '@tanstack/react-query'
import { fetchStrategyDetail, fetchStrategyList } from './strategy'
import { useInvalid } from './use-base'

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,
  })
}