theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improve refreshRemoteModels

Browse Source
This commit is contained in:
josc146
2023-12-22 14:44:27 +08:00
parent 2e0a542f33
commit 280194647c
4 changed files with 31 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
frontend/src/utils/index.tsx
View File

@@ -27,6 +27,7 @@ import logo from '../assets/images/logo.png';
import { Preset } from '../types/presets';
import { botName, Conversation, MessageType, userName } from '../types/chat';
import { v4 as uuid } from 'uuid';
import { findLastIndex } from 'lodash-es';
export type Cache = {
version: string
@@ -145,7 +146,7 @@ function initLastUnfinishedModelDownloads() {
export async function refreshRemoteModels(cache: {
models: ModelSourceItem[]
}) {
}, filter: boolean = true, initUnfinishedModels: boolean = false) {
const manifestUrls = commonStore.modelSourceManifestList.split(/[,，;；\n]/);
const requests = manifestUrls.filter(url => url.endsWith('.json')).map(
url => fetch(url, { cache: 'no-cache' }).then(r => r.json()));
@@ -162,18 +163,16 @@ export async function refreshRemoteModels(cache: {
});
cache.models = cache.models.filter((model, index, self) => {
return modelSuffix.some((ext => model.name.endsWith(ext)))
&& index === self.findIndex(
m => m.name === model.name || (m.SHA256 && m.SHA256 === model.SHA256 && m.size === model.size));
});
commonStore.setModelSourceList(cache.models);
await saveCache().catch(() => {
&& index === findLastIndex(self,
m => m.name === model.name || (!!m.SHA256 && m.SHA256 === model.SHA256 && m.size === model.size));
});
await refreshLocalModels(cache, filter, initUnfinishedModels);
}
export const refreshModels = async (readCache: boolean = false, initUnfinishedModels: boolean = false) => {
const cache = await refreshBuiltInModels(readCache);
await refreshLocalModels(cache, false, initUnfinishedModels);
await refreshRemoteModels(cache);
await refreshRemoteModels(cache, false, initUnfinishedModels);
};
export const getStrategy = (modelConfig: ModelConfig | undefined = undefined) => {