Spaces:
Sleeping
Sleeping
File size: 2,988 Bytes
f871fed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query'
import { modelsApi } from '@/lib/api/models'
import { useToast } from '@/lib/hooks/use-toast'
import { CreateModelRequest, ModelDefaults } from '@/lib/types/models'
export const MODEL_QUERY_KEYS = {
models: ['models'] as const,
model: (id: string) => ['models', id] as const,
defaults: ['models', 'defaults'] as const,
providers: ['models', 'providers'] as const,
}
export function useModels() {
return useQuery({
queryKey: MODEL_QUERY_KEYS.models,
queryFn: () => modelsApi.list(),
})
}
export function useModel(id: string) {
return useQuery({
queryKey: MODEL_QUERY_KEYS.model(id),
queryFn: () => modelsApi.get(id),
enabled: !!id,
})
}
export function useCreateModel() {
const queryClient = useQueryClient()
const { toast } = useToast()
return useMutation({
mutationFn: (data: CreateModelRequest) => modelsApi.create(data),
onSuccess: () => {
queryClient.invalidateQueries({ queryKey: MODEL_QUERY_KEYS.models })
toast({
title: 'Success',
description: 'Model created successfully',
})
},
onError: (error: unknown) => {
const errorMessage = (error as { response?: { data?: { detail?: string } } })?.response?.data?.detail || 'Failed to create model'
toast({
title: 'Error',
description: errorMessage,
variant: 'destructive',
})
},
})
}
export function useDeleteModel() {
const queryClient = useQueryClient()
const { toast } = useToast()
return useMutation({
mutationFn: (id: string) => modelsApi.delete(id),
onSuccess: () => {
queryClient.invalidateQueries({ queryKey: MODEL_QUERY_KEYS.models })
queryClient.invalidateQueries({ queryKey: MODEL_QUERY_KEYS.defaults })
toast({
title: 'Success',
description: 'Model deleted successfully',
})
},
onError: () => {
toast({
title: 'Error',
description: 'Failed to delete model',
variant: 'destructive',
})
},
})
}
export function useModelDefaults() {
return useQuery({
queryKey: MODEL_QUERY_KEYS.defaults,
queryFn: () => modelsApi.getDefaults(),
})
}
export function useUpdateModelDefaults() {
const queryClient = useQueryClient()
const { toast } = useToast()
return useMutation({
mutationFn: (data: Partial<ModelDefaults>) => modelsApi.updateDefaults(data),
onSuccess: () => {
queryClient.invalidateQueries({ queryKey: MODEL_QUERY_KEYS.defaults })
toast({
title: 'Success',
description: 'Default models updated successfully',
})
},
onError: () => {
toast({
title: 'Error',
description: 'Failed to update default models',
variant: 'destructive',
})
},
})
}
export function useProviders() {
return useQuery({
queryKey: MODEL_QUERY_KEYS.providers,
queryFn: () => modelsApi.getProviders(),
})
}
|