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

feat: use footnote for perplexity search results (#9851)

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

![截屏2025-01-22 16.49.25.png](https://graphite-user-uploaded-assets-prod.s3.amazonaws.com/sJGviKxfE3Ap685cl5bj/49febcdf-e403-4a2e-ba99-da36df34e08c.png)
This commit is contained in:
akumatus
2025-01-22 10:54:00 +00:00
parent f8a515e89a
commit 862a9d0bc4
5 changed files with 69 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
packages/backend/server/src/plugins/copilot/providers/perplexity.ts
View File

@@ -63,7 +63,10 @@ export class CitationParser {
private numberToken: string[] = [];
private citations: string[] = [];
public parse(content: string, citations: string[]) {
this.citations = citations;
let result = '';
const contentArray = content.split('');
for (const [index, char] of contentArray.entries()) {
@@ -85,7 +88,7 @@ export class CitationParser {
cIndex <= citations.length &&
contentArray[index + 1] !== this.PARENTHESES_OPEN
) {
const content = `[[${cIndex}](${citations[cIndex - 1]})]`;
const content = `[^${cIndex}]`;
result += content;
this.resetToken();
} else {
@@ -116,13 +119,26 @@ export class CitationParser {
return result;
}
public flush() {
const content = this.getFullContent();
public end() {
return this.flush() + this.getFootnotes();
}
private flush() {
const content = this.getTokenContent();
this.resetToken();
return content;
}
private getFullContent() {
private getFootnotes() {
const footnotes = this.citations.map((citation, index) => {
return `[^${index + 1}]: {"type":"url","url":"${encodeURIComponent(
citation
)}"}`;
});
return '\n\n' + footnotes.join('\n\n');
}
private getTokenContent() {
return this.startToken.concat(this.numberToken, this.endToken).join('');
}
@@ -208,7 +224,7 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
const { content } = data.choices[0].message;
const { citations } = data;
let result = parser.parse(content, citations);
result += parser.flush();
result += parser.end();
return result;
}
} catch (e: any) {
@@ -277,7 +293,7 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
}
},
flush(controller) {
controller.enqueue(parser.flush());
controller.enqueue(parser.end());
controller.enqueue(null);
},
})