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

Render prompt manager after swiping with updated tokens

Browse Source 
Including world info and extension prompts
This commit is contained in:
maver
2023-06-24 19:55:39 +02:00
parent 2f4424e6c6
commit 28fa2f5f57
2 changed files with 22 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
public/scripts/PromptManager.js
View File

@ -1,6 +1,6 @@
import {DraggablePromptListModule as DraggableList} from "./DraggableList.js";
import {event_types, eventSource, substituteParams} from "../script.js";
import {IdentifierNotFoundError, TokenHandler} from "./openai.js";
import {TokenHandler} from "./openai.js";
class Prompt {
identifier; role; content; name; system_prompt;
@ -240,14 +240,26 @@ PromptManagerModule.prototype.init = function (moduleConfiguration, serviceSetti
});
};
PromptManagerModule.prototype.render = function () {
/**
* Main rendering function
*
* @param afterTryGenerate - Whether a dry run should be attempted before rendering
*/
PromptManagerModule.prototype.render = function (afterTryGenerate = true) {
if (null === this.activeCharacter) return;
this.error = null;
this.tryGenerate().then(() => {
if (true === afterTryGenerate) {
this.tryGenerate().then(() => {
this.renderPromptManager();
this.renderPromptManagerListItems()
this.makeDraggable();
});
} else {
this.renderPromptManager();
this.renderPromptManagerListItems()
this.makeDraggable();
});
}
}
/**
@ -590,6 +602,7 @@ PromptManagerModule.prototype.getPromptCollection = function () {
}
PromptManagerModule.prototype.populateTokenHandler = function(messageCollection) {
this.tokenHandler.resetCounts();
const counts = this.tokenHandler.getCounts();
messageCollection.getCollection().forEach((message) => {
counts[message.identifier] = message.getTokens();