fix: llm result usage none (#1940)

a0f74cdd · takatost · GitHub · 296bf443 · a0f74cdd
Unverified Commit a0f74cdd authored Jan 05, 2024 by takatost Committed by GitHub Jan 05, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

large_language_model.py ...el_runtime/model_providers/__base/large_language_model.py +1 -1

No files found.
--- a/api/core/model_runtime/model_providers/__base/large_language_model.py
+++ b/api/core/model_runtime/model_providers/__base/large_language_model.py
@@ -165,7 +165,7 @@ class LargeLanguageModel(AIModel):
                model=real_model,
                prompt_messages=prompt_messages,
                message=prompt_message,
-                usage=usage,
+                usage=usage if usage else LLMUsage.empty_usage(),
                system_fingerprint=system_fingerprint
            ),
            credentials=credentials,