rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix injection order (again)

This commit is contained in:
Cohee 2023-10-11 22:56:17 +03:00
parent e2f0162e5a
commit 84098ae933
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
public/scripts/openai.js
View File

@ -456,12 +456,14 @@ function formatWorldInfo(value) {
* @param {Prompt[]} prompts - Array containing injection prompts.
*/
function populationInjectionPrompts(prompts) {
for (let i = MAX_INJECTION_DEPTH; i >= 0; i--) {
let totalInsertedMessages = 0;
for (let i = 0; i <= MAX_INJECTION_DEPTH; i++) {
// Get prompts for current depth
const depthPrompts = prompts.filter(prompt => prompt.injection_depth === i && prompt.content);
// Order of priority (most important go lower)
const roles = [ 'system', 'user', 'assistant'];
const roles = ['system', 'user', 'assistant'];
const roleMessages = [];
for (const role of roles) {
@ -478,7 +480,9 @@ function populationInjectionPrompts(prompts) {
}
if (roleMessages.length) {
openai_msgs.splice(i, 0, ...roleMessages);
const injectIdx = i + totalInsertedMessages;
openai_msgs.splice(injectIdx, 0, ...roleMessages);
totalInsertedMessages += roleMessages.length;
}
}