Improve KAI lite import

8574a5dd27e3574c997d954ba166292c5cf7ba56

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: f6b9cd970d3446057b49216c2001e8f36dca4b54
1 files changed, +4 -4Ignore whitespace
src/endpoints/chats.js+4 -4
@@ -203,19 +203,19 @@ function importKoboldLiteChat(_userName, _characterName, data) {
203203
204 /** @type {function(string): object} */204 /** @type {function(string): object} */
205 function processKoboldMessage(msg) {205 function processKoboldMessage(msg) {
206 const isUser = msg.includes(inputToken) || msg.includes(outputToken);206 const isUser = msg.includes(inputToken);
207 return {207 return {
208 name: isUser ? header.user_name : header.character_name,208 name: isUser ? header.user_name : header.character_name,
209 is_user: isUser,209 is_user: isUser,
210 mes: msg.replace(inputToken, '').replace(outputToken, '').trim(),210 mes: msg.replaceAll(inputToken, '').replaceAll(outputToken, '').trim(),
211 send_date: Date.now(),211 send_date: Date.now(),
212 };212 };
213 }213 }
214214
215 // Create the header215 // Create the header
216 const header = {216 const header = {
217 user_name: data.savedsettings.chatname,217 user_name: String(data.savedsettings.chatname),
218 character_name: data.savedsettings.chatopponent,218 character_name: String(data.savedsettings.chatopponent).split('||$||')[0],
219 };219 };
220 // Format messages220 // Format messages
221 const formattedMessages = data.actions.map(processKoboldMessage);221 const formattedMessages = data.actions.map(processKoboldMessage);