From 0eb98a38f28d39b5f1a8219749ecadea9e5cec58 Mon Sep 17 00:00:00 2001 From: Shakker Nerd Date: Sat, 21 Dec 2024 01:05:41 +0000 Subject: [PATCH] fix: add missing claude vertex case to handleProvider --- packages/core/src/generation.ts | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/packages/core/src/generation.ts b/packages/core/src/generation.ts index dba21e51123..167ee9447ff 100644 --- a/packages/core/src/generation.ts +++ b/packages/core/src/generation.ts @@ -1066,8 +1066,12 @@ export const generateImage = async ( num_inference_steps: modelSettings?.steps ?? 50, guidance_scale: data.guidanceScale || 3.5, num_images: data.count, - enable_safety_checker: runtime.getSetting("FAL_AI_ENABLE_SAFETY_CHECKER") === "true", - safety_tolerance: Number(runtime.getSetting("FAL_AI_SAFETY_TOLERANCE") || "2"), + enable_safety_checker: + runtime.getSetting("FAL_AI_ENABLE_SAFETY_CHECKER") === + "true", + safety_tolerance: Number( + runtime.getSetting("FAL_AI_SAFETY_TOLERANCE") || "2" + ), output_format: "png" as const, seed: data.seed ?? 6252023, ...(runtime.getSetting("FAL_AI_LORA_PATH") @@ -1368,6 +1372,7 @@ export async function handleProvider( case ModelProviderName.AKASH_CHAT_API: return await handleOpenAI(options); case ModelProviderName.ANTHROPIC: + case ModelProviderName.CLAUDE_VERTEX: return await handleAnthropic(options); case ModelProviderName.GROK: return await handleGrok(options);