diff --git a/src/lib/EditMessage.svelte b/src/lib/EditMessage.svelte index 5904e00..65df4f6 100644 --- a/src/lib/EditMessage.svelte +++ b/src/lib/EditMessage.svelte @@ -89,6 +89,7 @@ const continueIncomplete = () => { editing = false + truncateFromMessage(chatId, message.uuid) $continueMessage = message.uuid } diff --git a/src/lib/providers/petals/request.svelte b/src/lib/providers/petals/request.svelte index 629925b..33eb7be 100644 --- a/src/lib/providers/petals/request.svelte +++ b/src/lib/providers/petals/request.svelte @@ -140,11 +140,13 @@ export const chatRequest = async ( doLead = false } // const inputArray = buildInputArray(rMessages).map(m => m.content) - const lInputArray = buildInputArray(rMessages.slice(0, -1)).map(m => m.content) + const lInputArray = doLead + ? buildInputArray(rMessages.slice(0, -1)).map(m => m.content) + : buildInputArray(rMessages.slice()).map(m => m.content) const nInputArray = buildInputArray(rMessages.slice(-1)).map(m => m.content) const leadPrompt = (leadPromptSequence && doLead) ? delimiter + leadPromptSequence : '' const lastPrompt = startSequence + lInputArray.join(delimiter) - const nextPrompt = nInputArray.slice(-1).join('') + leadPrompt + const nextPrompt = doLead ? nInputArray.slice(-1).join('') + leadPrompt : '' // set up the request chatResponse.onFinish(() => { @@ -170,7 +172,7 @@ export const chatRequest = async ( } midDel = midDel.length ? delimiter.slice(0, 0 - midDel.length) : delimiter - let inputPrompt = midDel + let inputPrompt = doLead ? midDel : '' const getNewWs = ():Promise => new Promise((resolve, reject) => { // console.warn('requesting new ws') @@ -192,7 +194,7 @@ export const chatRequest = async ( throw err } // console.warn('got new ws') - inputPrompt = lastPrompt + delimiter + inputPrompt = lastPrompt + (doLead ? delimiter : '') providerData.knownBuffer = '' providerData.ws = nws resolve(nws)