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

fix: error handling - pull remote models list raise an error on fail (#4615)

Browse Source 
* chore: pull models list on /models update

* chore: correct engine extension name

* chore: fix undefined error

* chore: revert auto populating remote models on update
This commit is contained in:
Louis 2025-02-09 19:40:32 +07:00 committed by GitHub
parent 05b124f624
commit 45edb9cc51
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
extensions/engine-management-extension/src/index.ts
View File

@ -25,10 +25,10 @@ interface ModelList {
data: Model[] data: Model[]
} }
/** /**
* JSONEngineManagementExtension is a EngineManagementExtension implementation that provides * JanEngineManagementExtension is a EngineManagementExtension implementation that provides
* functionality for managing engines. * functionality for managing engines.
*/ */
export default class JSONEngineManagementExtension extends EngineManagementExtension { export default class JanEngineManagementExtension extends EngineManagementExtension {
queue = new PQueue({ concurrency: 1 }) queue = new PQueue({ concurrency: 1 })
/** /**
@ -356,17 +356,18 @@ export default class JSONEngineManagementExtension extends EngineManagementExten
private populateRemoteModels = async (engineConfig: EngineConfig) => { private populateRemoteModels = async (engineConfig: EngineConfig) => {
return this.getRemoteModels(engineConfig.engine) return this.getRemoteModels(engineConfig.engine)
.then((models: ModelList) => { .then((models: ModelList) => {
Promise.all( if (models?.data)
models.data?.map((model) => Promise.all(
this.addRemoteModel({ models.data.map((model) =>
...model, this.addRemoteModel({
engine: engineConfig.engine as InferenceEngine, ...model,
model: model.model ?? model.id, engine: engineConfig.engine as InferenceEngine,
}).catch(console.info) model: model.model ?? model.id,
) }).catch(console.info)
).then(() => { )
events.emit(ModelEvent.OnModelsUpdate, { fetch: true }) ).then(() => {
}) events.emit(ModelEvent.OnModelsUpdate, { fetch: true })
})
}) })
.catch(console.info) .catch(console.info)
} }