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

Add safety mode param

Browse Source
This commit is contained in:
kwaroran
2024-09-09 03:48:41 +09:00
parent 614087ae97
commit db39b64841
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1622,9 +1622,6 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
//reformat chat //reformat chat
let body = applyParameters({ let body = applyParameters({
message: lastChatPrompt, message: lastChatPrompt,
chat_history: formated.map((v) => { chat_history: formated.map((v) => {
@@ -1657,6 +1654,10 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
'top_p': 'p', 'top_p': 'p',
}) })
if(aiModel !== 'cohere-command-r-03-2024' && aiModel !== 'cohere-command-r-plus-04-2024'){
body.safety_mode = "NONE"
}
if(preamble){ if(preamble){
if(body.chat_history.length > 0){ if(body.chat_history.length > 0){
// @ts-ignore // @ts-ignore