33 lines
1.1 KiB
TypeScript
33 lines
1.1 KiB
TypeScript
import { useEffect } from 'react'
|
|
import { ModelManagementService } from '@janhq/core'
|
|
import { useAtom } from 'jotai'
|
|
import { downloadedModelAtom } from '@helpers/atoms/DownloadedModel.atom'
|
|
import { extensionPoints } from '../../electron/core/plugin-manager/execution'
|
|
import { executeSerial } from '@services/pluginService'
|
|
|
|
export function useGetDownloadedModels() {
|
|
const [downloadedModels, setDownloadedModels] = useAtom(downloadedModelAtom)
|
|
|
|
useEffect(() => {
|
|
getDownloadedModels().then((downloadedModels) => {
|
|
setDownloadedModels(downloadedModels)
|
|
})
|
|
}, [setDownloadedModels])
|
|
|
|
return { downloadedModels }
|
|
}
|
|
|
|
export async function getDownloadedModels(): Promise<AssistantModel[]> {
|
|
if (!extensionPoints.get(ModelManagementService.GetFinishedDownloadModels)) {
|
|
return []
|
|
}
|
|
const downloadedModels: AssistantModel[] = await executeSerial(
|
|
ModelManagementService.GetFinishedDownloadModels
|
|
)
|
|
return downloadedModels ?? []
|
|
}
|
|
|
|
export async function getConfiguredModels(): Promise<Product[]> {
|
|
return executeSerial(ModelManagementService.GetConfiguredModels)
|
|
}
|