a50ea4a634
* refactor: deprecate extension type implementation * chore: update README.md * refactor: EventName to use the events defined in /types
32 lines
808 B
TypeScript
32 lines
808 B
TypeScript
import { useEffect, useState } from 'react'
|
|
|
|
import { ExtensionTypeEnum, ModelExtension } from '@janhq/core'
|
|
import { Model } from '@janhq/core'
|
|
|
|
import { extensionManager } from '@/extension/ExtensionManager'
|
|
|
|
export function useGetConfiguredModels() {
|
|
const [loading, setLoading] = useState<boolean>(false)
|
|
const [models, setModels] = useState<Model[]>([])
|
|
|
|
const getConfiguredModels = async (): Promise<Model[]> => {
|
|
const models = await extensionManager
|
|
.get<ModelExtension>(ExtensionTypeEnum.Model)
|
|
?.getConfiguredModels()
|
|
return models ?? []
|
|
}
|
|
|
|
async function fetchModels() {
|
|
setLoading(true)
|
|
const models = await getConfiguredModels()
|
|
setLoading(false)
|
|
setModels(models)
|
|
}
|
|
|
|
useEffect(() => {
|
|
fetchModels()
|
|
}, [])
|
|
|
|
return { loading, models }
|
|
}
|