Fix example prefixes in Gemini with no system instruction #2997

This commit is contained in:
Cohee
2024-10-26 17:25:42 +03:00
parent 5b1a4fc723
commit 2c7e9729bc

View File

@@ -396,11 +396,23 @@ export function convertGooglePrompt(messages, model, useSysPrompt = false, charN
// similar story as claude // similar story as claude
if (message.name) { if (message.name) {
if (Array.isArray(message.content)) { if (userName && message.name === 'example_user') {
message.content[0].text = `${message.name}: ${message.content[0].text}`; message.name = userName;
} else {
message.content = `${message.name}: ${message.content}`;
} }
if (charName && message.name === 'example_assistant') {
message.name = charName;
}
if (Array.isArray(message.content)) {
if (!message.content[0].text.startsWith(`${message.name}: `)) {
message.content[0].text = `${message.name}: ${message.content[0].text}`;
}
} else {
if (!message.content.startsWith(`${message.name}: `)) {
message.content = `${message.name}: ${message.content}`;
}
}
delete message.name; delete message.name;
} }