#1891 Fix JWT Scale generation

This commit is contained in:
Cohee 2024-03-04 22:24:50 +02:00
parent 6f5dbc2a52
commit 1cce26d631
1 changed files with 4 additions and 0 deletions

View File

@ -4511,6 +4511,10 @@ function parseAndSaveLogprobs(data, continueFrom) {
* @returns {string} Extracted message * @returns {string} Extracted message
*/ */
function extractMessageFromData(data) { function extractMessageFromData(data) {
if (typeof data === 'string') {
return data;
}
switch (main_api) { switch (main_api) {
case 'kobold': case 'kobold':
return data.results[0].text; return data.results[0].text;