fix prompt log

This commit is contained in:
JzoNg 2024-04-02 12:34:40 +08:00
parent cf22842554
commit fd8ed95209
3 changed files with 27 additions and 15 deletions

View File

@ -96,11 +96,15 @@ const getFormattedChatList = (messages: ChatMessage[]) => {
message_files: item.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
log: [
...item.message,
{
role: 'assistant',
text: item.answer,
files: item.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
},
...(item.message[item.message.length - 1].role !== 'assistant'
? [
{
role: 'assistant',
text: item.answer,
files: item.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
},
]
: []),
],
workflow_run_id: item.workflow_run_id,
more: {

View File

@ -200,11 +200,15 @@ const GenerationItem: FC<IGenerationItemProps> = ({
...data,
log: [
...data.message,
{
role: 'assistant',
text: data.answer,
files: data.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
},
...(data.message[data.message.length - 1].role !== 'assistant'
? [
{
role: 'assistant',
text: data.answer,
files: data.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
},
]
: []),
],
}
setCurrentLogItem(logItem)

View File

@ -324,11 +324,15 @@ export const useChat = (
...draft[index],
log: [
...newResponseItem.message,
{
role: 'assistant',
text: newResponseItem.answer,
files: newResponseItem.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
},
...(newResponseItem.message[newResponseItem.message.length - 1].role !== 'assistant'
? [
{
role: 'assistant',
text: newResponseItem.answer,
files: newResponseItem.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
},
]
: []),
],
more: {
time: dayjs.unix(newResponseItem.created_at).format('hh:mm A'),