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

try fix 上下文上传文件聊天不连贯问题

Browse Source
This commit is contained in:
Yanyutin753 2024-04-02 15:21:03 +08:00
parent 1af80ca83a
commit f453c075e2
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -495,6 +495,19 @@ function messagesPrepare(messages: any[], refs: any[]) {
return !message.content.some(v => (typeof v === 'object' && ['file', 'image_url'].includes(v['type']))); return !message.content.some(v => (typeof v === 'object' && ['file', 'image_url'].includes(v['type'])));
}); });
// 检查最新消息是否含有"type": "image_url"或"type": "file",如果有则注入消息
let latestMessage = validMessages[validMessages.length - 1];
let hasFileOrImage = Array.isArray(latestMessage.content)
&& latestMessage.content.some(v => (typeof v === 'object' && ['file', 'image_url'].includes(v['type'])));
if (hasFileOrImage) {
// newMessage是一个用于指示文件的消息，是的模型能正常回答出相应的问题
let newMessage = {
"content": "持续关注用户最新发送的文件和消息的结尾",
"role": "system"
};
validMessages.splice(validMessages.length - 1, 0, newMessage);
}
const content = const content =
validMessages.reduce((content, message) => { validMessages.reduce((content, message) => {
if (_.isArray(message.content)) { if (_.isArray(message.content)) {