Change Non-streaming Handler

2024-02-22 15:51:11 -06:00 · 2024-02-22 15:51:11 -06:00 · f82740a238
parent f569424f3e
commit f82740a238
1 changed files with 1 additions and 1 deletions
--- a/src/endpoints/backends/text-completions.js
+++ b/src/endpoints/backends/text-completions.js
@ -300,7 +300,7 @@ router.post('/generate', jsonParser, async function (request, response) {
                }

                // Map InfermaticAI response to OAI completions format
-                if (request.body.api_type === TEXTGEN_TYPES.INFERMATICAI) {
+                if (completionsReply.url.includes('https://api.totalgpt.ai')) {
                    data['choices'] = (data?.choices || []).map(choice => ({ text: choice.message.content }));
                }