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

[feat] add openrouter

Browse Source
This commit is contained in:
kwaroran
2023-07-09 21:59:47 +09:00
parent d4157d3ab6
commit 77502416b5
3 changed files with 46 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -99,6 +99,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
case 'gpt4_32k_0613':
case 'gpt35_0301':
case 'gpt4_0301':
case 'openrouter':
case 'reverse_proxy':{
for(let i=0;i<formated.length;i++){
if(formated[i].role !== 'function'){
@@ -139,9 +140,10 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
const oaiFunctionCall = oaiFunctions ? (arg.useEmotion ? {"name": "set_emotion"} : "auto") : undefined
const requestModel = aiModel === 'reverse_proxy' ? db.proxyRequestModel : aiModel
const requestModel = (aiModel === 'reverse_proxy' || aiModel === 'openrouter') ? db.proxyRequestModel : aiModel
const body = ({
model: requestModel === 'gpt35' ? 'gpt-3.5-turbo'
model: aiModel === 'openrouter' ? db.openrouterRequestModel :
requestModel === 'gpt35' ? 'gpt-3.5-turbo'
: requestModel === 'gpt35_0613' ? 'gpt-3.5-turbo-0613'
: requestModel === 'gpt35_16k' ? 'gpt-3.5-turbo-16k'
: requestModel === 'gpt35_16k_0613' ? 'gpt-3.5-turbo-16k-0613'
@@ -179,16 +181,23 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
}
}
let headers = {
"Authorization": "Bearer " + aiModel === 'reverse_proxy' ? db.proxyKey :
aiModel === 'openrouter' ? db.openrouterKey : db.openAIKey,
"Content-Type": "application/json"
}
if(aiModel === 'openrouter'){
headers["X-Title"] = 'RisuAI'
headers["HTTP-Referer"] = 'https://risuai.xyz'
}
if(db.useStreaming && arg.useStreaming){
body.stream = true
const da = ((!isTauri) && (!isNodeServer))
? await fetch(`/proxy2`, {
body: JSON.stringify(body),
headers: {
"risu-header": encodeURIComponent(JSON.stringify({
"Authorization": "Bearer " + ((aiModel === 'reverse_proxy') ? db.proxyKey : db.openAIKey),
"Content-Type": "application/json"
})),
"risu-header": encodeURIComponent(JSON.stringify(headers)),
"risu-url": encodeURIComponent(replacerURL),
"Content-Type": "application/json"
},
@@ -198,10 +207,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
: await fetch(replacerURL, {
body: JSON.stringify(body),
method: "POST",
headers: {
"Authorization": "Bearer " + ((aiModel === 'reverse_proxy') ? db.proxyKey : db.openAIKey),
"Content-Type": "application/json"
},
headers: headers,
signal: abortSignal
})
@@ -259,10 +265,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
const res = await globalFetch(replacerURL, {
body: body,
headers: {
"Authorization": "Bearer " + ((aiModel === 'reverse_proxy') ? db.proxyKey : db.openAIKey),
"Content-Type": "application/json"
},
headers: headers,
abortSignal
})

6
src/ts/storage/database.ts
View File

@@ -267,6 +267,9 @@ export function setDatabase(data:Database){
}
data.classicMaxWidth ??= false
data.ooba ??= cloneDeep(defaultOoba)
data.ainconfig ??= cloneDeep(defaultAIN)
data.openrouterKey ??= ''
data.openrouterRequestModel ??= 'openai/gpt-3.5-turbo'
changeLanguage(data.language)
DataBase.set(data)
}
@@ -539,6 +542,9 @@ export interface Database{
proxyRequestModel:string
ooba:OobaSettings
ainconfig: AINsettings
openrouterRequestModel:string
openrouterKey:string
}
interface hordeConfig{