From 6c614f0c1f433e6b2dd7e04788b49006c6a2413b Mon Sep 17 00:00:00 2001
From: Yeuoly <45712896+Yeuoly@users.noreply.github.com>
Date: Wed, 24 Jan 2024 15:34:17 +0800
Subject: [PATCH] fix: empty usage (#2168)

---
 api/core/features/assistant_cot_runner.py | 2 +-
 api/core/features/assistant_fc_runner.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/core/features/assistant_cot_runner.py b/api/core/features/assistant_cot_runner.py
index 0d6725d051..6fcc02d857 100644
--- a/api/core/features/assistant_cot_runner.py
+++ b/api/core/features/assistant_cot_runner.py
@@ -298,7 +298,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
             message=AssistantPromptMessage(
                 content=final_answer
             ),
-            usage=llm_usage['usage'],
+            usage=llm_usage['usage'] if llm_usage['usage'] else LLMUsage.empty_usage(),
             system_fingerprint=''
         ), PublishFrom.APPLICATION_MANAGER)
 
diff --git a/api/core/features/assistant_fc_runner.py b/api/core/features/assistant_fc_runner.py
index 7f18607e1f..03c4e87015 100644
--- a/api/core/features/assistant_fc_runner.py
+++ b/api/core/features/assistant_fc_runner.py
@@ -276,7 +276,7 @@ class AssistantFunctionCallApplicationRunner(BaseAssistantApplicationRunner):
             message=AssistantPromptMessage(
                 content=final_answer,
             ),
-            usage=llm_usage['usage'],
+            usage=llm_usage['usage'] if llm_usage['usage'] else LLMUsage.empty_usage(),
             system_fingerprint=''
         ), PublishFrom.APPLICATION_MANAGER)