From 5b49a2cda6d57c9299566084c299c43a73346fa5 Mon Sep 17 00:00:00 2001
From: Nick Wong <nick.wong@hashicorp.com>
Date: Wed, 16 Apr 2025 11:29:40 -0400
Subject: [PATCH] fix condition to not default to claude haiku when custom
 model name is provided

---
 .../nodes/chatmodels/AWSBedrock/FlowiseAWSChatBedrock.ts        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/components/nodes/chatmodels/AWSBedrock/FlowiseAWSChatBedrock.ts b/packages/components/nodes/chatmodels/AWSBedrock/FlowiseAWSChatBedrock.ts
index 4dec41fd1f7..8a2d2cda340 100644
--- a/packages/components/nodes/chatmodels/AWSBedrock/FlowiseAWSChatBedrock.ts
+++ b/packages/components/nodes/chatmodels/AWSBedrock/FlowiseAWSChatBedrock.ts
@@ -27,7 +27,7 @@ export class BedrockChat extends LCBedrockChat implements IVisionChatModal {
     }
 
     setVisionModel(): void {
-        if (!this.model.startsWith('claude-3')) {
+        if (!this.model.includes('claude-3')) {
             this.model = DEFAULT_IMAGE_MODEL
             this.maxTokens = this.configuredMaxToken ? this.configuredMaxToken : DEFAULT_IMAGE_MAX_TOKEN
         }