Fix formatting changes

Author: bontalaa

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

@@ -667,8 +667,17 @@ impl From<amzn_codewhisperer_streaming_client::types::ChatResponseStream> for Ch
                 cache_write_input_tokens: token_usage.as_ref().and_then(|t| t.cache_write_input_tokens),
             },
             amzn_codewhisperer_streaming_client::types::ChatResponseStream::MeteringEvent(
-                amzn_codewhisperer_streaming_client::types::MeteringEvent { usage, unit, unit_plural, .. },
-            ) => ChatResponseStream::MeteringEvent { usage, unit, unit_plural },
+                amzn_codewhisperer_streaming_client::types::MeteringEvent {
+                    usage,
+                    unit,
+                    unit_plural,
+                    ..
+                },
+            ) => ChatResponseStream::MeteringEvent {
+                usage,
+                unit,
+                unit_plural,
+            },
             amzn_codewhisperer_streaming_client::types::ChatResponseStream::ToolUseEvent(
                 amzn_codewhisperer_streaming_client::types::ToolUseEvent {
                     tool_use_id,
@@ -735,8 +744,17 @@ impl From<amzn_qdeveloper_streaming_client::types::ChatResponseStream> for ChatR
                 cache_write_input_tokens: token_usage.as_ref().and_then(|t| t.cache_write_input_tokens),
             },
             amzn_qdeveloper_streaming_client::types::ChatResponseStream::MeteringEvent(
-                amzn_qdeveloper_streaming_client::types::MeteringEvent { usage, unit, unit_plural, .. },
-            ) => ChatResponseStream::MeteringEvent { usage, unit, unit_plural },
+                amzn_qdeveloper_streaming_client::types::MeteringEvent {
+                    usage,
+                    unit,
+                    unit_plural,
+                    ..
+                },
+            ) => ChatResponseStream::MeteringEvent {
+                usage,
+                unit,
+                unit_plural,
+            },
             amzn_qdeveloper_streaming_client::types::ChatResponseStream::ToolUseEvent(
                 amzn_qdeveloper_streaming_client::types::ToolUseEvent {
                     tool_use_id,

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

@@ -163,7 +163,11 @@ impl UserTurnMetadata {
     }
 
     pub fn add_usage(&mut self, value: f64, unit: String, unit_plural: String) {
-        self.usage_info.push(MeteringUsageInfo { value, unit, unit_plural });
+        self.usage_info.push(MeteringUsageInfo {
+            value,
+            unit,
+            unit_plural,
+        });
     }
 
     pub fn total_usage(&self) -> Vec<MeteringUsageInfo> {

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

@@ -15,8 +15,8 @@ pub mod context;
 mod conversation;
 mod input_source;
 mod message;
-mod turn_summary;
 mod parse;
+mod turn_summary;
 use std::path::MAIN_SEPARATOR;
 pub mod checkpoint;
 mod line_tracker;
@@ -829,7 +829,7 @@ impl ChatSession {
                 if self.tool_uses.is_empty() {
                     turn_summary::display_turn_usage_summary(&mut self.stderr, &self.conversation.user_turn_metadata)?;
                 }
-                
+
                 match (self.interactive, self.tool_uses.is_empty()) {
                     (false, true) => {
                         self.inner = Some(ChatState::Exit);
@@ -2813,8 +2813,14 @@ impl ChatSession {
                     trace!("Consumed: {:?}", msg_event);
 
                     match msg_event {
-                        parser::ResponseEvent::MeteringUsage { value, unit, unit_plural } => {
-                            self.conversation.user_turn_metadata.add_usage(value, unit.clone(), unit_plural.clone());
+                        parser::ResponseEvent::MeteringUsage {
+                            value,
+                            unit,
+                            unit_plural,
+                        } => {
+                            self.conversation
+                                .user_turn_metadata
+                                .add_usage(value, unit.clone(), unit_plural.clone());
                         },
                         parser::ResponseEvent::ToolUseStart { name } => {
                             // We need to flush the buffer here, otherwise text will not be

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

@@ -622,18 +622,25 @@ impl ResponseParser {
                         ChatResponseStream::ToolUseEvent { input, .. } => {
                             self.received_response_size += input.as_ref().map(String::len).unwrap_or_default();
                         },
-                        ChatResponseStream::MeteringEvent { usage, unit, unit_plural } => {
+                        ChatResponseStream::MeteringEvent {
+                            usage,
+                            unit,
+                            unit_plural,
+                        } => {
                             info!("GenerateAssistanceResponse - MeteringEvent");
                             if let (Some(value), Some(unit), Some(unit_plural)) = (usage, unit, unit_plural) {
-                                let _ = self.event_tx.send(Ok(ResponseEvent::MeteringUsage { 
-                                    value: *value, 
-                                    unit: unit.clone(), 
-                                    unit_plural: unit_plural.clone() 
-                                })).await;
+                                let _ = self
+                                    .event_tx
+                                    .send(Ok(ResponseEvent::MeteringUsage {
+                                        value: *value,
+                                        unit: unit.clone(),
+                                        unit_plural: unit_plural.clone(),
+                                    }))
+                                    .await;
                             }
                         },
                         _ => {
-                                warn!(?r, "received unexpected event from the response stream");
+                            warn!(?r, "received unexpected event from the response stream");
                         },
                     }
                 }
@@ -702,7 +709,11 @@ pub enum ResponseEvent {
         request_metadata: RequestMetadata,
     },
     /// Metering usage consumed for this request
-    MeteringUsage { value: f64, unit: String, unit_plural: String },
+    MeteringUsage {
+        value: f64,
+        unit: String,
+        unit_plural: String,
+    },
 }
 
 /// Metadata about the sent request and associated response stream.

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

@@ -1,6 +1,10 @@
-use crossterm::{execute, style};
 use std::io::Write;
 
+use crossterm::{
+    execute,
+    style,
+};
+
 use crate::cli::chat::conversation::UserTurnMetadata;
 
 pub fn format_number(value: f64) -> String {
@@ -15,7 +19,7 @@ pub fn capitalize_first(s: &str) -> String {
             let mut result = first.to_uppercase().collect::<String>();
             result.push_str(chars.as_str());
             result
-        }
+        },
     }
 }
 
@@ -36,19 +40,19 @@ pub fn display_turn_usage_summary<W: Write>(
 ) -> Result<(), std::io::Error> {
     let totals = user_turn_metadata.total_usage();
     let mut parts = Vec::new();
-    
+
     // Add usage info
     for usage in totals {
         let formatted = format_number(usage.value);
         let capitalized_unit = capitalize_first(&usage.unit_plural);
         parts.push(format!("{capitalized_unit} used: {formatted}"));
     }
-    
+
     // Add elapsed time
     if let Some(elapsed_ms) = user_turn_metadata.total_elapsed_time_ms() {
         parts.push(format!("Elapsed time: {}", format_elapsed_time(elapsed_ms)));
     }
-    
+
     if !parts.is_empty() {
         execute!(
             stderr,