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

Merge branch 'staging' into welcome-screen

Browse Source
This commit is contained in:
Cohee
2025-05-13 19:52:40 +03:00
parent b261354280 cf28df381c
commit 249cb7777c
4 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
public/scripts/extensions/memory/index.js
View File

@ -28,6 +28,7 @@ import { ARGUMENT_TYPE, SlashCommandArgument, SlashCommandNamedArgument } from '
import { MacrosParser } from '../../macros.js';
import { countWebLlmTokens, generateWebLlmChatPrompt, getWebLlmContextSize, isWebLlmSupported } from '../shared.js';
import { commonEnumProviders } from '../../slash-commands/SlashCommandCommonEnumsProvider.js';
import { removeReasoningFromString } from '../../reasoning.js';
export { MODULE_NAME };
const MODULE_NAME = '1_memory';
@ -504,7 +505,7 @@ async function summarizeCallback(args, text) {
case summary_sources.extras:
return await callExtrasSummarizeAPI(text);
case summary_sources.main:
return await generateRaw(text, '', false, false, prompt, extension_settings.memory.overrideResponseLength);
return removeReasoningFromString(await generateRaw(text, '', false, false, prompt, extension_settings.memory.overrideResponseLength));
case summary_sources.webllm: {
const messages = [{ role: 'system', content: prompt }, { role: 'user', content: text }].filter(m => m.content);
const params = extension_settings.memory.overrideResponseLength > 0 ? { max_tokens: extension_settings.memory.overrideResponseLength } : {};
@ -699,7 +700,8 @@ async function summarizeChatMain(context, force, skipWIAN) {
return null;
}
summary = await generateRaw(rawPrompt, '', false, false, prompt, extension_settings.memory.overrideResponseLength);
const rawSummary = await generateRaw(rawPrompt, '', false, false, prompt, extension_settings.memory.overrideResponseLength);
summary = removeReasoningFromString(rawSummary);
index = lastUsedIndex;
} finally {
inApiCall = false;

3
public/scripts/extensions/vectors/index.js
View File

@ -36,6 +36,7 @@ import { slashCommandReturnHelper } from '../../slash-commands/SlashCommandRetur
import { callGenericPopup, POPUP_RESULT, POPUP_TYPE } from '../../popup.js';
import { generateWebLlmChatPrompt, isWebLlmSupported } from '../shared.js';
import { WebLlmVectorProvider } from './webllm.js';
import { removeReasoningFromString } from '../../reasoning.js';
/**
* @typedef {object} HashedMessage
@ -260,7 +261,7 @@ async function summarizeExtra(element) {
* @returns {Promise<boolean>} Sucess
*/
async function summarizeMain(element) {
element.text = await generateRaw(element.text, '', false, false, settings.summary_prompt);
element.text = removeReasoningFromString(await generateRaw(element.text, '', false, false, settings.summary_prompt));
return true;
}

9
public/scripts/openai.js
View File

@ -765,7 +765,10 @@ async function populationInjectionPrompts(prompts, messages) {
const wrap = false;
// Group prompts by priority
const orderGroups = {};
const extensionPromptsOrder = '0';
const orderGroups = {
[extensionPromptsOrder]: [],
};
for (const prompt of depthPrompts) {
const order = prompt.injection_order || 0;
if (!orderGroups[order]) {
@ -788,7 +791,9 @@ async function populationInjectionPrompts(prompts, messages) {
.join(separator);
// Get extension prompt
const extensionPrompt = await getExtensionPrompt(extension_prompt_types.IN_CHAT, i, separator, roleTypes[role], wrap);
const extensionPrompt = order === extensionPromptsOrder
? await getExtensionPrompt(extension_prompt_types.IN_CHAT, i, separator, roleTypes[role], wrap)
: '';
const jointPrompt = [rolePrompts, extensionPrompt].filter(x => x).map(x => x.trim()).join(separator);
if (jointPrompt && jointPrompt.length) {

3
src/endpoints/stable-diffusion.js
View File

@ -824,7 +824,8 @@ pollinations.post('/generate', async (request, response) => {
height: String(request.body.height ?? 1024),
nologo: String(true),
nofeed: String(true),
referer: 'sillytavern',
private: String(true),
referrer: 'sillytavern',
});
promptUrl.search = params.toString();