@peopl-health/nexus 2.4.11-logs → 2.4.11-logs-msg

package/lib/helpers/processHelper.js

@@ -61,27 +61,37 @@ const processImageFileCore = async (fileName, reply) => {
   let imageAnalysis = null;
   let url = null;
   const messagesChat = [];
+  const timings = {
+    analysis_ms: 0,
+    url_generation_ms: 0
+  };
 
   const isSticker = reply.media?.mediaType === 'sticker' || 
                     fileName.toLowerCase().includes('sticker/') ||
                     fileName.toLowerCase().includes('/sticker/');
 
   try {
-    imageAnalysis = await withTracing(
+    const { result: analysis, duration: analysisDuration } = await withTracing(
       analyzeImage,
       'analyze_image',
-      () => ({ 'image.is_sticker': isSticker, 'image.message_id': reply.message_id })
+      () => ({ 'image.is_sticker': isSticker, 'image.message_id': reply.message_id }),
+      { returnTiming: true }
     )(fileName, isSticker, reply.media?.contentType);
+    imageAnalysis = analysis;
+    timings.analysis_ms = analysisDuration;
     
     const invalidAnalysis = ['NOT_MEDICAL', 'QUALITY_INSUFFICIENT'];
     
     // Generate presigned URL only if medically relevant AND not a sticker
     if (imageAnalysis?.medical_relevance && !isSticker) {
-      url = await withTracing(
+      const { result: presignedUrl, duration: urlDuration } = await withTracing(
         generatePresignedUrl,
         'generate_presigned_url',
-        () => ({ 'url.bucket': reply.media.bucketName })
+        () => ({ 'url.bucket': reply.media.bucketName }),
+        { returnTiming: true }
       )(reply.media.bucketName, reply.media.key);
+      url = presignedUrl;
+      timings.url_generation_ms = urlDuration;
     }
     
     // Add appropriate text based on analysis
@@ -107,7 +117,8 @@ const processImageFileCore = async (fileName, reply) => {
       is_sticker: isSticker,
       medical_relevance: imageAnalysis?.medical_relevance,
       has_table: imageAnalysis?.has_table,
-      analysis_type: imageAnalysis?.medical_analysis ? 'medical' : 'general'
+      analysis_type: imageAnalysis?.medical_analysis ? 'medical' : 'general',
+      ...timings
     });
     
     logger.debug('processImageFile_analysis', { imageAnalysis });
@@ -124,7 +135,7 @@ const processImageFileCore = async (fileName, reply) => {
     });
   }
 
-  return { messagesChat, url };
+  return { messagesChat, url, timings };
 };
 
 const processImageFile = withTracing(
@@ -138,17 +149,22 @@ const processImageFile = withTracing(
 
 const processAudioFileCore = async (fileName, provider) => {
   const messagesChat = [];
+  const timings = {
+    transcribe_ms: 0
+  };
   
   try {
-    const audioTranscript = await withTracing(
+    const { result: audioTranscript, duration: transcribeDuration } = await withTracing(
       async () => provider.transcribeAudio({
         file: fs.createReadStream(fileName),
         responseFormat: 'text',
         language: 'es'
       }),
       'transcribe_audio',
-      () => ({ 'audio.file_name': fileName ? fileName.split('/').pop().replace(/^[^-]+-[^-]+-/, 'xxx-xxx-') : 'unknown' })
+      () => ({ 'audio.file_name': fileName ? fileName.split('/').pop().replace(/^[^-]+-[^-]+-/, 'xxx-xxx-') : 'unknown' }),
+      { returnTiming: true }
     )();
+    timings.transcribe_ms = transcribeDuration;
     
     const transcriptText = audioTranscript?.text || audioTranscript;
     messagesChat.push({
@@ -159,7 +175,8 @@ const processAudioFileCore = async (fileName, provider) => {
     logger.info('processAudioFile', {
       fileName: fileName ? fileName.split('/').pop().replace(/^[^-]+-[^-]+-/, 'xxx-xxx-') : 'unknown',
       transcription_success: true,
-      transcript_length: transcriptText?.length || 0
+      transcript_length: transcriptText?.length || 0,
+      ...timings
     });
     
     logger.debug('processAudioFile_transcript', { transcriptText });
@@ -174,7 +191,7 @@ const processAudioFileCore = async (fileName, provider) => {
     });
   }
 
-  return messagesChat;
+  return { messagesChat, timings };
 };
 
 const processAudioFile = withTracing(
@@ -189,16 +206,24 @@ const processMediaFilesCore = async (code, reply, provider) => {
   let url = null;
   const messagesChat = [];
   const tempFiles = [];
+  const timings = {
+    download_ms: 0,
+    image_analysis_ms: 0,
+    audio_transcription_ms: 0,
+    url_generation_ms: 0
+  };
 
   if (!reply.is_media) {
-    return { messagesChat, url, tempFiles };
+    return { messagesChat, url, tempFiles, timings };
   }
 
-  const fileNames = await withTracing(
+  const { result: fileNames, duration: downloadDuration } = await withTracing(
     downloadMediaAndCreateFile,
     'download_media',
-    () => ({ 'media.message_id': reply.message_id, 'media.type': reply.media?.mediaType })
+    () => ({ 'media.message_id': reply.message_id, 'media.type': reply.media?.mediaType }),
+    { returnTiming: true }
   )(code, reply);
+  timings.download_ms = downloadDuration;
   tempFiles.push(...fileNames);
 
   for (const fileName of fileNames) {
@@ -223,21 +248,33 @@ const processMediaFilesCore = async (code, reply, provider) => {
                         fileName.toLowerCase().includes('/sticker/');
 
     if (isImageLike) {
-      const { messagesChat: imageMessages, url: imageUrl } = await processImageFile(fileName, reply);
+      const { messagesChat: imageMessages, url: imageUrl, timings: imageTimings } = await processImageFile(fileName, reply);
+      
       messagesChat.push(...imageMessages);
       if (imageUrl) url = imageUrl;
+      
+      if (imageTimings) {
+        timings.image_analysis_ms += imageTimings.analysis_ms || 0;
+        timings.url_generation_ms += imageTimings.url_generation_ms || 0;
+      }
     } else if (fileName.includes('audio')) {
-      const audioMessages = await processAudioFile(fileName, provider);
+      const { messagesChat: audioMessages, timings: audioTimings } = await processAudioFile(fileName, provider);
+      
       messagesChat.push(...audioMessages);
+      
+      if (audioTimings) {
+        timings.audio_transcription_ms += audioTimings.transcribe_ms || 0;
+      }
     }
   }
 
   logger.info('processMediaFiles_complete', {
     message_id: reply.message_id,
-    file_count: fileNames.length
+    file_count: fileNames.length,
+    ...timings
   });
 
-  return { messagesChat, url, tempFiles };
+  return { messagesChat, url, tempFiles, timings };
 };
 
 const processMediaFiles = withTracing(
@@ -251,6 +288,13 @@ const processMediaFiles = withTracing(
 
 const processThreadMessageCore = async (code, replies, provider) => {
   const replyArray = Array.isArray(replies) ? replies : [replies];
+  const timings = {
+    download_ms: 0,
+    image_analysis_ms: 0,
+    audio_transcription_ms: 0,
+    url_generation_ms: 0,
+    total_media_ms: 0
+  };
   
   const results = await Promise.all(
     replyArray.map(async (reply, i) => {
@@ -262,9 +306,17 @@ const processThreadMessageCore = async (code, replies, provider) => {
         const textMessages = processTextMessage(reply);
         const mediaResult = await processMediaFiles(code, reply, provider);
         
-        const { messagesChat: mediaMessages, url, tempFiles: mediaFiles } = mediaResult;
+        const { messagesChat: mediaMessages, url, tempFiles: mediaFiles, timings: mediaTimings } = mediaResult;
         tempFiles = mediaFiles;
         
+        if (mediaTimings) {
+          timings.download_ms += mediaTimings.download_ms || 0;
+          timings.image_analysis_ms += mediaTimings.image_analysis_ms || 0;
+          timings.audio_transcription_ms += mediaTimings.audio_transcription_ms || 0;
+          timings.url_generation_ms += mediaTimings.url_generation_ms || 0;
+          timings.total_media_ms += (mediaTimings.download_ms + mediaTimings.image_analysis_ms + mediaTimings.audio_transcription_ms + mediaTimings.url_generation_ms);
+        }
+        
         const allMessages = [...textMessages, ...mediaMessages];
         const role = reply.origin === 'patient' ? 'user' : 'assistant';
         const messages = allMessages.map(content => ({ role, content }));
@@ -290,10 +342,11 @@ const processThreadMessageCore = async (code, replies, provider) => {
   );
   
   logger.info('processThreadMessage_complete', {
-    message_count: replyArray.length
+    message_count: replyArray.length,
+    ...timings
   });
   
-  return results;
+  return { results, timings };
 };
 
 const processThreadMessage = withTracing(

package/lib/services/assistantService.js

@@ -314,7 +314,7 @@ const replyAssistantCore = async (code, message_ = null, thread_ = null, runOpti
   
   logger.info(`[replyAssistantCore] Processing ${patientReply.length} messages in parallel`);
   
-  const { result: processResults, duration: processMessagesMs } = await withTracing(
+  const { result: processResult, duration: processMessagesMs } = await withTracing(
     processThreadMessage,
     'process_thread_messages',
     (code, patientReply, provider) => ({
@@ -323,8 +323,22 @@ const replyAssistantCore = async (code, message_ = null, thread_ = null, runOpti
     }),
     { returnTiming: true }
   )(code, patientReply, provider);
+  
+  const { results: processResults, timings: processTimings } = processResult;
   timings.process_messages_ms = processMessagesMs;
   
+  logger.debug('[replyAssistantCore] Process timings breakdown', { processTimings });
+  
+  if (processTimings) {
+    timings.process_messages_breakdown = {
+      download_ms: processTimings.download_ms || 0,
+      image_analysis_ms: processTimings.image_analysis_ms || 0,
+      audio_transcription_ms: processTimings.audio_transcription_ms || 0,
+      url_generation_ms: processTimings.url_generation_ms || 0,
+      total_media_ms: processTimings.total_media_ms || 0
+    };
+  }
+  
   const patientMsg = processResults.some(r => r.isPatient);
   const urls = processResults.filter(r => r.url).map(r => ({ url: r.url }));
   const allMessagesToAdd = processResults.flatMap(r => r.messages || []);
@@ -401,6 +415,11 @@ const replyAssistantCore = async (code, message_ = null, thread_ = null, runOpti
   });
 
   if (output && predictionTimeMs) {
+    logger.debug('[replyAssistantCore] Storing metrics with timing_breakdown', { 
+      timing_breakdown: timings,
+      has_breakdown: !!timings.process_messages_breakdown
+    });
+    
     await PredictionMetrics.create({
       message_id: `${code}-${Date.now()}`,
       numero: code,

package/package.json

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