feat: show usage on completion (tailcallhq#1448)

Co-authored-by: Tushar Mathur <tusharmath@gmail.com>

Author: manthanabc

crates/forge_main/src/info.rs

@@ -15,6 +15,7 @@ pub enum Section {
  Items(String, Option<String>),
 }
 
+#[derive(Default)]
 pub struct Info {
  sections: Vec<Section>,
 }
@@ -118,10 +119,12 @@ impl From<&Metrics> for Info {
  Some(d) => humantime::format_duration(Duration::from_secs(d.as_secs())).to_string(),
  None => "0s".to_string(),
  };
- let mut info = Info::new().add_title(format!("TASK COMPLETED [{duration}]"));
+ let mut info = Info::new().add_title(format!("TASK COMPLETED [in {duration}]"));
 
  // Add file changes section inspired by the example format
- if !metrics.files_changed.is_empty() {
+ if metrics.files_changed.is_empty() {
+            info = info.add_key("[No Changes Produced]");
+ } else {
  // First, calculate the maximum filename length for proper alignment
  let max_filename_len = metrics
  .files_changed

crates/forge_main/src/ui.rs

@@ -856,10 +856,23 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
  }
 
  async fn on_completion(&mut self, metrics: Metrics) -> anyhow::Result<()> {
- self.spinner.stop(None)?;
+ self.spinner.start(Some("Loading Summary"))?;
+
+ let mut info = Info::default();
 
  // Show summary
- self.writeln(Info::from(&metrics))?;
+        info = info.extend(Info::from(&metrics));
+
+ // Fetch Usage
+        info = info.extend(get_usage(&self.state));
+
+ if let Ok(Some(usage)) = self.api.user_usage().await {
+            info = info.extend(Info::from(&usage));
+ }
+
+ self.writeln(info)?;
+
+ self.spinner.stop(None)?;
 
  let prompt_text = "Start a new conversation?";
  let should_start_new_chat = ForgeSelect::confirm(prompt_text)