Sfoglia il codice sorgente

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 mesi fa
parent
commit
f535a2aa71
Nessun account collegato all'indirizzo email del committer

+ 3
- 3
api/core/model_runtime/model_providers/__base/large_language_model.py Vedi File

:return: result generator :return: result generator
""" """
callbacks = callbacks or [] callbacks = callbacks or []
prompt_message = AssistantPromptMessage(content="")
assistant_message = AssistantPromptMessage(content="")
usage = None usage = None
system_fingerprint = None system_fingerprint = None
real_model = model real_model = model
callbacks=callbacks, callbacks=callbacks,
) )


prompt_message.content += chunk.delta.message.content
assistant_message.content += chunk.delta.message.content
real_model = chunk.model real_model = chunk.model
if chunk.delta.usage: if chunk.delta.usage:
usage = chunk.delta.usage usage = chunk.delta.usage
result=LLMResult( result=LLMResult(
model=real_model, model=real_model,
prompt_messages=prompt_messages, prompt_messages=prompt_messages,
message=prompt_message,
message=assistant_message,
usage=usage or LLMUsage.empty_usage(), usage=usage or LLMUsage.empty_usage(),
system_fingerprint=system_fingerprint, system_fingerprint=system_fingerprint,
), ),

Loading…
Annulla
Salva