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
84 lines
2.4 KiB
TypeScript
84 lines
2.4 KiB
TypeScript
import { useTranslation } from "react-i18next"
|
|
|
|
import type { ModelInfo } from "@/api/models"
|
|
import {
|
|
Select,
|
|
SelectContent,
|
|
SelectGroup,
|
|
SelectItem,
|
|
SelectLabel,
|
|
SelectSeparator,
|
|
SelectTrigger,
|
|
SelectValue,
|
|
} from "@/components/ui/select"
|
|
|
|
interface ModelSelectorProps {
|
|
defaultModelName: string
|
|
apiKeyModels: ModelInfo[]
|
|
oauthModels: ModelInfo[]
|
|
localModels: ModelInfo[]
|
|
onValueChange: (modelName: string) => void
|
|
}
|
|
|
|
export function ModelSelector({
|
|
defaultModelName,
|
|
apiKeyModels,
|
|
oauthModels,
|
|
localModels,
|
|
onValueChange,
|
|
}: ModelSelectorProps) {
|
|
const { t } = useTranslation()
|
|
|
|
return (
|
|
<Select value={defaultModelName} onValueChange={onValueChange}>
|
|
<SelectTrigger
|
|
size="sm"
|
|
className="text-muted-foreground hover:text-foreground focus-visible:border-input h-8 max-w-[160px] min-w-[80px] bg-transparent shadow-none focus-visible:ring-0 sm:max-w-[220px]"
|
|
>
|
|
<SelectValue placeholder={t("chat.noModel")} />
|
|
</SelectTrigger>
|
|
<SelectContent position="popper" align="start">
|
|
{apiKeyModels.length > 0 && (
|
|
<SelectGroup>
|
|
<SelectLabel>{t("chat.modelGroup.apikey")}</SelectLabel>
|
|
{apiKeyModels.map((model) => (
|
|
<SelectItem key={model.index} value={model.model_name}>
|
|
{model.model_name}
|
|
</SelectItem>
|
|
))}
|
|
</SelectGroup>
|
|
)}
|
|
{apiKeyModels.length > 0 &&
|
|
(oauthModels.length > 0 || localModels.length > 0) && (
|
|
<SelectSeparator />
|
|
)}
|
|
|
|
{oauthModels.length > 0 && (
|
|
<SelectGroup>
|
|
<SelectLabel>{t("chat.modelGroup.oauth")}</SelectLabel>
|
|
{oauthModels.map((model) => (
|
|
<SelectItem key={model.index} value={model.model_name}>
|
|
{model.model_name}
|
|
</SelectItem>
|
|
))}
|
|
</SelectGroup>
|
|
)}
|
|
{oauthModels.length > 0 &&
|
|
(localModels.length > 0 || apiKeyModels.length > 0) && (
|
|
<SelectSeparator />
|
|
)}
|
|
|
|
{localModels.length > 0 && (
|
|
<SelectGroup>
|
|
<SelectLabel>{t("chat.modelGroup.local")}</SelectLabel>
|
|
{localModels.map((model) => (
|
|
<SelectItem key={model.index} value={model.model_name}>
|
|
{model.model_name}
|
|
</SelectItem>
|
|
))}
|
|
</SelectGroup>
|
|
)}
|
|
</SelectContent>
|
|
</Select>
|
|
)
|
|
}
|