fix: fix bug of metrics missing

2025-03-02 11:51:11 +08:00 · 2025-03-02 11:51:11 +08:00 · 1c60375d71
commit 1c60375d71
parent c9699609ed
1 changed files with 4 additions and 0 deletions
--- 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'