@peopl-health/nexus 2.4.2 → 2.4.4

package/lib/adapters/TwilioProvider.js

@@ -1,7 +1,8 @@
 const { MessageProvider } = require('../core/MessageProvider');
 const axios = require('axios');
 const runtimeConfig = require('../config/runtimeConfig');
-const { uploadMediaToS3, sanitizeFileName, getFileExtension } = require('../helpers/mediaHelper');
+const { uploadMediaToS3, getFileExtension } = require('../helpers/mediaHelper');
+const { sanitizeMediaFilename } = require('../utils/sanitizer');
 const { generatePresignedUrl } = require('../config/awsConfig');
 const { validateMedia, getMediaType } = require('../utils/mediaValidator');
 const { v4: uuidv4 } = require('uuid');
@@ -299,7 +300,7 @@ class TwilioProvider extends MessageProvider {
       })();
 
       const baseName = existingFileName ? existingFileName.split('.').slice(0, -1).join('.') : `${mediaType}_${Date.now()}`;
-      const sanitizedBase = sanitizeFileName(baseName) || `${mediaType}_${Date.now()}`;
+      const sanitizedBase = sanitizeMediaFilename(baseName) || `${mediaType}_${Date.now()}`;
       const extension = getFileExtension(contentType) || 'bin';
       const uploadId = uuidv4();
       const key = await uploadMediaToS3(buffer, uploadId, sanitizedBase, bucketName, contentType, mediaType);

package/lib/assistants/BaseAssistant.js

@@ -1,7 +1,7 @@
 const llmConfig = require('../config/llmConfig');
 const { Thread } = require('../models/threadModel');
 const { Message } = require('../models/messageModel');
-const { formatMessage } = require('../helpers/assistantHelper');
+const { formatMessage } = require('../helpers/messageHelper');
 const { createProvider } = require('../providers/createProvider');
 
 const DEFAULT_MAX_HISTORICAL_MESSAGES = parseInt(process.env.MAX_HISTORICAL_MESSAGES || '50', 10);

package/lib/helpers/assistantHelper.js

@@ -111,6 +111,7 @@ const runAssistantWithRetries = async (thread, assistant, runConfig, patientRepl
     assistant.setReplies(patientReply);
   }
 
+  const startTime = Date.now();
   let run, output, completed;
   let retries = 0;
   const maxRetries = DEFAULT_MAX_RETRIES;
@@ -131,11 +132,14 @@ const runAssistantWithRetries = async (thread, assistant, runConfig, patientRepl
     if (retries < maxRetries) await new Promise(resolve => setTimeout(resolve, 2000));
   } while (retries < maxRetries && (!completed || !output));
 
+  const predictionTimeMs = Date.now() - startTime;
+
   if (run?.last_error) console.log('[runAssistantWithRetries] RUN LAST ERROR:', run.last_error);
   console.log('[runAssistantWithRetries] RUN STATUS', completed);
   console.log('[runAssistantWithRetries] OUTPUT', output);
+  console.log('[runAssistantWithRetries] TIMING', { predictionTimeMs, retries });
   
-  return { run, output, completed, retries };
+  return { run, output, completed, retries, predictionTimeMs };
 };
 
 module.exports = {

package/lib/helpers/filesHelper.js

@@ -15,7 +15,7 @@ async function convertPdfToImages(pdfName) {
   const pdfPath = path.join(outputDir, `${sanitizedName}.pdf`);
   const outputPattern = path.join(outputDir, sanitizedName);
 
-  await fs.promises.mkdir(outputDir, { recursive: true });
+  await fs.mkdir(outputDir, { recursive: true });
 
   return new Promise((resolve, reject) => {
     const args = ['-jpeg', pdfPath, outputPattern];
@@ -49,7 +49,7 @@ async function combineImagesToPDF(config) {
   } = config;
   
   const inputDir = path.join(__dirname, 'assets', 'tmp');
-  const files = await fs.promises.readdir(inputDir);
+  const files = await fs.readdir(inputDir);
     
   const imageFiles = files.filter(file => {
     const ext = path.extname(file).toLowerCase().substring(1);
@@ -81,7 +81,7 @@ async function combineImagesToPDF(config) {
       const filePath = path.join(inputDir, file);
       console.log(`Processing file ${index + 1}/${imageFiles.length}: ${file}`);
             
-      const imageBuffer = await fs.promises.readFile(filePath);
+      const imageBuffer = await fs.readFile(filePath);
       const pngBuffer = await sharp(imageBuffer)
         .toFormat('png')
         .toBuffer();
@@ -116,7 +116,7 @@ const cleanupFiles = async (files) => {
   
   await Promise.all(files.map(async (filePath) => {
     try {
-      await fs.promises.unlink(filePath);
+      await fs.unlink(filePath);
     } catch (error) {
       if (error?.code !== 'ENOENT') {
         const safeFileName = filePath ? filePath.split('/').pop().replace(/^[^-]+-[^-]+-/, 'xxx-xxx-') : 'unknown';
@@ -154,7 +154,7 @@ async function downloadMediaAndCreateFile(code, reply) {
   const sourceFile = `${sanitizedCode}-${sanitizedSubType}-${sanitizedFileName}`;
   const downloadPath = path.join(__dirname, 'assets', 'tmp', sourceFile);
   
-  await fs.promises.mkdir(path.dirname(downloadPath), { recursive: true });
+  await fs.mkdir(path.dirname(downloadPath), { recursive: true });
   await downloadFileFromS3(bucketName, key, downloadPath);
 
   const { name: baseName } = path.parse(sourceFile);
@@ -163,7 +163,7 @@ async function downloadMediaAndCreateFile(code, reply) {
     : [downloadPath];
 
   if (subType === 'document' ||  subType === 'application') {
-    await fs.promises.unlink(downloadPath);
+    await fs.unlink(downloadPath);
   }
 
   return fileNames;

package/lib/models/predictionMetricsModel.js

@@ -0,0 +1,20 @@
+const mongoose = require('mongoose');
+
+const predictionMetricsSchema = new mongoose.Schema({
+  message_id: { type: String, required: true, index: true },
+  numero: { type: String, required: true, index: true },
+  assistant_id: { type: String, required: true, index: true },
+  thread_id: { type: String, required: true },
+  prediction_time_ms: { type: Number, required: true },
+  retry_count: { type: Number, required: true, default: 1 },
+  completed: { type: Boolean, default: true },
+  error: { type: String, default: null }
+}, { timestamps: true });
+
+predictionMetricsSchema.index({ createdAt: -1 });
+predictionMetricsSchema.index({ assistant_id: 1, createdAt: -1 });
+predictionMetricsSchema.index({ numero: 1, createdAt: -1 });
+
+const PredictionMetrics = mongoose.model('PredictionMetrics', predictionMetricsSchema);
+
+module.exports = { PredictionMetrics, predictionMetricsSchema };

package/lib/services/assistantService.js

@@ -6,6 +6,7 @@ const { BaseAssistant } = require('../assistants/BaseAssistant');
 const { createProvider } = require('../providers/createProvider');
 
 const { Thread } = require('../models/threadModel.js');
+const { PredictionMetrics } = require('../models/predictionMetricsModel');
 
 const { getCurRow } = require('../helpers/assistantHelper.js');
 const { runAssistantAndWait, runAssistantWithRetries } = require('../helpers/assistantHelper.js');
@@ -316,7 +317,7 @@ const replyAssistantCore = async (code, message_ = null, thread_ = null, runOpti
   if (!patientMsg || thread.stopped) return null;
 
   const assistant = getAssistantById(thread.getAssistantId(), thread);
-  const { run, output, completed, retries } = await withTracing(
+  const { run, output, completed, retries, predictionTimeMs } = await withTracing(
     runAssistantWithRetries,
     'run_assistant_with_retries',
     (thread, assistant, runConfig, patientReply) => ({
@@ -326,6 +327,19 @@ const replyAssistantCore = async (code, message_ = null, thread_ = null, runOpti
     })
   )(thread, assistant, runOptions, patientReply);
 
+  // Store prediction metrics
+  if (output && predictionTimeMs) {
+    await PredictionMetrics.create({
+      message_id: `${code}-${Date.now()}`,
+      numero: code,
+      assistant_id: thread.getAssistantId(),
+      thread_id: thread.getConversationId(),
+      prediction_time_ms: predictionTimeMs,
+      retry_count: retries,
+      completed: completed
+    }).catch(err => console.error('[replyAssistantCore] Failed to store metrics:', err));
+  }
+
   return output;
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@peopl-health/nexus",
-  "version": "2.4.2",
+  "version": "2.4.4",
   "description": "Core messaging and assistant library for WhatsApp communication platforms",
   "keywords": [
     "whatsapp",