From 1412ac9889046d8d9d125a0d4a9f9093a47eb12a Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Fri, 20 Sep 2024 11:12:15 +0800
Subject: [PATCH] fix: improved api call validation

- Improved API call validation to account for additional usage properties.
---
 src/renderer/src/services/api.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/renderer/src/services/api.ts b/src/renderer/src/services/api.ts
index a7ffb0ceac..6722d19d8f 100644
--- a/src/renderer/src/services/api.ts
+++ b/src/renderer/src/services/api.ts
@@ -77,7 +77,7 @@ export async function fetchChatCompletion({
 
     message.status = 'success'
 
-    if (!message.usage) {
+    if (!message.usage || !message?.usage?.completion_tokens) {
       message.usage = await estimateMessagesUsage({
         assistant,
         messages: [..._messages, message]