fix(image): allow custom model ids
This commit is contained in:
@@ -3574,11 +3574,10 @@ function ImageGenerationSettings({
|
|||||||
IMAGE_SIZE_OPTIONS.map((value) => ({ name: value, label: value })),
|
IMAGE_SIZE_OPTIONS.map((value) => ({ name: value, label: value })),
|
||||||
form.defaultImageSize,
|
form.defaultImageSize,
|
||||||
);
|
);
|
||||||
const modelOptions = optionRowsWithCurrent(
|
const modelOptions = (selectedProvider?.models ?? []).map((model) => ({
|
||||||
(selectedProvider?.models ?? []).map((model) => ({ name: model, label: model })),
|
name: model,
|
||||||
form.model,
|
label: model,
|
||||||
);
|
}));
|
||||||
const hasModelCatalog = Boolean(selectedProvider?.models?.length);
|
|
||||||
|
|
||||||
const selectProvider = (provider: string) => {
|
const selectProvider = (provider: string) => {
|
||||||
const nextProvider = settings.image_generation.providers.find((row) => row.name === provider);
|
const nextProvider = settings.image_generation.providers.find((row) => row.name === provider);
|
||||||
@@ -3649,22 +3648,21 @@ function ImageGenerationSettings({
|
|||||||
title={tx("settings.rows.imageModel", "Image model")}
|
title={tx("settings.rows.imageModel", "Image model")}
|
||||||
description={tx("settings.help.imageModel", "Model name sent to the selected image provider.")}
|
description={tx("settings.help.imageModel", "Model name sent to the selected image provider.")}
|
||||||
>
|
>
|
||||||
{hasModelCatalog ? (
|
<EditableOptionPicker
|
||||||
<ProviderPicker
|
options={modelOptions}
|
||||||
providers={modelOptions}
|
value={form.model}
|
||||||
value={form.model}
|
emptyLabel={tx("settings.image.selectModel", "Select image model")}
|
||||||
emptyLabel={tx("settings.image.selectModel", "Select image model")}
|
searchPlaceholder={tx(
|
||||||
onChange={(model) => onChangeForm((prev) => ({ ...prev, model }))}
|
"settings.image.searchOrTypeModel",
|
||||||
triggerClassName="w-[min(300px,70vw)]"
|
"Search or type model ID",
|
||||||
contentClassName="w-[320px]"
|
)}
|
||||||
/>
|
emptyMessage={tx(
|
||||||
) : (
|
"settings.image.typeModelId",
|
||||||
<Input
|
"Type the model ID supported by this provider.",
|
||||||
value={form.model}
|
)}
|
||||||
onChange={(event) => onChangeForm((prev) => ({ ...prev, model: event.target.value }))}
|
useCustomLabel={tx("settings.models.useCustomModel", "Use")}
|
||||||
className="h-8 w-[min(300px,70vw)] rounded-full text-[13px]"
|
onChange={(model) => onChangeForm((prev) => ({ ...prev, model }))}
|
||||||
/>
|
/>
|
||||||
)}
|
|
||||||
</SettingsRow>
|
</SettingsRow>
|
||||||
<SettingsRow
|
<SettingsRow
|
||||||
title={tx("settings.rows.defaultAspectRatio", "Default aspect")}
|
title={tx("settings.rows.defaultAspectRatio", "Default aspect")}
|
||||||
@@ -7735,6 +7733,137 @@ function ProviderPicker({
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function EditableOptionPicker({
|
||||||
|
options,
|
||||||
|
value,
|
||||||
|
emptyLabel,
|
||||||
|
searchPlaceholder,
|
||||||
|
emptyMessage,
|
||||||
|
useCustomLabel,
|
||||||
|
onChange,
|
||||||
|
}: {
|
||||||
|
options: Array<{ name: string; label: string }>;
|
||||||
|
value: string;
|
||||||
|
emptyLabel: string;
|
||||||
|
searchPlaceholder: string;
|
||||||
|
emptyMessage: string;
|
||||||
|
useCustomLabel: string;
|
||||||
|
onChange: (value: string) => void;
|
||||||
|
}) {
|
||||||
|
const [open, setOpen] = useState(false);
|
||||||
|
const [query, setQuery] = useState("");
|
||||||
|
const normalizedQuery = query.trim().toLowerCase();
|
||||||
|
const customCandidate = query.trim();
|
||||||
|
const exactMatch = options.some((option) => option.name === customCandidate);
|
||||||
|
const visibleOptions = options.filter((option) =>
|
||||||
|
[option.name, option.label].some((field) => field.toLowerCase().includes(normalizedQuery)),
|
||||||
|
);
|
||||||
|
|
||||||
|
const selectValue = (nextValue: string) => {
|
||||||
|
onChange(nextValue);
|
||||||
|
setQuery("");
|
||||||
|
setOpen(false);
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<DropdownMenu
|
||||||
|
open={open}
|
||||||
|
onOpenChange={(nextOpen) => {
|
||||||
|
setOpen(nextOpen);
|
||||||
|
if (!nextOpen) setQuery("");
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<DropdownMenuTrigger asChild>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
variant="outline"
|
||||||
|
className={cn(
|
||||||
|
"h-8 w-[min(300px,70vw)] justify-between rounded-full border-input bg-background px-3 text-[13px] font-normal shadow-none",
|
||||||
|
"hover:bg-accent/55 focus-visible:ring-2 focus-visible:ring-ring",
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
<span className={cn("truncate", !value && "text-muted-foreground")}>
|
||||||
|
{value || emptyLabel}
|
||||||
|
</span>
|
||||||
|
<ChevronDown className="ml-2 h-3.5 w-3.5 shrink-0 text-muted-foreground" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
</DropdownMenuTrigger>
|
||||||
|
<DropdownMenuContent
|
||||||
|
align="end"
|
||||||
|
className="w-[320px] max-w-[calc(100vw-2rem)] p-1.5"
|
||||||
|
>
|
||||||
|
<div className="p-1 pb-1.5">
|
||||||
|
<div className="relative">
|
||||||
|
<Search
|
||||||
|
className="pointer-events-none absolute left-3 top-1/2 h-3.5 w-3.5 -translate-y-1/2 text-muted-foreground"
|
||||||
|
aria-hidden
|
||||||
|
/>
|
||||||
|
<Input
|
||||||
|
autoFocus
|
||||||
|
value={query}
|
||||||
|
onChange={(event) => setQuery(event.target.value)}
|
||||||
|
onKeyDown={(event) => {
|
||||||
|
event.stopPropagation();
|
||||||
|
if (event.key === "Enter" && customCandidate) {
|
||||||
|
event.preventDefault();
|
||||||
|
selectValue(customCandidate);
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
placeholder={searchPlaceholder}
|
||||||
|
aria-label={searchPlaceholder}
|
||||||
|
className="h-8 rounded-full pl-8 pr-3 text-[12px]"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{visibleOptions.length ? (
|
||||||
|
<div className="max-h-[14rem] overflow-y-auto pr-0.5 scrollbar-thin scrollbar-track-transparent">
|
||||||
|
{visibleOptions.map((option) => {
|
||||||
|
const selected = option.name === value;
|
||||||
|
return (
|
||||||
|
<DropdownMenuItem
|
||||||
|
key={option.name}
|
||||||
|
onSelect={() => selectValue(option.name)}
|
||||||
|
className={cn(
|
||||||
|
"flex cursor-default items-center justify-between gap-2 rounded-[12px] px-2.5 py-2 text-[13px]",
|
||||||
|
"focus:bg-muted/85 focus:text-foreground",
|
||||||
|
selected && "bg-muted/80 text-foreground focus:bg-muted",
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
<span className="min-w-0 truncate">{option.label}</span>
|
||||||
|
{selected ? <Check className="h-3.5 w-3.5 shrink-0" aria-hidden /> : null}
|
||||||
|
</DropdownMenuItem>
|
||||||
|
);
|
||||||
|
})}
|
||||||
|
</div>
|
||||||
|
) : !customCandidate ? (
|
||||||
|
<div className="px-3 py-2 text-[11px] leading-4 text-muted-foreground">
|
||||||
|
{emptyMessage}
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
|
|
||||||
|
{customCandidate && !exactMatch ? (
|
||||||
|
<>
|
||||||
|
{visibleOptions.length ? <DropdownMenuSeparator /> : null}
|
||||||
|
<DropdownMenuItem
|
||||||
|
onSelect={() => selectValue(customCandidate)}
|
||||||
|
className="flex cursor-default items-center gap-2 rounded-[12px] px-2 py-1.5 text-[12px] focus:bg-muted/85"
|
||||||
|
>
|
||||||
|
<span className="grid h-5 w-5 shrink-0 place-items-center rounded-md bg-muted/80 text-muted-foreground">
|
||||||
|
<Pencil className="h-3 w-3" aria-hidden />
|
||||||
|
</span>
|
||||||
|
<span className="min-w-0 truncate">
|
||||||
|
{useCustomLabel}{" "}
|
||||||
|
<span className="font-medium text-foreground">“{customCandidate}”</span>
|
||||||
|
</span>
|
||||||
|
</DropdownMenuItem>
|
||||||
|
</>
|
||||||
|
) : null}
|
||||||
|
</DropdownMenuContent>
|
||||||
|
</DropdownMenu>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
function ModelIdPicker({
|
function ModelIdPicker({
|
||||||
token,
|
token,
|
||||||
settings,
|
settings,
|
||||||
|
|||||||
@@ -2323,9 +2323,23 @@ describe("SettingsView Apps catalog", () => {
|
|||||||
expect(screen.getByRole("button", { name: "imagen-4.0-generate-001" })).toBeInTheDocument(),
|
expect(screen.getByRole("button", { name: "imagen-4.0-generate-001" })).toBeInTheDocument(),
|
||||||
);
|
);
|
||||||
|
|
||||||
|
fireEvent.pointerDown(screen.getByRole("button", { name: "imagen-4.0-generate-001" }));
|
||||||
|
const modelInput = await screen.findByRole("textbox", { name: "Search or type model ID" });
|
||||||
|
fireEvent.change(modelInput, { target: { value: "imagen-5-preview" } });
|
||||||
|
fireEvent.click(await screen.findByRole("menuitem", { name: "Use “imagen-5-preview”" }));
|
||||||
|
expect(await screen.findByRole("button", { name: "imagen-5-preview" })).toBeInTheDocument();
|
||||||
|
|
||||||
fireEvent.pointerDown(screen.getByRole("button", { name: "Gemini" }));
|
fireEvent.pointerDown(screen.getByRole("button", { name: "Gemini" }));
|
||||||
fireEvent.click(await screen.findByRole("menuitem", { name: "Custom" }));
|
fireEvent.click(await screen.findByRole("menuitem", { name: "Custom" }));
|
||||||
expect(screen.getByDisplayValue("imagen-4.0-generate-001")).toBeInTheDocument();
|
expect(screen.getByRole("button", { name: "imagen-5-preview" })).toBeInTheDocument();
|
||||||
|
|
||||||
|
fireEvent.pointerDown(screen.getByRole("button", { name: "imagen-5-preview" }));
|
||||||
|
const customProviderInput = await screen.findByRole("textbox", {
|
||||||
|
name: "Search or type model ID",
|
||||||
|
});
|
||||||
|
fireEvent.change(customProviderInput, { target: { value: "private/image-v2" } });
|
||||||
|
fireEvent.keyDown(customProviderInput, { key: "Enter" });
|
||||||
|
expect(await screen.findByRole("button", { name: "private/image-v2" })).toBeInTheDocument();
|
||||||
});
|
});
|
||||||
|
|
||||||
it("keeps the default model distinct from the active named configuration", async () => {
|
it("keeps the default model distinct from the active named configuration", async () => {
|
||||||
|
|||||||
Reference in New Issue
Block a user