瀏覽代碼

chore: prompt_message is actually assistant_message which is a bit am… (#13839)

Co-authored-by: 刘江波 <jiangbo721@163.com>
tags/1.0.0
jiangbo721 8 月之前
父節點
當前提交
f535a2aa71
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 3 行新增3 行删除
  1. 3
    3
      api/core/model_runtime/model_providers/__base/large_language_model.py

+ 3
- 3
api/core/model_runtime/model_providers/__base/large_language_model.py 查看文件

@@ -228,7 +228,7 @@ class LargeLanguageModel(AIModel):
:return: result generator
"""
callbacks = callbacks or []
prompt_message = AssistantPromptMessage(content="")
assistant_message = AssistantPromptMessage(content="")
usage = None
system_fingerprint = None
real_model = model
@@ -250,7 +250,7 @@ class LargeLanguageModel(AIModel):
callbacks=callbacks,
)

prompt_message.content += chunk.delta.message.content
assistant_message.content += chunk.delta.message.content
real_model = chunk.model
if chunk.delta.usage:
usage = chunk.delta.usage
@@ -265,7 +265,7 @@ class LargeLanguageModel(AIModel):
result=LLMResult(
model=real_model,
prompt_messages=prompt_messages,
message=prompt_message,
message=assistant_message,
usage=usage or LLMUsage.empty_usage(),
system_fingerprint=system_fingerprint,
),

Loading…
取消
儲存