Parcourir la source

fix: llm result usage none (#1940)

takatost il y a 1 an
Parent
commit
a0f74cdd9d

+ 1 - 1
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,