From ccddd4bdc000b2cda863bd25317caa8b76e46ba9 Mon Sep 17 00:00:00 2001
From: leo <longshuang@msn.cn>
Date: Wed, 25 Mar 2026 16:43:39 +0800
Subject: [PATCH] feature: show token usage in AI assistant result

Signed-off-by: leo <longshuang@msn.cn>
---
 src/AI/Agent.cs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/AI/Agent.cs b/src/AI/Agent.cs
index b8a08971..368c7669 100644
--- a/src/AI/Agent.cs
+++ b/src/AI/Agent.cs
@@ -47,11 +47,15 @@ namespace SourceGit.AI
                 {
                     case ChatFinishReason.Stop:
                         onUpdate?.Invoke(string.Empty);
-                        onUpdate?.Invoke("[Assistant]:");
+                        onUpdate?.Invoke("# Assistant");
                         if (completion.Content.Count > 0)
                             onUpdate?.Invoke(completion.Content[0].Text);
                         else
                             onUpdate?.Invoke("[No content was generated.]");
+
+                        onUpdate?.Invoke(string.Empty);
+                        onUpdate?.Invoke("# Token Usage");
+                        onUpdate?.Invoke($"Total: {completion.Usage.TotalTokenCount}. Input: {completion.Usage.InputTokenCount}. Output: {completion.Usage.OutputTokenCount}");
                         break;
                     case ChatFinishReason.Length:
                         throw new Exception("The response was cut off because it reached the maximum length. Consider increasing the max tokens limit.");