fix: tensorrt-llm model list

This commit is contained in:
Louis 2024-11-21 16:58:15 +07:00
parent ef3c090360
commit f74994a735

View File

@ -89,13 +89,13 @@ const useModels = () => {
const cachedModels = ModelManager.instance().models.values().toArray() const cachedModels = ModelManager.instance().models.values().toArray()
const toUpdate = [ const toUpdate = [
...downloadedModels, ...downloadedModels,
...cachedModels.filter( ...cachedModels.filter((e) => !isLocalEngine(e.engine) && !downloadedModels.some((g: Model) => g.id === e.id)
(e: Model) => !downloadedModels.some((g: Model) => g.id === e.id)
), ),
] ]
setDownloadedModels(toUpdate) setDownloadedModels(toUpdate)
}, [downloadedModels, setDownloadedModels]) setExtensionModels(cachedModels)
}, [downloadedModels, setDownloadedModels, setExtensionModels])
const getModels = async (): Promise<Model[]> => const getModels = async (): Promise<Model[]> =>
extensionManager extensionManager