Refactor classifier to share code

This commit is contained in:
Jordan Wages 2025-06-24 16:10:38 -05:00
commit 312de63b2e
3 changed files with 71 additions and 22 deletions

View file

@ -35,7 +35,7 @@ let logger;
})();
// Listen for messages from UI/devtools
browser.runtime.onMessage.addListener((msg) => {
browser.runtime.onMessage.addListener(async (msg) => {
logger.aiLog("onMessage received", {debug: true}, msg);
if (msg?.type === "aiFilter:test") {
@ -45,7 +45,7 @@ browser.runtime.onMessage.addListener((msg) => {
try {
logger.aiLog("Calling browser.aiFilter.classify()", {debug: true});
const result = browser.aiFilter.classify(text, criterion);
const result = await browser.aiFilter.classify(text, criterion);
logger.aiLog("classify() returned", {debug: true}, result);
return { match: result };
}