Add Jamba Support

This commit is contained in:
kwaroran
2024-09-13 08:54:29 +09:00
parent 567a6e3dea
commit a9ddf66001
4 changed files with 15 additions and 0 deletions

View File

@@ -238,6 +238,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
case 'gpt4o-chatgpt':
case 'gpt4o1-preview':
case 'gpt4o1-mini':
case 'jamba-1.5-large':
case 'jamba-1.5-medium':
case 'reverse_proxy':{
let formatedChat:OpenAIChatExtra[] = []
for(let i=0;i<formated.length;i++){
@@ -613,6 +615,10 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
if(risuIdentify){
headers["X-Proxy-Risu"] = 'RisuAI'
}
if(aiModel.startsWith('jamba')){
headers['Authorization'] = 'Bearer ' + db.ai21Key
replacerURL = 'https://api.ai21.com/studio/v1/chat/completions'
}
if(multiGen){
// @ts-ignore
body.n = db.genTime