diff --git a/public/script.js b/public/script.js index 1dde4307d..6de4f2961 100644 --- a/public/script.js +++ b/public/script.js @@ -2198,10 +2198,9 @@ async function Generate(type, { automatic_trigger, force_name2, resolve, reject, // Determine token limit let this_max_context = getMaxContextSize(); - if (extension_settings.chromadb.n_results !== 0) { - await runGenerationInterceptors(coreChat, this_max_context); - console.log(`Core/all messages: ${coreChat.length}/${chat.length}`); - } + // Always run the extension interceptors. + await runGenerationInterceptors(coreChat, this_max_context); + console.log(`Core/all messages: ${coreChat.length}/${chat.length}`); let storyString = ""; diff --git a/public/scripts/extensions.js b/public/scripts/extensions.js index a4a6d5de9..1c7f2052c 100644 --- a/public/scripts/extensions.js +++ b/public/scripts/extensions.js @@ -456,6 +456,6 @@ $(document).ready(async function () { $("#extensions_connect").on('click', connectClickHandler); $("#extensions_autoconnect").on('input', autoConnectInputHandler); $("#extensions_details").on('click', showExtensionsDetails); - $(document).on('click', '.disable_extension', onDisableExtensionClick); - $(document).on('click', '.enable_extension', onEnableExtensionClick); + $(document).on('click', '.toggle_disable', onDisableExtensionClick); + $(document).on('click', '.toggle_enable', onEnableExtensionClick); }); diff --git a/public/scripts/extensions/infinity-context/index.js b/public/scripts/extensions/infinity-context/index.js index f00f266dc..0e804ac75 100644 --- a/public/scripts/extensions/infinity-context/index.js +++ b/public/scripts/extensions/infinity-context/index.js @@ -622,11 +622,11 @@ window.chromadb_interceptGeneration = async (chat, maxContext) => { queryBlob = lastMessage.mes; } else { - for (let msg of chat) { + for (let msg of chat.slice(-extension_settings.chromadb.keep_context)) { queryBlob += `${msg.mes}\n` } } - console.log("ChromDB Query text:", queryBlob); + console.log("CHROMADB: Query text:", queryBlob); if (recallStrategy === 'multichat') { console.log("Utilizing multichat") @@ -642,7 +642,7 @@ window.chromadb_interceptGeneration = async (chat, maxContext) => { else { queriedMessages.sort((a, b) => b.distance - a.distance); } - console.log(queriedMessages); + console.debug("CHROMADB: Query results: %o", queriedMessages); let newChat = []; @@ -702,6 +702,7 @@ window.chromadb_interceptGeneration = async (chat, maxContext) => { } const promptBlob = wrapperMsg.replace('{{memories}}', allMemoryBlob); + console.debug("CHROMADB: prompt blob: %o", promptBlob); context.setExtensionPrompt(MODULE_NAME, promptBlob, extension_prompt_types.AFTER_SCENARIO); } if (selectedStrategy === 'custom') {