Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: model id is optional on import

Browse Source
This commit is contained in:
Louis 2024-11-01 17:09:56 +07:00
parent a986c6de2d
commit b913af9f88
No known key found for this signature in database
GPG Key ID: 44FA9F4D33C37DE2
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
web/hooks/useImportModel.ts
View File

@ -4,8 +4,10 @@ import {
ExtensionTypeEnum, ExtensionTypeEnum,
ImportingModel, ImportingModel,
Model, Model,
ModelEvent,
ModelExtension, ModelExtension,
OptionType, OptionType,
events,
fs, fs,
} from '@janhq/core' } from '@janhq/core'
@ -58,18 +60,19 @@ const useImportModel = () => {
const importModels = useCallback( const importModels = useCallback(
(models: ImportingModel[], optionType: OptionType) => { (models: ImportingModel[], optionType: OptionType) => {
models models.map((model) => {
.filter((e) => !!e.modelId) const modelId = model.modelId ?? model.path.split('/').pop()
.map((model) => { if (modelId) {
if (model.modelId) { addDownloadingModel(modelId)
const modelId = model.modelId extensionManager
addDownloadingModel(modelId) .get<ModelExtension>(ExtensionTypeEnum.Model)
extensionManager ?.importModel(modelId, model.path, model.name)
.get<ModelExtension>(ExtensionTypeEnum.Model) .finally(() => {
?.importModel(model.modelId, model.path, model.name) removeDownloadingModel(modelId)
.finally(() => removeDownloadingModel(modelId)) events.emit(ModelEvent.OnModelsUpdate, {})
} })
}) }
})
}, },
[addDownloadingModel, removeDownloadingModel] [addDownloadingModel, removeDownloadingModel]
) )