diff --git a/frontend/libs/common/src/lib/enums/GenerationProvider.ts b/frontend/libs/common/src/lib/enums/GenerationProvider.ts index 2bdd04368c..68abeb5d01 100644 --- a/frontend/libs/common/src/lib/enums/GenerationProvider.ts +++ b/frontend/libs/common/src/lib/enums/GenerationProvider.ts @@ -4,4 +4,5 @@ export enum GenerationProvider { Fal = "fal", Midjourney = "midjourney", Sora = "sora", + WorldLabs = "world_labs", } diff --git a/frontend/libs/components/provider-setup-modal/src/lib/provider-setup-modal.tsx b/frontend/libs/components/provider-setup-modal/src/lib/provider-setup-modal.tsx index 8585bf5431..d9f0b690d3 100644 --- a/frontend/libs/components/provider-setup-modal/src/lib/provider-setup-modal.tsx +++ b/frontend/libs/components/provider-setup-modal/src/lib/provider-setup-modal.tsx @@ -44,8 +44,11 @@ export function ProviderSetupModal({ const buttonOnClick = async () => { switch (provider) { + case GenerationProvider.WorldLabs: + await invoke("worldlabs_open_login_command"); + break; case GenerationProvider.Grok: - await invoke("grok_open_login_command"); // TODO: Rename in Tauri + await invoke("grok_open_login_command"); break; case GenerationProvider.Midjourney: await invoke("midjourney_open_login_command"); @@ -123,6 +126,8 @@ function getServiceProviderName(provider: GenerationProvider) : string { return "Midjourney"; case GenerationProvider.Sora: return "Sora"; + case GenerationProvider.WorldLabs: + return "WorldLabs"; case GenerationProvider.Artcraft: default: return "Artcraft";