Change insertion strategy to an extension block

This commit is contained in:
Cohee
2023-09-08 01:26:26 +03:00
parent 9d45c0a018
commit 96df705409
6 changed files with 51 additions and 54 deletions

View File

@@ -739,7 +739,7 @@ window.chromadb_interceptGeneration = async (chat, maxContext) => {
// No memories? No prompt.
const promptBlob = (tokenApprox == 0) ? "" : wrapperMsg.replace('{{memories}}', allMemoryBlob);
console.debug("CHROMADB: prompt blob: %o", promptBlob);
context.setExtensionPrompt(MODULE_NAME, promptBlob, extension_prompt_types.AFTER_SCENARIO);
context.setExtensionPrompt(MODULE_NAME, promptBlob, extension_prompt_types.IN_PROMPT);
}
if (selectedStrategy === 'custom') {
const context = getContext();