[feat] fix horde support

This commit is contained in:
kwaroran
2024-01-03 19:36:47 +09:00
parent 99743f581a
commit 23ef56cb33
3 changed files with 41 additions and 35 deletions

View File

@@ -1,20 +1,30 @@
import { sleep } from "../util"
let modelList:string[]|'loading' = null
interface HordeModel {
"performance": number,
"queued": number,
"jobs": number,
"eta": Number,
"type": "text",
"name": "aphrodite\/Undi95\/Toppy-M-7B",
"count": number
}
let modelList:HordeModel[]|'loading' = null
//until horde is ready
// modelList = []
export async function getHordeModels():Promise<string[]> {
export async function getHordeModels():Promise<HordeModel[]> {
if(modelList === null){
try {
modelList = 'loading'
const models = await fetch("https://stablehorde.net/api/v2/status/models?type=text")
modelList = ((await models.json()).map((a) => {
return a.name
}) as string[])
return modelList
const res = await models.json()
modelList = res
return res
} catch (error) {
modelList = null
return []