Merge pull request #3726 from SillyTavern/feat/expressions-remove-reasoning Clean reasoning from LLM/webLLM classify response on expression classification
Signed| @@ -17,6 +17,7 @@ import { slashCommandReturnHelper } from '../../slash-commands/SlashCommandRetur | |||
| 17 | import { generateWebLlmChatPrompt, isWebLlmSupported } from '../shared.js'; | 17 | import { generateWebLlmChatPrompt, isWebLlmSupported } from '../shared.js'; |
| 18 | import { Popup, POPUP_RESULT } from '../../popup.js'; | 18 | import { Popup, POPUP_RESULT } from '../../popup.js'; |
| 19 | import { t } from '../../i18n.js'; | 19 | import { t } from '../../i18n.js'; |
| 20 | import { removeReasoningFromString } from '../../reasoning.js'; | ||
| 20 | export { MODULE_NAME }; | 21 | export { MODULE_NAME }; |
| 21 | 22 | ||
| 22 | /** | 23 | /** |
| @@ -928,6 +929,9 @@ function parseLlmResponse(emotionResponse, labels) { | |||
| 928 | 929 | ||
| 929 | return response; | 930 | return response; |
| 930 | } catch { | 931 | } catch { |
| 932 | // Clean possible reasoning from response | ||
| 933 | emotionResponse = removeReasoningFromString(emotionResponse); | ||
| 934 | |||
| 931 | const fuse = new Fuse(labels, { includeScore: true }); | 935 | const fuse = new Fuse(labels, { includeScore: true }); |
| 932 | console.debug('Using fuzzy search in labels:', labels); | 936 | console.debug('Using fuzzy search in labels:', labels); |
| 933 | const result = fuse.search(emotionResponse); | 937 | const result = fuse.search(emotionResponse); |