From 890cf8162781ed1e007dc86e506cc4cfb55387eb Mon Sep 17 00:00:00 2001
From: joenunezb <joenunezb@gmail.com>
Date: Tue, 23 Apr 2024 03:56:50 -0700
Subject: [PATCH] Fix: InformaticAI response without message in choices

---
 src/endpoints/backends/text-completions.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/endpoints/backends/text-completions.js b/src/endpoints/backends/text-completions.js
index 22806cbf0..0e9598827 100644
--- a/src/endpoints/backends/text-completions.js
+++ b/src/endpoints/backends/text-completions.js
@@ -325,7 +325,7 @@ router.post('/generate', jsonParser, async function (request, response) {
 
                 // Map InfermaticAI response to OAI completions format
                 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);