87257819f6
- track boot and config default models in gateway status/events - preserve running, starting, and restarting states during health checks - add safer gateway restart handling with stronger backend test coverage - expose restart-required UI and refresh model state after default model update
91 lines
2.2 KiB
TypeScript
91 lines
2.2 KiB
TypeScript
import { refreshGatewayState } from "@/store/gateway"
|
|
|
|
// API client for model list management.
|
|
|
|
export interface ModelInfo {
|
|
index: number
|
|
model_name: string
|
|
model: string
|
|
api_base?: string
|
|
api_key: string
|
|
proxy?: string
|
|
auth_method?: string
|
|
// Advanced fields
|
|
connect_mode?: string
|
|
workspace?: string
|
|
rpm?: number
|
|
max_tokens_field?: string
|
|
request_timeout?: number
|
|
thinking_level?: string
|
|
// Meta
|
|
configured: boolean
|
|
is_default: boolean
|
|
}
|
|
|
|
interface ModelsListResponse {
|
|
models: ModelInfo[]
|
|
total: number
|
|
default_model: string
|
|
}
|
|
|
|
interface ModelActionResponse {
|
|
status: string
|
|
index?: number
|
|
default_model?: string
|
|
}
|
|
|
|
const BASE_URL = ""
|
|
|
|
async function request<T>(path: string, options?: RequestInit): Promise<T> {
|
|
const res = await fetch(`${BASE_URL}${path}`, options)
|
|
if (!res.ok) {
|
|
throw new Error(`API error: ${res.status} ${res.statusText}`)
|
|
}
|
|
return res.json() as Promise<T>
|
|
}
|
|
|
|
export async function getModels(): Promise<ModelsListResponse> {
|
|
return request<ModelsListResponse>("/api/models")
|
|
}
|
|
|
|
export async function addModel(
|
|
model: Partial<ModelInfo>,
|
|
): Promise<ModelActionResponse> {
|
|
return request<ModelActionResponse>("/api/models", {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json" },
|
|
body: JSON.stringify(model),
|
|
})
|
|
}
|
|
|
|
export async function updateModel(
|
|
index: number,
|
|
model: Partial<ModelInfo>,
|
|
): Promise<ModelActionResponse> {
|
|
return request<ModelActionResponse>(`/api/models/${index}`, {
|
|
method: "PUT",
|
|
headers: { "Content-Type": "application/json" },
|
|
body: JSON.stringify(model),
|
|
})
|
|
}
|
|
|
|
export async function deleteModel(index: number): Promise<ModelActionResponse> {
|
|
return request<ModelActionResponse>(`/api/models/${index}`, {
|
|
method: "DELETE",
|
|
})
|
|
}
|
|
|
|
export async function setDefaultModel(
|
|
modelName: string,
|
|
): Promise<ModelActionResponse> {
|
|
const response = await request<ModelActionResponse>("/api/models/default", {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json" },
|
|
body: JSON.stringify({ model_name: modelName }),
|
|
})
|
|
|
|
await refreshGatewayState()
|
|
return response
|
|
}
|
|
|
|
export type { ModelsListResponse, ModelActionResponse }
|