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

fix: downloaded models should account for remote models

Browse Source
This commit is contained in:
Louis 2024-10-21 17:11:25 +07:00
parent b5edc12b28
commit 8129c2319f
No known key found for this signature in database
GPG Key ID: 44FA9F4D33C37DE2
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
web/hooks/useModels.ts
View File

@ -11,6 +11,8 @@ import {
import { useSetAtom } from 'jotai' import { useSetAtom } from 'jotai'
import { isLocalEngine } from '@/utils/modelEngine'
import { extensionManager } from '@/extension' import { extensionManager } from '@/extension'
import { import {
configuredModelsAtom, configuredModelsAtom,
@ -28,8 +30,12 @@ const useModels = () => {
const getData = useCallback(() => { const getData = useCallback(() => {
const getDownloadedModels = async () => { const getDownloadedModels = async () => {
const models = await getModels() const localModels = await getModels()
setDownloadedModels(models) const remoteModels = ModelManager.instance()
.models.values()
.toArray()
.filter((e) => !isLocalEngine(e.engine))
setDownloadedModels([...localModels, ...remoteModels])
} }
const getExtensionModels = async () => { const getExtensionModels = async () => {