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

probably not gonna work well

Browse Source
This commit is contained in:
ouoertheo
2023-06-03 14:41:59 -05:00
parent 460030d600
commit 8ac368e2ef
1 changed files with 20 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
public/scripts/extensions/objective/index.js
View File

@ -38,9 +38,7 @@ const objectivePrompts = {
` `
} }
const injectPrompts = { const extensionPrompt = "Your current task is [{{task}}]. Balance existing roleplay with completing this task."
"task": "Your current task is [{{task}}]. Balance existing roleplay with completing this task."
}
// Background prompt generation // Background prompt generation
async function generateQuietPrompt(quiet_prompt) { async function generateQuietPrompt(quiet_prompt) {
@ -126,7 +124,7 @@ async function checkTaskCompleted() {
// Check response if task complete // Check response if task complete
if (taskResponse.includes("true")) { if (taskResponse.includes("true")) {
console.info(`Character determined task '${JSON.stringify(currentTask)} is completed.`) console.info(`Character determined task '${JSON.stringify(currentTask)} is completed.`)
completeTask(getTask()) currentTask.completeTask()
} else if (!(taskResponse.includes("false"))) { } else if (!(taskResponse.includes("false"))) {
console.warn(`checkTaskCompleted response did not contain true or false. taskResponse: ${taskResponse}`) console.warn(`checkTaskCompleted response did not contain true or false. taskResponse: ${taskResponse}`)
} else { } else {
@ -140,17 +138,23 @@ function setCurrentTask(index = null) {
const context = getContext(); const context = getContext();
currentTask = {}; currentTask = {};
// Set current task to either the next incomplete task, or the index if valid value is specified
if (index === null) { if (index === null) {
currentTask = globalTasks.find(task => !task.completed) || {}; currentTask = globalTasks.find(task => !task.completed) || {};
} else if (index >= 0 && index < globalTasks.length) { } else {
if (index >= 0 && index < globalTasks.length){
toastr.error(`Invalide task index ${index} specified. Must be between 0 and ${globalTasks.length}`)
return
}
currentTask = globalTasks[index]; currentTask = globalTasks[index];
} }
const { description } = currentTask; // Get the task description and add to extension prompt
const injectPromptsTask = injectPrompts["task"].replace(/{{task}}/gi, description); const { description } = currentTask.description;
if (description) { if (description) {
context.setExtensionPrompt(MODULE_NAME, injectPromptsTask, 1, $('#objective-chat-depth').val()); const extensionPromptText = extensionPrompt.replace(/{{task}}/gi, description);
context.setExtensionPrompt(MODULE_NAME, extensionPromptText, 1, $('#objective-chat-depth').val());
console.info(`Current task in context.extensionPrompts.Objective is ${JSON.stringify(context.extensionPrompts.Objective)}`); console.info(`Current task in context.extensionPrompts.Objective is ${JSON.stringify(context.extensionPrompts.Objective)}`);
} else { } else {
context.setExtensionPrompt(MODULE_NAME, ''); context.setExtensionPrompt(MODULE_NAME, '');
@ -212,10 +216,10 @@ class ObjectiveTask {
// Add event listeners and set properties // Add event listeners and set properties
$(`#objective-task-complete-${this.index}`).prop('checked', this.completed); $(`#objective-task-complete-${this.index}`).prop('checked', this.completed);
$(`#objective-task-complete-${this.index}`).on('click', this.onCompleteClick); $(`#objective-task-complete-${this.index}`).on('click', () => (this.onCompleteClick()));
$(`#objective-task-description-${this.index}`).on('keyup', this.onDescriptionUpdate); $(`#objective-task-description-${this.index}`).on('keyup', () => (this.onDescriptionUpdate()));
$(`#objective-task-delete-${this.index}`).on('click', this.onDeleteClick); $(`#objective-task-delete-${this.index}`).on('click', () => (this.onDeleteClick()));
$(`#objective-task-add-${this.index}`).on('click', this.onAddClick); $(`#objective-task-add-${this.index}`).on('click', () => (this.onAddClick()));
} }
onCompleteClick(){ onCompleteClick(){
@ -237,7 +241,7 @@ class ObjectiveTask {
updateUiTaskList(); updateUiTaskList();
} }
toString() { toSaveState() {
return { return {
"index":this.index, "index":this.index,
"description":this.description, "description":this.description,
@ -256,7 +260,7 @@ class TaskManager {
if (this.tasks){} if (this.tasks){}
} }
deleteTask (index) deleteTask (index) {}
} }
//###############################// //###############################//
@ -286,7 +290,7 @@ function saveState() {
} }
// Convert globalTasks for saving // Convert globalTasks for saving
const tasks = globalTasks.map(taskString => {return taskString.toString()}) const tasks = globalTasks.map(task => {return task.toSaveState()})
chat_metadata['objective'] = { chat_metadata['objective'] = {
objective: globalObjective, objective: globalObjective,
@ -389,7 +393,7 @@ function loadSettings() {
// Update globals // Update globals
globalObjective = chat_metadata['objective'].objective globalObjective = chat_metadata['objective'].objective
globalTasks = chat_metadata['objective'].tasks globalTasks = chat_metadata['objective'].tasks.map((task, index) => {return new ObjectiveTask(index, task.description)})
checkCounter = chat_metadata['objective'].checkFrequency checkCounter = chat_metadata['objective'].checkFrequency
// Update UI elements // Update UI elements