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

Add world info support for OAI. Fix regens

Browse Source
This commit is contained in:
SillyLossy
2023-03-22 01:16:16 +02:00
parent edab2e84b9
commit 7cce26bd22
3 changed files with 29 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
public/script.js
View File

@ -16,7 +16,7 @@ import {
world_info_data,
world_info_depth,
world_info,
checkWorldInfo,
getWorldInfoPrompt,
selectImportedWorldInfo,
setWorldInfoSettings,
deleteWorldInfo,
@ -55,7 +55,8 @@ import {
setOpenAIOnlineStatus,
generateOpenAIPromptCache,
oai_settings,
is_get_status_openai
is_get_status_openai,
openai_msgs,
} from "./scripts/openai.js";
import {
@ -900,18 +901,6 @@ function getExtensionPrompt() {
return extension_prompt;
}
function getWorldInfoPrompt(chat2) {
let worldInfoString = "", worldInfoBefore = "", worldInfoAfter = "";
if (world_info && world_info_data) {
const activatedWorldInfo = checkWorldInfo(chat2);
worldInfoBefore = activatedWorldInfo.worldInfoBefore;
worldInfoAfter = activatedWorldInfo.worldInfoAfter;
worldInfoString = worldInfoBefore + worldInfoAfter;
}
return { worldInfoString, worldInfoBefore, worldInfoAfter };
}
function baseChatReplace(value, name1, name2) {
if (value !== undefined && value.length > 0) {
if (is_pygmalion) {
@ -1461,7 +1450,7 @@ async function Generate(type, automatic_trigger, force_name2) {//encode("dsfs").
if (main_api == 'openai') {
let prompt = prepareOpenAIMessages(name2, storyString);
let prompt = prepareOpenAIMessages(name2, storyString, worldInfoBefore, worldInfoAfter, extension_prompt);
sendOpenAIRequest(prompt).then(onSuccess).catch(onError);
}
else {