Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-13 12:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): prompt message tunning (#6747)

Browse Source 
fix AFF-1003
This commit is contained in:
pengx17
2024-04-30 05:10:00 +00:00
committed by Peng Xiao
parent 2420b2849f
commit eb7904bf62
3 changed files with 190 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
packages/backend/server/src/plugins/copilot/session.ts
View File

@@ -110,9 +110,30 @@ export class ChatSession implements AsyncDisposable {
finish(params: PromptParams): PromptMessage[] {
const messages = this.takeMessages();
const firstMessage = messages.at(0);
// if the message in prompt config contains {{content}},
// we should combine it with the user message in the prompt
if (
messages.length === 1 &&
firstMessage?.content &&
this.state.prompt.paramKeys.includes('content')
) {
const normalizedParams = {
...params,
...firstMessage.params,
content: firstMessage.content,
};
const finished = this.state.prompt.finish(
normalizedParams,
this.config.sessionId
);
finished[0].attachments = firstMessage.attachments;
return finished;
}
return [
...this.state.prompt.finish(
Object.keys(params).length ? params : messages[0]?.params || {},
Object.keys(params).length ? params : firstMessage?.params || {},
this.config.sessionId
),
...messages.filter(m => m.content?.trim() || m.attachments?.length),