[feat] new oai handling

This commit is contained in:
kwaroran
2023-11-10 17:51:41 +09:00
parent dff566485c
commit 501167dd57
2 changed files with 37 additions and 3 deletions

View File

@@ -15,6 +15,7 @@ import { risuChatParser } from "../parser";
import { SignatureV4 } from "@smithy/signature-v4";
import { HttpRequest } from "@smithy/protocol-http";
import { Sha256 } from "@aws-crypto/sha256-js";
import { v4 } from "uuid";
@@ -130,13 +131,25 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
if(formated[i].role !== 'function'){
if(arg.isGroupChat && formated[i].name){
formated[i].content = formated[i].name + ": " + formated[i].content
formated[i].name = undefined
}
if(!(formated[i].name && formated[i].name.startsWith('example_') && db.newOAIHandle)){
formated[i].name = undefined
}
if(db.newOAIHandle && formated[i].memo && formated[i].memo.startsWith('NewChat')){
formated[i].content === ''
}
formated[i].name = undefined
delete formated[i].memo
delete formated[i].removable
}
}
if(db.newOAIHandle){
formated = formated.filter(m => {
return m.content !== ''
})
}
for(let i=0;i<biasString.length;i++){
const bia = biasString[i]
const tokens = await tokenizeNum(bia[0])
@@ -213,6 +226,11 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
body.seed = db.generationSeed
}
if(db.newOAIHandle){
// @ts-ignore
body.user = getOpenUserString()
}
let replacerURL = aiModel === 'openrouter' ? "https://openrouter.ai/api/v1/chat/completions" :
(aiModel === 'reverse_proxy') ? (db.forceReplaceUrl) : ('https://api.openai.com/v1/chat/completions')
@@ -1142,4 +1160,19 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
}
}
}
}
}
let userString = ''
let requestedTimes = 999
let refreshTime = 0
function getOpenUserString(){
if(refreshTime < Date.now() && requestedTimes > 2 ){
refreshTime = Date.now() + (300000 * Math.random()) + 60000
userString = v4()
requestedTimes = 0
}
requestedTimes += 1
console.log(userString)
return userString
}

View File

@@ -318,6 +318,7 @@ export function setDatabase(data:Database){
data.autofillRequestUrl ??= true
data.customProxyRequestModel ??= ''
data.generationSeed ??= -1
data.newOAIHandle ??= true
changeLanguage(data.language)
DataBase.set(data)
}
@@ -492,6 +493,7 @@ export interface Database{
autofillRequestUrl:boolean
customProxyRequestModel:string
generationSeed:number
newOAIHandle:boolean
}
export interface customscript{
@@ -920,7 +922,6 @@ export function changeToPreset(id =0, savecurrent = true){
export function setPreset(db:Database, newPres: botPreset){
db.apiType = newPres.apiType ?? db.apiType
db.openAIKey = newPres.openAIKey ?? db.openAIKey
db.mainPrompt = newPres.mainPrompt ?? db.mainPrompt
db.jailbreak = newPres.jailbreak ?? db.jailbreak
db.globalNote = newPres.globalNote ?? db.globalNote