jason/ThunderAI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Enhance tag extraction functionality in getTagsFromResponse method with improved JSON parsing and backward compatibility. see #365

Browse source
This commit is contained in:
mic 2025-06-15 00:22:10 +02:00
parent 07cf0774f9
commit 61480c0dc9
1 changed files with 30 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
js/mzta-utils-prompt.js
View file

@ -79,7 +79,36 @@ export const taPromptUtils = {
return chatgpt_lang;
},
/**
* Extracts tags from the response text.
* @param {string} response_text - The response text from which to extract tags.
* @returns {Array} An array of tags extracted from the response text.
*
* The response text should be a JSON with this structure:
* {
* "tags": ["tag1", "tag2", ...]
* }
*
* For backwords compatibility, if the response text is not a valid JSON,
* it will try to split the text by commas and return the resulting array.
*/
getTagsFromResponse(response_text){
let tags = [];
if(response_text && response_text.length > 0){
try {
// Try to parse the response text as JSON
let response_json = JSON.parse(response_text);
if(response_json && Array.isArray(response_json.tags)){
tags = response_json.tags;
} else if(response_json && response_json.tags && typeof response_json.tags === 'string'){
// If tags is a string, split it by commas
tags = response_json.tags.split(',').map(tag => tag.trim());
}
} catch (e) {
// If parsing fails, fallback to splitting by commas
tags = response_text.split(',').map(tag => tag.trim());
}
}
return tags;
}
};