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-04-26 16:19:57 +08:00
parent c70a4a4102
commit 600a42938c
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -486,10 +486,6 @@ function extractRefFileUrls(messages: any[]) {
* 消息预处理
*
* 由于接口只取第一条消息，此处会将多条消息合并为一条，实现多轮对话效果
* 使用”你“这个角色回复”我“这个角色，以第一人称对话\n
* 我:旧消息1
* 你:旧消息2
* 我:新消息
*
* @param messages 参考gpt系列消息格式，多轮对话请完整提供上下文
*/
@ -520,20 +516,25 @@ function messagesPrepare(messages: any[], refs: any[]) {
const content = (
messages.reduce((content, message) => {
const role = message.role
.replace("system", "<|sytstem|>")
.replace("assistant", "<|assistant|>")
.replace("user", "<|user|>");
if (_.isArray(message.content)) {
return (
message.content.reduce((_content, v) => {
if (!_.isObject(v) || v["type"] != "text") return _content;
return _content + ("<|user|>\n" + v["text"] || "") + "\n";
return _content + (`${role}\n` + v["text"] || "") + "\n";
}, content)
);
}
return (content += `${message.role
.replace("system", "<|sytstem|>")
.replace("assistant", "<|assistant|>")
.replace("user", "<|user|>")}\n${message.content}\n`);
return (content += `${role}\n${message.content}\n`);
}, "") + "<|assistant|>\n"
).replace(/\!\[.+\]\(.+\)/g, "");
)
// 移除MD图像URL避免幻觉
.replace(/\!\[.+\]\(.+\)/g, "")
// 移除临时路径避免在新会话引发幻觉
.replace(/\/mnt\/data\/.+/g, "");
const fileRefs = refs.filter((ref) => !ref.width && !ref.height);
const imageRefs = refs
.filter((ref) => ref.width || ref.height)
@ -541,6 +542,7 @@ function messagesPrepare(messages: any[], refs: any[]) {
ref.image_url = ref.file_url;
return ref;
});
content
logger.info("\n对话合并：\n" + content);
return [
{