ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2026-04-26 02:07:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

chat file url white list (#6053)

Browse Source 
* chat file url white list

* perf: white list

---------

Co-authored-by: archer <545436317@qq.com>
This commit is contained in:
heheer
2025-12-08 17:10:23 +08:00
committed by GitHub
parent 2ccb5b50c6
commit 5aaf123135
9 changed files with 106 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/service/core/ai/llm/utils.ts
+26 -19
View File
@@ -14,6 +14,7 @@ import { addLog } from '../../../common/system/log';
import { getImageBase64 } from '../../../common/file/image/utils';
import { getS3ChatSource } from '../../../common/s3/sources/chat';
import { isInternalAddress } from '../../../common/system/utils';
import { getErrText } from '@fastgpt/global/common/error/utils';
export const filterGPTMessageByMaxContext = async ({
messages = [],
@@ -166,26 +167,32 @@ export const loadRequestMessages = async ({
process.env.MULTIPLE_DATA_TO_BASE64 !== 'false' ||
isInternalAddress(imgUrl)
) {
const url = await (async () => {
if (item.key) {
try {
return await getS3ChatSource().createGetChatFileURL({
key: item.key,
external: false
});
} catch (error) {}
}
return imgUrl;
})();
const { completeBase64: base64 } = await getImageBase64(url);
try {
const url = await (async () => {
if (item.key) {
try {
return await getS3ChatSource().createGetChatFileURL({
key: item.key,
external: false
});
} catch (error) {}
}
return imgUrl;
})();
const { completeBase64: base64 } = await getImageBase64(url);
return {
...item,
image_url: {
...item.image_url,
url: base64
}
};
return {
...item,
image_url: {
...item.image_url,
url: base64
}
};
} catch (error) {
return Promise.reject(
`Cannot load image ${imgUrl}, because ${getErrText(error)}`
);
}
}
// 检查下这个图片是否可以被访问，如果不行的话，则过滤掉