Function calling for Mistral

This commit is contained in:
Cohee
2024-05-25 17:31:40 +03:00
parent b545185f1a
commit cb381595f9
3 changed files with 9 additions and 3 deletions

View File

@@ -486,6 +486,11 @@ async function sendMistralAIRequest(request, response) {
'random_seed': request.body.seed === -1 ? undefined : request.body.seed,
};
if (Array.isArray(request.body.tools) && request.body.tools.length > 0) {
requestBody['tools'] = request.body.tools;
requestBody['tool_choice'] = request.body.tool_choice === 'required' ? 'any' : 'auto';
}
const config = {
method: 'POST',
headers: {