Merge pull request #1293 from ai16z/patch/handle_provider

fix: add missing claude vertex case to handleProvider
elizaOS · Dec 21, 2024 · dfab4c1 · dfab4c1
2 parents 907c5a5 + 0eb98a3
commit dfab4c1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git 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);