fix: inherits model decoration metadata from Jan

This commit is contained in:
Louis 2024-10-22 16:16:43 +07:00
parent 981675f365
commit e8a9e8e28c
No known key found for this signature in database
GPG Key ID: 44FA9F4D33C37DE2

View File

@ -30,10 +30,17 @@ const useModels = () => {
const getData = useCallback(() => {
const getDownloadedModels = async () => {
const localModels = await getModels()
const hubModels = ModelManager.instance().models.values().toArray()
const localModels = (await getModels()).map((e) => ({
...e,
name: ModelManager.instance().models.get(e.id)?.name ?? e.name,
metadata:
ModelManager.instance().models.get(e.id)?.metadata ?? e.metadata,
}))
const remoteModels = hubModels.filter((e) => !isLocalEngine(e.engine))
const remoteModels = ModelManager.instance()
.models.values()
.toArray()
.filter((e) => !isLocalEngine(e.engine))
setDownloadedModels([...localModels, ...remoteModels])
}