mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-06 20:15:17 +02:00
refactor: replace ModelConfigDialog with a shared component and update related imports for better organization and clarity
This commit is contained in:
Anish Sarkar
parent
f4adfb54fc
commit
d88236d43b
5 changed files with 441 additions and 586 deletions
|
|
@ -26,7 +26,7 @@ export const createNewLLMConfigMutationAtom = atomWithMutation((get) => {
|
||||||
return newLLMConfigApiService.createConfig(request);
|
return newLLMConfigApiService.createConfig(request);
|
||||||
},
|
},
|
||||||
onSuccess: () => {
|
onSuccess: () => {
|
||||||
toast.success("Configuration created successfully");
|
toast.success("LLM model created");
|
||||||
queryClient.invalidateQueries({
|
queryClient.invalidateQueries({
|
||||||
queryKey: cacheKeys.newLLMConfigs.all(Number(searchSpaceId)),
|
queryKey: cacheKeys.newLLMConfigs.all(Number(searchSpaceId)),
|
||||||
});
|
});
|
||||||
|
|
@ -50,7 +50,7 @@ export const updateNewLLMConfigMutationAtom = atomWithMutation((get) => {
|
||||||
return newLLMConfigApiService.updateConfig(request);
|
return newLLMConfigApiService.updateConfig(request);
|
||||||
},
|
},
|
||||||
onSuccess: (_: UpdateNewLLMConfigResponse, request: UpdateNewLLMConfigRequest) => {
|
onSuccess: (_: UpdateNewLLMConfigResponse, request: UpdateNewLLMConfigRequest) => {
|
||||||
toast.success("Configuration updated successfully");
|
toast.success("LLM model updated");
|
||||||
queryClient.invalidateQueries({
|
queryClient.invalidateQueries({
|
||||||
queryKey: cacheKeys.newLLMConfigs.all(Number(searchSpaceId)),
|
queryKey: cacheKeys.newLLMConfigs.all(Number(searchSpaceId)),
|
||||||
});
|
});
|
||||||
|
|
@ -77,7 +77,7 @@ export const deleteNewLLMConfigMutationAtom = atomWithMutation((get) => {
|
||||||
return newLLMConfigApiService.deleteConfig(request);
|
return newLLMConfigApiService.deleteConfig(request);
|
||||||
},
|
},
|
||||||
onSuccess: (_, request: DeleteNewLLMConfigRequest) => {
|
onSuccess: (_, request: DeleteNewLLMConfigRequest) => {
|
||||||
toast.success("Configuration deleted successfully");
|
toast.success("LLM model deleted");
|
||||||
queryClient.setQueryData(
|
queryClient.setQueryData(
|
||||||
cacheKeys.newLLMConfigs.all(Number(searchSpaceId)),
|
cacheKeys.newLLMConfigs.all(Number(searchSpaceId)),
|
||||||
(oldData: GetNewLLMConfigsResponse | undefined) => {
|
(oldData: GetNewLLMConfigsResponse | undefined) => {
|
||||||
|
|
|
||||||
|
|
@ -8,7 +8,7 @@ import type {
|
||||||
NewLLMConfigPublic,
|
NewLLMConfigPublic,
|
||||||
} from "@/contracts/types/new-llm-config.types";
|
} from "@/contracts/types/new-llm-config.types";
|
||||||
import { ImageConfigDialog } from "@/components/shared/image-config-dialog";
|
import { ImageConfigDialog } from "@/components/shared/image-config-dialog";
|
||||||
import { ModelConfigDialog } from "./model-config-dialog";
|
import { ModelConfigDialog } from "@/components/shared/model-config-dialog";
|
||||||
import { ModelSelector } from "./model-selector";
|
import { ModelSelector } from "./model-selector";
|
||||||
|
|
||||||
interface ChatHeaderProps {
|
interface ChatHeaderProps {
|
||||||
|
|
|
||||||
|
|
@ -1,489 +0,0 @@
|
||||||
"use client";
|
|
||||||
|
|
||||||
import { useAtomValue } from "jotai";
|
|
||||||
import { AlertCircle, X, Zap } from "lucide-react";
|
|
||||||
import { AnimatePresence, motion } from "motion/react";
|
|
||||||
import { useCallback, useEffect, useRef, useState } from "react";
|
|
||||||
import { createPortal } from "react-dom";
|
|
||||||
import { toast } from "sonner";
|
|
||||||
import {
|
|
||||||
createNewLLMConfigMutationAtom,
|
|
||||||
updateLLMPreferencesMutationAtom,
|
|
||||||
updateNewLLMConfigMutationAtom,
|
|
||||||
} from "@/atoms/new-llm-config/new-llm-config-mutation.atoms";
|
|
||||||
import { LLMConfigForm, type LLMConfigFormData } from "@/components/shared/llm-config-form";
|
|
||||||
import { Alert, AlertDescription } from "@/components/ui/alert";
|
|
||||||
import { Badge } from "@/components/ui/badge";
|
|
||||||
import { Button } from "@/components/ui/button";
|
|
||||||
import { Spinner } from "@/components/ui/spinner";
|
|
||||||
import type {
|
|
||||||
GlobalNewLLMConfig,
|
|
||||||
LiteLLMProvider,
|
|
||||||
NewLLMConfigPublic,
|
|
||||||
} from "@/contracts/types/new-llm-config.types";
|
|
||||||
import { cn } from "@/lib/utils";
|
|
||||||
|
|
||||||
interface ModelConfigDialogProps {
|
|
||||||
open: boolean;
|
|
||||||
onOpenChange: (open: boolean) => void;
|
|
||||||
config: NewLLMConfigPublic | GlobalNewLLMConfig | null;
|
|
||||||
isGlobal: boolean;
|
|
||||||
searchSpaceId: number;
|
|
||||||
mode: "create" | "edit" | "view";
|
|
||||||
}
|
|
||||||
|
|
||||||
export function ModelConfigDialog({
|
|
||||||
open,
|
|
||||||
onOpenChange,
|
|
||||||
config,
|
|
||||||
isGlobal,
|
|
||||||
searchSpaceId,
|
|
||||||
mode,
|
|
||||||
}: ModelConfigDialogProps) {
|
|
||||||
const [isSubmitting, setIsSubmitting] = useState(false);
|
|
||||||
const [mounted, setMounted] = useState(false);
|
|
||||||
const [scrollPos, setScrollPos] = useState<"top" | "middle" | "bottom">("top");
|
|
||||||
const scrollRef = useRef<HTMLDivElement>(null);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
setMounted(true);
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
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 { mutateAsync: createConfig } = useAtomValue(createNewLLMConfigMutationAtom);
|
|
||||||
const { mutateAsync: updateConfig } = useAtomValue(updateNewLLMConfigMutationAtom);
|
|
||||||
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 isAutoMode = config && "is_auto_mode" in config && config.is_auto_mode;
|
|
||||||
|
|
||||||
const getTitle = () => {
|
|
||||||
if (mode === "create") return "Add New Configuration";
|
|
||||||
if (isAutoMode) return "Auto Mode (Fastest)";
|
|
||||||
if (isGlobal) return "View Global Configuration";
|
|
||||||
return "Edit Configuration";
|
|
||||||
};
|
|
||||||
|
|
||||||
const getSubtitle = () => {
|
|
||||||
if (mode === "create") return "Set up a new LLM provider for this search space";
|
|
||||||
if (isAutoMode) return "Automatically routes requests across providers";
|
|
||||||
if (isGlobal) return "Read-only global configuration";
|
|
||||||
return "Update your configuration settings";
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleSubmit = useCallback(
|
|
||||||
async (data: LLMConfigFormData) => {
|
|
||||||
setIsSubmitting(true);
|
|
||||||
try {
|
|
||||||
if (mode === "create") {
|
|
||||||
const result = await createConfig({
|
|
||||||
...data,
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (result?.id) {
|
|
||||||
await updatePreferences({
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
data: {
|
|
||||||
agent_llm_id: result.id,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
toast.success("Configuration created and assigned!");
|
|
||||||
onOpenChange(false);
|
|
||||||
} else if (!isGlobal && config) {
|
|
||||||
await updateConfig({
|
|
||||||
id: config.id,
|
|
||||||
data: {
|
|
||||||
name: data.name,
|
|
||||||
description: data.description,
|
|
||||||
provider: data.provider,
|
|
||||||
custom_provider: data.custom_provider,
|
|
||||||
model_name: data.model_name,
|
|
||||||
api_key: data.api_key,
|
|
||||||
api_base: data.api_base,
|
|
||||||
litellm_params: data.litellm_params,
|
|
||||||
system_instructions: data.system_instructions,
|
|
||||||
use_default_system_instructions: data.use_default_system_instructions,
|
|
||||||
citations_enabled: data.citations_enabled,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
toast.success("Configuration updated!");
|
|
||||||
onOpenChange(false);
|
|
||||||
}
|
|
||||||
} catch (error) {
|
|
||||||
console.error("Failed to save configuration:", error);
|
|
||||||
toast.error("Failed to save configuration");
|
|
||||||
} finally {
|
|
||||||
setIsSubmitting(false);
|
|
||||||
}
|
|
||||||
},
|
|
||||||
[
|
|
||||||
mode,
|
|
||||||
isGlobal,
|
|
||||||
config,
|
|
||||||
searchSpaceId,
|
|
||||||
createConfig,
|
|
||||||
updateConfig,
|
|
||||||
updatePreferences,
|
|
||||||
onOpenChange,
|
|
||||||
]
|
|
||||||
);
|
|
||||||
|
|
||||||
const handleUseGlobalConfig = useCallback(async () => {
|
|
||||||
if (!config || !isGlobal) return;
|
|
||||||
setIsSubmitting(true);
|
|
||||||
try {
|
|
||||||
await updatePreferences({
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
data: {
|
|
||||||
agent_llm_id: config.id,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
toast.success(`Now using ${config.name}`);
|
|
||||||
onOpenChange(false);
|
|
||||||
} catch (error) {
|
|
||||||
console.error("Failed to set model:", error);
|
|
||||||
toast.error("Failed to set model");
|
|
||||||
} finally {
|
|
||||||
setIsSubmitting(false);
|
|
||||||
}
|
|
||||||
}, [config, isGlobal, searchSpaceId, updatePreferences, onOpenChange]);
|
|
||||||
|
|
||||||
if (!mounted) return null;
|
|
||||||
|
|
||||||
const dialogContent = (
|
|
||||||
<AnimatePresence>
|
|
||||||
{open && (
|
|
||||||
<>
|
|
||||||
{/* Backdrop */}
|
|
||||||
<motion.div
|
|
||||||
initial={{ opacity: 0 }}
|
|
||||||
animate={{ opacity: 1 }}
|
|
||||||
exit={{ opacity: 0 }}
|
|
||||||
transition={{ duration: 0.15 }}
|
|
||||||
className="fixed inset-0 z-[24] bg-black/50 backdrop-blur-sm"
|
|
||||||
onClick={() => onOpenChange(false)}
|
|
||||||
/>
|
|
||||||
|
|
||||||
{/* Dialog */}
|
|
||||||
<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-[25] 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>
|
|
||||||
)}
|
|
||||||
{!isGlobal && mode !== "create" && !isAutoMode && (
|
|
||||||
<Badge variant="outline" className="text-[10px]">
|
|
||||||
Custom
|
|
||||||
</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 automatically distributes requests across all available LLM
|
|
||||||
providers to optimize performance and avoid rate limits.
|
|
||||||
</AlertDescription>
|
|
||||||
</Alert>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{isGlobal && !isAutoMode && mode !== "create" && (
|
|
||||||
<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 settings, create a new
|
|
||||||
configuration based on this template.
|
|
||||||
</AlertDescription>
|
|
||||||
</Alert>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{mode === "create" ? (
|
|
||||||
<LLMConfigForm
|
|
||||||
searchSpaceId={searchSpaceId}
|
|
||||||
onSubmit={handleSubmit}
|
|
||||||
isSubmitting={isSubmitting}
|
|
||||||
mode="create"
|
|
||||||
formId="model-config-form"
|
|
||||||
hideActions
|
|
||||||
/>
|
|
||||||
) : isAutoMode && config ? (
|
|
||||||
<div className="space-y-6">
|
|
||||||
<div className="space-y-4">
|
|
||||||
<div className="space-y-1.5">
|
|
||||||
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
|
||||||
How It Works
|
|
||||||
</div>
|
|
||||||
<p className="text-sm text-muted-foreground">{config.description}</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="h-px bg-border/50" />
|
|
||||||
|
|
||||||
<div className="space-y-3">
|
|
||||||
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
|
||||||
Key Benefits
|
|
||||||
</div>
|
|
||||||
<div className="space-y-2">
|
|
||||||
<div className="flex items-start gap-3 p-3 rounded-lg bg-violet-50 dark:bg-violet-900/20 border border-violet-200 dark:border-violet-800/50">
|
|
||||||
<Zap className="size-4 text-violet-600 dark:text-violet-400 mt-0.5 shrink-0" />
|
|
||||||
<div>
|
|
||||||
<p className="text-sm font-medium text-violet-900 dark:text-violet-100">
|
|
||||||
Automatic (Fastest)
|
|
||||||
</p>
|
|
||||||
<p className="text-xs text-violet-700 dark:text-violet-300">
|
|
||||||
Distributes requests across all configured LLM providers
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div className="flex items-start gap-3 p-3 rounded-lg bg-violet-50 dark:bg-violet-900/20 border border-violet-200 dark:border-violet-800/50">
|
|
||||||
<Zap className="size-4 text-violet-600 dark:text-violet-400 mt-0.5 shrink-0" />
|
|
||||||
<div>
|
|
||||||
<p className="text-sm font-medium text-violet-900 dark:text-violet-100">
|
|
||||||
Rate Limit Protection
|
|
||||||
</p>
|
|
||||||
<p className="text-xs text-violet-700 dark:text-violet-300">
|
|
||||||
Automatically handles rate limits with cooldowns and retries
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div className="flex items-start gap-3 p-3 rounded-lg bg-violet-50 dark:bg-violet-900/20 border border-violet-200 dark:border-violet-800/50">
|
|
||||||
<Zap className="size-4 text-violet-600 dark:text-violet-400 mt-0.5 shrink-0" />
|
|
||||||
<div>
|
|
||||||
<p className="text-sm font-medium text-violet-900 dark:text-violet-100">
|
|
||||||
Automatic Failover
|
|
||||||
</p>
|
|
||||||
<p className="text-xs text-violet-700 dark:text-violet-300">
|
|
||||||
Falls back to other providers if one becomes unavailable
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
) : isGlobal && config ? (
|
|
||||||
<div className="space-y-6">
|
|
||||||
<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">
|
|
||||||
Configuration 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>
|
|
||||||
|
|
||||||
<div className="h-px bg-border/50" />
|
|
||||||
|
|
||||||
<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 className="h-px bg-border/50" />
|
|
||||||
|
|
||||||
<div className="grid gap-4 sm:grid-cols-2">
|
|
||||||
<div className="space-y-2">
|
|
||||||
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
|
||||||
Citations
|
|
||||||
</div>
|
|
||||||
<Badge
|
|
||||||
variant={config.citations_enabled ? "default" : "secondary"}
|
|
||||||
className="w-fit"
|
|
||||||
>
|
|
||||||
{config.citations_enabled ? "Enabled" : "Disabled"}
|
|
||||||
</Badge>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{config.system_instructions && (
|
|
||||||
<>
|
|
||||||
<div className="h-px bg-border/50" />
|
|
||||||
<div className="space-y-1.5">
|
|
||||||
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
|
||||||
System Instructions
|
|
||||||
</div>
|
|
||||||
<div className="p-3 rounded-lg bg-muted/50 border border-border/50">
|
|
||||||
<p className="text-xs font-mono text-muted-foreground whitespace-pre-wrap line-clamp-10">
|
|
||||||
{config.system_instructions}
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
) : config ? (
|
|
||||||
<LLMConfigForm
|
|
||||||
searchSpaceId={searchSpaceId}
|
|
||||||
initialData={{
|
|
||||||
name: config.name,
|
|
||||||
description: config.description,
|
|
||||||
provider: config.provider as LiteLLMProvider,
|
|
||||||
custom_provider: config.custom_provider,
|
|
||||||
model_name: config.model_name,
|
|
||||||
api_key: "api_key" in config ? (config.api_key as string) : "",
|
|
||||||
api_base: config.api_base,
|
|
||||||
litellm_params: config.litellm_params,
|
|
||||||
system_instructions: config.system_instructions,
|
|
||||||
use_default_system_instructions: config.use_default_system_instructions,
|
|
||||||
citations_enabled: config.citations_enabled,
|
|
||||||
search_space_id: searchSpaceId,
|
|
||||||
}}
|
|
||||||
onSubmit={handleSubmit}
|
|
||||||
isSubmitting={isSubmitting}
|
|
||||||
mode="edit"
|
|
||||||
formId="model-config-form"
|
|
||||||
hideActions
|
|
||||||
/>
|
|
||||||
) : null}
|
|
||||||
</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" || (!isGlobal && !isAutoMode && config) ? (
|
|
||||||
<Button
|
|
||||||
type="submit"
|
|
||||||
form="model-config-form"
|
|
||||||
disabled={isSubmitting}
|
|
||||||
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;
|
|
||||||
}
|
|
||||||
|
|
@ -12,18 +12,16 @@ import {
|
||||||
Trash2,
|
Trash2,
|
||||||
Wand2,
|
Wand2,
|
||||||
} from "lucide-react";
|
} from "lucide-react";
|
||||||
import { useCallback, useMemo, useState } from "react";
|
import { useMemo, useState } from "react";
|
||||||
import { membersAtom, myAccessAtom } from "@/atoms/members/members-query.atoms";
|
import { membersAtom, myAccessAtom } from "@/atoms/members/members-query.atoms";
|
||||||
import {
|
import {
|
||||||
createNewLLMConfigMutationAtom,
|
|
||||||
deleteNewLLMConfigMutationAtom,
|
deleteNewLLMConfigMutationAtom,
|
||||||
updateNewLLMConfigMutationAtom,
|
|
||||||
} from "@/atoms/new-llm-config/new-llm-config-mutation.atoms";
|
} from "@/atoms/new-llm-config/new-llm-config-mutation.atoms";
|
||||||
import {
|
import {
|
||||||
globalNewLLMConfigsAtom,
|
globalNewLLMConfigsAtom,
|
||||||
newLLMConfigsAtom,
|
newLLMConfigsAtom,
|
||||||
} from "@/atoms/new-llm-config/new-llm-config-query.atoms";
|
} from "@/atoms/new-llm-config/new-llm-config-query.atoms";
|
||||||
import { LLMConfigForm, type LLMConfigFormData } from "@/components/shared/llm-config-form";
|
import { ModelConfigDialog } from "@/components/shared/model-config-dialog";
|
||||||
import { Alert, AlertDescription } from "@/components/ui/alert";
|
import { Alert, AlertDescription } from "@/components/ui/alert";
|
||||||
import {
|
import {
|
||||||
AlertDialog,
|
AlertDialog,
|
||||||
|
|
@ -39,13 +37,6 @@ import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar";
|
||||||
import { Badge } from "@/components/ui/badge";
|
import { Badge } from "@/components/ui/badge";
|
||||||
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 {
|
|
||||||
Dialog,
|
|
||||||
DialogContent,
|
|
||||||
DialogDescription,
|
|
||||||
DialogHeader,
|
|
||||||
DialogTitle,
|
|
||||||
} from "@/components/ui/dialog";
|
|
||||||
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";
|
||||||
|
|
@ -69,12 +60,6 @@ function getInitials(name: string): string {
|
||||||
export function ModelConfigManager({ searchSpaceId }: ModelConfigManagerProps) {
|
export function ModelConfigManager({ searchSpaceId }: ModelConfigManagerProps) {
|
||||||
const isDesktop = useMediaQuery("(min-width: 768px)");
|
const isDesktop = useMediaQuery("(min-width: 768px)");
|
||||||
// Mutations
|
// Mutations
|
||||||
const { mutateAsync: createConfig, isPending: isCreating } = useAtomValue(
|
|
||||||
createNewLLMConfigMutationAtom
|
|
||||||
);
|
|
||||||
const { mutateAsync: updateConfig, isPending: isUpdating } = useAtomValue(
|
|
||||||
updateNewLLMConfigMutationAtom
|
|
||||||
);
|
|
||||||
const { mutateAsync: deleteConfig, isPending: isDeleting } = useAtomValue(
|
const { mutateAsync: deleteConfig, isPending: isDeleting } = useAtomValue(
|
||||||
deleteNewLLMConfigMutationAtom
|
deleteNewLLMConfigMutationAtom
|
||||||
);
|
);
|
||||||
|
|
@ -128,29 +113,6 @@ export function ModelConfigManager({ searchSpaceId }: ModelConfigManagerProps) {
|
||||||
const [editingConfig, setEditingConfig] = useState<NewLLMConfig | null>(null);
|
const [editingConfig, setEditingConfig] = useState<NewLLMConfig | null>(null);
|
||||||
const [configToDelete, setConfigToDelete] = useState<NewLLMConfig | null>(null);
|
const [configToDelete, setConfigToDelete] = useState<NewLLMConfig | null>(null);
|
||||||
|
|
||||||
const isSubmitting = isCreating || isUpdating;
|
|
||||||
|
|
||||||
const handleFormSubmit = useCallback(
|
|
||||||
async (formData: LLMConfigFormData) => {
|
|
||||||
try {
|
|
||||||
if (editingConfig) {
|
|
||||||
const { search_space_id, ...updateData } = formData;
|
|
||||||
await updateConfig({
|
|
||||||
id: editingConfig.id,
|
|
||||||
data: updateData,
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
await createConfig(formData);
|
|
||||||
}
|
|
||||||
setIsDialogOpen(false);
|
|
||||||
setEditingConfig(null);
|
|
||||||
} catch {
|
|
||||||
// Error is displayed inside the dialog by the form
|
|
||||||
}
|
|
||||||
},
|
|
||||||
[editingConfig, createConfig, updateConfig]
|
|
||||||
);
|
|
||||||
|
|
||||||
const handleDelete = async () => {
|
const handleDelete = async () => {
|
||||||
if (!configToDelete) return;
|
if (!configToDelete) return;
|
||||||
try {
|
try {
|
||||||
|
|
@ -171,11 +133,6 @@ export function ModelConfigManager({ searchSpaceId }: ModelConfigManagerProps) {
|
||||||
setIsDialogOpen(true);
|
setIsDialogOpen(true);
|
||||||
};
|
};
|
||||||
|
|
||||||
const closeDialog = () => {
|
|
||||||
setIsDialogOpen(false);
|
|
||||||
setEditingConfig(null);
|
|
||||||
};
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="space-y-5 md:space-y-6">
|
<div className="space-y-5 md:space-y-6">
|
||||||
{/* Header actions */}
|
{/* Header actions */}
|
||||||
|
|
@ -457,54 +414,17 @@ export function ModelConfigManager({ searchSpaceId }: ModelConfigManagerProps) {
|
||||||
)}
|
)}
|
||||||
|
|
||||||
{/* Add/Edit Configuration Dialog */}
|
{/* Add/Edit Configuration Dialog */}
|
||||||
<Dialog open={isDialogOpen} onOpenChange={(open) => !open && closeDialog()}>
|
<ModelConfigDialog
|
||||||
<DialogContent
|
open={isDialogOpen}
|
||||||
className="max-w-2xl max-h-[90vh] overflow-y-auto"
|
onOpenChange={(open) => {
|
||||||
onOpenAutoFocus={(e) => e.preventDefault()}
|
setIsDialogOpen(open);
|
||||||
>
|
if (!open) setEditingConfig(null);
|
||||||
<DialogHeader>
|
}}
|
||||||
<DialogTitle>
|
config={editingConfig}
|
||||||
{editingConfig ? "Edit Configuration" : "Create New Configuration"}
|
isGlobal={false}
|
||||||
</DialogTitle>
|
searchSpaceId={searchSpaceId}
|
||||||
<DialogDescription>
|
mode={editingConfig ? "edit" : "create"}
|
||||||
{editingConfig
|
/>
|
||||||
? "Update your AI model and prompt configuration"
|
|
||||||
: "Set up a new AI model with custom prompts and citation settings"}
|
|
||||||
</DialogDescription>
|
|
||||||
</DialogHeader>
|
|
||||||
|
|
||||||
<LLMConfigForm
|
|
||||||
key={editingConfig ? `edit-${editingConfig.id}` : "create"}
|
|
||||||
searchSpaceId={searchSpaceId}
|
|
||||||
initialData={
|
|
||||||
editingConfig
|
|
||||||
? {
|
|
||||||
name: editingConfig.name,
|
|
||||||
description: editingConfig.description || "",
|
|
||||||
provider: editingConfig.provider,
|
|
||||||
custom_provider: editingConfig.custom_provider || "",
|
|
||||||
model_name: editingConfig.model_name,
|
|
||||||
api_key: editingConfig.api_key,
|
|
||||||
api_base: editingConfig.api_base || "",
|
|
||||||
litellm_params: editingConfig.litellm_params || {},
|
|
||||||
system_instructions: editingConfig.system_instructions || "",
|
|
||||||
use_default_system_instructions: editingConfig.use_default_system_instructions,
|
|
||||||
citations_enabled: editingConfig.citations_enabled,
|
|
||||||
}
|
|
||||||
: {
|
|
||||||
citations_enabled: true,
|
|
||||||
use_default_system_instructions: true,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
onSubmit={handleFormSubmit}
|
|
||||||
onCancel={closeDialog}
|
|
||||||
isSubmitting={isSubmitting}
|
|
||||||
mode={editingConfig ? "edit" : "create"}
|
|
||||||
showAdvanced={true}
|
|
||||||
compact={true}
|
|
||||||
/>
|
|
||||||
</DialogContent>
|
|
||||||
</Dialog>
|
|
||||||
|
|
||||||
{/* Delete Confirmation Dialog */}
|
{/* Delete Confirmation Dialog */}
|
||||||
<AlertDialog
|
<AlertDialog
|
||||||
|
|
|
||||||
424
surfsense_web/components/shared/model-config-dialog.tsx
Normal file
424
surfsense_web/components/shared/model-config-dialog.tsx
Normal file
|
|
@ -0,0 +1,424 @@
|
||||||
|
"use client";
|
||||||
|
|
||||||
|
import { useAtomValue } from "jotai";
|
||||||
|
import { AlertCircle, Zap } from "lucide-react";
|
||||||
|
import { useCallback, useRef, useState } from "react";
|
||||||
|
import { toast } from "sonner";
|
||||||
|
import {
|
||||||
|
createNewLLMConfigMutationAtom,
|
||||||
|
updateLLMPreferencesMutationAtom,
|
||||||
|
updateNewLLMConfigMutationAtom,
|
||||||
|
} from "@/atoms/new-llm-config/new-llm-config-mutation.atoms";
|
||||||
|
import { LLMConfigForm, type LLMConfigFormData } from "@/components/shared/llm-config-form";
|
||||||
|
import { Alert, AlertDescription } from "@/components/ui/alert";
|
||||||
|
import { Badge } from "@/components/ui/badge";
|
||||||
|
import { Button } from "@/components/ui/button";
|
||||||
|
import { Dialog, DialogContent, DialogTitle } from "@/components/ui/dialog";
|
||||||
|
import { Spinner } from "@/components/ui/spinner";
|
||||||
|
import type {
|
||||||
|
GlobalNewLLMConfig,
|
||||||
|
LiteLLMProvider,
|
||||||
|
NewLLMConfigPublic,
|
||||||
|
} from "@/contracts/types/new-llm-config.types";
|
||||||
|
|
||||||
|
interface ModelConfigDialogProps {
|
||||||
|
open: boolean;
|
||||||
|
onOpenChange: (open: boolean) => void;
|
||||||
|
config: NewLLMConfigPublic | GlobalNewLLMConfig | null;
|
||||||
|
isGlobal: boolean;
|
||||||
|
searchSpaceId: number;
|
||||||
|
mode: "create" | "edit" | "view";
|
||||||
|
}
|
||||||
|
|
||||||
|
export function ModelConfigDialog({
|
||||||
|
open,
|
||||||
|
onOpenChange,
|
||||||
|
config,
|
||||||
|
isGlobal,
|
||||||
|
searchSpaceId,
|
||||||
|
mode,
|
||||||
|
}: ModelConfigDialogProps) {
|
||||||
|
const [isSubmitting, setIsSubmitting] = useState(false);
|
||||||
|
const [scrollPos, setScrollPos] = useState<"top" | "middle" | "bottom">("top");
|
||||||
|
const scrollRef = useRef<HTMLDivElement>(null);
|
||||||
|
|
||||||
|
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 { mutateAsync: createConfig } = useAtomValue(createNewLLMConfigMutationAtom);
|
||||||
|
const { mutateAsync: updateConfig } = useAtomValue(updateNewLLMConfigMutationAtom);
|
||||||
|
const { mutateAsync: updatePreferences } = useAtomValue(updateLLMPreferencesMutationAtom);
|
||||||
|
|
||||||
|
const isAutoMode = config && "is_auto_mode" in config && config.is_auto_mode;
|
||||||
|
|
||||||
|
const getTitle = () => {
|
||||||
|
if (mode === "create") return "Add New Configuration";
|
||||||
|
if (isAutoMode) return "Auto Mode (Fastest)";
|
||||||
|
if (isGlobal) return "View Global Configuration";
|
||||||
|
return "Edit Configuration";
|
||||||
|
};
|
||||||
|
|
||||||
|
const getSubtitle = () => {
|
||||||
|
if (mode === "create") return "Set up a new LLM provider for this search space";
|
||||||
|
if (isAutoMode) return "Automatically routes requests across providers";
|
||||||
|
if (isGlobal) return "Read-only global configuration";
|
||||||
|
return "Update your configuration settings";
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleSubmit = useCallback(
|
||||||
|
async (data: LLMConfigFormData) => {
|
||||||
|
setIsSubmitting(true);
|
||||||
|
try {
|
||||||
|
if (mode === "create") {
|
||||||
|
const result = await createConfig({
|
||||||
|
...data,
|
||||||
|
search_space_id: searchSpaceId,
|
||||||
|
});
|
||||||
|
|
||||||
|
if (result?.id) {
|
||||||
|
await updatePreferences({
|
||||||
|
search_space_id: searchSpaceId,
|
||||||
|
data: {
|
||||||
|
agent_llm_id: result.id,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
onOpenChange(false);
|
||||||
|
} else if (!isGlobal && config) {
|
||||||
|
await updateConfig({
|
||||||
|
id: config.id,
|
||||||
|
data: {
|
||||||
|
name: data.name,
|
||||||
|
description: data.description,
|
||||||
|
provider: data.provider,
|
||||||
|
custom_provider: data.custom_provider,
|
||||||
|
model_name: data.model_name,
|
||||||
|
api_key: data.api_key,
|
||||||
|
api_base: data.api_base,
|
||||||
|
litellm_params: data.litellm_params,
|
||||||
|
system_instructions: data.system_instructions,
|
||||||
|
use_default_system_instructions: data.use_default_system_instructions,
|
||||||
|
citations_enabled: data.citations_enabled,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
onOpenChange(false);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Failed to save configuration:", error);
|
||||||
|
} finally {
|
||||||
|
setIsSubmitting(false);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[
|
||||||
|
mode,
|
||||||
|
isGlobal,
|
||||||
|
config,
|
||||||
|
searchSpaceId,
|
||||||
|
createConfig,
|
||||||
|
updateConfig,
|
||||||
|
updatePreferences,
|
||||||
|
onOpenChange,
|
||||||
|
]
|
||||||
|
);
|
||||||
|
|
||||||
|
const handleUseGlobalConfig = useCallback(async () => {
|
||||||
|
if (!config || !isGlobal) return;
|
||||||
|
setIsSubmitting(true);
|
||||||
|
try {
|
||||||
|
await updatePreferences({
|
||||||
|
search_space_id: searchSpaceId,
|
||||||
|
data: {
|
||||||
|
agent_llm_id: config.id,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
toast.success(`Now using ${config.name}`);
|
||||||
|
onOpenChange(false);
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Failed to set model:", error);
|
||||||
|
} finally {
|
||||||
|
setIsSubmitting(false);
|
||||||
|
}
|
||||||
|
}, [config, isGlobal, searchSpaceId, updatePreferences, onOpenChange]);
|
||||||
|
|
||||||
|
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>
|
||||||
|
)}
|
||||||
|
{!isGlobal && mode !== "create" && !isAutoMode && (
|
||||||
|
<Badge variant="outline" className="text-[10px]">
|
||||||
|
Custom
|
||||||
|
</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 automatically distributes requests across all available LLM
|
||||||
|
providers to optimize performance and avoid rate limits.
|
||||||
|
</AlertDescription>
|
||||||
|
</Alert>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{isGlobal && !isAutoMode && mode !== "create" && (
|
||||||
|
<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 settings, create a new
|
||||||
|
configuration based on this template.
|
||||||
|
</AlertDescription>
|
||||||
|
</Alert>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{mode === "create" ? (
|
||||||
|
<LLMConfigForm
|
||||||
|
searchSpaceId={searchSpaceId}
|
||||||
|
onSubmit={handleSubmit}
|
||||||
|
isSubmitting={isSubmitting}
|
||||||
|
mode="create"
|
||||||
|
formId="model-config-form"
|
||||||
|
hideActions
|
||||||
|
/>
|
||||||
|
) : isAutoMode && config ? (
|
||||||
|
<div className="space-y-6">
|
||||||
|
<div className="space-y-4">
|
||||||
|
<div className="space-y-1.5">
|
||||||
|
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
||||||
|
How It Works
|
||||||
|
</div>
|
||||||
|
<p className="text-sm text-muted-foreground">{config.description}</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="h-px bg-border/50" />
|
||||||
|
|
||||||
|
<div className="space-y-3">
|
||||||
|
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
||||||
|
Key Benefits
|
||||||
|
</div>
|
||||||
|
<div className="space-y-2">
|
||||||
|
<div className="flex items-start gap-3 p-3 rounded-lg bg-violet-50 dark:bg-violet-900/20 border border-violet-200 dark:border-violet-800/50">
|
||||||
|
<Zap className="size-4 text-violet-600 dark:text-violet-400 mt-0.5 shrink-0" />
|
||||||
|
<div>
|
||||||
|
<p className="text-sm font-medium text-violet-900 dark:text-violet-100">
|
||||||
|
Automatic (Fastest)
|
||||||
|
</p>
|
||||||
|
<p className="text-xs text-violet-700 dark:text-violet-300">
|
||||||
|
Distributes requests across all configured LLM providers
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div className="flex items-start gap-3 p-3 rounded-lg bg-violet-50 dark:bg-violet-900/20 border border-violet-200 dark:border-violet-800/50">
|
||||||
|
<Zap className="size-4 text-violet-600 dark:text-violet-400 mt-0.5 shrink-0" />
|
||||||
|
<div>
|
||||||
|
<p className="text-sm font-medium text-violet-900 dark:text-violet-100">
|
||||||
|
Rate Limit Protection
|
||||||
|
</p>
|
||||||
|
<p className="text-xs text-violet-700 dark:text-violet-300">
|
||||||
|
Automatically handles rate limits with cooldowns and retries
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div className="flex items-start gap-3 p-3 rounded-lg bg-violet-50 dark:bg-violet-900/20 border border-violet-200 dark:border-violet-800/50">
|
||||||
|
<Zap className="size-4 text-violet-600 dark:text-violet-400 mt-0.5 shrink-0" />
|
||||||
|
<div>
|
||||||
|
<p className="text-sm font-medium text-violet-900 dark:text-violet-100">
|
||||||
|
Automatic Failover
|
||||||
|
</p>
|
||||||
|
<p className="text-xs text-violet-700 dark:text-violet-300">
|
||||||
|
Falls back to other providers if one becomes unavailable
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
) : isGlobal && config ? (
|
||||||
|
<div className="space-y-6">
|
||||||
|
<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">
|
||||||
|
Configuration 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>
|
||||||
|
|
||||||
|
<div className="h-px bg-border/50" />
|
||||||
|
|
||||||
|
<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 className="h-px bg-border/50" />
|
||||||
|
|
||||||
|
<div className="grid gap-4 sm:grid-cols-2">
|
||||||
|
<div className="space-y-2">
|
||||||
|
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
||||||
|
Citations
|
||||||
|
</div>
|
||||||
|
<Badge
|
||||||
|
variant={config.citations_enabled ? "default" : "secondary"}
|
||||||
|
className="w-fit"
|
||||||
|
>
|
||||||
|
{config.citations_enabled ? "Enabled" : "Disabled"}
|
||||||
|
</Badge>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{config.system_instructions && (
|
||||||
|
<>
|
||||||
|
<div className="h-px bg-border/50" />
|
||||||
|
<div className="space-y-1.5">
|
||||||
|
<div className="text-xs font-medium text-muted-foreground uppercase tracking-wider">
|
||||||
|
System Instructions
|
||||||
|
</div>
|
||||||
|
<div className="p-3 rounded-lg bg-muted/50 border border-border/50">
|
||||||
|
<p className="text-xs font-mono text-muted-foreground whitespace-pre-wrap line-clamp-10">
|
||||||
|
{config.system_instructions}
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
) : config ? (
|
||||||
|
<LLMConfigForm
|
||||||
|
searchSpaceId={searchSpaceId}
|
||||||
|
initialData={{
|
||||||
|
name: config.name,
|
||||||
|
description: config.description,
|
||||||
|
provider: config.provider as LiteLLMProvider,
|
||||||
|
custom_provider: config.custom_provider,
|
||||||
|
model_name: config.model_name,
|
||||||
|
api_key: "api_key" in config ? (config.api_key as string) : "",
|
||||||
|
api_base: config.api_base,
|
||||||
|
litellm_params: config.litellm_params,
|
||||||
|
system_instructions: config.system_instructions,
|
||||||
|
use_default_system_instructions: config.use_default_system_instructions,
|
||||||
|
citations_enabled: config.citations_enabled,
|
||||||
|
search_space_id: searchSpaceId,
|
||||||
|
}}
|
||||||
|
onSubmit={handleSubmit}
|
||||||
|
isSubmitting={isSubmitting}
|
||||||
|
mode="edit"
|
||||||
|
formId="model-config-form"
|
||||||
|
hideActions
|
||||||
|
/>
|
||||||
|
) : null}
|
||||||
|
</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" || (!isGlobal && !isAutoMode && config) ? (
|
||||||
|
<Button
|
||||||
|
type="submit"
|
||||||
|
form="model-config-form"
|
||||||
|
disabled={isSubmitting}
|
||||||
|
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