remove model id mapping in client

Author: evanliu048

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

@@ -57,11 +57,10 @@ pub async fn select_model(os: &mut Os, session: &mut ChatSession) -> Result<Opti
     let labels: Vec<String> = models
         .iter()
         .map(|model| {
-            let display_name = get_display_name(model.model_id());
             if Some(model.model_id()) == active_model_id {
-                format!("{} (active)", display_name)
+                format!("{} (active)", model.model_id())
             } else {
-                display_name.to_owned()
+                model.model_id().to_owned()
             }
         })
         .collect();
@@ -89,13 +88,12 @@ pub async fn select_model(os: &mut Os, session: &mut ChatSession) -> Result<Opti
     if let Some(index) = selection {
         let selected = &models[index];
         let model_id_str = selected.model_id.clone();
-        session.conversation.model = Some(model_id_str);
-        let display_name = get_display_name(selected.model_id());
+        session.conversation.model = Some(model_id_str.clone());
 
         queue!(
             session.stderr,
             style::Print("\n"),
-            style::Print(format!(" Using {}\n\n", display_name)),
+            style::Print(format!(" Using {}\n\n", model_id_str)),
             style::ResetColor,
             style::SetForegroundColor(Color::Reset),
             style::SetBackgroundColor(Color::Reset),
@@ -129,15 +127,3 @@ pub async fn default_model_id(os: &Os) -> String {
     // Default to 4.0
     "CLAUDE_SONNET_4_20250514_V1_0".to_string()
 }
-
-pub fn get_display_name(model_id: &str) -> &str {
-    match model_id {
-        "CLAUDE_SONNET_4_20250514_V1_0" => "claude-4-sonnet",
-        "CLAUDE_3_7_SONNET_20250219_V1_0" => "claude-3.7-sonnet",
-        "CLAUDE_3_5_SONNET_20240620_V1_0" => "claude-3.5-sonnet-v1",
-        "CLAUDE_3_5_SONNET_20241022_V2_0" => "claude-3.5-sonnet-v2",
-        "CLAUDE_3_5_HAIKU_20241022_V1_0" => "claude-3.5-haiku",
-        "NOVA_PRO_V1_0" => "nova-pro",
-        _ => model_id,
-    }
-}

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

@@ -128,10 +128,7 @@ use crate::auth::AuthError;
 use crate::auth::builder_id::is_idc_user;
 use crate::cli::agent::Agents;
 use crate::cli::chat::cli::SlashCommand;
-use crate::cli::chat::cli::model::{
-    default_model_id,
-    get_display_name,
-};
+use crate::cli::chat::cli::model::default_model_id;
 use crate::cli::chat::cli::prompts::{
     GetPromptError,
     PromptsSubcommand,
@@ -298,21 +295,17 @@ impl ChatArgs {
             {
                 Some(m.model_id.clone())
             } else {
-                let available = models
-                    .iter()
-                    .map(|m| get_display_name(m.model_id()).to_string())
-                    .collect::<Vec<_>>()
-                    .join(", ");
+                let available = models.iter().map(|m| m.model_id.clone()).collect::<Vec<_>>().join(", ");
                 bail!("Model '{}' does not exist. Available models: {}", requested, available);
             }
         } else if let Some(saved) = os
             .database
             .settings
             .get_string(Setting::ChatDefaultModel)
-            .and_then(|name| {
+            .and_then(|model_id| {
                 models
                     .iter()
-                    .find(|m| get_display_name(m.model_id()) == name)
+                    .find(|m| m.model_id() == model_id)
                     .map(|m| m.model_id.clone())
             })
         {
@@ -1165,11 +1158,10 @@ impl ChatSession {
         self.stderr.flush()?;
 
         if let Some(ref id) = self.conversation.model {
-            let display_name = get_display_name(id);
             execute!(
                 self.stderr,
                 style::SetForegroundColor(Color::Cyan),
-                style::Print(format!("🤖 You are chatting with {}\n", display_name)),
+                style::Print(format!("🤖 You are chatting with {}\n", id)),
                 style::SetForegroundColor(Color::Reset),
                 style::Print("\n")
             )?;