fix: add machine model dialog

This commit is contained in:
bennykok 2024-01-29 13:38:00 +08:00
parent 194715920e
commit 24e95a1954
2 changed files with 4 additions and 2 deletions

View File

@ -31,6 +31,7 @@ export function InsertModal<
tooltip?: string; tooltip?: string;
disabled?: boolean; disabled?: boolean;
title: React.ReactNode; title: React.ReactNode;
buttonTitle?: React.ReactNode;
description: string; description: string;
dialogClassName?: string; dialogClassName?: string;
serverAction: (data: z.infer<Z>) => Promise<unknown>; serverAction: (data: z.infer<Z>) => Promise<unknown>;
@ -54,7 +55,7 @@ export function InsertModal<
setOpen(true); setOpen(true);
}} }}
> >
{props.title} {props.buttonTitle ?? props.title}
</Button> </Button>
</TooltipTrigger> </TooltipTrigger>
<TooltipContent> <TooltipContent>

View File

@ -396,11 +396,12 @@ export function MachineList({
? `Max ${machineMaxCount} ComfyUI machine for your account, upgrade to unlock more cnfiguration.` ? `Max ${machineMaxCount} ComfyUI machine for your account, upgrade to unlock more cnfiguration.`
: `Max ${machineMaxCount} ComfyUI machine for your account` : `Max ${machineMaxCount} ComfyUI machine for your account`
} }
title={ buttonTitle={
<> <>
New Machine {locked ? <Lock size={14} /> : <Plus size={14} />} New Machine {locked ? <Lock size={14} /> : <Plus size={14} />}
</> </>
} }
title={"New Machine"}
description="Add custom ComfyUI machines to your account." description="Add custom ComfyUI machines to your account."
serverAction={addCustomMachine} serverAction={addCustomMachine}
formSchema={addCustomMachineSchema} formSchema={addCustomMachineSchema}