From 09a1d51e9a1ab597c81a7ddd5022055cb124b490 Mon Sep 17 00:00:00 2001
From: Shardul Vaidya <31039336+5herlocked@users.noreply.github.com>
Date: Wed, 4 Jun 2025 01:57:31 -0400
Subject: [PATCH] bedrock: Fix Claude 4 output token bug (#31599)

Release Notes:

- Fixed an issue preventing the use of Claude 4 Thinking models with Bedrock
---
 crates/bedrock/src/models.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/crates/bedrock/src/models.rs b/crates/bedrock/src/models.rs
index e4c786cbc0..3a2a6d9db5 100644
--- a/crates/bedrock/src/models.rs
+++ b/crates/bedrock/src/models.rs
@@ -253,7 +253,9 @@ impl Model {
             | Self::Claude3_5Haiku
             | Self::Claude3_7Sonnet
             | Self::ClaudeSonnet4
-            | Self::ClaudeOpus4 => 200_000,
+            | Self::ClaudeOpus4
+            | Self::ClaudeSonnet4Thinking
+            | Self::ClaudeOpus4Thinking => 200_000,
             Self::AmazonNovaPremier => 1_000_000,
             Self::PalmyraWriterX5 => 1_000_000,
             Self::PalmyraWriterX4 => 128_000,