56 lines
1.3 KiB
TypeScript
56 lines
1.3 KiB
TypeScript
import type { AiModel } from "@/lib/db/schema";
|
|
|
|
export type ModelFrontendAvailabilityStatus =
|
|
| "available"
|
|
| "maintenance"
|
|
| "testing";
|
|
|
|
export type ModelFrontendAvailability = {
|
|
availabilityStatus: ModelFrontendAvailabilityStatus;
|
|
availabilityLabel: string | null;
|
|
selectable: boolean;
|
|
disabledReason: string | null;
|
|
};
|
|
|
|
type AvailabilityModel = Pick<
|
|
AiModel,
|
|
| "isEnabled"
|
|
| "healthStatus"
|
|
| "healthAutoDisabledAt"
|
|
| "healthObservationStartedAt"
|
|
>;
|
|
|
|
export function getAiModelFrontendAvailability(
|
|
model: AvailabilityModel
|
|
): ModelFrontendAvailability {
|
|
const healthStatus = model.healthStatus || "healthy";
|
|
|
|
if (
|
|
healthStatus === "auto_disabled" ||
|
|
(model.isEnabled === false && Boolean(model.healthAutoDisabledAt))
|
|
) {
|
|
return {
|
|
availabilityStatus: "maintenance",
|
|
availabilityLabel: "维护中",
|
|
selectable: false,
|
|
disabledReason: "模型维护中，请稍后再试",
|
|
};
|
|
}
|
|
|
|
if (healthStatus === "degraded" && model.healthObservationStartedAt) {
|
|
return {
|
|
availabilityStatus: "testing",
|
|
availabilityLabel: "测试中",
|
|
selectable: true,
|
|
disabledReason: null,
|
|
};
|
|
}
|
|
|
|
return {
|
|
availabilityStatus: "available",
|
|
availabilityLabel: null,
|
|
selectable: true,
|
|
disabledReason: null,
|
|
};
|
|
}
|