wagesj45/Sortana
1
0
Fork 0
Code Issues Pull requests Projects Releases 5 Packages Wiki Activity Actions

Merge pull request #21 from wagesj45/codex/implement-ai-based-message-classification

Browse source 
Add new mail classification handler
This commit is contained in:
Jordan Wages 2025-06-24 17:31:31 -05:00 • committed by GitHub
commit ddec4e245f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 26 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

26
background.js
View file

@ -11,10 +11,17 @@
"use strict";
let logger;
let AiClassifier;
// Startup
(async () => {
logger = await import(browser.runtime.getURL("logger.js"));
logger.aiLog("background.js loaded – ready to classify", {debug: true});
try {
({ AiClassifier } = ChromeUtils.import("resource://aifilter/modules/AiClassifier.jsm"));
logger.aiLog("AiClassifier imported", {debug: true});
} catch (e) {
logger.aiLog("failed to import AiClassifier", {level: 'error'}, e);
}
try {
const store = await browser.storage.local.get(["endpoint", "templateName", "customTemplate", "customSystemPrompt", "aiParams", "debugLogging"]);
logger.setDebug(store.debugLogging);
@ -60,6 +67,25 @@ browser.runtime.onMessage.addListener(async (msg) => {
}
});
// Automatically classify new messages
browser.messages.onNewMailReceived.addListener(async (folder, messages) => {
logger.aiLog("onNewMailReceived", {debug: true}, messages);
for (const msg of (messages?.messages || messages || [])) {
const id = msg.id ?? msg;
try {
const full = await browser.messages.getFull(id);
const text = full?.parts?.[0]?.body || "";
const criterion = (await browser.storage.local.get("autoCriterion")).autoCriterion || "";
const matched = await AiClassifier.classifyText(text, criterion);
if (matched) {
await browser.messages.update(id, {tags: ["$label1"]});
}
} catch (e) {
logger.aiLog("failed to classify new mail", {level: 'error'}, e);
}
}
});
// Catch any unhandled rejections
window.addEventListener("unhandledrejection", ev => {
logger.aiLog("Unhandled promise rejection", {level: 'error'}, ev.reason);