29 lines
853 B
TypeScript
29 lines
853 B
TypeScript
import { ExtensionTypeEnum, ModelExtension, Model } from '@janhq/core'
|
|
|
|
import { useAtom } from 'jotai'
|
|
|
|
import { toaster } from '@/containers/Toast'
|
|
|
|
import { extensionManager } from '@/extension/ExtensionManager'
|
|
import { downloadedModelsAtom } from '@/helpers/atoms/Model.atom'
|
|
|
|
export default function useDeleteModel() {
|
|
const [downloadedModels, setDownloadedModels] = useAtom(downloadedModelsAtom)
|
|
|
|
const deleteModel = async (model: Model) => {
|
|
await extensionManager
|
|
.get<ModelExtension>(ExtensionTypeEnum.Model)
|
|
?.deleteModel(model.id)
|
|
|
|
// reload models
|
|
setDownloadedModels(downloadedModels.filter((e) => e.id !== model.id))
|
|
toaster({
|
|
title: 'Model Deletion Successful',
|
|
description: `The model ${model.id} has been successfully deleted.`,
|
|
type: 'success',
|
|
})
|
|
}
|
|
|
|
return { deleteModel }
|
|
}
|