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

图像解析支持

Browse Source
This commit is contained in:
Vinlic 2024-03-17 18:18:11 +08:00
parent 9868897d07
commit cbb14b0198
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -316,14 +316,23 @@ function messagesPrepare(messages: any[], refs: any[]) {
const role = message.role.replace('user', '我').replace('assistant', '你') || '我';
return content += `${role}:${message.content}\n你:`;
}, '');
const fileRefs = refs.filter(ref => !ref.width && !ref.height);
const imageRefs = refs.filter(ref => ref.width || ref.height).map(ref => {
ref.image_url = ref.file_url;
return ref;
});
return [
{
role: 'user',
content: [
{ type: 'text', text: headPrompt + content },
...(refs.length == 0 ? [] : [{
...(fileRefs.length == 0 ? [] : [{
type: 'file',
file: refs
file: fileRefs
}]),
...(imageRefs.length == 0 ? [] : [{
type: 'image',
image: imageRefs
}])
]
}