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

Add support for empty prompt replacement

This commit is contained in:
maver 2023-07-03 19:05:45 +02:00
parent 632fa770be
commit 4509df0fd9
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
public/scripts/openai.js
View File

@ -405,8 +405,16 @@ function populateChatHistory(prompts, chatCompletion) {
chatCompletion.reserveBudget(mainChatMessage);
const lastChatPrompt = openai_msgs[openai_msgs.length - 1];
const message = new Message('user', oai_settings.send_if_empty, 'emptyUserMessageReplacement');
if (lastChatPrompt && lastChatPrompt.role === 'assistant' && oai_settings.send_if_empty && chatCompletion.canAfford(message)) {
chatCompletion.insert(message, 'chatHistory');
}
// Insert chat messages as long as there is budget available
[...openai_msgs].reverse().every((prompt, index) => {
[...openai_msgs].reverse().every((chatPrompt, index) => {
// We do not want to mutate the prompt
const prompt = new Prompt(chatPrompt);
prompt.identifier = 'chatHistory-' + index;
const chatMessage = Message.fromPrompt(promptManager.preparePrompt(prompt));
@ -675,7 +683,6 @@ function prepareOpenAIMessages({
const messages = chatCompletion.getMessages();
promptManager.populateTokenHandler(messages);
promptManager.setMessages(messages);
console.log(messages)
// All information are up-to-date, render without dry-run.
if (false === dryRun) promptManager.render(false);