From 1cce26d631460cbf638182f1deef26086cc025e4 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Mon, 4 Mar 2024 22:24:50 +0200
Subject: [PATCH] #1891 Fix JWT Scale generation

---
 public/script.js | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/public/script.js b/public/script.js
index c4b2435bb..e68e3084a 100644
--- a/public/script.js
+++ b/public/script.js
@@ -4511,6 +4511,10 @@ function parseAndSaveLogprobs(data, continueFrom) {
  * @returns {string} Extracted message
  */
 function extractMessageFromData(data) {
+    if (typeof data === 'string') {
+        return data;
+    }
+
     switch (main_api) {
         case 'kobold':
             return data.results[0].text;