From ed96940e82f30c93d0fda8839419e0f6cb87c671 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Sun, 2 Mar 2025 19:41:06 +0800
Subject: [PATCH] fix: Improve metrics completion tokens handling

---
 src/renderer/src/services/ApiService.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/renderer/src/services/ApiService.ts b/src/renderer/src/services/ApiService.ts
index e0473d29..f00ce1b3 100644
--- a/src/renderer/src/services/ApiService.ts
+++ b/src/renderer/src/services/ApiService.ts
@@ -114,9 +114,11 @@ 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
+      // Set metrics.completion_tokens
+      if (message.metrics && message?.usage?.completion_tokens) {
+        if (!message.metrics?.completion_tokens) {
+          message.metrics.completion_tokens = message.usage.completion_tokens
+        }
       }
     }
   } catch (error: any) {