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

feat(core): convert ai think tag to markdown divider (#10459)

Browse Source 
Support issue [AF-2282](https://linear.app/affine-design/issue/AF-2282).
This commit is contained in:
akumatus
2025-02-27 04:43:11 +00:00
parent 8cec22cc64
commit f43a848e18
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -221,11 +221,12 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
message: data.detail[0].msg || 'Unexpected perplexity response',
});
} else {
const parser = new CitationParser();
const citationParser = new CitationParser();
const { content } = data.choices[0].message;
const { citations } = data;
let result = parser.parse(content, citations);
result += parser.end();
let result = content.replaceAll(/<\/?think>\n/g, '\n---\n');
result = citationParser.parse(result, citations);
result += citationParser.end();
return result;
}
} catch (e: any) {
@@ -264,7 +265,7 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
params
);
if (response.body) {
const parser = new CitationParser();
const citationParser = new CitationParser();
const provider = this.type;
const eventStream = response.body
.pipeThrough(new TextDecoderStream())
@@ -289,12 +290,13 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
}
const { content } = data.choices[0].delta;
const { citations } = data;
const result = parser.parse(content, citations);
let result = content.replaceAll(/<\/?think>\n?/g, '\n---\n');
result = citationParser.parse(result, citations);
controller.enqueue(result);
}
},
flush(controller) {
controller.enqueue(parser.end());
controller.enqueue(citationParser.end());
controller.enqueue(null);
},
})