fix: llm result usage none (#1940)

2024-01-05 12:47:10 +08:00
parent 296bf443a8
commit a0f74cdd9d
1 changed files with 1 additions and 1 deletions
--- 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,