rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Mistral: sanitize tool call id

Browse Source
This commit is contained in:
Cohee
2024-10-06 20:07:43 +03:00
parent 2e36f45e24
commit 2a25f75fa5
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/prompt-converters.js
View File

@ -522,8 +522,18 @@ function convertMistralMessages(messages, charName = '', userName = '') {
lastMsg.prefix = true; lastMsg.prefix = true;
} }
const sanitizeToolId = (id) => id.replace(/[^a-zA-Z0-9]/g, '').slice(0, 9);
// Doesn't support completion names, so prepend if not already done by the frontend (e.g. for group chats). // Doesn't support completion names, so prepend if not already done by the frontend (e.g. for group chats).
messages.forEach(msg => { messages.forEach(msg => {
if ('tool_calls' in msg && Array.isArray(msg.tool_calls)) {
msg.tool_calls.forEach(tool => {
tool.id = sanitizeToolId(tool.id);
});
}
if ('tool_call_id' in msg && msg.role === 'tool') {
msg.tool_call_id = sanitizeToolId(msg.tool_call_id);
}
if (msg.role === 'system' && msg.name === 'example_assistant') { if (msg.role === 'system' && msg.name === 'example_assistant') {
if (charName && !msg.content.startsWith(`${charName}: `)) { if (charName && !msg.content.startsWith(`${charName}: `)) {
msg.content = `${charName}: ${msg.content}`; msg.content = `${charName}: ${msg.content}`;