mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-17 18:35:19 +02:00
refactor: move ImageConfigDialog to shared components and update imports in chat-header and image-model-manager for better organization
This commit is contained in:
Anish Sarkar
parent
db26373594
commit
430372a4ff
5 changed files with 527 additions and 916 deletions
|
|
@ -24,13 +24,13 @@ export const createImageGenConfigMutationAtom = atomWithMutation((get) => {
|
||||||
return imageGenConfigApiService.createConfig(request);
|
return imageGenConfigApiService.createConfig(request);
|
||||||
},
|
},
|
||||||
onSuccess: () => {
|
onSuccess: () => {
|
||||||
toast.success("Image model configuration created");
|
toast.success("Image model created");
|
||||||
queryClient.invalidateQueries({
|
queryClient.invalidateQueries({
|
||||||
queryKey: cacheKeys.imageGenConfigs.all(Number(searchSpaceId)),
|
queryKey: cacheKeys.imageGenConfigs.all(Number(searchSpaceId)),
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
onError: (error: Error) => {
|
onError: (error: Error) => {
|
||||||
toast.error(error.message || "Failed to create image model configuration");
|
toast.error(error.message || "Failed to create image model");
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
});
|
});
|
||||||
|
|
@ -48,7 +48,7 @@ export const updateImageGenConfigMutationAtom = atomWithMutation((get) => {
|
||||||
return imageGenConfigApiService.updateConfig(request);
|
return imageGenConfigApiService.updateConfig(request);
|
||||||
},
|
},
|
||||||
onSuccess: (_: UpdateImageGenConfigResponse, request: UpdateImageGenConfigRequest) => {
|
onSuccess: (_: UpdateImageGenConfigResponse, request: UpdateImageGenConfigRequest) => {
|
||||||
toast.success("Image model configuration updated");
|
toast.success("Image model updated");
|
||||||
queryClient.invalidateQueries({
|
queryClient.invalidateQueries({
|
||||||
queryKey: cacheKeys.imageGenConfigs.all(Number(searchSpaceId)),
|
queryKey: cacheKeys.imageGenConfigs.all(Number(searchSpaceId)),
|
||||||
});
|
});
|
||||||
|
|
@ -57,7 +57,7 @@ export const updateImageGenConfigMutationAtom = atomWithMutation((get) => {
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
onError: (error: Error) => {
|
onError: (error: Error) => {
|
||||||
toast.error(error.message || "Failed to update image model configuration");
|
toast.error(error.message || "Failed to update image model");
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
});
|
});
|
||||||
|
|
@ -75,7 +75,7 @@ export const deleteImageGenConfigMutationAtom = atomWithMutation((get) => {
|
||||||
return imageGenConfigApiService.deleteConfig(id);
|
return imageGenConfigApiService.deleteConfig(id);
|
||||||
},
|
},
|
||||||
onSuccess: (_, id: number) => {
|
onSuccess: (_, id: number) => {
|
||||||
toast.success("Image model configuration deleted");
|
toast.success("Image model deleted");
|
||||||
queryClient.setQueryData(
|
queryClient.setQueryData(
|
||||||
cacheKeys.imageGenConfigs.all(Number(searchSpaceId)),
|
cacheKeys.imageGenConfigs.all(Number(searchSpaceId)),
|
||||||
(oldData: GetImageGenConfigsResponse | undefined) => {
|
(oldData: GetImageGenConfigsResponse | undefined) => {
|
||||||
|
|
@ -85,7 +85,7 @@ export const deleteImageGenConfigMutationAtom = atomWithMutation((get) => {
|
||||||
);
|
);
|
||||||
},
|
},
|
||||||
onError: (error: Error) => {
|
onError: (error: Error) => {
|
||||||
toast.error(error.message || "Failed to delete image model configuration");
|
toast.error(error.message || "Failed to delete image model");
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
});
|
});
|
||||||
|
|
|
||||||
|
|
@ -7,7 +7,7 @@ import type {
|
||||||
ImageGenerationConfig,
|
ImageGenerationConfig,
|
||||||
NewLLMConfigPublic,
|
NewLLMConfigPublic,
|
||||||
} from "@/contracts/types/new-llm-config.types";
|
} from "@/contracts/types/new-llm-config.types";
|
||||||
import { ImageConfigDialog } from "./image-config-dialog";
|
import { ImageConfigDialog } from "@/components/shared/image-config-dialog";
|
||||||
import { ModelConfigDialog } from "./model-config-dialog";
|
import { ModelConfigDialog } from "./model-config-dialog";
|
||||||
import { ModelSelector } from "./model-selector";
|
import { ModelSelector } from "./model-selector";
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,558 +0,0 @@
|
||||||
"use client";
|
|
||||||
|
|
||||||
import { useAtomValue } from "jotai";
|
|
||||||
import { AlertCircle, Check, ChevronsUpDown, X } from "lucide-react";
|
|
||||||
import { AnimatePresence, motion } from "motion/react";
|
|
||||||
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
|
|
||||||
import { createPortal } from "react-dom";
|
|
||||||
import { toast } from "sonner";
|
|
||||||
import {
|
|
||||||
createImageGenConfigMutationAtom,
|
|
||||||
updateImageGenConfigMutationAtom,
|
|
||||||
} from "@/atoms/image-gen-config/image-gen-config-mutation.atoms";
|
|
||||||
import { updateLLMPreferencesMutationAtom } from "@/atoms/new-llm-config/new-llm-config-mutation.atoms";
|
|
||||||
import { Alert, AlertDescription } from "@/components/ui/alert";
|
|
||||||
import { Badge } from "@/components/ui/badge";
|
|
||||||
import { Button } from "@/components/ui/button";
|
|
||||||
import {
|
|
||||||
Command,
|
|
||||||
CommandEmpty,
|
|
||||||
CommandGroup,
|
|
||||||
CommandInput,
|
|
||||||
CommandItem,
|
|
||||||
CommandList,
|
|
||||||
} from "@/components/ui/command";
|
|
||||||
import { Input } from "@/components/ui/input";
|
|
||||||
import { Label } from "@/components/ui/label";
|
|
||||||
import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover";
|
|
||||||
import {
|
|
||||||
Select,
|
|
||||||
SelectContent,
|
|
||||||
SelectItem,
|
|
||||||
SelectTrigger,
|
|
||||||
SelectValue,
|
|
||||||
} from "@/components/ui/select";
|
|
||||||
import { Separator } from "@/components/ui/separator";
|
|
||||||
import { Spinner } from "@/components/ui/spinner";
|
|
||||||
import { IMAGE_GEN_MODELS, IMAGE_GEN_PROVIDERS } from "@/contracts/enums/image-gen-providers";
|
|
||||||
import type {
|
|
||||||
GlobalImageGenConfig,
|
|
||||||
ImageGenerationConfig,
|
|
||||||
ImageGenProvider,
|
|
||||||
} from "@/contracts/types/new-llm-config.types";
|
|
||||||
import { cn } from "@/lib/utils";
|
|
||||||
|
|
||||||
interface ImageConfigDialogProps {
|
|
||||||
open: boolean;
|
|
||||||
onOpenChange: (open: boolean) => void;
|
|
||||||
config: ImageGenerationConfig | GlobalImageGenConfig | null;
|
|
||||||
isGlobal: boolean;
|
|
||||||
searchSpaceId: number;
|
|
||||||
mode: "create" | "edit" | "view";
|
|
||||||
}
|
|
||||||
|
|
||||||
const INITIAL_FORM = {
|
|
||||||
name: "",
|
|
||||||
description: "",
|
|
||||||
provider: "",
|
|
||||||
model_name: "",
|
|
||||||
api_key: "",
|
|
||||||
api_base: "",
|
|
||||||
api_version: "",
|
|
||||||
};
|
|
||||||
|
|
||||||
export function ImageConfigDialog({
|
|
||||||
open,
|
|
||||||
onOpenChange,
|
|
||||||
config,
|
|
||||||
isGlobal,
|
|
||||||
searchSpaceId,
|
|
||||||
mode,
|
|
||||||
}: ImageConfigDialogProps) {
|
|
||||||
const [isSubmitting, setIsSubmitting] = useState(false);
|
|
||||||
const [mounted, setMounted] = useState(false);
|
|
||||||
const [formData, setFormData] = useState(INITIAL_FORM);
|
|
||||||
const [modelComboboxOpen, setModelComboboxOpen] = useState(false);
|
|
||||||
const [scrollPos, setScrollPos] = useState<"top" | "middle" | "bottom">("top");
|
|
||||||
const scrollRef = useRef<HTMLDivElement>(null);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
setMounted(true);
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
if (open) {
|
|
||||||
if (mode === "edit" && config && !isGlobal) {
|
|
||||||
setFormData({
|
|
||||||
name: config.name || "",
|
|
||||||
description: config.description || "",
|
|
||||||
provider: config.provider || "",
|
|
||||||
model_name: config.model_name || "",
|
|
||||||
api_key: (config as ImageGenerationConfig).api_key || "",
|
|
||||||
api_base: config.api_base || "",
|
|
||||||
api_version: config.api_version || "",
|
|
||||||
});
|
|
||||||
} else if (mode === "create") {
|
|
||||||
setFormData(INITIAL_FORM);
|
|
||||||
}
|
|
||||||
setScrollPos("top");
|
|
||||||
}
|
|
||||||
}, [open, mode, config, isGlobal]);
|
|
||||||
|
|
||||||
const { mutateAsync: createConfig } = useAtomValue(createImageGenConfigMutationAtom);
|
|
||||||
const { mutateAsync: updateConfig } = useAtomValue(updateImageGenConfigMutationAtom);
|
|
||||||
const { mutateAsync: updatePreferences } = useAtomValue(updateLLMPreferencesMutationAtom);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
const handleEscape = (e: KeyboardEvent) => {
|
|
||||||
if (e.key === "Escape" && open) onOpenChange(false);
|
|
||||||
};
|
|
||||||
window.addEventListener("keydown", handleEscape);
|
|
||||||
return () => window.removeEventListener("keydown", handleEscape);
|
|
||||||
}, [open, onOpenChange]);
|
|
||||||
|
|
||||||
const handleScroll = useCallback((e: React.UIEvent<HTMLDivElement>) => {
|
|
||||||
const el = e.currentTarget;
|
|
||||||
const atTop = el.scrollTop <= 2;
|
|
||||||
const atBottom = el.scrollHeight - el.scrollTop - el.clientHeight <= 2;
|
|
||||||
setScrollPos(atTop ? "top" : atBottom ? "bottom" : "middle");
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
const isAutoMode = config && "is_auto_mode" in config && config.is_auto_mode;
|
|
||||||
|
|
||||||
const suggestedModels = useMemo(() => {
|
|
||||||
if (!formData.provider) return [];
|
|
||||||
return IMAGE_GEN_MODELS.filter((m) => m.provider === formData.provider);
|
|
||||||
}, [formData.provider]);
|
|
||||||
|
|
||||||
const getTitle = () => {
|
|
||||||
if (mode === "create") return "Add Image Model";
|
|
||||||
if (isAutoMode) return "Auto Mode (Fastest)";
|
|
||||||
if (isGlobal) return "View Global Image Model";
|
|
||||||
return "Edit Image Model";
|
|
||||||
};
|
|
||||||
|
|
||||||
const getSubtitle = () => {
|
|
||||||
if (mode === "create") return "Set up a new image generation provider";
|
|
||||||
if (isAutoMode) return "Automatically routes requests across providers";
|
|
||||||
if (isGlobal) return "Read-only global configuration";
|
|
||||||
return "Update your image model settings";
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleSubmit = useCallback(async () => {
|
|
||||||
setIsSubmitting(true);
|
|
||||||
try {
|
|
||||||
if (mode === "create") {
|
|
||||||
const result = await createConfig({
|
|
||||||
name: formData.name,
|
|
||||||
provider: formData.provider as ImageGenProvider,
|
|
||||||
model_name: formData.model_name,
|
|
||||||
api_key: formData.api_key,
|
|
||||||
api_base: formData.api_base || undefined,
|
|
||||||
api_version: formData.api_version || undefined,
|
|
||||||
description: formData.description || undefined,
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
});
|
|
||||||
if (result?.id) {
|
|
||||||
await updatePreferences({
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
data: { image_generation_config_id: result.id },
|
|
||||||
});
|
|
||||||
}
|
|
||||||
toast.success("Image model created and assigned!");
|
|
||||||
onOpenChange(false);
|
|
||||||
} else if (!isGlobal && config) {
|
|
||||||
await updateConfig({
|
|
||||||
id: config.id,
|
|
||||||
data: {
|
|
||||||
name: formData.name,
|
|
||||||
description: formData.description || undefined,
|
|
||||||
provider: formData.provider as ImageGenProvider,
|
|
||||||
model_name: formData.model_name,
|
|
||||||
api_key: formData.api_key,
|
|
||||||
api_base: formData.api_base || undefined,
|
|
||||||
api_version: formData.api_version || undefined,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
toast.success("Image model updated!");
|
|
||||||
onOpenChange(false);
|
|
||||||
}
|
|
||||||
} catch (error) {
|
|
||||||
console.error("Failed to save image config:", error);
|
|
||||||
toast.error("Failed to save image model");
|
|
||||||
} finally {
|
|
||||||
setIsSubmitting(false);
|
|
||||||
}
|
|
||||||
}, [
|
|
||||||
mode,
|
|
||||||
isGlobal,
|
|
||||||
config,
|
|
||||||
formData,
|
|
||||||
searchSpaceId,
|
|
||||||
createConfig,
|
|
||||||
updateConfig,
|
|
||||||
updatePreferences,
|
|
||||||
onOpenChange,
|
|
||||||
]);
|
|
||||||
|
|
||||||
const handleUseGlobalConfig = useCallback(async () => {
|
|
||||||
if (!config || !isGlobal) return;
|
|
||||||
setIsSubmitting(true);
|
|
||||||
try {
|
|
||||||
await updatePreferences({
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
data: { image_generation_config_id: config.id },
|
|
||||||
});
|
|
||||||
toast.success(`Now using ${config.name}`);
|
|
||||||
onOpenChange(false);
|
|
||||||
} catch (error) {
|
|
||||||
console.error("Failed to set image model:", error);
|
|
||||||
toast.error("Failed to set image model");
|
|
||||||
} finally {
|
|
||||||
setIsSubmitting(false);
|
|
||||||
}
|
|
||||||
}, [config, isGlobal, searchSpaceId, updatePreferences, onOpenChange]);
|
|
||||||
|
|
||||||
const isFormValid = formData.name && formData.provider && formData.model_name && formData.api_key;
|
|
||||||
const selectedProvider = IMAGE_GEN_PROVIDERS.find((p) => p.value === formData.provider);
|
|
||||||
|
|
||||||
if (!mounted) return null;
|
|
||||||
|
|
||||||
const dialogContent = (
|
|
||||||
<AnimatePresence>
|
|
||||||
{open && (
|
|
||||||
<>
|
|
||||||
<motion.div
|
|
||||||
initial={{ opacity: 0 }}
|
|
||||||
animate={{ opacity: 1 }}
|
|
||||||
exit={{ opacity: 0 }}
|
|
||||||
transition={{ duration: 0.15 }}
|
|
||||||
className="fixed inset-0 z-50 bg-black/50 backdrop-blur-sm"
|
|
||||||
onClick={() => onOpenChange(false)}
|
|
||||||
/>
|
|
||||||
|
|
||||||
<motion.div
|
|
||||||
initial={{ opacity: 0, scale: 0.96 }}
|
|
||||||
animate={{ opacity: 1, scale: 1 }}
|
|
||||||
exit={{ opacity: 0, scale: 0.96 }}
|
|
||||||
transition={{ duration: 0.15, ease: "easeOut" }}
|
|
||||||
className="fixed inset-0 z-50 flex items-center justify-center p-4 sm:p-6"
|
|
||||||
>
|
|
||||||
<div
|
|
||||||
role="dialog"
|
|
||||||
aria-modal="true"
|
|
||||||
className={cn(
|
|
||||||
"relative w-full max-w-lg h-[85vh]",
|
|
||||||
"rounded-xl bg-background shadow-2xl",
|
|
||||||
"dark:bg-neutral-900",
|
|
||||||
"flex flex-col overflow-hidden"
|
|
||||||
)}
|
|
||||||
onClick={(e) => e.stopPropagation()}
|
|
||||||
onKeyDown={(e) => {
|
|
||||||
if (e.key === "Escape") onOpenChange(false);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
{/* Header */}
|
|
||||||
<div className="flex items-start justify-between px-6 pt-6 pb-4">
|
|
||||||
<div className="space-y-1 pr-8">
|
|
||||||
<div className="flex items-center gap-2">
|
|
||||||
<h2 className="text-lg font-semibold tracking-tight">{getTitle()}</h2>
|
|
||||||
{isAutoMode && (
|
|
||||||
<Badge variant="secondary" className="text-[10px]">
|
|
||||||
Recommended
|
|
||||||
</Badge>
|
|
||||||
)}
|
|
||||||
{isGlobal && !isAutoMode && mode !== "create" && (
|
|
||||||
<Badge variant="secondary" className="text-[10px]">
|
|
||||||
Global
|
|
||||||
</Badge>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
<p className="text-sm text-muted-foreground">{getSubtitle()}</p>
|
|
||||||
{config && !isAutoMode && mode !== "create" && (
|
|
||||||
<p className="text-xs font-mono text-muted-foreground/70">
|
|
||||||
{config.model_name}
|
|
||||||
</p>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
<Button
|
|
||||||
variant="ghost"
|
|
||||||
size="icon"
|
|
||||||
onClick={() => onOpenChange(false)}
|
|
||||||
className="absolute right-4 top-4 h-8 w-8 rounded-full text-muted-foreground hover:text-foreground"
|
|
||||||
>
|
|
||||||
<X className="h-4 w-4" />
|
|
||||||
<span className="sr-only">Close</span>
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{/* Scrollable content */}
|
|
||||||
<div
|
|
||||||
ref={scrollRef}
|
|
||||||
onScroll={handleScroll}
|
|
||||||
className="flex-1 overflow-y-auto px-6 py-5"
|
|
||||||
style={{
|
|
||||||
maskImage: `linear-gradient(to bottom, ${scrollPos === "top" ? "black" : "transparent"}, black 16px, black calc(100% - 16px), ${scrollPos === "bottom" ? "black" : "transparent"})`,
|
|
||||||
WebkitMaskImage: `linear-gradient(to bottom, ${scrollPos === "top" ? "black" : "transparent"}, black 16px, black calc(100% - 16px), ${scrollPos === "bottom" ? "black" : "transparent"})`,
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
{isAutoMode && (
|
|
||||||
<Alert className="mb-5 border-violet-500/30 bg-violet-500/5">
|
|
||||||
<AlertDescription className="text-sm text-violet-700 dark:text-violet-400">
|
|
||||||
Auto mode distributes image generation requests across all configured
|
|
||||||
providers for optimal performance and rate limit protection.
|
|
||||||
</AlertDescription>
|
|
||||||
</Alert>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{isGlobal && !isAutoMode && config && (
|
|
||||||
<>
|
|
||||||
<Alert className="mb-5 border-amber-500/30 bg-amber-500/5">
|
|
||||||
<AlertCircle className="size-4 text-amber-500" />
|
|
||||||
<AlertDescription className="text-sm text-amber-700 dark:text-amber-400">
|
|
||||||
Global configurations are read-only. To customize, create a new model.
|
|
||||||
</AlertDescription>
|
|
||||||
</Alert>
|
|
||||||
<div className="space-y-4">
|
|
||||||
<div className="grid gap-4 sm:grid-cols-2">
|
|
||||||
<div className="space-y-1.5">
|
|
||||||
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
|
||||||
Name
|
|
||||||
</div>
|
|
||||||
<p className="text-sm font-medium">{config.name}</p>
|
|
||||||
</div>
|
|
||||||
{config.description && (
|
|
||||||
<div className="space-y-1.5">
|
|
||||||
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
|
||||||
Description
|
|
||||||
</div>
|
|
||||||
<p className="text-sm text-muted-foreground">{config.description}</p>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
<Separator />
|
|
||||||
<div className="grid gap-4 sm:grid-cols-2">
|
|
||||||
<div className="space-y-1.5">
|
|
||||||
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
|
||||||
Provider
|
|
||||||
</div>
|
|
||||||
<p className="text-sm font-medium">{config.provider}</p>
|
|
||||||
</div>
|
|
||||||
<div className="space-y-1.5">
|
|
||||||
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
|
||||||
Model
|
|
||||||
</div>
|
|
||||||
<p className="text-sm font-medium font-mono">{config.model_name}</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{(mode === "create" || (mode === "edit" && !isGlobal)) && (
|
|
||||||
<div className="space-y-4">
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">Name *</Label>
|
|
||||||
<Input
|
|
||||||
placeholder="e.g., My DALL-E 3"
|
|
||||||
value={formData.name}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, name: e.target.value }))}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">Description</Label>
|
|
||||||
<Input
|
|
||||||
placeholder="Optional description"
|
|
||||||
value={formData.description}
|
|
||||||
onChange={(e) =>
|
|
||||||
setFormData((p) => ({ ...p, description: e.target.value }))
|
|
||||||
}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Separator />
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">Provider *</Label>
|
|
||||||
<Select
|
|
||||||
value={formData.provider}
|
|
||||||
onValueChange={(val) =>
|
|
||||||
setFormData((p) => ({ ...p, provider: val, model_name: "" }))
|
|
||||||
}
|
|
||||||
>
|
|
||||||
<SelectTrigger>
|
|
||||||
<SelectValue placeholder="Select a provider" />
|
|
||||||
</SelectTrigger>
|
|
||||||
<SelectContent className="bg-muted dark:border-neutral-700">
|
|
||||||
{IMAGE_GEN_PROVIDERS.map((p) => (
|
|
||||||
<SelectItem key={p.value} value={p.value} description={p.example}>
|
|
||||||
{p.label}
|
|
||||||
</SelectItem>
|
|
||||||
))}
|
|
||||||
</SelectContent>
|
|
||||||
</Select>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">Model Name *</Label>
|
|
||||||
{suggestedModels.length > 0 ? (
|
|
||||||
<Popover open={modelComboboxOpen} onOpenChange={setModelComboboxOpen}>
|
|
||||||
<PopoverTrigger asChild>
|
|
||||||
<Button
|
|
||||||
variant="outline"
|
|
||||||
role="combobox"
|
|
||||||
className="w-full justify-between font-normal bg-transparent"
|
|
||||||
>
|
|
||||||
{formData.model_name || "Select or type a model..."}
|
|
||||||
<ChevronsUpDown className="ml-2 h-4 w-4 shrink-0 opacity-50" />
|
|
||||||
</Button>
|
|
||||||
</PopoverTrigger>
|
|
||||||
<PopoverContent
|
|
||||||
className="w-full p-0 bg-muted dark:border-neutral-700"
|
|
||||||
align="start"
|
|
||||||
>
|
|
||||||
<Command className="bg-transparent">
|
|
||||||
<CommandInput
|
|
||||||
placeholder="Search or type model..."
|
|
||||||
value={formData.model_name}
|
|
||||||
onValueChange={(val) =>
|
|
||||||
setFormData((p) => ({ ...p, model_name: val }))
|
|
||||||
}
|
|
||||||
/>
|
|
||||||
<CommandList>
|
|
||||||
<CommandEmpty>
|
|
||||||
<span className="text-xs text-muted-foreground">
|
|
||||||
Type a custom model name
|
|
||||||
</span>
|
|
||||||
</CommandEmpty>
|
|
||||||
<CommandGroup>
|
|
||||||
{suggestedModels.map((m) => (
|
|
||||||
<CommandItem
|
|
||||||
key={m.value}
|
|
||||||
value={m.value}
|
|
||||||
onSelect={() => {
|
|
||||||
setFormData((p) => ({ ...p, model_name: m.value }));
|
|
||||||
setModelComboboxOpen(false);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Check
|
|
||||||
className={cn(
|
|
||||||
"mr-2 h-4 w-4",
|
|
||||||
formData.model_name === m.value
|
|
||||||
? "opacity-100"
|
|
||||||
: "opacity-0"
|
|
||||||
)}
|
|
||||||
/>
|
|
||||||
<span className="font-mono text-sm">{m.value}</span>
|
|
||||||
<span className="ml-2 text-xs text-muted-foreground">
|
|
||||||
{m.label}
|
|
||||||
</span>
|
|
||||||
</CommandItem>
|
|
||||||
))}
|
|
||||||
</CommandGroup>
|
|
||||||
</CommandList>
|
|
||||||
</Command>
|
|
||||||
</PopoverContent>
|
|
||||||
</Popover>
|
|
||||||
) : (
|
|
||||||
<Input
|
|
||||||
placeholder="e.g., dall-e-3"
|
|
||||||
value={formData.model_name}
|
|
||||||
onChange={(e) =>
|
|
||||||
setFormData((p) => ({ ...p, model_name: e.target.value }))
|
|
||||||
}
|
|
||||||
/>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">API Key *</Label>
|
|
||||||
<Input
|
|
||||||
type="password"
|
|
||||||
placeholder="sk-..."
|
|
||||||
value={formData.api_key}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, api_key: e.target.value }))}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">API Base URL</Label>
|
|
||||||
<Input
|
|
||||||
placeholder={selectedProvider?.apiBase || "Optional"}
|
|
||||||
value={formData.api_base}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, api_base: e.target.value }))}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{formData.provider === "AZURE_OPENAI" && (
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">API Version (Azure)</Label>
|
|
||||||
<Input
|
|
||||||
placeholder="2024-02-15-preview"
|
|
||||||
value={formData.api_version}
|
|
||||||
onChange={(e) =>
|
|
||||||
setFormData((p) => ({ ...p, api_version: e.target.value }))
|
|
||||||
}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{/* Fixed footer */}
|
|
||||||
<div className="shrink-0 px-6 py-4 flex items-center justify-end gap-3">
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="secondary"
|
|
||||||
onClick={() => onOpenChange(false)}
|
|
||||||
disabled={isSubmitting}
|
|
||||||
className="text-sm h-9"
|
|
||||||
>
|
|
||||||
Cancel
|
|
||||||
</Button>
|
|
||||||
{mode === "create" || (mode === "edit" && !isGlobal) ? (
|
|
||||||
<Button
|
|
||||||
onClick={handleSubmit}
|
|
||||||
disabled={isSubmitting || !isFormValid}
|
|
||||||
className="text-sm h-9 min-w-[120px]"
|
|
||||||
>
|
|
||||||
{isSubmitting ? (
|
|
||||||
<>
|
|
||||||
<Spinner size="sm" />
|
|
||||||
{mode === "edit" ? "Saving" : "Creating"}
|
|
||||||
</>
|
|
||||||
) : mode === "edit" ? (
|
|
||||||
"Save Changes"
|
|
||||||
) : (
|
|
||||||
"Create & Use"
|
|
||||||
)}
|
|
||||||
</Button>
|
|
||||||
) : isAutoMode ? (
|
|
||||||
<Button
|
|
||||||
className="text-sm h-9 gap-2 bg-gradient-to-r from-violet-500 to-purple-600 hover:from-violet-600 hover:to-purple-700"
|
|
||||||
onClick={handleUseGlobalConfig}
|
|
||||||
disabled={isSubmitting}
|
|
||||||
>
|
|
||||||
{isSubmitting ? "Loading..." : "Use Auto Mode"}
|
|
||||||
</Button>
|
|
||||||
) : isGlobal && config ? (
|
|
||||||
<Button
|
|
||||||
className="text-sm h-9 gap-2"
|
|
||||||
onClick={handleUseGlobalConfig}
|
|
||||||
disabled={isSubmitting}
|
|
||||||
>
|
|
||||||
{isSubmitting ? "Loading..." : "Use This Model"}
|
|
||||||
</Button>
|
|
||||||
) : null}
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</motion.div>
|
|
||||||
</>
|
|
||||||
)}
|
|
||||||
</AnimatePresence>
|
|
||||||
);
|
|
||||||
|
|
||||||
return typeof document !== "undefined" ? createPortal(dialogContent, document.body) : null;
|
|
||||||
}
|
|
||||||
|
|
@ -3,29 +3,22 @@
|
||||||
import { useAtomValue } from "jotai";
|
import { useAtomValue } from "jotai";
|
||||||
import {
|
import {
|
||||||
AlertCircle,
|
AlertCircle,
|
||||||
Check,
|
|
||||||
ChevronsUpDown,
|
|
||||||
Edit3,
|
Edit3,
|
||||||
Info,
|
Info,
|
||||||
Key,
|
|
||||||
Plus,
|
Plus,
|
||||||
RefreshCw,
|
RefreshCw,
|
||||||
Trash2,
|
Trash2,
|
||||||
Wand2,
|
Wand2,
|
||||||
} from "lucide-react";
|
} from "lucide-react";
|
||||||
import { useCallback, useMemo, useState } from "react";
|
import { useMemo, useState } from "react";
|
||||||
import { toast } from "sonner";
|
|
||||||
import {
|
import {
|
||||||
createImageGenConfigMutationAtom,
|
|
||||||
deleteImageGenConfigMutationAtom,
|
deleteImageGenConfigMutationAtom,
|
||||||
updateImageGenConfigMutationAtom,
|
|
||||||
} from "@/atoms/image-gen-config/image-gen-config-mutation.atoms";
|
} from "@/atoms/image-gen-config/image-gen-config-mutation.atoms";
|
||||||
import {
|
import {
|
||||||
globalImageGenConfigsAtom,
|
globalImageGenConfigsAtom,
|
||||||
imageGenConfigsAtom,
|
imageGenConfigsAtom,
|
||||||
} from "@/atoms/image-gen-config/image-gen-config-query.atoms";
|
} from "@/atoms/image-gen-config/image-gen-config-query.atoms";
|
||||||
import { membersAtom, myAccessAtom } from "@/atoms/members/members-query.atoms";
|
import { membersAtom, myAccessAtom } from "@/atoms/members/members-query.atoms";
|
||||||
import { updateLLMPreferencesMutationAtom } from "@/atoms/new-llm-config/new-llm-config-mutation.atoms";
|
|
||||||
import { Alert, AlertDescription } from "@/components/ui/alert";
|
import { Alert, AlertDescription } from "@/components/ui/alert";
|
||||||
import {
|
import {
|
||||||
AlertDialog,
|
AlertDialog,
|
||||||
|
|
@ -40,43 +33,14 @@ import {
|
||||||
import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar";
|
import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar";
|
||||||
import { Button } from "@/components/ui/button";
|
import { Button } from "@/components/ui/button";
|
||||||
import { Card, CardContent } from "@/components/ui/card";
|
import { Card, CardContent } from "@/components/ui/card";
|
||||||
import {
|
|
||||||
Command,
|
|
||||||
CommandEmpty,
|
|
||||||
CommandGroup,
|
|
||||||
CommandInput,
|
|
||||||
CommandItem,
|
|
||||||
CommandList,
|
|
||||||
} from "@/components/ui/command";
|
|
||||||
import {
|
|
||||||
Dialog,
|
|
||||||
DialogContent,
|
|
||||||
DialogDescription,
|
|
||||||
DialogHeader,
|
|
||||||
DialogTitle,
|
|
||||||
} from "@/components/ui/dialog";
|
|
||||||
import { Input } from "@/components/ui/input";
|
|
||||||
import { Label } from "@/components/ui/label";
|
|
||||||
import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover";
|
|
||||||
import {
|
|
||||||
Select,
|
|
||||||
SelectContent,
|
|
||||||
SelectItem,
|
|
||||||
SelectTrigger,
|
|
||||||
SelectValue,
|
|
||||||
} from "@/components/ui/select";
|
|
||||||
import { Separator } from "@/components/ui/separator";
|
|
||||||
import { Skeleton } from "@/components/ui/skeleton";
|
import { Skeleton } from "@/components/ui/skeleton";
|
||||||
import { Spinner } from "@/components/ui/spinner";
|
import { Spinner } from "@/components/ui/spinner";
|
||||||
import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from "@/components/ui/tooltip";
|
import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from "@/components/ui/tooltip";
|
||||||
import {
|
|
||||||
getImageGenModelsByProvider,
|
|
||||||
IMAGE_GEN_PROVIDERS,
|
|
||||||
} from "@/contracts/enums/image-gen-providers";
|
|
||||||
import type { ImageGenerationConfig } from "@/contracts/types/new-llm-config.types";
|
import type { ImageGenerationConfig } from "@/contracts/types/new-llm-config.types";
|
||||||
import { useMediaQuery } from "@/hooks/use-media-query";
|
import { useMediaQuery } from "@/hooks/use-media-query";
|
||||||
import { getProviderIcon } from "@/lib/provider-icons";
|
import { getProviderIcon } from "@/lib/provider-icons";
|
||||||
import { cn } from "@/lib/utils";
|
import { cn } from "@/lib/utils";
|
||||||
|
import { ImageConfigDialog } from "@/components/shared/image-config-dialog";
|
||||||
|
|
||||||
interface ImageModelManagerProps {
|
interface ImageModelManagerProps {
|
||||||
searchSpaceId: number;
|
searchSpaceId: number;
|
||||||
|
|
@ -92,23 +56,12 @@ function getInitials(name: string): string {
|
||||||
|
|
||||||
export function ImageModelManager({ searchSpaceId }: ImageModelManagerProps) {
|
export function ImageModelManager({ searchSpaceId }: ImageModelManagerProps) {
|
||||||
const isDesktop = useMediaQuery("(min-width: 768px)");
|
const isDesktop = useMediaQuery("(min-width: 768px)");
|
||||||
// Image gen config atoms
|
|
||||||
const {
|
|
||||||
mutateAsync: createConfig,
|
|
||||||
isPending: isCreating,
|
|
||||||
error: createError,
|
|
||||||
} = useAtomValue(createImageGenConfigMutationAtom);
|
|
||||||
const {
|
|
||||||
mutateAsync: updateConfig,
|
|
||||||
isPending: isUpdating,
|
|
||||||
error: updateError,
|
|
||||||
} = useAtomValue(updateImageGenConfigMutationAtom);
|
|
||||||
const {
|
const {
|
||||||
mutateAsync: deleteConfig,
|
mutateAsync: deleteConfig,
|
||||||
isPending: isDeleting,
|
isPending: isDeleting,
|
||||||
error: deleteError,
|
error: deleteError,
|
||||||
} = useAtomValue(deleteImageGenConfigMutationAtom);
|
} = useAtomValue(deleteImageGenConfigMutationAtom);
|
||||||
const { mutateAsync: updatePreferences } = useAtomValue(updateLLMPreferencesMutationAtom);
|
|
||||||
|
|
||||||
const {
|
const {
|
||||||
data: userConfigs,
|
data: userConfigs,
|
||||||
|
|
@ -119,7 +72,6 @@ export function ImageModelManager({ searchSpaceId }: ImageModelManagerProps) {
|
||||||
const { data: globalConfigs = [], isFetching: globalLoading } =
|
const { data: globalConfigs = [], isFetching: globalLoading } =
|
||||||
useAtomValue(globalImageGenConfigsAtom);
|
useAtomValue(globalImageGenConfigsAtom);
|
||||||
|
|
||||||
// Members for user resolution
|
|
||||||
const { data: members } = useAtomValue(membersAtom);
|
const { data: members } = useAtomValue(membersAtom);
|
||||||
const memberMap = useMemo(() => {
|
const memberMap = useMemo(() => {
|
||||||
const map = new Map<string, { name: string; email?: string; avatarUrl?: string }>();
|
const map = new Map<string, { name: string; email?: string; avatarUrl?: string }>();
|
||||||
|
|
@ -135,7 +87,6 @@ export function ImageModelManager({ searchSpaceId }: ImageModelManagerProps) {
|
||||||
return map;
|
return map;
|
||||||
}, [members]);
|
}, [members]);
|
||||||
|
|
||||||
// Permissions
|
|
||||||
const { data: access } = useAtomValue(myAccessAtom);
|
const { data: access } = useAtomValue(myAccessAtom);
|
||||||
const canCreate = useMemo(() => {
|
const canCreate = useMemo(() => {
|
||||||
if (!access) return false;
|
if (!access) return false;
|
||||||
|
|
@ -147,92 +98,25 @@ export function ImageModelManager({ searchSpaceId }: ImageModelManagerProps) {
|
||||||
if (access.is_owner) return true;
|
if (access.is_owner) return true;
|
||||||
return access.permissions?.includes("image_generations:delete") ?? false;
|
return access.permissions?.includes("image_generations:delete") ?? false;
|
||||||
}, [access]);
|
}, [access]);
|
||||||
// Backend uses image_generations:create for update as well
|
|
||||||
const canUpdate = canCreate;
|
const canUpdate = canCreate;
|
||||||
const isReadOnly = !canCreate && !canDelete;
|
const isReadOnly = !canCreate && !canDelete;
|
||||||
|
|
||||||
// Local state
|
|
||||||
const [isDialogOpen, setIsDialogOpen] = useState(false);
|
const [isDialogOpen, setIsDialogOpen] = useState(false);
|
||||||
const [editingConfig, setEditingConfig] = useState<ImageGenerationConfig | null>(null);
|
const [editingConfig, setEditingConfig] = useState<ImageGenerationConfig | null>(null);
|
||||||
const [configToDelete, setConfigToDelete] = useState<ImageGenerationConfig | null>(null);
|
const [configToDelete, setConfigToDelete] = useState<ImageGenerationConfig | null>(null);
|
||||||
|
|
||||||
const isSubmitting = isCreating || isUpdating;
|
|
||||||
const isLoading = configsLoading || globalLoading;
|
const isLoading = configsLoading || globalLoading;
|
||||||
const errors = [createError, updateError, deleteError, fetchError].filter(Boolean) as Error[];
|
const errors = [deleteError, fetchError].filter(Boolean) as Error[];
|
||||||
|
|
||||||
// Form state for create/edit dialog
|
const openEditDialog = (config: ImageGenerationConfig) => {
|
||||||
const [formData, setFormData] = useState({
|
setEditingConfig(config);
|
||||||
name: "",
|
setIsDialogOpen(true);
|
||||||
description: "",
|
|
||||||
provider: "",
|
|
||||||
custom_provider: "",
|
|
||||||
model_name: "",
|
|
||||||
api_key: "",
|
|
||||||
api_base: "",
|
|
||||||
api_version: "",
|
|
||||||
});
|
|
||||||
const [modelComboboxOpen, setModelComboboxOpen] = useState(false);
|
|
||||||
|
|
||||||
const resetForm = () => {
|
|
||||||
setFormData({
|
|
||||||
name: "",
|
|
||||||
description: "",
|
|
||||||
provider: "",
|
|
||||||
custom_provider: "",
|
|
||||||
model_name: "",
|
|
||||||
api_key: "",
|
|
||||||
api_base: "",
|
|
||||||
api_version: "",
|
|
||||||
});
|
|
||||||
};
|
};
|
||||||
|
|
||||||
const handleFormSubmit = useCallback(async () => {
|
const openNewDialog = () => {
|
||||||
if (!formData.name || !formData.provider || !formData.model_name || !formData.api_key) {
|
|
||||||
toast.error("Please fill in all required fields");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
try {
|
|
||||||
if (editingConfig) {
|
|
||||||
await updateConfig({
|
|
||||||
id: editingConfig.id,
|
|
||||||
data: {
|
|
||||||
name: formData.name,
|
|
||||||
description: formData.description || undefined,
|
|
||||||
provider: formData.provider as any,
|
|
||||||
custom_provider: formData.custom_provider || undefined,
|
|
||||||
model_name: formData.model_name,
|
|
||||||
api_key: formData.api_key,
|
|
||||||
api_base: formData.api_base || undefined,
|
|
||||||
api_version: formData.api_version || undefined,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
const result = await createConfig({
|
|
||||||
name: formData.name,
|
|
||||||
description: formData.description || undefined,
|
|
||||||
provider: formData.provider as any,
|
|
||||||
custom_provider: formData.custom_provider || undefined,
|
|
||||||
model_name: formData.model_name,
|
|
||||||
api_key: formData.api_key,
|
|
||||||
api_base: formData.api_base || undefined,
|
|
||||||
api_version: formData.api_version || undefined,
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
});
|
|
||||||
// Auto-assign newly created config
|
|
||||||
if (result?.id) {
|
|
||||||
await updatePreferences({
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
data: { image_generation_config_id: result.id },
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
setIsDialogOpen(false);
|
|
||||||
setEditingConfig(null);
|
setEditingConfig(null);
|
||||||
resetForm();
|
setIsDialogOpen(true);
|
||||||
} catch {
|
};
|
||||||
// Error handled by mutation
|
|
||||||
}
|
|
||||||
}, [editingConfig, formData, searchSpaceId, createConfig, updateConfig, updatePreferences]);
|
|
||||||
|
|
||||||
const handleDelete = async () => {
|
const handleDelete = async () => {
|
||||||
if (!configToDelete) return;
|
if (!configToDelete) return;
|
||||||
|
|
@ -244,30 +128,6 @@ export function ImageModelManager({ searchSpaceId }: ImageModelManagerProps) {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
const openEditDialog = (config: ImageGenerationConfig) => {
|
|
||||||
setEditingConfig(config);
|
|
||||||
setFormData({
|
|
||||||
name: config.name,
|
|
||||||
description: config.description || "",
|
|
||||||
provider: config.provider,
|
|
||||||
custom_provider: config.custom_provider || "",
|
|
||||||
model_name: config.model_name,
|
|
||||||
api_key: config.api_key,
|
|
||||||
api_base: config.api_base || "",
|
|
||||||
api_version: config.api_version || "",
|
|
||||||
});
|
|
||||||
setIsDialogOpen(true);
|
|
||||||
};
|
|
||||||
|
|
||||||
const openNewDialog = () => {
|
|
||||||
setEditingConfig(null);
|
|
||||||
resetForm();
|
|
||||||
setIsDialogOpen(true);
|
|
||||||
};
|
|
||||||
|
|
||||||
const selectedProvider = IMAGE_GEN_PROVIDERS.find((p) => p.value === formData.provider);
|
|
||||||
const suggestedModels = getImageGenModelsByProvider(formData.provider);
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="space-y-4 md:space-y-6">
|
<div className="space-y-4 md:space-y-6">
|
||||||
{/* Header */}
|
{/* Header */}
|
||||||
|
|
@ -348,31 +208,26 @@ export function ImageModelManager({ searchSpaceId }: ImageModelManagerProps) {
|
||||||
{/* Loading Skeleton */}
|
{/* Loading Skeleton */}
|
||||||
{isLoading && (
|
{isLoading && (
|
||||||
<div className="space-y-4 md:space-y-6">
|
<div className="space-y-4 md:space-y-6">
|
||||||
{/* Your Image Models Section Skeleton */}
|
|
||||||
<div className="space-y-4">
|
<div className="space-y-4">
|
||||||
<div className="flex items-center justify-between">
|
<div className="flex items-center justify-between">
|
||||||
<Skeleton className="h-6 md:h-7 w-40 md:w-48" />
|
<Skeleton className="h-6 md:h-7 w-40 md:w-48" />
|
||||||
<Skeleton className="h-8 md:h-9 w-32 md:w-36 rounded-md" />
|
<Skeleton className="h-8 md:h-9 w-32 md:w-36 rounded-md" />
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{/* Cards Grid Skeleton */}
|
|
||||||
<div className="grid gap-3 grid-cols-1 sm:grid-cols-2 xl:grid-cols-3">
|
<div className="grid gap-3 grid-cols-1 sm:grid-cols-2 xl:grid-cols-3">
|
||||||
{["skeleton-a", "skeleton-b", "skeleton-c"].map((key) => (
|
{["skeleton-a", "skeleton-b", "skeleton-c"].map((key) => (
|
||||||
<Card key={key} className="border-border/60">
|
<Card key={key} className="border-border/60">
|
||||||
<CardContent className="p-4 flex flex-col gap-3">
|
<CardContent className="p-4 flex flex-col gap-3">
|
||||||
{/* Header */}
|
|
||||||
<div className="flex items-start justify-between gap-2">
|
<div className="flex items-start justify-between gap-2">
|
||||||
<div className="space-y-1.5 flex-1 min-w-0">
|
<div className="space-y-1.5 flex-1 min-w-0">
|
||||||
<Skeleton className="h-4 w-28 md:w-32" />
|
<Skeleton className="h-4 w-28 md:w-32" />
|
||||||
<Skeleton className="h-3 w-40 md:w-48" />
|
<Skeleton className="h-3 w-40 md:w-48" />
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
{/* Provider + Model */}
|
|
||||||
<div className="flex items-center gap-2">
|
<div className="flex items-center gap-2">
|
||||||
<Skeleton className="h-5 w-16 rounded-full" />
|
<Skeleton className="h-5 w-16 rounded-full" />
|
||||||
<Skeleton className="h-5 w-24 rounded-md" />
|
<Skeleton className="h-5 w-24 rounded-md" />
|
||||||
</div>
|
</div>
|
||||||
{/* Footer */}
|
|
||||||
<div className="flex items-center gap-2 pt-2 border-t border-border/40">
|
<div className="flex items-center gap-2 pt-2 border-t border-border/40">
|
||||||
<Skeleton className="h-3 w-20" />
|
<Skeleton className="h-3 w-20" />
|
||||||
<Skeleton className="h-4 w-4 rounded-full" />
|
<Skeleton className="h-4 w-4 rounded-full" />
|
||||||
|
|
@ -529,204 +384,18 @@ export function ImageModelManager({ searchSpaceId }: ImageModelManagerProps) {
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
{/* Create/Edit Dialog */}
|
{/* Create/Edit Dialog — shared component */}
|
||||||
<Dialog
|
<ImageConfigDialog
|
||||||
open={isDialogOpen}
|
open={isDialogOpen}
|
||||||
onOpenChange={(open) => {
|
onOpenChange={(open) => {
|
||||||
if (!open) {
|
setIsDialogOpen(open);
|
||||||
setIsDialogOpen(false);
|
if (!open) setEditingConfig(null);
|
||||||
setEditingConfig(null);
|
|
||||||
resetForm();
|
|
||||||
}
|
|
||||||
}}
|
}}
|
||||||
>
|
config={editingConfig}
|
||||||
<DialogContent
|
isGlobal={false}
|
||||||
className="max-w-lg max-h-[90vh] overflow-y-auto"
|
searchSpaceId={searchSpaceId}
|
||||||
onOpenAutoFocus={(e) => e.preventDefault()}
|
mode={editingConfig ? "edit" : "create"}
|
||||||
>
|
|
||||||
<DialogHeader>
|
|
||||||
<DialogTitle>{editingConfig ? "Edit Image Model" : "Add Image Model"}</DialogTitle>
|
|
||||||
<DialogDescription>
|
|
||||||
{editingConfig
|
|
||||||
? "Update your image generation model"
|
|
||||||
: "Configure a new image generation model (DALL-E 3, GPT Image 1, etc.)"}
|
|
||||||
</DialogDescription>
|
|
||||||
</DialogHeader>
|
|
||||||
|
|
||||||
<div className="space-y-4 pt-2">
|
|
||||||
{/* Name */}
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">Name *</Label>
|
|
||||||
<Input
|
|
||||||
placeholder="e.g., My DALL-E 3"
|
|
||||||
value={formData.name}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, name: e.target.value }))}
|
|
||||||
/>
|
/>
|
||||||
</div>
|
|
||||||
|
|
||||||
{/* Description */}
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">Description</Label>
|
|
||||||
<Input
|
|
||||||
placeholder="Optional description"
|
|
||||||
value={formData.description}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, description: e.target.value }))}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Separator />
|
|
||||||
|
|
||||||
{/* Provider */}
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">Provider *</Label>
|
|
||||||
<Select
|
|
||||||
value={formData.provider}
|
|
||||||
onValueChange={(val) =>
|
|
||||||
setFormData((p) => ({ ...p, provider: val, model_name: "" }))
|
|
||||||
}
|
|
||||||
>
|
|
||||||
<SelectTrigger>
|
|
||||||
<SelectValue placeholder="Select a provider" />
|
|
||||||
</SelectTrigger>
|
|
||||||
<SelectContent>
|
|
||||||
{IMAGE_GEN_PROVIDERS.map((p) => (
|
|
||||||
<SelectItem key={p.value} value={p.value}>
|
|
||||||
<div className="flex flex-col">
|
|
||||||
<span className="font-medium">{p.label}</span>
|
|
||||||
<span className="text-xs text-muted-foreground">{p.example}</span>
|
|
||||||
</div>
|
|
||||||
</SelectItem>
|
|
||||||
))}
|
|
||||||
</SelectContent>
|
|
||||||
</Select>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{/* Model Name */}
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">Model Name *</Label>
|
|
||||||
{suggestedModels.length > 0 ? (
|
|
||||||
<Popover open={modelComboboxOpen} onOpenChange={setModelComboboxOpen}>
|
|
||||||
<PopoverTrigger asChild>
|
|
||||||
<Button
|
|
||||||
variant="outline"
|
|
||||||
role="combobox"
|
|
||||||
className="w-full justify-between font-normal bg-transparent hover:bg-transparent hover:text-foreground"
|
|
||||||
>
|
|
||||||
{formData.model_name || "Select a model"}
|
|
||||||
<ChevronsUpDown className="ml-2 h-4 w-4 shrink-0 opacity-50" />
|
|
||||||
</Button>
|
|
||||||
</PopoverTrigger>
|
|
||||||
<PopoverContent className="w-full p-0" align="start">
|
|
||||||
<Command className="bg-transparent">
|
|
||||||
<CommandInput
|
|
||||||
placeholder="Search a model name"
|
|
||||||
value={formData.model_name}
|
|
||||||
onValueChange={(val) => setFormData((p) => ({ ...p, model_name: val }))}
|
|
||||||
/>
|
|
||||||
<CommandList>
|
|
||||||
<CommandEmpty>
|
|
||||||
<span className="text-xs text-muted-foreground">
|
|
||||||
Type a custom model name
|
|
||||||
</span>
|
|
||||||
</CommandEmpty>
|
|
||||||
<CommandGroup>
|
|
||||||
{suggestedModels.map((m) => (
|
|
||||||
<CommandItem
|
|
||||||
key={m.value}
|
|
||||||
value={m.value}
|
|
||||||
onSelect={() => {
|
|
||||||
setFormData((p) => ({ ...p, model_name: m.value }));
|
|
||||||
setModelComboboxOpen(false);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Check
|
|
||||||
className={cn(
|
|
||||||
"mr-2 h-4 w-4",
|
|
||||||
formData.model_name === m.value ? "opacity-100" : "opacity-0"
|
|
||||||
)}
|
|
||||||
/>
|
|
||||||
<span className="font-mono text-sm">{m.value}</span>
|
|
||||||
<span className="ml-2 text-xs text-muted-foreground">{m.label}</span>
|
|
||||||
</CommandItem>
|
|
||||||
))}
|
|
||||||
</CommandGroup>
|
|
||||||
</CommandList>
|
|
||||||
</Command>
|
|
||||||
</PopoverContent>
|
|
||||||
</Popover>
|
|
||||||
) : (
|
|
||||||
<Input
|
|
||||||
placeholder="e.g., dall-e-3"
|
|
||||||
value={formData.model_name}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, model_name: e.target.value }))}
|
|
||||||
/>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{/* API Key */}
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium flex items-center gap-1.5">
|
|
||||||
<Key className="h-3.5 w-3.5" /> API Key *
|
|
||||||
</Label>
|
|
||||||
<Input
|
|
||||||
type="password"
|
|
||||||
placeholder="sk-..."
|
|
||||||
value={formData.api_key}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, api_key: e.target.value }))}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{/* API Base (optional) */}
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">API Base URL</Label>
|
|
||||||
<Input
|
|
||||||
placeholder={selectedProvider?.apiBase || "Optional"}
|
|
||||||
value={formData.api_base}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, api_base: e.target.value }))}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{/* API Version (Azure) */}
|
|
||||||
{formData.provider === "AZURE_OPENAI" && (
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label className="text-sm font-medium">API Version (Azure)</Label>
|
|
||||||
<Input
|
|
||||||
placeholder="2024-02-15-preview"
|
|
||||||
value={formData.api_version}
|
|
||||||
onChange={(e) => setFormData((p) => ({ ...p, api_version: e.target.value }))}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{/* Actions */}
|
|
||||||
<div className="flex justify-end gap-3 pt-4 border-t">
|
|
||||||
<Button
|
|
||||||
variant="secondary"
|
|
||||||
onClick={() => {
|
|
||||||
setIsDialogOpen(false);
|
|
||||||
setEditingConfig(null);
|
|
||||||
resetForm();
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
Cancel
|
|
||||||
</Button>
|
|
||||||
<Button
|
|
||||||
onClick={handleFormSubmit}
|
|
||||||
disabled={
|
|
||||||
isSubmitting ||
|
|
||||||
!formData.name ||
|
|
||||||
!formData.provider ||
|
|
||||||
!formData.model_name ||
|
|
||||||
!formData.api_key
|
|
||||||
}
|
|
||||||
>
|
|
||||||
{isSubmitting ? <Spinner size="sm" className="mr-2" /> : null}
|
|
||||||
{editingConfig ? "Save Changes" : "Create & Use"}
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</DialogContent>
|
|
||||||
</Dialog>
|
|
||||||
|
|
||||||
{/* Delete Confirmation */}
|
{/* Delete Confirmation */}
|
||||||
<AlertDialog
|
<AlertDialog
|
||||||
|
|
|
||||||
500
surfsense_web/components/shared/image-config-dialog.tsx
Normal file
500
surfsense_web/components/shared/image-config-dialog.tsx
Normal file
|
|
@ -0,0 +1,500 @@
|
||||||
|
"use client";
|
||||||
|
|
||||||
|
import { useAtomValue } from "jotai";
|
||||||
|
import { AlertCircle, Check, ChevronsUpDown } from "lucide-react";
|
||||||
|
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
|
||||||
|
import { toast } from "sonner";
|
||||||
|
import {
|
||||||
|
createImageGenConfigMutationAtom,
|
||||||
|
updateImageGenConfigMutationAtom,
|
||||||
|
} from "@/atoms/image-gen-config/image-gen-config-mutation.atoms";
|
||||||
|
import { updateLLMPreferencesMutationAtom } from "@/atoms/new-llm-config/new-llm-config-mutation.atoms";
|
||||||
|
import { Alert, AlertDescription } from "@/components/ui/alert";
|
||||||
|
import { Badge } from "@/components/ui/badge";
|
||||||
|
import { Button } from "@/components/ui/button";
|
||||||
|
import {
|
||||||
|
Command,
|
||||||
|
CommandEmpty,
|
||||||
|
CommandGroup,
|
||||||
|
CommandInput,
|
||||||
|
CommandItem,
|
||||||
|
CommandList,
|
||||||
|
} from "@/components/ui/command";
|
||||||
|
import { Dialog, DialogContent, DialogTitle } from "@/components/ui/dialog";
|
||||||
|
import { Input } from "@/components/ui/input";
|
||||||
|
import { Label } from "@/components/ui/label";
|
||||||
|
import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover";
|
||||||
|
import {
|
||||||
|
Select,
|
||||||
|
SelectContent,
|
||||||
|
SelectItem,
|
||||||
|
SelectTrigger,
|
||||||
|
SelectValue,
|
||||||
|
} from "@/components/ui/select";
|
||||||
|
import { Separator } from "@/components/ui/separator";
|
||||||
|
import { Spinner } from "@/components/ui/spinner";
|
||||||
|
import { IMAGE_GEN_MODELS, IMAGE_GEN_PROVIDERS } from "@/contracts/enums/image-gen-providers";
|
||||||
|
import type {
|
||||||
|
GlobalImageGenConfig,
|
||||||
|
ImageGenerationConfig,
|
||||||
|
ImageGenProvider,
|
||||||
|
} from "@/contracts/types/new-llm-config.types";
|
||||||
|
import { cn } from "@/lib/utils";
|
||||||
|
|
||||||
|
interface ImageConfigDialogProps {
|
||||||
|
open: boolean;
|
||||||
|
onOpenChange: (open: boolean) => void;
|
||||||
|
config: ImageGenerationConfig | GlobalImageGenConfig | null;
|
||||||
|
isGlobal: boolean;
|
||||||
|
searchSpaceId: number;
|
||||||
|
mode: "create" | "edit" | "view";
|
||||||
|
}
|
||||||
|
|
||||||
|
const INITIAL_FORM = {
|
||||||
|
name: "",
|
||||||
|
description: "",
|
||||||
|
provider: "",
|
||||||
|
model_name: "",
|
||||||
|
api_key: "",
|
||||||
|
api_base: "",
|
||||||
|
api_version: "",
|
||||||
|
};
|
||||||
|
|
||||||
|
export function ImageConfigDialog({
|
||||||
|
open,
|
||||||
|
onOpenChange,
|
||||||
|
config,
|
||||||
|
isGlobal,
|
||||||
|
searchSpaceId,
|
||||||
|
mode,
|
||||||
|
}: ImageConfigDialogProps) {
|
||||||
|
const [isSubmitting, setIsSubmitting] = useState(false);
|
||||||
|
const [formData, setFormData] = useState(INITIAL_FORM);
|
||||||
|
const [modelComboboxOpen, setModelComboboxOpen] = useState(false);
|
||||||
|
const [scrollPos, setScrollPos] = useState<"top" | "middle" | "bottom">("top");
|
||||||
|
const scrollRef = useRef<HTMLDivElement>(null);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (open) {
|
||||||
|
if (mode === "edit" && config && !isGlobal) {
|
||||||
|
setFormData({
|
||||||
|
name: config.name || "",
|
||||||
|
description: config.description || "",
|
||||||
|
provider: config.provider || "",
|
||||||
|
model_name: config.model_name || "",
|
||||||
|
api_key: (config as ImageGenerationConfig).api_key || "",
|
||||||
|
api_base: config.api_base || "",
|
||||||
|
api_version: config.api_version || "",
|
||||||
|
});
|
||||||
|
} else if (mode === "create") {
|
||||||
|
setFormData(INITIAL_FORM);
|
||||||
|
}
|
||||||
|
setScrollPos("top");
|
||||||
|
}
|
||||||
|
}, [open, mode, config, isGlobal]);
|
||||||
|
|
||||||
|
const { mutateAsync: createConfig } = useAtomValue(createImageGenConfigMutationAtom);
|
||||||
|
const { mutateAsync: updateConfig } = useAtomValue(updateImageGenConfigMutationAtom);
|
||||||
|
const { mutateAsync: updatePreferences } = useAtomValue(updateLLMPreferencesMutationAtom);
|
||||||
|
|
||||||
|
const handleScroll = useCallback((e: React.UIEvent<HTMLDivElement>) => {
|
||||||
|
const el = e.currentTarget;
|
||||||
|
const atTop = el.scrollTop <= 2;
|
||||||
|
const atBottom = el.scrollHeight - el.scrollTop - el.clientHeight <= 2;
|
||||||
|
setScrollPos(atTop ? "top" : atBottom ? "bottom" : "middle");
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const isAutoMode = config && "is_auto_mode" in config && config.is_auto_mode;
|
||||||
|
|
||||||
|
const suggestedModels = useMemo(() => {
|
||||||
|
if (!formData.provider) return [];
|
||||||
|
return IMAGE_GEN_MODELS.filter((m) => m.provider === formData.provider);
|
||||||
|
}, [formData.provider]);
|
||||||
|
|
||||||
|
const getTitle = () => {
|
||||||
|
if (mode === "create") return "Add Image Model";
|
||||||
|
if (isAutoMode) return "Auto Mode (Fastest)";
|
||||||
|
if (isGlobal) return "View Global Image Model";
|
||||||
|
return "Edit Image Model";
|
||||||
|
};
|
||||||
|
|
||||||
|
const getSubtitle = () => {
|
||||||
|
if (mode === "create") return "Set up a new image generation provider";
|
||||||
|
if (isAutoMode) return "Automatically routes requests across providers";
|
||||||
|
if (isGlobal) return "Read-only global configuration";
|
||||||
|
return "Update your image model settings";
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleSubmit = useCallback(async () => {
|
||||||
|
setIsSubmitting(true);
|
||||||
|
try {
|
||||||
|
if (mode === "create") {
|
||||||
|
const result = await createConfig({
|
||||||
|
name: formData.name,
|
||||||
|
provider: formData.provider as ImageGenProvider,
|
||||||
|
model_name: formData.model_name,
|
||||||
|
api_key: formData.api_key,
|
||||||
|
api_base: formData.api_base || undefined,
|
||||||
|
api_version: formData.api_version || undefined,
|
||||||
|
description: formData.description || undefined,
|
||||||
|
search_space_id: searchSpaceId,
|
||||||
|
});
|
||||||
|
if (result?.id) {
|
||||||
|
await updatePreferences({
|
||||||
|
search_space_id: searchSpaceId,
|
||||||
|
data: { image_generation_config_id: result.id },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
onOpenChange(false);
|
||||||
|
} else if (!isGlobal && config) {
|
||||||
|
await updateConfig({
|
||||||
|
id: config.id,
|
||||||
|
data: {
|
||||||
|
name: formData.name,
|
||||||
|
description: formData.description || undefined,
|
||||||
|
provider: formData.provider as ImageGenProvider,
|
||||||
|
model_name: formData.model_name,
|
||||||
|
api_key: formData.api_key,
|
||||||
|
api_base: formData.api_base || undefined,
|
||||||
|
api_version: formData.api_version || undefined,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
onOpenChange(false);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Failed to save image config:", error);
|
||||||
|
toast.error("Failed to save image model");
|
||||||
|
} finally {
|
||||||
|
setIsSubmitting(false);
|
||||||
|
}
|
||||||
|
}, [
|
||||||
|
mode,
|
||||||
|
isGlobal,
|
||||||
|
config,
|
||||||
|
formData,
|
||||||
|
searchSpaceId,
|
||||||
|
createConfig,
|
||||||
|
updateConfig,
|
||||||
|
updatePreferences,
|
||||||
|
onOpenChange,
|
||||||
|
]);
|
||||||
|
|
||||||
|
const handleUseGlobalConfig = useCallback(async () => {
|
||||||
|
if (!config || !isGlobal) return;
|
||||||
|
setIsSubmitting(true);
|
||||||
|
try {
|
||||||
|
await updatePreferences({
|
||||||
|
search_space_id: searchSpaceId,
|
||||||
|
data: { image_generation_config_id: config.id },
|
||||||
|
});
|
||||||
|
toast.success(`Now using ${config.name}`);
|
||||||
|
onOpenChange(false);
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Failed to set image model:", error);
|
||||||
|
toast.error("Failed to set image model");
|
||||||
|
} finally {
|
||||||
|
setIsSubmitting(false);
|
||||||
|
}
|
||||||
|
}, [config, isGlobal, searchSpaceId, updatePreferences, onOpenChange]);
|
||||||
|
|
||||||
|
const isFormValid = formData.name && formData.provider && formData.model_name && formData.api_key;
|
||||||
|
const selectedProvider = IMAGE_GEN_PROVIDERS.find((p) => p.value === formData.provider);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Dialog open={open} onOpenChange={onOpenChange}>
|
||||||
|
<DialogContent
|
||||||
|
className="max-w-lg h-[85vh] flex flex-col p-0 gap-0 overflow-hidden"
|
||||||
|
onOpenAutoFocus={(e) => e.preventDefault()}
|
||||||
|
>
|
||||||
|
<DialogTitle className="sr-only">{getTitle()}</DialogTitle>
|
||||||
|
|
||||||
|
{/* Header */}
|
||||||
|
<div className="flex items-start justify-between px-6 pt-6 pb-4 pr-14">
|
||||||
|
<div className="space-y-1">
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<h2 className="text-lg font-semibold tracking-tight">{getTitle()}</h2>
|
||||||
|
{isAutoMode && (
|
||||||
|
<Badge variant="secondary" className="text-[10px]">
|
||||||
|
Recommended
|
||||||
|
</Badge>
|
||||||
|
)}
|
||||||
|
{isGlobal && !isAutoMode && mode !== "create" && (
|
||||||
|
<Badge variant="secondary" className="text-[10px]">
|
||||||
|
Global
|
||||||
|
</Badge>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<p className="text-sm text-muted-foreground">{getSubtitle()}</p>
|
||||||
|
{config && !isAutoMode && mode !== "create" && (
|
||||||
|
<p className="text-xs font-mono text-muted-foreground/70">
|
||||||
|
{config.model_name}
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Scrollable content */}
|
||||||
|
<div
|
||||||
|
ref={scrollRef}
|
||||||
|
onScroll={handleScroll}
|
||||||
|
className="flex-1 overflow-y-auto px-6 py-5"
|
||||||
|
style={{
|
||||||
|
maskImage: `linear-gradient(to bottom, ${scrollPos === "top" ? "black" : "transparent"}, black 16px, black calc(100% - 16px), ${scrollPos === "bottom" ? "black" : "transparent"})`,
|
||||||
|
WebkitMaskImage: `linear-gradient(to bottom, ${scrollPos === "top" ? "black" : "transparent"}, black 16px, black calc(100% - 16px), ${scrollPos === "bottom" ? "black" : "transparent"})`,
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
{isAutoMode && (
|
||||||
|
<Alert className="mb-5 border-violet-500/30 bg-violet-500/5">
|
||||||
|
<AlertDescription className="text-sm text-violet-700 dark:text-violet-400">
|
||||||
|
Auto mode distributes image generation requests across all configured
|
||||||
|
providers for optimal performance and rate limit protection.
|
||||||
|
</AlertDescription>
|
||||||
|
</Alert>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{isGlobal && !isAutoMode && config && (
|
||||||
|
<>
|
||||||
|
<Alert className="mb-5 border-amber-500/30 bg-amber-500/5">
|
||||||
|
<AlertCircle className="size-4 text-amber-500" />
|
||||||
|
<AlertDescription className="text-sm text-amber-700 dark:text-amber-400">
|
||||||
|
Global configurations are read-only. To customize, create a new model.
|
||||||
|
</AlertDescription>
|
||||||
|
</Alert>
|
||||||
|
<div className="space-y-4">
|
||||||
|
<div className="grid gap-4 sm:grid-cols-2">
|
||||||
|
<div className="space-y-1.5">
|
||||||
|
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
||||||
|
Name
|
||||||
|
</div>
|
||||||
|
<p className="text-sm font-medium">{config.name}</p>
|
||||||
|
</div>
|
||||||
|
{config.description && (
|
||||||
|
<div className="space-y-1.5">
|
||||||
|
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
||||||
|
Description
|
||||||
|
</div>
|
||||||
|
<p className="text-sm text-muted-foreground">{config.description}</p>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<Separator />
|
||||||
|
<div className="grid gap-4 sm:grid-cols-2">
|
||||||
|
<div className="space-y-1.5">
|
||||||
|
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
||||||
|
Provider
|
||||||
|
</div>
|
||||||
|
<p className="text-sm font-medium">{config.provider}</p>
|
||||||
|
</div>
|
||||||
|
<div className="space-y-1.5">
|
||||||
|
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
||||||
|
Model
|
||||||
|
</div>
|
||||||
|
<p className="text-sm font-medium font-mono">{config.model_name}</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{(mode === "create" || (mode === "edit" && !isGlobal)) && (
|
||||||
|
<div className="space-y-4">
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label className="text-sm font-medium">Name *</Label>
|
||||||
|
<Input
|
||||||
|
placeholder="e.g., My DALL-E 3"
|
||||||
|
value={formData.name}
|
||||||
|
onChange={(e) => setFormData((p) => ({ ...p, name: e.target.value }))}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label className="text-sm font-medium">Description</Label>
|
||||||
|
<Input
|
||||||
|
placeholder="Optional description"
|
||||||
|
value={formData.description}
|
||||||
|
onChange={(e) =>
|
||||||
|
setFormData((p) => ({ ...p, description: e.target.value }))
|
||||||
|
}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<Separator />
|
||||||
|
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label className="text-sm font-medium">Provider *</Label>
|
||||||
|
<Select
|
||||||
|
value={formData.provider}
|
||||||
|
onValueChange={(val) =>
|
||||||
|
setFormData((p) => ({ ...p, provider: val, model_name: "" }))
|
||||||
|
}
|
||||||
|
>
|
||||||
|
<SelectTrigger>
|
||||||
|
<SelectValue placeholder="Select a provider" />
|
||||||
|
</SelectTrigger>
|
||||||
|
<SelectContent>
|
||||||
|
{IMAGE_GEN_PROVIDERS.map((p) => (
|
||||||
|
<SelectItem key={p.value} value={p.value} description={p.example}>
|
||||||
|
{p.label}
|
||||||
|
</SelectItem>
|
||||||
|
))}
|
||||||
|
</SelectContent>
|
||||||
|
</Select>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label className="text-sm font-medium">Model Name *</Label>
|
||||||
|
{suggestedModels.length > 0 ? (
|
||||||
|
<Popover open={modelComboboxOpen} onOpenChange={setModelComboboxOpen}>
|
||||||
|
<PopoverTrigger asChild>
|
||||||
|
<Button
|
||||||
|
variant="outline"
|
||||||
|
role="combobox"
|
||||||
|
className="w-full justify-between font-normal bg-transparent"
|
||||||
|
>
|
||||||
|
{formData.model_name || "Select or type a model..."}
|
||||||
|
<ChevronsUpDown className="ml-2 h-4 w-4 shrink-0 opacity-50" />
|
||||||
|
</Button>
|
||||||
|
</PopoverTrigger>
|
||||||
|
<PopoverContent
|
||||||
|
className="w-full p-0"
|
||||||
|
align="start"
|
||||||
|
>
|
||||||
|
<Command className="bg-transparent">
|
||||||
|
<CommandInput
|
||||||
|
placeholder="Search or type model..."
|
||||||
|
value={formData.model_name}
|
||||||
|
onValueChange={(val) =>
|
||||||
|
setFormData((p) => ({ ...p, model_name: val }))
|
||||||
|
}
|
||||||
|
/>
|
||||||
|
<CommandList>
|
||||||
|
<CommandEmpty>
|
||||||
|
<span className="text-xs text-muted-foreground">
|
||||||
|
Type a custom model name
|
||||||
|
</span>
|
||||||
|
</CommandEmpty>
|
||||||
|
<CommandGroup>
|
||||||
|
{suggestedModels.map((m) => (
|
||||||
|
<CommandItem
|
||||||
|
key={m.value}
|
||||||
|
value={m.value}
|
||||||
|
onSelect={() => {
|
||||||
|
setFormData((p) => ({ ...p, model_name: m.value }));
|
||||||
|
setModelComboboxOpen(false);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<Check
|
||||||
|
className={cn(
|
||||||
|
"mr-2 h-4 w-4",
|
||||||
|
formData.model_name === m.value
|
||||||
|
? "opacity-100"
|
||||||
|
: "opacity-0"
|
||||||
|
)}
|
||||||
|
/>
|
||||||
|
<span className="font-mono text-sm">{m.value}</span>
|
||||||
|
<span className="ml-2 text-xs text-muted-foreground">
|
||||||
|
{m.label}
|
||||||
|
</span>
|
||||||
|
</CommandItem>
|
||||||
|
))}
|
||||||
|
</CommandGroup>
|
||||||
|
</CommandList>
|
||||||
|
</Command>
|
||||||
|
</PopoverContent>
|
||||||
|
</Popover>
|
||||||
|
) : (
|
||||||
|
<Input
|
||||||
|
placeholder="e.g., dall-e-3"
|
||||||
|
value={formData.model_name}
|
||||||
|
onChange={(e) =>
|
||||||
|
setFormData((p) => ({ ...p, model_name: e.target.value }))
|
||||||
|
}
|
||||||
|
/>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label className="text-sm font-medium">API Key *</Label>
|
||||||
|
<Input
|
||||||
|
type="password"
|
||||||
|
placeholder="sk-..."
|
||||||
|
value={formData.api_key}
|
||||||
|
onChange={(e) => setFormData((p) => ({ ...p, api_key: e.target.value }))}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label className="text-sm font-medium">API Base URL</Label>
|
||||||
|
<Input
|
||||||
|
placeholder={selectedProvider?.apiBase || "Optional"}
|
||||||
|
value={formData.api_base}
|
||||||
|
onChange={(e) => setFormData((p) => ({ ...p, api_base: e.target.value }))}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{formData.provider === "AZURE_OPENAI" && (
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label className="text-sm font-medium">API Version (Azure)</Label>
|
||||||
|
<Input
|
||||||
|
placeholder="2024-02-15-preview"
|
||||||
|
value={formData.api_version}
|
||||||
|
onChange={(e) =>
|
||||||
|
setFormData((p) => ({ ...p, api_version: e.target.value }))
|
||||||
|
}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* Fixed footer */}
|
||||||
|
<div className="shrink-0 px-6 py-4 flex items-center justify-end gap-3">
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
variant="secondary"
|
||||||
|
onClick={() => onOpenChange(false)}
|
||||||
|
disabled={isSubmitting}
|
||||||
|
className="text-sm h-9"
|
||||||
|
>
|
||||||
|
Cancel
|
||||||
|
</Button>
|
||||||
|
{mode === "create" || (mode === "edit" && !isGlobal) ? (
|
||||||
|
<Button
|
||||||
|
onClick={handleSubmit}
|
||||||
|
disabled={isSubmitting || !isFormValid}
|
||||||
|
className="text-sm h-9 min-w-[120px]"
|
||||||
|
>
|
||||||
|
{isSubmitting ? (
|
||||||
|
<>
|
||||||
|
<Spinner size="sm" />
|
||||||
|
{mode === "edit" ? "Saving" : "Creating"}
|
||||||
|
</>
|
||||||
|
) : mode === "edit" ? (
|
||||||
|
"Save Changes"
|
||||||
|
) : (
|
||||||
|
"Create & Use"
|
||||||
|
)}
|
||||||
|
</Button>
|
||||||
|
) : isAutoMode ? (
|
||||||
|
<Button
|
||||||
|
className="text-sm h-9 gap-2 bg-gradient-to-r from-violet-500 to-purple-600 hover:from-violet-600 hover:to-purple-700"
|
||||||
|
onClick={handleUseGlobalConfig}
|
||||||
|
disabled={isSubmitting}
|
||||||
|
>
|
||||||
|
{isSubmitting ? "Loading..." : "Use Auto Mode"}
|
||||||
|
</Button>
|
||||||
|
) : isGlobal && config ? (
|
||||||
|
<Button
|
||||||
|
className="text-sm h-9 gap-2"
|
||||||
|
onClick={handleUseGlobalConfig}
|
||||||
|
disabled={isSubmitting}
|
||||||
|
>
|
||||||
|
{isSubmitting ? "Loading..." : "Use This Model"}
|
||||||
|
</Button>
|
||||||
|
) : null}
|
||||||
|
</div>
|
||||||
|
</DialogContent>
|
||||||
|
</Dialog>
|
||||||
|
);
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue