From c976afac31dedbbba60c508aa591ab35d72f23a6 Mon Sep 17 00:00:00 2001
From: divit <27228526+delgado3d@users.noreply.github.com>
Date: Mon, 7 Jul 2025 09:53:08 +0530
Subject: [PATCH] fix: remove apac bedrock region and add us region (#36881)

* fix: remove apac bedrock region and add us region

* fix: update test
---
 apps/studio/lib/ai/bedrock.ts    | 4 ++--
 apps/studio/lib/ai/model.test.ts | 2 +-
 apps/studio/lib/ai/model.ts      | 6 +++---
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/apps/studio/lib/ai/bedrock.ts b/apps/studio/lib/ai/bedrock.ts
index c3513fe2612..daa1619ad37 100644
--- a/apps/studio/lib/ai/bedrock.ts
+++ b/apps/studio/lib/ai/bedrock.ts
@@ -35,9 +35,9 @@ async function vercelOidcProvider() {
 }
 
 export const bedrockRegionMap = {
-  us: 'us-east-1',
+  us1: 'us-east-1',
+  us2: 'us-east-2',
   eu: 'eu-central-1',
-  apac: 'ap-southeast-1',
 } as const
 
 export type BedrockRegion = keyof typeof bedrockRegionMap
diff --git a/apps/studio/lib/ai/model.test.ts b/apps/studio/lib/ai/model.test.ts
index 20faba45a6c..7be70a0962a 100644
--- a/apps/studio/lib/ai/model.test.ts
+++ b/apps/studio/lib/ai/model.test.ts
@@ -32,7 +32,7 @@ describe('getModel', () => {
     console.log('Model:', model)
 
     expect(model).toEqual('bedrock-model')
-    expect(bedrockModule.bedrockForRegion).toHaveBeenCalledWith('us')
+    expect(bedrockModule.bedrockForRegion).toHaveBeenCalledWith('us1')
     expect(error).toBeUndefined()
   })
 
diff --git a/apps/studio/lib/ai/model.ts b/apps/studio/lib/ai/model.ts
index 3c6e8b7e1f9..cd9713f449f 100644
--- a/apps/studio/lib/ai/model.ts
+++ b/apps/studio/lib/ai/model.ts
@@ -9,9 +9,9 @@ import {
 
 export const modelsByProvider = {
   bedrock: {
-    us: 'us.anthropic.claude-3-7-sonnet-20250219-v1:0',
+    us1: 'us.anthropic.claude-3-7-sonnet-20250219-v1:0',
+    us2: 'us.anthropic.claude-3-7-sonnet-20250219-v1:0',
     eu: 'eu.anthropic.claude-3-7-sonnet-20250219-v1:0',
-    apac: 'apac.anthropic.claude-3-7-sonnet-20250219-v1:0',
   },
   openai: 'gpt-4.1-2025-04-14',
 }
@@ -43,7 +43,7 @@ export async function getModel(routingKey?: string): Promise<ModelResponse> {
 
   if (hasAwsCredentials) {
     // Select the Bedrock region based on the routing key
-    const bedrockRegion: BedrockRegion = routingKey ? await selectBedrockRegion(routingKey) : 'us'
+    const bedrockRegion: BedrockRegion = routingKey ? await selectBedrockRegion(routingKey) : 'us1'
     const bedrock = bedrockForRegion(bedrockRegion)
     const modelName = modelsByProvider.bedrock[bedrockRegion]