aiWorker.js

Go to the documentation of this file.

// AI Worker: Ticket, Alert, Email, Remediation, Contract/Policy
// Now uses llama.cpp HTTP API instead of Ollama spawn
const Redis = require("ioredis");
const { callLlama } = require("./services/llamaCppService");
 
const redisConfig = require('./config/redis');
const sub = new Redis(redisConfig);
const pub = new Redis(redisConfig);
 
// Subscribe to all relevant channels
sub.subscribe("tickets:new", "alerts:new", "emails:inbound", "remediation:new", "contracts:policy");
 
sub.on("message", async (channel, message) => {
    const data = JSON.parse(message);
    let type = "";
    let outputKey = "";
    
    switch(channel) {
        case "tickets:new":
            type = "ticket";
            outputKey = `tickets:ai:analysis:${data.id}`;
            break;
        case "alerts:new":
            type = "alert";
            outputKey = `alerts:ai:analysis:${data.id}`;
            break;
        case "emails:inbound":
            type = "email";
            outputKey = `emails:ai:analysis:${data.id}`;
            break;
        case "remediation:new":
            type = "remediation";
            outputKey = `remediation:ai:analysis:${data.id}`;
            break;
        case "contracts:policy":
            type = "contract";
            outputKey = `contracts:ai:analysis:${data.id}`;
            break;
        default:
            return;
    }
    
    try {
        console.log(`[AI Worker] Processing ${channel} for ID: ${data.id}`);
        const output = await callLlama(data.text, type);
        await pub.publish(outputKey, JSON.stringify({ output }));
        console.log(`[AI Worker] Published result to ${outputKey}`);
    } catch (error) {
        console.error(`[AI Worker] Error processing ${channel}:`, error.message);
        await pub.publish(outputKey, JSON.stringify({ 
            error: error.message,
            output: "AI processing failed. Please try again later."
        }));
    }
});
 
console.log(`[AI Worker] Connected to Redis: ${process.env.REDIS_HOST || 'localhost'}:${process.env.REDIS_PORT || 6379}`);
console.log(`[AI Worker] Using llama.cpp endpoint: ${process.env.LLAMA_CPP_ENDPOINT || 'http://localhost:8080'}`);
console.log("AI Worker running: listening to Redis channels for automation.");