22 lines
807 B
TypeScript
22 lines
807 B
TypeScript
import { execute, executeSerial } from '@/_services/pluginService'
|
|
import { ModelManagementService } from '@janhq/core'
|
|
import { useSetAtom } from 'jotai'
|
|
import { downloadedModelAtom } from '@/_helpers/atoms/DownloadedModel.atom'
|
|
import { getDownloadedModels } from './useGetDownloadedModels'
|
|
import { AssistantModel } from '@/_models/AssistantModel'
|
|
|
|
export default function useDeleteModel() {
|
|
const setDownloadedModels = useSetAtom(downloadedModelAtom)
|
|
|
|
const deleteModel = async (model: AssistantModel) => {
|
|
execute(ModelManagementService.DeleteDownloadModel, model._id)
|
|
await executeSerial(ModelManagementService.DeleteModel, model._id)
|
|
|
|
// reload models
|
|
const downloadedModels = await getDownloadedModels()
|
|
setDownloadedModels(downloadedModels)
|
|
}
|
|
|
|
return { deleteModel }
|
|
}
|