chore: update default model to sonnet 3.7 for internal user (#376)

* update default model

* update model

* add is_amzn_user method

* add fra check

Author: evanliu048

crates/chat-cli/src/auth/builder_id.rs

@@ -458,6 +458,12 @@ impl BuilderIdToken {
             Some(_) => TokenType::IamIdentityCenter,
         }
     }
+
+    /// Check if the token is for the internal amzn start URL (`https://amzn.awsapps.com/start`),
+    /// this implies the user will use midway for private specs
+    pub fn is_amzn_user(&self) -> bool {
+        matches!(&self.start_url, Some(url) if url == AMZN_START_URL)
+    }
 }
 
 pub enum PollCreateToken {

crates/chat-cli/src/auth/consts.rs

@@ -21,5 +21,8 @@ pub(crate) const CLIENT_TYPE: &str = "public";
 // The start URL for public builder ID users
 pub const START_URL: &str = "https://view.awsapps.com/start";
 
+// The start URL for internal amzn users
+pub const AMZN_START_URL: &str = "https://amzn.awsapps.com/start";
+
 pub(crate) const DEVICE_GRANT_TYPE: &str = "urn:ietf:params:oauth:grant-type:device_code";
 pub(crate) const REFRESH_GRANT_TYPE: &str = "refresh_token";

crates/chat-cli/src/cli/chat/cli/model.rs

@@ -9,11 +9,16 @@ use crossterm::{
 };
 use dialoguer::Select;
 
+use crate::auth::builder_id::{
+    BuilderIdToken,
+    TokenType,
+};
 use crate::cli::chat::{
     ChatError,
     ChatSession,
     ChatState,
 };
+use crate::os::Os;
 
 pub struct ModelOption {
     pub name: &'static str,
@@ -97,7 +102,23 @@ impl ModelArgs {
     }
 }
 
-/// Currently, Sonnet 4 is set as the default model for non-FRA users.
-pub fn default_model_id() -> &'static str {
-    "CLAUDE_3_7_SONNET_20250219_V1_0"
+/// Returns Claude 3.7 for: Amazon IDC users, FRA region users
+/// Returns Claude 4.0 for: Builder ID users, other regions
+pub async fn default_model_id(os: &Os) -> &'static str {
+    // Check FRA region first
+    if let Ok(Some(profile)) = os.database.get_auth_profile() {
+        if profile.arn.split(':').nth(3) == Some("eu-central-1") {
+            return "CLAUDE_3_7_SONNET_20250219_V1_0";
+        }
+    }
+
+    // Check if Amazon IDC user
+    if let Ok(Some(token)) = BuilderIdToken::load(&os.database).await {
+        if matches!(token.token_type(), TokenType::IamIdentityCenter) && token.is_amzn_user() {
+            return "CLAUDE_3_7_SONNET_20250219_V1_0";
+        }
+    }
+
+    // Default to 4.0
+    "CLAUDE_SONNET_4_20250514_V1_0"
 }

crates/chat-cli/src/cli/chat/mod.rs

@@ -524,19 +524,26 @@ impl ChatSession {
         tool_permissions: ToolPermissions,
         interactive: bool,
     ) -> Result<Self> {
-        let valid_model_id = model_id
-            .or_else(|| {
-                os.database
+        let valid_model_id = match model_id {
+            Some(id) => id,
+            None => {
+                let from_settings = os
+                    .database
                     .settings
                     .get_string(Setting::ChatDefaultModel)
                     .and_then(|model_name| {
                         MODEL_OPTIONS
                             .iter()
                             .find(|opt| opt.name == model_name)
                             .map(|opt| opt.model_id.to_owned())
-                    })
-            })
-            .unwrap_or_else(|| default_model_id().to_owned());
+                    });
+
+                match from_settings {
+                    Some(id) => id,
+                    None => default_model_id(os).await.to_owned(),
+                }
+            },
+        };
 
         // Reload prior conversation
         let mut existing_conversation = false;