Update store.js

修复阿里/清华模型无法使用的问题
This commit is contained in:
vastxie 2024-02-23 17:14:21 +08:00 committed by GitHub
parent 6598d27d43
commit 4de54d8fec
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -31,16 +31,8 @@ class NineStore {
let messages = []; let messages = [];
let nextNumTokensEstimate = 0; let nextNumTokensEstimate = 0;
if (systemMessage) { if (systemMessage) {
const specialModels = ['gemini-pro', 'ERNIE', 'hunyuan'];
const isSpecialModel = model && specialModels.some(specialModel => model.includes(specialModel));
if (isSpecialModel) {
messages.push({ role: 'user', content: systemMessage, name });
messages.push({ role: 'assistant', content: "好的", name });
}
else {
messages.push({ role: 'system', content: systemMessage, name }); messages.push({ role: 'system', content: systemMessage, name });
} }
}
const systemMessageOffset = messages.length; const systemMessageOffset = messages.length;
let round = 0; let round = 0;
if (model === 'gpt-4-vision-preview' && fileInfo) { if (model === 'gpt-4-vision-preview' && fileInfo) {