35 lines
1.0 KiB
TypeScript
35 lines
1.0 KiB
TypeScript
import { get } from './base'
|
|
import type {
|
|
StrategyPluginDetail,
|
|
} from '@/app/components/plugins/types'
|
|
import { useInvalid } from './use-base'
|
|
import {
|
|
useQuery,
|
|
} from '@tanstack/react-query'
|
|
|
|
const NAME_SPACE = 'agent_strategy'
|
|
|
|
const useStrategyListKey = [NAME_SPACE, 'strategyList']
|
|
export const useStrategyProviders = () => {
|
|
return useQuery<StrategyPluginDetail[]>({
|
|
queryKey: useStrategyListKey,
|
|
queryFn: () => get<StrategyPluginDetail[]>('/workspaces/current/agent-providers'),
|
|
})
|
|
}
|
|
|
|
export const useInvalidateStrategyProviders = () => {
|
|
return useInvalid(useStrategyListKey)
|
|
}
|
|
|
|
export const useStrategyProviderDetail = (agentProvider: string) => {
|
|
return useQuery<StrategyPluginDetail>({
|
|
queryKey: [NAME_SPACE, 'detail', agentProvider],
|
|
queryFn: () => get<StrategyPluginDetail>(`/workspaces/current/agent-provider/${agentProvider}`),
|
|
enabled: !!agentProvider,
|
|
})
|
|
}
|
|
|
|
export const useInvalidateStrategyProviderDetail = (agentProvider: string) => {
|
|
return useInvalid([NAME_SPACE, 'detail', agentProvider])
|
|
}
|