rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2024-12-11 08:57:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

Filter system messages from raw last SD gens

Browse Source
This commit is contained in:
Cohee 2023-08-29 18:33:30 +03:00
parent 44f88c61ff
commit eb7a8f9ae4
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
public/scripts/extensions/stable-diffusion/index.js
View File

@ -539,8 +539,21 @@ function processReply(str) {
function getRawLastMessage() {
const getLastUsableMessage = () => {
for (const message of context.chat.slice().reverse()) {
if (message.is_system) {
continue;
}
return message.mes;
}
toastr.warning('No usable messages found.', 'Stable Diffusion');
throw new Error('No usable messages found.');
}
const context = getContext();
const lastMessage = context.chat.slice(-1)[0].mes,
const lastMessage = getLastUsableMessage(),
characterDescription = context.characters[context.characterId].description,
situation = context.characters[context.characterId].scenario;
return `((${processReply(lastMessage)})), (${processReply(situation)}:0.7), (${processReply(characterDescription)}:0.5)`