Add support for empty prompt replacement
This commit is contained in:
parent
632fa770be
commit
4509df0fd9
|
@ -405,8 +405,16 @@ function populateChatHistory(prompts, chatCompletion) {
|
||||||
|
|
||||||
chatCompletion.reserveBudget(mainChatMessage);
|
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
|
// 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;
|
prompt.identifier = 'chatHistory-' + index;
|
||||||
const chatMessage = Message.fromPrompt(promptManager.preparePrompt(prompt));
|
const chatMessage = Message.fromPrompt(promptManager.preparePrompt(prompt));
|
||||||
|
|
||||||
|
@ -675,7 +683,6 @@ function prepareOpenAIMessages({
|
||||||
const messages = chatCompletion.getMessages();
|
const messages = chatCompletion.getMessages();
|
||||||
promptManager.populateTokenHandler(messages);
|
promptManager.populateTokenHandler(messages);
|
||||||
promptManager.setMessages(messages);
|
promptManager.setMessages(messages);
|
||||||
console.log(messages)
|
|
||||||
|
|
||||||
// All information are up-to-date, render without dry-run.
|
// All information are up-to-date, render without dry-run.
|
||||||
if (false === dryRun) promptManager.render(false);
|
if (false === dryRun) promptManager.render(false);
|
||||||
|
|
Loading…
Reference in New Issue