rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct usage of fuzzy search

Browse Source
This commit is contained in:
Matěj Račinský
2024-05-18 17:05:55 +02:00
parent 0653dad5c5
commit bd1bfee941
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
public/scripts/extensions/expressions/index.js
View File

@ -1020,12 +1020,12 @@ function parseLlmResponse(emotionResponse, labels) {
const parsedEmotion = JSON.parse(emotionResponse); const parsedEmotion = JSON.parse(emotionResponse);
return parsedEmotion?.emotion ?? fallbackExpression; return parsedEmotion?.emotion ?? fallbackExpression;
} catch { } catch {
const fuse = new Fuse([emotionResponse]); const fuse = new Fuse(labels, {includeScore: true});
for (const label of labels) { console.debug("Using fuzzy search in labels: " + labels.join(', '))
const result = fuse.search(label); const result = fuse.search(emotionResponse);
if (result.length > 0) { if (result.length > 0) {
return label; console.debug("fuzzy search found: " + result[0].item + " as closest for the LLM response: " + emotionResponse)
} return result[0].item;
} }
} }