From 3227b33cb1ed3247da1fecefc17eb253db1b64fe Mon Sep 17 00:00:00 2001 From: kwaroran Date: Fri, 4 Aug 2023 17:41:05 +0900 Subject: [PATCH] [fix] prompt template --- src/ts/process/index.ts | 41 +++++++++++++++++++++++++++++++++++++---- 1 file changed, 37 insertions(+), 4 deletions(-) diff --git a/src/ts/process/index.ts b/src/ts/process/index.ts index 41f9b03c..581eeda2 100644 --- a/src/ts/process/index.ts +++ b/src/ts/process/index.ts @@ -318,8 +318,24 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n break } case 'chat':{ - const start = card.rangeStart - const end = (card.rangeEnd === 'end') ? unformated.chats.length : card.rangeEnd + let start = card.rangeStart + let end = (card.rangeEnd === 'end') ? unformated.chats.length : card.rangeEnd + if(start < 0){ + start = unformated.chats.length + start + if(start < 0){ + start = 0 + } + } + if(end < 0){ + end = unformated.chats.length + end + if(end < 0){ + end = 0 + } + } + + if(start >= end){ + break + } const chats = unformated.chats.slice(start, end) await tokenizeChatArray(chats) break @@ -561,8 +577,25 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n break } case 'chat':{ - const start = card.rangeStart - const end = (card.rangeEnd === 'end') ? unformated.chats.length : card.rangeEnd + let start = card.rangeStart + let end = (card.rangeEnd === 'end') ? unformated.chats.length : card.rangeEnd + if(start < 0){ + start = unformated.chats.length + start + if(start < 0){ + start = 0 + } + } + if(end < 0){ + end = unformated.chats.length + end + if(end < 0){ + end = 0 + } + } + + if(start >= end){ + break + } + const chats = unformated.chats.slice(start, end) pushPrompts(chats) break