From 1c60375d71c92a2b572140d3c59f5c2f0d11639c Mon Sep 17 00:00:00 2001
From: ZhuangYumin <zhuangyumin@sjtu.edu.cn>
Date: Sun, 2 Mar 2025 11:51:11 +0800
Subject: [PATCH] fix: fix bug of metrics missing

---
 src/renderer/src/services/ApiService.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/renderer/src/services/ApiService.ts b/src/renderer/src/services/ApiService.ts
index 741be5a2..e0473d29 100644
--- a/src/renderer/src/services/ApiService.ts
+++ b/src/renderer/src/services/ApiService.ts
@@ -114,6 +114,10 @@ export async function fetchChatCompletion({
         assistant,
         messages: [..._messages, message]
       })
+      // 设置 metrics.completion_tokens
+      if (message.metrics && message.usage.completion_tokens && !message.metrics.completion_tokens) {
+        message.metrics.completion_tokens = message.usage.completion_tokens
+      }
     }
   } catch (error: any) {
     message.status = 'error'