chore: improve conversation switching message (#1791)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>

Author: tusharmath

crates/forge_main/src/cli.rs

@@ -96,6 +96,10 @@ pub enum TopLevelCommand {
 
     /// Show current configuration, active model, and environment status
     Info {
+        /// Optional conversation ID to show info for a specific session
+        #[arg(long, alias = "cid")]
+        conversation_id: Option<String>,
+
         /// Output in machine-readable format (porcelain)
         #[arg(long)]
         porcelain: bool,
@@ -679,7 +683,7 @@ mod tests {
     fn test_info_command_without_porcelain() {
         let fixture = Cli::parse_from(["forge", "info"]);
         let actual = match fixture.subcommands {
-            Some(TopLevelCommand::Info { porcelain }) => porcelain,
+            Some(TopLevelCommand::Info { porcelain, .. }) => porcelain,
             _ => true,
         };
         let expected = false;
@@ -690,13 +694,48 @@ mod tests {
     fn test_info_command_with_porcelain() {
         let fixture = Cli::parse_from(["forge", "info", "--porcelain"]);
         let actual = match fixture.subcommands {
-            Some(TopLevelCommand::Info { porcelain }) => porcelain,
+            Some(TopLevelCommand::Info { porcelain, .. }) => porcelain,
             _ => false,
         };
         let expected = true;
         assert_eq!(actual, expected);
     }
 
+    #[test]
+    fn test_info_command_with_conversation_id() {
+        let fixture = Cli::parse_from(["forge", "info", "--conversation-id", "abc123"]);
+        let actual = match fixture.subcommands {
+            Some(TopLevelCommand::Info { conversation_id, .. }) => conversation_id,
+            _ => None,
+        };
+        let expected = Some("abc123".to_string());
+        assert_eq!(actual, expected);
+    }
+
+    #[test]
+    fn test_info_command_with_cid_alias() {
+        let fixture = Cli::parse_from(["forge", "info", "--cid", "xyz789"]);
+        let actual = match fixture.subcommands {
+            Some(TopLevelCommand::Info { conversation_id, .. }) => conversation_id,
+            _ => None,
+        };
+        let expected = Some("xyz789".to_string());
+        assert_eq!(actual, expected);
+    }
+
+    #[test]
+    fn test_info_command_with_conversation_id_and_porcelain() {
+        let fixture = Cli::parse_from(["forge", "info", "--cid", "test123", "--porcelain"]);
+        let (conversation_id, porcelain) = match fixture.subcommands {
+            Some(TopLevelCommand::Info { conversation_id, porcelain }) => {
+                (conversation_id, porcelain)
+            }
+            _ => (None, false),
+        };
+        assert_eq!(conversation_id, Some("test123".to_string()));
+        assert_eq!(porcelain, true);
+    }
+
     #[test]
     fn test_list_agents_without_porcelain() {
         let fixture = Cli::parse_from(["forge", "list", "agents"]);

crates/forge_main/src/info.rs

@@ -208,15 +208,10 @@ impl fmt::Display for Info {
                 Section::Items(key, value) => {
                     if let Some(key) = key {
                         if let Some(width) = width {
-                            writeln!(
-                                f,
-                                "  {} {}",
-                                format!("{key:<width$}:").yellow().bold(),
-                                value
-                            )?;
+                            writeln!(f, "  {} {}", format!("{key:<width$}:").cyan().bold(), value)?;
                         } else {
                             // No section width (items without a title)
-                            writeln!(f, "  {}: {}", key.yellow().bold(), value)?;
+                            writeln!(f, "  {}: {}", key.cyan().bold(), value)?;
                         }
                     } else {
                         // Show value-only items

crates/forge_main/src/ui.rs

@@ -359,11 +359,16 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
                     self.writeln_title(TitleFormat::info("MCP reloaded"))?;
                 }
             },
-            TopLevelCommand::Info { porcelain } => {
+            TopLevelCommand::Info { porcelain, conversation_id } => {
                 // Make sure to init model
                 self.on_new().await?;
 
-                self.on_info(porcelain, None).await?;
+                let conversation_id = conversation_id
+                    .as_deref()
+                    .map(ConversationId::parse)
+                    .transpose()?;
+
+                self.on_info(porcelain, conversation_id).await?;
                 return Ok(());
             }
             TopLevelCommand::Banner => {
@@ -463,8 +468,7 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
 
                 self.validate_session_exists(&conversation_id).await?;
 
-                self.on_info(session_group.porcelain, Some(conversation_id))
-                    .await?;
+                self.on_show_conv_info(conversation_id).await?;
             }
         }
 
@@ -1550,7 +1554,7 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
             }
             ChatResponse::TaskComplete => {
                 if let Some(conversation_id) = self.state.conversation_id {
-                    self.on_completion(conversation_id).await?;
+                    self.on_show_conv_info(conversation_id).await?;
                 }
             }
         }
@@ -1570,7 +1574,7 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
         Ok(())
     }
 
-    async fn on_completion(&mut self, conversation_id: ConversationId) -> anyhow::Result<()> {
+    async fn on_show_conv_info(&mut self, conversation_id: ConversationId) -> anyhow::Result<()> {
         if !should_show_completion_prompt() {
             return Ok(());
         }

shell-plugin/forge.plugin.zsh

@@ -195,7 +195,7 @@ function _forge_action_new() {
 function _forge_action_info() {
     echo
     if [[ -n "$FORGE_CONVERSATION_ID" ]]; then
-        _forge_exec session info "$FORGE_CONVERSATION_ID"
+        _forge_exec info --cid "$FORGE_CONVERSATION_ID"
     else
         _forge_exec info
     fi
@@ -256,18 +256,16 @@ function _forge_action_conversation() {
 
             # Set the selected conversation as active (in parent shell)
             FORGE_CONVERSATION_ID="$conversation_id"
-
-            # Show conversation info
+            # Show conversation content
             echo
-            _forge_exec session info "$conversation_id"       
-
+            _forge_exec session show "$conversation_id"
+            
+            # Show conversation info
+            _forge_exec session info "$conversation_id"
+            
             # Print log about conversation switching
-            echo
             echo "\033[36m⏺\033[0m \033[90m[$(date '+%H:%M:%S')] Switched to conversation \033[1m${conversation_id}\033[0m"
-
-             # Show conversation content
-            echo
-            _forge_exec session show "$conversation_id"     
+            
         fi
     else
         echo "\033[31m✗\033[0m No conversations found"