mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor loadLoreBookPrompt function in lorebook.ts

Browse Source
This commit is contained in:
kwaroran
2024-04-15 23:58:57 +09:00
parent 706e208f53
commit d93fd7b191
1 changed files with 53 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

104
src/ts/process/request.ts
View File

@@ -1796,64 +1796,66 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
let reader = res.body.getReader() let reader = res.body.getReader()
const decoder = new TextDecoder() const decoder = new TextDecoder()
const parser = createParser(async (e) => { const parser = createParser(async (e) => {
if(e.type === 'event'){ try {
switch(e.event){ if(e.type === 'event'){
case 'content_block_delta': { switch(e.event){
if(e.data){ case 'content_block_delta': {
text += JSON.parse(e.data).delta?.text if(e.data){
controller.enqueue({ text += JSON.parse(e.data).delta?.text
"0": text controller.enqueue({
}) "0": text
}
break
}
case 'error': {
if(e.data){
const errormsg:string = JSON.parse(e.data).error?.message
if(errormsg && errormsg.toLocaleLowerCase().includes('overload') && db.antiClaudeOverload){
console.log('Overload detected, retrying...')
reader.cancel()
rerequesting = true
await sleep(2000)
body.max_tokens -= await tokenize(text)
if(body.max_tokens < 0){
body.max_tokens = 0
}
if(body.messages.at(-1)?.role !== 'assistant'){
body.messages.push({
role: 'assistant',
content: ''
})
}
body.messages[body.messages.length-1].content += text
const res = await fetchNative(replacerURL, {
body: JSON.stringify(body),
headers: {
"Content-Type": "application/json",
"x-api-key": apiKey,
"anthropic-version": "2023-06-01",
"accept": "application/json",
},
method: "POST",
chatId: arg.chatId
}) })
if(res.status !== 200){ }
breakError = 'Error: ' + await textifyReadableStream(res.body) break
}
case 'error': {
if(e.data){
const errormsg:string = JSON.parse(e.data).error?.message
if(errormsg && errormsg.toLocaleLowerCase().includes('overload') && db.antiClaudeOverload){
console.log('Overload detected, retrying...')
reader.cancel()
rerequesting = true
await sleep(2000)
body.max_tokens -= await tokenize(text)
if(body.max_tokens < 0){
body.max_tokens = 0
}
if(body.messages.at(-1)?.role !== 'assistant'){
body.messages.push({
role: 'assistant',
content: ''
})
}
body.messages[body.messages.length-1].content += text
const res = await fetchNative(replacerURL, {
body: JSON.stringify(body),
headers: {
"Content-Type": "application/json",
"x-api-key": apiKey,
"anthropic-version": "2023-06-01",
"accept": "application/json",
},
method: "POST",
chatId: arg.chatId
})
if(res.status !== 200){
breakError = 'Error: ' + await textifyReadableStream(res.body)
break
}
reader = res.body.getReader()
rerequesting = false
break break
} }
reader = res.body.getReader() text += "Error:" + JSON.parse(e.data).error?.message
rerequesting = false controller.enqueue({
break "0": text
})
} }
text += "Error:" + JSON.parse(e.data).error?.message break
controller.enqueue({
"0": text
})
} }
break
} }
} }
} } catch (error) {}
}) })
while(true){ while(true){
if(rerequesting){ if(rerequesting){