27 lines
752 B
TypeScript
27 lines
752 B
TypeScript
import { useEffect } from 'react'
|
|
|
|
import { ExtensionType, ModelExtension, Model } from '@janhq/core'
|
|
|
|
import { atom, useAtom } from 'jotai'
|
|
|
|
import { extensionManager } from '@/extension/ExtensionManager'
|
|
|
|
const downloadedModelsAtom = atom<Model[]>([])
|
|
|
|
export function useGetDownloadedModels() {
|
|
const [downloadedModels, setDownloadedModels] = useAtom(downloadedModelsAtom)
|
|
|
|
useEffect(() => {
|
|
getDownloadedModels().then((downloadedModels) => {
|
|
setDownloadedModels(downloadedModels)
|
|
})
|
|
}, [setDownloadedModels])
|
|
|
|
return { downloadedModels, setDownloadedModels }
|
|
}
|
|
|
|
export const getDownloadedModels = async (): Promise<Model[]> =>
|
|
extensionManager
|
|
.get<ModelExtension>(ExtensionType.Model)
|
|
?.getDownloadedModels() ?? []
|