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

feat(core): cite source documents in the AI answer (#9863)

Browse Source 
Support issue [BS-2424](https://linear.app/affine-design/issue/BS-2424).

### What changed?
- Add relevant document prompt templates.
- Add citation rules in system prompts.
- Change message `params` type to `Record<string, any>`
- Add unit test.

<div class='graphite__hidden'>
          <div>🎥 Video uploaded on Graphite:</div>
            <a href="https://app.graphite.dev/media/video/sJGviKxfE3Ap685cl5bj/ec24e664-039e-4fab-bd26-b3312f011daf.mov">
              <img src="https://app.graphite.dev/api/v1/graphite/video/thumbnail/sJGviKxfE3Ap685cl5bj/ec24e664-039e-4fab-bd26-b3312f011daf.mov">
            </a>
          </div>
<video src="https://graphite-user-uploaded-assets-prod.s3.amazonaws.com/sJGviKxfE3Ap685cl5bj/ec24e664-039e-4fab-bd26-b3312f011daf.mov">录屏2025-01-23 10.40.38.mov</video>
This commit is contained in:
akumatus
2025-01-24 04:04:00 +00:00
parent 48c26017ae
commit 95cf2e047f
13 changed files with 131 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
packages/backend/server/src/__tests__/copilot.spec.ts
View File

@@ -629,6 +629,57 @@ test('should revert message correctly', async t => {
}
});
test('should handle params correctly in chat session', async t => {
const { prompt, session } = t.context;
await prompt.set('prompt', 'model', [
{ role: 'system', content: 'hello {{word}}' },
]);
const sessionId = await session.create({
docId: 'test',
workspaceId: 'test',
userId,
promptName: 'prompt',
});
const s = (await session.get(sessionId))!;
// Case 1: When params is provided directly
{
const directParams = { word: 'direct' };
const messages = s.finish(directParams);
t.is(messages[0].content, 'hello direct', 'should use provided params');
}
// Case 2: When no params provided but last message has params
{
s.push({
role: 'user',
content: 'test message',
params: { word: 'fromMessage' },
createdAt: new Date(),
});
const messages = s.finish({});
t.is(
messages[0].content,
'hello fromMessage',
'should use params from last message'
);
}
// Case 3: When neither params provided nor last message has params
{
s.push({
role: 'user',
content: 'test message without params',
createdAt: new Date(),
});
const messages = s.finish({});
t.is(messages[0].content, 'hello ', 'should use empty params');
}
});
// ==================== provider ====================
test('should be able to get provider', async t => {
@@ -1102,7 +1153,7 @@ test('CitationParser should replace citation placeholders with URLs', t => {
'This is [a] test sentence with [citations [^1]] and [^2] and [3].',
`[^1]: {"type":"url","url":"${encodeURIComponent(citations[0])}"}`,
`[^2]: {"type":"url","url":"${encodeURIComponent(citations[1])}"}`,
].join('\n\n');
].join('\n');
t.is(result, expected);
});
@@ -1145,7 +1196,7 @@ test('CitationParser should replace chunks of citation placeholders with URLs',
`[^5]: {"type":"url","url":"${encodeURIComponent(citations[4])}"}`,
`[^6]: {"type":"url","url":"${encodeURIComponent(citations[5])}"}`,
`[^7]: {"type":"url","url":"${encodeURIComponent(citations[6])}"}`,
].join('\n\n');
].join('\n');
t.is(result, expected);
});
@@ -1166,7 +1217,7 @@ test('CitationParser should not replace citation already with URLs', t => {
`[^1]: {"type":"url","url":"${encodeURIComponent(citations[0])}"}`,
`[^2]: {"type":"url","url":"${encodeURIComponent(citations[1])}"}`,
`[^3]: {"type":"url","url":"${encodeURIComponent(citations[2])}"}`,
].join('\n\n');
].join('\n');
t.is(result, expected);
});
@@ -1193,6 +1244,6 @@ test('CitationParser should not replace chunks of citation already with URLs', t
`[^1]: {"type":"url","url":"${encodeURIComponent(citations[0])}"}`,
`[^2]: {"type":"url","url":"${encodeURIComponent(citations[1])}"}`,
`[^3]: {"type":"url","url":"${encodeURIComponent(citations[2])}"}`,
].join('\n\n');
].join('\n');
t.is(result, expected);
});