administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-07-15 00:11:15 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

refresh model dropdown after changing provider or base url

Browse Source
This commit is contained in:
Josh Hawkins 2026-04-05 11:32:05 -05:00
parent d8c35d5a0f
commit f7f73e0de4
1 changed files with 32 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
web/src/components/config-form/theme/widgets/GenAIModelWidget.tsx
View File

@ -1,6 +1,6 @@
// Combobox widget for genai *.model fields. // Combobox widget for genai *.model fields.
// Fetches available models from the provider's backend and shows them in a dropdown. // Fetches available models from the provider's backend and shows them in a dropdown.
import { useState, useMemo } from "react"; import { useState, useMemo, useEffect, useRef } from "react";
import type { WidgetProps } from "@rjsf/utils"; import type { WidgetProps } from "@rjsf/utils";
import { useTranslation } from "react-i18next"; import { useTranslation } from "react-i18next";
import useSWR from "swr"; import useSWR from "swr";
@ -19,6 +19,7 @@ import {
PopoverContent, PopoverContent,
PopoverTrigger, PopoverTrigger,
} from "@/components/ui/popover"; } from "@/components/ui/popover";
import type { ConfigFormContext } from "@/types/configForm";
import { getSizedFieldClassName } from "../utils"; import { getSizedFieldClassName } from "../utils";
/** /**
@ -37,17 +38,45 @@ function getProviderKey(widgetId: string): string | undefined {
} }
export function GenAIModelWidget(props: WidgetProps) { export function GenAIModelWidget(props: WidgetProps) {
const { id, value, disabled, readonly, onChange, options } = props; const { id, value, disabled, readonly, onChange, options, registry } = props;
const { t } = useTranslation(["views/settings"]); const { t } = useTranslation(["views/settings"]);
const [open, setOpen] = useState(false); const [open, setOpen] = useState(false);
const fieldClassName = getSizedFieldClassName(options, "sm"); const fieldClassName = getSizedFieldClassName(options, "sm");
const providerKey = useMemo(() => getProviderKey(id), [id]); const providerKey = useMemo(() => getProviderKey(id), [id]);
const { data: allModels } = useSWR<Record<string, string[]>>("genai/models", { const formContext = registry?.formContext as ConfigFormContext | undefined;
// Build a fingerprint from the saved config's provider + base_url so the
// SWR key changes (and models are refetched) whenever those fields are saved.
const configFingerprint = useMemo(() => {
if (!providerKey) return "";
const genai = (
formContext?.fullConfig as Record<string, unknown> | undefined
)?.genai;
if (!genai || typeof genai !== "object" || Array.isArray(genai)) return "";
const entry = (genai as Record<string, unknown>)[providerKey];
if (!entry || typeof entry !== "object" || Array.isArray(entry)) return "";
const e = entry as Record<string, unknown>;
return `${e.provider ?? ""}|${e.base_url ?? ""}`;
}, [providerKey, formContext?.fullConfig]);
const { data: allModels, mutate: mutateModels } = useSWR<
Record<string, string[]>
>("genai/models", {
revalidateOnFocus: false, revalidateOnFocus: false,
}); });
// Revalidate models when the saved config fingerprint changes (e.g. after
// switching provider or base_url and saving).
const prevFingerprint = useRef(configFingerprint);
useEffect(() => {
if (configFingerprint !== prevFingerprint.current) {
prevFingerprint.current = configFingerprint;
mutateModels();
}
}, [configFingerprint, mutateModels]);
const models = useMemo(() => { const models = useMemo(() => {
if (!allModels || !providerKey) return []; if (!allModels || !providerKey) return [];
return allModels[providerKey] ?? []; return allModels[providerKey] ?? [];