MigrateRepos/glm-free-api
1
0
Fork 0
mirror of https://github.com/LLM-Red-Team/glm-free-api.git synced 2024-12-22 17:09:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化非流式搜索结果展示

Browse Source
This commit is contained in:
Vinlic 2024-05-06 11:19:13 +08:00
parent cb044beae0
commit 53f3365872
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/api/controllers/chat.ts
View File

@ -731,6 +731,7 @@ async function receiveStream(stream: any): Promise<any> {
let codeTemp = "";
let lastExecutionOutput = "";
let textOffset = 0;
let refContent = '';
const parser = createParser((event) => {
try {
if (event.type !== "event") return;
@ -772,14 +773,9 @@ async function receiveStream(stream: any): Promise<any> {
meta_data &&
_.isArray(meta_data.metadata_list)
) {
const searchText =
meta_data.metadata_list.reduce(
(meta, v) => meta + `检索 ${v.title}(${v.url}) ...`,
""
) + "\n";
textOffset += searchText.length;
toolCall = true;
return innerStr + searchText;
refContent = meta_data.metadata_list.reduce((meta, v) => {
return meta + `${v.title} - ${v.url}\n`;
}, refContent);
} else if (
type == "image" &&
_.isArray(image) &&
@ -839,7 +835,7 @@ async function receiveStream(stream: any): Promise<any> {
data.choices[0].message.content += chunk;
} else {
data.choices[0].message.content =
data.choices[0].message.content.replace(/【\d+†来源】/g, "");
data.choices[0].message.content.replace(/【\d+†(来源|source)】/g, "") + (refContent ? `\n\n搜索结果来自：\n${refContent.replace(/\n$/, '')}` : '');
resolve(data);
}
} catch (err) {