ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-23 13:03:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: base64 image undefined (#1231)

Browse Source
This commit is contained in:
xiaotian
2024-04-17 22:56:52 +08:00
committed by GitHub
parent 3d046974b8
commit 78d50e157f
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
packages/service/core/chat/utils.ts
View File

@@ -6,6 +6,7 @@ import type {
} from '@fastgpt/global/core/ai/type.d'; } from '@fastgpt/global/core/ai/type.d';
import axios from 'axios'; import axios from 'axios';
import { ChatCompletionRequestMessageRoleEnum } from '@fastgpt/global/core/ai/constants'; import { ChatCompletionRequestMessageRoleEnum } from '@fastgpt/global/core/ai/constants';
import { guessBase64ImageType } from '../../common/file/utils';
/* slice chat context by tokens */ /* slice chat context by tokens */
const filterEmptyMessages = (messages: ChatCompletionMessageParam[]) => { const filterEmptyMessages = (messages: ChatCompletionMessageParam[]) => {
@@ -242,7 +243,11 @@ export const loadChatImgToBase64 = async (content: string | ChatCompletionConten
responseType: 'arraybuffer' responseType: 'arraybuffer'
}); });
const base64 = Buffer.from(response.data).toString('base64'); const base64 = Buffer.from(response.data).toString('base64');
item.image_url.url = `data:${response.headers['content-type']};base64,${base64}`; let imageType = response.headers['content-type'];
if (imageType === undefined) {
imageType = guessBase64ImageType(base64);
}
item.image_url.url = `data:${imageType};base64,${base64}`;
return item; return item;
}) })
); );