rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Allow user to edit new chat, new group chat and new example chat prompts

Browse Source
This commit is contained in:
maver
2023-07-08 22:35:12 +02:00
parent 40079f7660
commit 9669199408
4 changed files with 158 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
public/scripts/openai.js
View File

@ -398,9 +398,12 @@ function formatWorldInfo(value) {
* @param {ChatCompletion} chatCompletion - An instance of ChatCompletion class that will be populated with the prompts.
*/
function populateChatHistory(prompts, chatCompletion) {
const newChat = promptManager.getSettings('utilityPrompts').newChat;
const newGroupChat = promptManager.getSettings('utilityPrompts').newGroupChat.replace('{{names}}', '${names}');
// Chat History
chatCompletion.add(new MessageCollection('chatHistory'), prompts.index('chatHistory'));
const mainChat = selected_group ? '[Start a new group chat. Group members: ${names}]' : '[Start a new Chat]';
const mainChat = selected_group ? newGroupChat : newChat;
const mainChatMessage = new Message('system', mainChat, 'newMainChat');
chatCompletion.reserveBudget(mainChatMessage);
@ -415,7 +418,7 @@ function populateChatHistory(prompts, chatCompletion) {
[...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-${openai_msgs.length - index}`;
const chatMessage = Message.fromPrompt(promptManager.preparePrompt(prompt));
if (true === promptManager.serviceSettings.names_in_completion && prompt.name)
@ -440,8 +443,9 @@ function populateChatHistory(prompts, chatCompletion) {
function populateDialogueExamples(prompts, chatCompletion) {
chatCompletion.add( new MessageCollection('dialogueExamples'), prompts.index('dialogueExamples'));
if (openai_msgs_example.length) {
const newExampleChat = promptManager.getSettings('utilityPrompts').newExampleChat;
// Insert chat message examples if there's enough budget if there is enough budget left for at least one example.
const dialogueExampleChat = new Message('system', '[Start a new Chat]', 'newChat');
const dialogueExampleChat = new Message('system', newExampleChat, 'newChat');
const prompt = openai_msgs_example[0];
const dialogueExample = new Message(prompt[0]?.role || 'system', prompt[0]?.content || '', 'dialogueExampleTest');
@ -1375,7 +1379,7 @@ class MessageCollection {
* @returns {Object} The found item, or undefined if no item was found.
*/
getItemByIdentifier(identifier) {
return this.collection.find(item => item.identifier === identifier);
return this.collection.find(item => item?.identifier === identifier);
}
/**