allow creating and listing custom language model (#134)

This commit is contained in:
Will Chen
2025-05-12 16:00:16 -07:00
committed by GitHub
parent c63781d7cc
commit 477015b43d
13 changed files with 925 additions and 291 deletions

View File

@@ -0,0 +1,194 @@
import { Info, KeyRound, Trash2 } from "lucide-react";
import { Alert, AlertDescription, AlertTitle } from "@/components/ui/alert";
import {
Accordion,
AccordionContent,
AccordionItem,
AccordionTrigger,
} from "@/components/ui/accordion";
import { Input } from "@/components/ui/input";
import { Button } from "@/components/ui/button";
import { UserSettings } from "@/lib/schemas";
// Helper function to mask ENV API keys (move or duplicate if needed elsewhere)
const maskEnvApiKey = (key: string | undefined): string => {
if (!key) return "Not Set";
if (key.length < 8) return "****";
return `${key.substring(0, 4)}...${key.substring(key.length - 4)}`;
};
interface ApiKeyConfigurationProps {
provider: string;
providerDisplayName: string;
settings: UserSettings | null | undefined;
envVars: Record<string, string | undefined>;
envVarName?: string;
isSaving: boolean;
saveError: string | null;
apiKeyInput: string;
onApiKeyInputChange: (value: string) => void;
onSaveKey: () => Promise<void>;
onDeleteKey: () => Promise<void>;
isDyad: boolean;
}
export function ApiKeyConfiguration({
provider,
providerDisplayName,
settings,
envVars,
envVarName,
isSaving,
saveError,
apiKeyInput,
onApiKeyInputChange,
onSaveKey,
onDeleteKey,
isDyad,
}: ApiKeyConfigurationProps) {
const envApiKey = envVarName ? envVars[envVarName] : undefined;
const userApiKey = settings?.providerSettings?.[provider]?.apiKey?.value;
const isValidUserKey =
!!userApiKey &&
!userApiKey.startsWith("Invalid Key") &&
userApiKey !== "Not Set";
const hasEnvKey = !!envApiKey;
const activeKeySource = isValidUserKey
? "settings"
: hasEnvKey
? "env"
: "none";
const defaultAccordionValue = [];
if (isValidUserKey || !hasEnvKey) {
defaultAccordionValue.push("settings-key");
}
if (!isDyad && hasEnvKey) {
defaultAccordionValue.push("env-key");
}
return (
<Accordion
type="multiple"
className="w-full space-y-4"
defaultValue={defaultAccordionValue}
>
<AccordionItem
value="settings-key"
className="border rounded-lg px-4 bg-(--background-lightest)"
>
<AccordionTrigger className="text-lg font-medium hover:no-underline cursor-pointer">
API Key from Settings
</AccordionTrigger>
<AccordionContent className="pt-4 ">
{isValidUserKey && (
<Alert variant="default" className="mb-4">
<KeyRound className="h-4 w-4" />
<AlertTitle className="flex justify-between items-center">
<span>Current Key (Settings)</span>
<Button
variant="destructive"
size="sm"
onClick={onDeleteKey}
disabled={isSaving}
className="flex items-center gap-1 h-7 px-2"
>
<Trash2 className="h-4 w-4" />
{isSaving ? "Deleting..." : "Delete"}
</Button>
</AlertTitle>
<AlertDescription>
<p className="font-mono text-sm">{userApiKey}</p>
{activeKeySource === "settings" && (
<p className="text-xs text-green-600 dark:text-green-400 mt-1">
This key is currently active.
</p>
)}
</AlertDescription>
</Alert>
)}
<div className="space-y-2">
<label
htmlFor="apiKeyInput"
className="block text-sm font-medium text-gray-700 dark:text-gray-300"
>
{isValidUserKey ? "Update" : "Set"} {providerDisplayName} API Key
</label>
<div className="flex items-start space-x-2">
<Input
id="apiKeyInput"
value={apiKeyInput}
onChange={(e) => onApiKeyInputChange(e.target.value)}
placeholder={`Enter new ${providerDisplayName} API Key here`}
className={`flex-grow ${saveError ? "border-red-500" : ""}`}
/>
<Button onClick={onSaveKey} disabled={isSaving || !apiKeyInput}>
{isSaving ? "Saving..." : "Save Key"}
</Button>
</div>
{saveError && <p className="text-xs text-red-600">{saveError}</p>}
<p className="text-xs text-gray-500 dark:text-gray-400">
Setting a key here will override the environment variable (if
set).
</p>
</div>
</AccordionContent>
</AccordionItem>
{!isDyad && envVarName && (
<AccordionItem
value="env-key"
className="border rounded-lg px-4 bg-(--background-lightest)"
>
<AccordionTrigger className="text-lg font-medium hover:no-underline cursor-pointer">
API Key from Environment Variable
</AccordionTrigger>
<AccordionContent className="pt-4">
{hasEnvKey ? (
<Alert variant="default">
<KeyRound className="h-4 w-4" />
<AlertTitle>Environment Variable Key ({envVarName})</AlertTitle>
<AlertDescription>
<p className="font-mono text-sm">
{maskEnvApiKey(envApiKey)}
</p>
{activeKeySource === "env" && (
<p className="text-xs text-green-600 dark:text-green-400 mt-1">
This key is currently active (no settings key set).
</p>
)}
{activeKeySource === "settings" && (
<p className="text-xs text-yellow-600 dark:text-yellow-400 mt-1">
This key is currently being overridden by the key set in
Settings.
</p>
)}
</AlertDescription>
</Alert>
) : (
<Alert variant="default">
<Info className="h-4 w-4" />
<AlertTitle>Environment Variable Not Set</AlertTitle>
<AlertDescription>
The{" "}
<code className="font-mono bg-gray-100 dark:bg-gray-800 px-1 rounded text-xs">
{envVarName}
</code>{" "}
environment variable is not set.
</AlertDescription>
</Alert>
)}
<p className="text-xs text-gray-500 dark:text-gray-400 mt-3">
This key is set outside the application. If present, it will be
used only if no key is configured in the Settings section above.
Requires app restart to detect changes.
</p>
</AccordionContent>
</AccordionItem>
)}
</Accordion>
);
}