Fix formatting for Claude

This commit is contained in:
kwaroran
2024-02-06 21:34:02 +09:00
parent 1ddb24c250
commit 8bcae0864b
2 changed files with 8 additions and 3 deletions

View File

@@ -1423,7 +1423,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
}
let latestRole = 'user'
let requestPrompt = formated.map((v, i) => {
let prefix = ''
switch (v.role){
@@ -1437,13 +1437,18 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
prefix = "\n\nSystem: "
break
}
latestRole = v.role
if(raiModel.startsWith('claude-2') && (!raiModel.startsWith('claude-2.0'))){
if(v.role === 'system' && i === 0){
prefix = ''
}
}
return prefix + v.content
}).join('') + '\n\nAssistant: '
}).join('')
if(latestRole !== 'assistant'){
requestPrompt += '\n\nAssistant: '
}
const bedrock = db.claudeAws