소스 검색

fix: llm result usage none (#1940)

takatost 1 년 전
부모
커밋
a0f74cdd9d
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      api/core/model_runtime/model_providers/__base/large_language_model.py

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