Cohere: unbreak requires for old models
| @@ -570,9 +570,13 @@ async function sendCohereRequest(request, response) { | |||
| 570 | documents: [], | 570 | documents: [], |
| 571 | tools: tools, | 571 | tools: tools, |
| 572 | search_queries_only: false, | 572 | search_queries_only: false, |
| 573 | safety_mode: 'NONE', | ||
| 574 | }; | 573 | }; |
| 575 | 574 | ||
| 575 | const canDoSafetyMode = String(request.body.model).endsWith('08-2024'); | ||
| 576 | if (canDoSafetyMode) { | ||
| 577 | requestBody.safety_mode = 'NONE'; | ||
| 578 | } | ||
| 579 | |||
| 576 | console.log('Cohere request:', requestBody); | 580 | console.log('Cohere request:', requestBody); |
| 577 | 581 | ||
| 578 | const config = { | 582 | const config = { |