Add Groq as AI provider + new inline AI tasks, extend database to support more prompt types

inventory-server/src/services/ai/index.js

@@ -1,28 +1,38 @@
 /**
  * AI Service
  *
- * Main entry point for AI functionality including embeddings.
- * Provides embedding generation and similarity search for product validation.
+ * Main entry point for AI functionality including:
+ * - Embeddings for taxonomy suggestions (OpenAI)
+ * - Chat completions for validation tasks (Groq)
+ * - Task registry for AI operations
  */
 
 const { OpenAIProvider } = require('./providers/openaiProvider');
+const { GroqProvider, MODELS: GROQ_MODELS } = require('./providers/groqProvider');
 const { TaxonomyEmbeddings } = require('./embeddings/taxonomyEmbeddings');
 const { cosineSimilarity, findTopMatches } = require('./embeddings/similarity');
+const { getRegistry, TASK_IDS, registerAllTasks } = require('./tasks');
 
 let initialized = false;
 let initializing = false;
 let openaiProvider = null;
+let groqProvider = null;
 let taxonomyEmbeddings = null;
 let logger = console;
 
+// Store pool reference for task access
+let appPool = null;
+
 /**
  * Initialize the AI service
  * @param {Object} options
- * @param {string} options.openaiApiKey - OpenAI API key
+ * @param {string} options.openaiApiKey - OpenAI API key (for embeddings)
+ * @param {string} [options.groqApiKey] - Groq API key (for chat completions)
  * @param {Object} options.mysqlConnection - MySQL connection for taxonomy data
+ * @param {Object} [options.pool] - PostgreSQL pool for prompt loading
  * @param {Object} [options.logger] - Logger instance
  */
-async function initialize({ openaiApiKey, mysqlConnection, logger: customLogger }) {
+async function initialize({ openaiApiKey, groqApiKey, mysqlConnection, pool, logger: customLogger }) {
   if (initialized) {
     return { success: true, message: 'Already initialized' };
   }
@@ -48,9 +58,22 @@ async function initialize({ openaiApiKey, mysqlConnection, logger: customLogger
 
     logger.info('[AI] Initializing AI service...');
 
-    // Create OpenAI provider
+    // Store pool reference for tasks
+    if (pool) {
+      appPool = pool;
+    }
+
+    // Create OpenAI provider (for embeddings)
     openaiProvider = new OpenAIProvider({ apiKey: openaiApiKey });
 
+    // Create Groq provider (for chat completions) if API key provided
+    if (groqApiKey) {
+      groqProvider = new GroqProvider({ apiKey: groqApiKey });
+      logger.info('[AI] Groq provider initialized for chat completions');
+    } else {
+      logger.warn('[AI] No Groq API key provided - chat completion tasks will not be available');
+    }
+
     // Create and initialize taxonomy embeddings
     taxonomyEmbeddings = new TaxonomyEmbeddings({
       provider: openaiProvider,
@@ -59,13 +82,23 @@ async function initialize({ openaiApiKey, mysqlConnection, logger: customLogger
 
     const stats = await taxonomyEmbeddings.initialize(mysqlConnection);
 
+    // Register validation tasks if Groq is available
+    if (groqProvider) {
+      registerValidationTasks();
+    }
+
     initialized = true;
-    logger.info('[AI] AI service initialized', stats);
+    logger.info('[AI] AI service initialized', {
+      ...stats,
+      groqEnabled: !!groqProvider,
+      tasksRegistered: getRegistry().list()
+    });
 
     return {
       success: true,
       message: 'Initialized',
-      stats
+      stats,
+      groqEnabled: !!groqProvider
     };
   } catch (error) {
     logger.error('[AI] Initialization failed:', error);
@@ -75,6 +108,15 @@ async function initialize({ openaiApiKey, mysqlConnection, logger: customLogger
   }
 }
 
+/**
+ * Register validation tasks with the task registry
+ * Called during initialization if Groq is available
+ */
+function registerValidationTasks() {
+  registerAllTasks(logger);
+  logger.info('[AI] Validation tasks registered');
+}
+
 /**
  * Check if service is ready
  */
@@ -245,28 +287,98 @@ function getTaxonomyData() {
  * Get service status
  */
 function getStatus() {
+  const registry = getRegistry();
+
   return {
     initialized,
     ready: isReady(),
-    hasProvider: !!openaiProvider,
+    hasOpenAI: !!openaiProvider,
+    hasGroq: !!groqProvider,
     hasTaxonomy: !!taxonomyEmbeddings,
     taxonomyStats: taxonomyEmbeddings ? {
       categories: taxonomyEmbeddings.categories?.length || 0,
       themes: taxonomyEmbeddings.themes?.length || 0,
       colors: taxonomyEmbeddings.colors?.length || 0
-    } : null
+    } : null,
+    tasks: {
+      registered: registry.list(),
+      count: registry.size()
+    }
   };
 }
 
+/**
+ * Run an AI task by ID
+ * @param {string} taskId - Task identifier from TASK_IDS
+ * @param {Object} payload - Task-specific input
+ * @returns {Promise<Object>} Task result
+ */
+async function runTask(taskId, payload = {}) {
+  if (!initialized) {
+    throw new Error('AI service not initialized');
+  }
+
+  if (!groqProvider) {
+    throw new Error('Groq provider not available - chat completion tasks require GROQ_API_KEY');
+  }
+
+  const registry = getRegistry();
+  return registry.runTask(taskId, {
+    ...payload,
+    // Inject dependencies tasks may need
+    provider: groqProvider,
+    pool: appPool,
+    logger
+  });
+}
+
+/**
+ * Get the Groq provider instance (for direct use if needed)
+ * @returns {GroqProvider|null}
+ */
+function getGroqProvider() {
+  return groqProvider;
+}
+
+/**
+ * Get the PostgreSQL pool (for tasks that need DB access)
+ * @returns {Object|null}
+ */
+function getPool() {
+  return appPool;
+}
+
+/**
+ * Check if chat completion tasks are available
+ * @returns {boolean}
+ */
+function hasChatCompletion() {
+  return !!groqProvider;
+}
+
 module.exports = {
+  // Initialization
   initialize,
   isReady,
+  getStatus,
+
+  // Embeddings (OpenAI)
   getProductEmbedding,
   getProductEmbeddings,
   findSimilarTaxonomy,
   getSuggestionsForProduct,
   getTaxonomyData,
-  getStatus,
+
+  // Chat completions (Groq)
+  runTask,
+  hasChatCompletion,
+  getGroqProvider,
+  getPool,
+
+  // Constants
+  TASK_IDS,
+  GROQ_MODELS,
+
   // Re-export utilities
   cosineSimilarity,
   findTopMatches