Merge pull request #2122 from joenunezb/fix-informaticai-missing-choices-message

Fix: Handle InformaticAI Endpoint response without message in response payload
This commit is contained in:
Cohee 2024-04-23 14:07:27 +03:00 committed by GitHub
commit d1f292f462
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -325,7 +325,7 @@ router.post('/generate', jsonParser, async function (request, response) {
// Map InfermaticAI response to OAI completions format // Map InfermaticAI response to OAI completions format
if (apiType === TEXTGEN_TYPES.INFERMATICAI) { if (apiType === TEXTGEN_TYPES.INFERMATICAI) {
data['choices'] = (data?.choices || []).map(choice => ({ text: choice.message.content })); data['choices'] = (data?.choices || []).map(choice => ({ text: choice?.message?.content || choice.text }));
} }
return response.send(data); return response.send(data);