feat(core): add insert positions for character-specific content

- Add 'before_char' and 'after_char' as new insert position options - Implement logic to handle character-specific content insertion - Enhance flexibility for integrating character-centric narratives and LoreBooks
ChatLunaLab · Nov 23, 2024 · c3a527e · c3a527e
1 parent d847275
commit c3a527e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/packages/core/src/llm-core/chain/prompt.ts b/packages/core/src/llm-core/chain/prompt.ts
@@ -391,6 +391,8 @@ Your goal is to craft a response that intelligently incorporates relevant knowle
         insertPosition:
             | PresetTemplate['loreBooks']['insertPosition']
             | PresetTemplate['authorsNote']['insertPosition']
+            | 'before_char'
+            | 'after_char'
     ) {
         if (insertPosition === 'in_chat') {
             return chatHistory.length - 1
@@ -414,9 +416,11 @@ Your goal is to craft a response that intelligently incorporates relevant knowle
 
         switch (insertPosition) {
             case 'before_char_defs':
+            case 'before_char':
                 return charDefIndex !== -1 ? charDefIndex : 1
 
             case 'after_char_defs':
+            case 'after_char':
                 if (scenarioIndex !== -1) return scenarioIndex + 1
                 return charDefIndex !== -1
                     ? charDefIndex + 1