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

Fix claude compatibility for custom

Browse Source
This commit is contained in:
kwaroran
2024-08-17 02:18:57 +09:00
parent e6cf4aa279
commit 407b0830c8
1 changed files with 25 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1687,6 +1687,11 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
content: Claude3ContentBlock[] content: Claude3ContentBlock[]
} }
interface Claude3ExtendedChat {
role: 'user'|'assistant'
content: Claude3ContentBlock[]|string
}
let claudeChat: Claude3Chat[] = [] let claudeChat: Claude3Chat[] = []
let systemPrompt:string = '' let systemPrompt:string = ''
@@ -1854,10 +1859,23 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
} }
} }
let finalChat:Claude3ExtendedChat[] = claudeChat
if(aiModel === 'reverse_proxy'){
finalChat = claudeChat.map((v) => {
if(v.content.length === 0 && v.content[0].type === 'text'){
return {
role: v.role,
content: v.content[0].text
}
}
})
}
let body = { let body = {
model: raiModel, model: raiModel,
messages: claudeChat, messages: finalChat,
system: systemPrompt.trim(), system: systemPrompt.trim(),
max_tokens: maxTokens, max_tokens: maxTokens,
temperature: temperature, temperature: temperature,
@@ -2037,9 +2055,12 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
}] }]
}) })
} }
const block = body.messages[body.messages.length-1].content[0] let block = body.messages[body.messages.length-1].content
if(block.type === 'text'){ if(typeof block === 'string'){
block.text += text body.messages[body.messages.length-1].content += text
}
else if(block[0].type === 'text'){
block[0].text += text
} }
const res = await fetchNative(replacerURL, { const res = await fetchNative(replacerURL, {
body: JSON.stringify(body), body: JSON.stringify(body),