@peopl-health/nexus 2.0.0 → 2.0.2

package/examples/test-batching.js

@@ -0,0 +1,767 @@
+const axios = require('axios');
+
+// Configuration
+const SERVER_URL = 'http://localhost:3000';
+const TEST_PHONE = 'whatsapp:+51976302758'; // Change this to your test number
+
+// Test messages to send
+const testMessages = [
+  { From: TEST_PHONE, Body: 'hola este mensaje' },
+  { From: TEST_PHONE, Body: 'es para hacer una prueba' },
+  { From: TEST_PHONE, Body: 'y que te diga que' },
+  { From: TEST_PHONE, Body: 'es mi cumpleaños :D' }
+];
+
+// Test messages with documents
+const testMessagesWithDocs = [
+  { 
+    From: TEST_PHONE, 
+    Body: 'Hola, aquí tienes un documento',
+    MediaUrl0: 'https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf',
+    NumMedia: '1',
+    MediaContentType0: 'application/pdf'
+  },
+  { 
+    From: TEST_PHONE, 
+    Body: 'Y aquí otro documento',
+    MediaUrl0: 'https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf',
+    NumMedia: '1',
+    MediaContentType0: 'application/pdf'
+  },
+  { From: TEST_PHONE, Body: '¿Puedes procesar estos documentos?' },
+  { From: TEST_PHONE, Body: 'Espero tu respuesta' }
+];
+
+// Mixed test messages (text + documents)
+const mixedTestMessages = [
+  { From: TEST_PHONE, Body: 'Mensaje de texto 1' },
+  { 
+    From: TEST_PHONE, 
+    Body: 'Documento adjunto',
+    MediaUrl0: 'https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf',
+    NumMedia: '1',
+    MediaContentType0: 'application/pdf'
+  },
+  { From: TEST_PHONE, Body: 'Mensaje de texto 2' },
+  { 
+    From: TEST_PHONE, 
+    Body: 'Otro documento',
+    MediaUrl0: 'https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf',
+    NumMedia: '1',
+    MediaContentType0: 'application/pdf'
+  },
+  { From: TEST_PHONE, Body: 'Mensaje final' }
+];
+
+// Test for media-first conflict scenario (the specific issue we're fixing)
+const mediaFirstTestMessages = [
+  { 
+    From: TEST_PHONE, 
+    Body: 'Documento primero',
+    MediaUrl0: 'https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf',
+    NumMedia: '1',
+    MediaContentType0: 'application/pdf'
+  },
+  { From: TEST_PHONE, Body: 'Mensaje después del documento' },
+  { From: TEST_PHONE, Body: '¿Qué es lo que puedes ver en el documento que te acabo de enviar?' }
+];
+
+// Test messages with images
+const testMessagesWithImages = [
+  { 
+    From: TEST_PHONE, 
+    Body: 'Aquí tienes una imagen',
+    MediaUrl0: 'https://picsum.photos/800/600?random=1',
+    NumMedia: '1',
+    MediaContentType0: 'image/jpeg'
+  },
+  { 
+    From: TEST_PHONE, 
+    Body: 'Y aquí otra imagen',
+    MediaUrl0: 'https://picsum.photos/800/600?random=2',
+    NumMedia: '1',
+    MediaContentType0: 'image/jpeg'
+  },
+  { From: TEST_PHONE, Body: '¿Puedes analizar estas imágenes?' },
+  { From: TEST_PHONE, Body: 'Espero tu respuesta sobre lo que ves' }
+];
+
+// Mixed test messages (text + images)
+const mixedImageTestMessages = [
+  { From: TEST_PHONE, Body: 'Mensaje de texto 1' },
+  { 
+    From: TEST_PHONE, 
+    Body: 'Imagen adjunta',
+    MediaUrl0: 'https://picsum.photos/800/600?random=3',
+    NumMedia: '1',
+    MediaContentType0: 'image/jpeg'
+  },
+  { From: TEST_PHONE, Body: 'Mensaje de texto 2' },
+  { 
+    From: TEST_PHONE, 
+    Body: 'Otra imagen',
+    MediaUrl0: 'https://picsum.photos/800/600?random=4',
+    NumMedia: '1',
+    MediaContentType0: 'image/jpeg'
+  },
+  { From: TEST_PHONE, Body: 'Mensaje final' }
+];
+
+// Test for image-first batching scenario
+const imageFirstTestMessages = [
+  { 
+    From: TEST_PHONE, 
+    Body: 'Imagen primero',
+    MediaUrl0: 'https://picsum.photos/800/600?random=5',
+    NumMedia: '1',
+    MediaContentType0: 'image/jpeg'
+  },
+  { From: TEST_PHONE, Body: 'Mensaje después de la imagen' },
+  { From: TEST_PHONE, Body: '¿Puedes procesar la imagen y responder?' }
+];
+
+// Delay between messages (in milliseconds)
+const MESSAGE_DELAY = 2000; // 2 seconds between messages
+
+async function sendMessage(message, index) {
+  try {
+    let messageType = '💬 Text';
+    if (message.MediaUrl0) {
+      if (message.MediaContentType0?.startsWith('image/')) {
+        messageType = '🖼️ Image';
+      } else if (message.MediaContentType0?.startsWith('application/')) {
+        messageType = '📄 Document';
+      } else {
+        messageType = '📎 Media';
+      }
+    }
+    
+    console.log(`📤 Sending message ${index + 1} (${messageType}): "${message.Body}"`);
+    
+    if (message.MediaUrl0) {
+      console.log(`   📎 Media: ${message.MediaUrl0}`);
+      console.log(`   📎 Type: ${message.MediaContentType0}`);
+    }
+    
+    const response = await axios.post(`${SERVER_URL}/webhook`, message, {
+      headers: {
+        'Content-Type': 'application/json'
+      }
+    });
+    
+    console.log(`✅ Message ${index + 1} sent successfully (Status: ${response.status})`);
+    return response.data;
+  } catch (error) {
+    console.error(`❌ Error sending message ${index + 1}:`, error.message);
+    return null;
+  }
+}
+
+async function testBatching() {
+  console.log('🧪 Starting Message Batching Test (Text Only)');
+  console.log('=====================================');
+  console.log(`Server URL: ${SERVER_URL}`);
+  console.log(`Test Phone: ${TEST_PHONE}`);
+  console.log(`Message Delay: ${MESSAGE_DELAY}ms`);
+  console.log('=====================================\n');
+
+  // Check if server is running
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    console.error('Please start the server with: node examples/basic-usage.js');
+    process.exit(1);
+  }
+
+  // Send messages with delays
+  console.log('📨 Sending test messages...\n');
+  
+  for (let i = 0; i < testMessages.length; i++) {
+    const message = testMessages[i];
+    
+    // Send the message
+    await sendMessage(message, i);
+    
+    // Wait before sending next message (except for the last one)
+    if (i < testMessages.length - 1) {
+      console.log(`⏳ Waiting ${MESSAGE_DELAY}ms before next message...\n`);
+      await new Promise(resolve => setTimeout(resolve, MESSAGE_DELAY));
+    }
+  }
+
+  console.log('\n🎯 Test completed!');
+  console.log('=====================================');
+  console.log('Expected behavior:');
+  console.log('- Messages should be batched together');
+  console.log('- You should see "Waiting X seconds for more messages" in server logs');
+  console.log('- After the wait time, you should see "Processing batched messages"');
+  console.log('- Only ONE assistant response should be generated for all messages');
+  console.log('=====================================\n');
+  
+  console.log('📋 To verify the test worked:');
+  console.log('1. Check server logs for batching messages');
+  console.log('2. Look for "Waiting X seconds for more messages"');
+  console.log('3. Look for "Processing batched messages"');
+  console.log('4. Verify only one assistant response was sent');
+}
+
+async function testBatchingWithDocuments() {
+  console.log('🧪 Starting Message Batching Test (With Documents)');
+  console.log('==================================================');
+  console.log(`Server URL: ${SERVER_URL}`);
+  console.log(`Test Phone: ${TEST_PHONE}`);
+  console.log(`Message Delay: ${MESSAGE_DELAY}ms`);
+  console.log('==================================================\n');
+
+  // Check if server is running
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    console.error('Please start the server with: node examples/basic-usage.js');
+    process.exit(1);
+  }
+
+  // Send messages with documents
+  console.log('📨 Sending test messages with documents...\n');
+  
+  for (let i = 0; i < testMessagesWithDocs.length; i++) {
+    const message = testMessagesWithDocs[i];
+    
+    // Send the message
+    await sendMessage(message, i);
+    
+    // Wait before sending next message (except for the last one)
+    if (i < testMessagesWithDocs.length - 1) {
+      console.log(`⏳ Waiting ${MESSAGE_DELAY}ms before next message...\n`);
+      await new Promise(resolve => setTimeout(resolve, MESSAGE_DELAY));
+    }
+  }
+
+  console.log('\n🎯 Document batching test completed!');
+  console.log('==================================================');
+  console.log('Expected behavior:');
+  console.log('- Messages with documents should be batched together');
+  console.log('- Documents should be processed and stored');
+  console.log('- You should see "Waiting X seconds for more messages" in server logs');
+  console.log('- After the wait time, you should see "Processing batched messages"');
+  console.log('- Only ONE assistant response should be generated for all messages');
+  console.log('==================================================\n');
+}
+
+async function testMixedBatching() {
+  console.log('🧪 Starting Mixed Message Batching Test (Text + Documents)');
+  console.log('==========================================================');
+  console.log(`Server URL: ${SERVER_URL}`);
+  console.log(`Test Phone: ${TEST_PHONE}`);
+  console.log(`Message Delay: ${MESSAGE_DELAY}ms`);
+  console.log('==========================================================\n');
+
+  // Check if server is running
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    console.error('Please start the server with: node examples/basic-usage.js');
+    process.exit(1);
+  }
+
+  // Send mixed messages
+  console.log('📨 Sending mixed test messages (text + documents)...\n');
+  console.log('🔍 Watch the server logs for batching messages!');
+  console.log('Expected server logs:');
+  console.log('  - "Batching config: { enabled: true, ... }"');
+  console.log('  - "Received additional message from whatsapp:+51976302758, resetting wait timer"');
+  console.log('  - "Waiting X seconds for more messages from whatsapp:+51976302758"');
+  console.log('  - "Processing batched messages from whatsapp:+51976302758"');
+  console.log('');
+  
+  for (let i = 0; i < mixedTestMessages.length; i++) {
+    const message = mixedTestMessages[i];
+    
+    // Send the message
+    await sendMessage(message, i);
+    
+    // Wait before sending next message (except for the last one)
+    if (i < mixedTestMessages.length - 1) {
+      console.log(`⏳ Waiting ${MESSAGE_DELAY}ms before next message...\n`);
+      await new Promise(resolve => setTimeout(resolve, MESSAGE_DELAY));
+    }
+  }
+
+  console.log('\n🎯 Mixed batching test completed!');
+  console.log('==========================================================');
+  console.log('Expected behavior:');
+  console.log('- Text and document messages should be batched together');
+  console.log('- Documents should be processed and stored');
+  console.log('- You should see "Waiting X seconds for more messages" in server logs');
+  console.log('- After the wait time, you should see "Processing batched messages"');
+  console.log('- Only ONE assistant response should be generated for all messages');
+  console.log('==========================================================\n');
+  
+  console.log('🔍 If you received 2 separate responses instead of 1:');
+  console.log('1. Check server logs for batching messages');
+  console.log('2. Verify batching is enabled: "Batching config: { enabled: true }"');
+  console.log('3. Look for "Waiting X seconds" messages');
+  console.log('4. Check if messages are being processed too quickly/slowly');
+  console.log('5. Look for "Processing batched messages from [chatId] (including media if any)"');
+  console.log('6. Verify no "Error handling batched messages" or run conflicts');
+}
+
+async function testMediaFirstBatching() {
+  console.log('🧪 Starting Media-First Batching Test (Document + Text)');
+  console.log('==========================================================');
+  console.log('This test specifically addresses the media-first conflict issue');
+  console.log('==========================================================\n');
+
+  // Check if server is running
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    console.error('Please start the server with: node examples/basic-usage.js');
+    process.exit(1);
+  }
+
+  // Send media-first messages
+  console.log('📨 Sending media-first test messages (document + text)...\n');
+  console.log('🔍 Watch the server logs for batching messages!');
+  console.log('Expected server logs:');
+  console.log('  - "Batching config: { enabled: true, ... }"');
+  console.log('  - "Received additional message from whatsapp:+51976302758, resetting wait timer"');
+  console.log('  - "Waiting X seconds for more messages from whatsapp:+51976302758"');
+  console.log('  - "Processing batched messages from whatsapp:+51976302758 (including media if any)"');
+  console.log('');
+  
+  for (let i = 0; i < mediaFirstTestMessages.length; i++) {
+    const message = mediaFirstTestMessages[i];
+    
+    // Send the message
+    await sendMessage(message, i);
+    
+    // Wait before sending next message (except for the last one)
+    if (i < mediaFirstTestMessages.length - 1) {
+      console.log(`⏳ Waiting ${MESSAGE_DELAY}ms before next message...\n`);
+      await new Promise(resolve => setTimeout(resolve, MESSAGE_DELAY));
+    }
+  }
+
+  console.log('\n🎯 Media-first batching test completed!');
+  console.log('==========================================================');
+  console.log('Expected behavior:');
+  console.log('- Document and text messages should be batched together');
+  console.log('- No "Error handling batched messages" or run conflicts');
+  console.log('- Only ONE assistant response should be generated');
+  console.log('- You should see "Processing batched messages from [chatId] (including media if any)"');
+  console.log('==========================================================\n');
+  
+  console.log('🔍 If you see errors or multiple responses:');
+  console.log('1. Check for "Error handling batched messages" in server logs');
+  console.log('2. Look for OpenAI run conflicts');
+  console.log('3. Verify media messages are being included in batching');
+  console.log('4. Check that messages are marked as processed correctly');
+}
+
+async function testBatchingDiagnostic() {
+  console.log('🔍 Batching Diagnostic Test');
+  console.log('============================');
+  console.log('This test will help diagnose why batching might not be working\n');
+
+  // Check server status
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    process.exit(1);
+  }
+
+  // Send a single message first to check immediate processing
+  console.log('📤 Step 1: Sending single message to test immediate processing...');
+  const singleMessage = { From: TEST_PHONE, Body: 'Single test message' };
+  await sendMessage(singleMessage, 0);
+  
+  console.log('\n⏳ Waiting 5 seconds...\n');
+  await new Promise(resolve => setTimeout(resolve, 5000));
+
+  // Send two messages quickly
+  console.log('📤 Step 2: Sending two messages quickly (should batch)...');
+  const quickMessages = [
+    { From: TEST_PHONE, Body: 'Quick message 1' },
+    { From: TEST_PHONE, Body: 'Quick message 2' }
+  ];
+
+  for (let i = 0; i < quickMessages.length; i++) {
+    await sendMessage(quickMessages[i], i);
+    if (i < quickMessages.length - 1) {
+      console.log('⏳ Waiting 1 second before next message...\n');
+      await new Promise(resolve => setTimeout(resolve, 1000));
+    }
+  }
+
+  console.log('\n🎯 Diagnostic test completed!');
+  console.log('============================');
+  console.log('Check server logs for:');
+  console.log('1. "Batching config: { enabled: true, ... }"');
+  console.log('2. "Received additional message from whatsapp:+51976302758, resetting wait timer"');
+  console.log('3. "Waiting X seconds for more messages from whatsapp:+51976302758"');
+  console.log('4. "Processing batched messages from whatsapp:+51976302758"');
+  console.log('');
+  console.log('If you see these logs but still get 2 responses, there might be an issue with:');
+  console.log('- Assistant processing logic');
+  console.log('- Message deduplication');
+  console.log('- Batching timeout handling');
+}
+
+async function testBatchingCorrectly() {
+  console.log('🧪 Correct Batching Test');
+  console.log('=========================');
+  console.log('This test shows batching working correctly\n');
+
+  // Check server status
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    process.exit(1);
+  }
+
+  // Send multiple messages quickly (all should be batched)
+  console.log('📤 Sending 3 messages quickly (all should be batched together)...');
+  const batchMessages = [
+    { From: TEST_PHONE, Body: 'Batch message 1' },
+    { From: TEST_PHONE, Body: 'Batch message 2' },
+    { From: TEST_PHONE, Body: 'Batch message 3' }
+  ];
+
+  for (let i = 0; i < batchMessages.length; i++) {
+    await sendMessage(batchMessages[i], i);
+    if (i < batchMessages.length - 1) {
+      console.log('⏳ Waiting 1 second before next message...\n');
+      await new Promise(resolve => setTimeout(resolve, 1000));
+    }
+  }
+
+  console.log('\n🎯 Correct batching test completed!');
+  console.log('===================================');
+  console.log('Expected behavior:');
+  console.log('- All 3 messages should be batched together');
+  console.log('- You should see "Waiting X seconds for more messages" in server logs');
+  console.log('- After the wait time, you should see "Processing batched messages"');
+  console.log('- Only ONE assistant response should be generated for all 3 messages');
+  console.log('===================================\n');
+  
+  console.log('🔍 Check server logs for:');
+  console.log('1. "Batching config: { enabled: true, ... }"');
+  console.log('2. "Received additional message from whatsapp:+51976302758, resetting wait timer"');
+  console.log('3. "Waiting X seconds for more messages from whatsapp:+51976302758"');
+  console.log('4. "Processing batched messages from whatsapp:+51976302758"');
+  console.log('5. Only ONE "Assistant initialized" message');
+  console.log('6. Only ONE response sent to WhatsApp');
+}
+
+async function testWithoutBatching() {
+  console.log('🧪 Testing WITHOUT Batching (Control Test)');
+  console.log('==========================================');
+  console.log('This test sends messages with longer delays to avoid batching');
+  console.log('Each message should be processed immediately\n');
+
+  const controlMessages = [
+    { From: TEST_PHONE, Body: 'Control message 1' },
+    { From: TEST_PHONE, Body: 'Control message 2' }
+  ];
+
+  for (let i = 0; i < controlMessages.length; i++) {
+    const message = controlMessages[i];
+    await sendMessage(message, i);
+    
+    if (i < controlMessages.length - 1) {
+      console.log('⏳ Waiting 20 seconds (longer than batch timeout)...\n');
+      await new Promise(resolve => setTimeout(resolve, 20000));
+    }
+  }
+
+  console.log('\n🎯 Control test completed!');
+  console.log('Expected: Each message should be processed immediately');
+}
+
+async function testSingleMessage() {
+  console.log('🧪 Testing Single Message (No Batching)');
+  console.log('======================================');
+  
+  const singleMessage = { From: TEST_PHONE, Body: 'Single message test' };
+  await sendMessage(singleMessage, 0);
+  
+  console.log('\n🎯 Single message test completed!');
+  console.log('Expected: Message should be processed immediately (no batching)');
+}
+
+async function testBatchingWithImages() {
+  console.log('🧪 Starting Message Batching Test (With Images)');
+  console.log('================================================');
+  console.log(`Server URL: ${SERVER_URL}`);
+  console.log(`Test Phone: ${TEST_PHONE}`);
+  console.log(`Message Delay: ${MESSAGE_DELAY}ms`);
+  console.log('================================================\n');
+
+  // Check if server is running
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    console.error('Please start the server with: node examples/basic-usage.js');
+    process.exit(1);
+  }
+
+  // Send messages with images
+  console.log('📨 Sending test messages with images...\n');
+  
+  for (let i = 0; i < testMessagesWithImages.length; i++) {
+    const message = testMessagesWithImages[i];
+    
+    // Send the message
+    await sendMessage(message, i);
+    
+    // Wait before sending next message (except for the last one)
+    if (i < testMessagesWithImages.length - 1) {
+      console.log(`⏳ Waiting ${MESSAGE_DELAY}ms before next message...\n`);
+      await new Promise(resolve => setTimeout(resolve, MESSAGE_DELAY));
+    }
+  }
+
+  console.log('\n🎯 Image batching test completed!');
+  console.log('================================================');
+  console.log('Expected behavior:');
+  console.log('- Messages with images should be batched together');
+  console.log('- Images should be processed and stored');
+  console.log('- You should see "Waiting X seconds for more messages" in server logs');
+  console.log('- After the wait time, you should see "Processing batched messages"');
+  console.log('- Only ONE assistant response should be generated for all messages');
+  console.log('================================================\n');
+}
+
+async function testMixedImageBatching() {
+  console.log('🧪 Starting Mixed Image Batching Test (Text + Images)');
+  console.log('====================================================');
+  console.log(`Server URL: ${SERVER_URL}`);
+  console.log(`Test Phone: ${TEST_PHONE}`);
+  console.log(`Message Delay: ${MESSAGE_DELAY}ms`);
+  console.log('====================================================\n');
+
+  // Check if server is running
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    console.error('Please start the server with: node examples/basic-usage.js');
+    process.exit(1);
+  }
+
+  // Send mixed messages
+  console.log('📨 Sending mixed test messages (text + images)...\n');
+  console.log('🔍 Watch the server logs for batching messages!');
+  console.log('Expected server logs:');
+  console.log('  - "Batching config: { enabled: true, ... }"');
+  console.log('  - "Received additional message from whatsapp:+51976302758, resetting wait timer"');
+  console.log('  - "Waiting X seconds for more messages from whatsapp:+51976302758"');
+  console.log('  - "Processing batched messages from whatsapp:+51976302758"');
+  console.log('');
+  
+  for (let i = 0; i < mixedImageTestMessages.length; i++) {
+    const message = mixedImageTestMessages[i];
+    
+    // Send the message
+    await sendMessage(message, i);
+    
+    // Wait before sending next message (except for the last one)
+    if (i < mixedImageTestMessages.length - 1) {
+      console.log(`⏳ Waiting ${MESSAGE_DELAY}ms before next message...\n`);
+      await new Promise(resolve => setTimeout(resolve, MESSAGE_DELAY));
+    }
+  }
+
+  console.log('\n🎯 Mixed image batching test completed!');
+  console.log('====================================================');
+  console.log('Expected behavior:');
+  console.log('- Text and image messages should be batched together');
+  console.log('- Images should be processed and stored');
+  console.log('- You should see "Waiting X seconds for more messages" in server logs');
+  console.log('- After the wait time, you should see "Processing batched messages"');
+  console.log('- Only ONE assistant response should be generated for all messages');
+  console.log('====================================================\n');
+  
+  console.log('🔍 If you received 2 separate responses instead of 1:');
+  console.log('1. Check server logs for batching messages');
+  console.log('2. Verify batching is enabled: "Batching config: { enabled: true }"');
+  console.log('3. Look for "Waiting X seconds" messages');
+  console.log('4. Check if messages are being processed too quickly/slowly');
+  console.log('5. Look for "Processing batched messages from [chatId] (including media if any)"');
+  console.log('6. Verify no "Error handling batched messages" or run conflicts');
+}
+
+async function testImageFirstBatching() {
+  console.log('🧪 Starting Image-First Batching Test (Image + Text)');
+  console.log('====================================================');
+  console.log('This test specifically addresses the image-first conflict issue');
+  console.log('====================================================\n');
+
+  // Check if server is running
+  try {
+    const statusResponse = await axios.get(`${SERVER_URL}/status`);
+    console.log('✅ Server is running');
+    console.log('Server Status:', JSON.stringify(statusResponse.data, null, 2));
+    console.log('');
+  } catch (error) {
+    console.error('❌ Server is not running or not accessible');
+    console.error('Please start the server with: node examples/basic-usage.js');
+    process.exit(1);
+  }
+
+  // Send image-first messages
+  console.log('📨 Sending image-first test messages (image + text)...\n');
+  console.log('🔍 Watch the server logs for batching messages!');
+  console.log('Expected server logs:');
+  console.log('  - "Batching config: { enabled: true, ... }"');
+  console.log('  - "Received additional message from whatsapp:+51976302758, resetting wait timer"');
+  console.log('  - "Waiting X seconds for more messages from whatsapp:+51976302758"');
+  console.log('  - "Processing batched messages from whatsapp:+51976302758 (including media if any)"');
+  console.log('');
+  
+  for (let i = 0; i < imageFirstTestMessages.length; i++) {
+    const message = imageFirstTestMessages[i];
+    
+    // Send the message
+    await sendMessage(message, i);
+    
+    // Wait before sending next message (except for the last one)
+    if (i < imageFirstTestMessages.length - 1) {
+      console.log(`⏳ Waiting ${MESSAGE_DELAY}ms before next message...\n`);
+      await new Promise(resolve => setTimeout(resolve, MESSAGE_DELAY));
+    }
+  }
+
+  console.log('\n🎯 Image-first batching test completed!');
+  console.log('====================================================');
+  console.log('Expected behavior:');
+  console.log('- Image and text messages should be batched together');
+  console.log('- No "Error handling batched messages" or run conflicts');
+  console.log('- Only ONE assistant response should be generated');
+  console.log('- You should see "Processing batched messages from [chatId] (including media if any)"');
+  console.log('====================================================\n');
+  
+  console.log('🔍 If you see errors or multiple responses:');
+  console.log('1. Check for "Error handling batched messages" in server logs');
+  console.log('2. Look for OpenAI run conflicts');
+  console.log('3. Verify image messages are being included in batching');
+  console.log('4. Check that messages are marked as processed correctly');
+}
+
+// Main execution
+async function main() {
+  const testType = process.argv[2] || 'batching';
+  
+  switch (testType) {
+    case 'batching':
+      await testBatching();
+      break;
+    case 'documents':
+      await testBatchingWithDocuments();
+      break;
+    case 'images':
+      await testBatchingWithImages();
+      break;
+    case 'mixed':
+      await testMixedBatching();
+      break;
+    case 'mixed-images':
+      await testMixedImageBatching();
+      break;
+    case 'media-first':
+      await testMediaFirstBatching();
+      break;
+    case 'image-first':
+      await testImageFirstBatching();
+      break;
+    case 'diagnostic':
+      await testBatchingDiagnostic();
+      break;
+    case 'correct':
+      await testBatchingCorrectly();
+      break;
+    case 'control':
+      await testWithoutBatching();
+      break;
+    case 'single':
+      await testSingleMessage();
+      break;
+    case 'all':
+      console.log('🧪 Running ALL Tests\n');
+      await testSingleMessage();
+      console.log('\n' + '='.repeat(50) + '\n');
+      await testBatching();
+      console.log('\n' + '='.repeat(50) + '\n');
+      await testBatchingWithDocuments();
+      console.log('\n' + '='.repeat(50) + '\n');
+      await testBatchingWithImages();
+      console.log('\n' + '='.repeat(50) + '\n');
+      await testMixedBatching();
+      console.log('\n' + '='.repeat(50) + '\n');
+      await testMixedImageBatching();
+      console.log('\n' + '='.repeat(50) + '\n');
+      await testMediaFirstBatching();
+      console.log('\n' + '='.repeat(50) + '\n');
+      await testImageFirstBatching();
+      console.log('\n' + '='.repeat(50) + '\n');
+      await testWithoutBatching();
+      break;
+    default:
+      console.log('Usage: node test-batching.js [batching|documents|images|mixed|mixed-images|media-first|image-first|diagnostic|correct|control|single|all]');
+      console.log('');
+      console.log('Tests:');
+      console.log('  batching      - Test message batching with text only (default)');
+      console.log('  documents     - Test message batching with documents only');
+      console.log('  images        - Test message batching with images only');
+      console.log('  mixed         - Test message batching with text + documents');
+      console.log('  mixed-images  - Test message batching with text + images');
+      console.log('  media-first   - Test media-first batching (fixes run conflicts)');
+      console.log('  image-first   - Test image-first batching (fixes run conflicts)');
+      console.log('  diagnostic    - Diagnostic test to debug batching issues');
+      console.log('  correct       - Correct batching test (3 messages, all batched)');
+      console.log('  control       - Test without batching (long delays)');
+      console.log('  single        - Test single message (no batching)');
+      console.log('  all           - Run all tests');
+      process.exit(1);
+  }
+}
+
+// Handle graceful shutdown
+process.on('SIGINT', () => {
+  console.log('\n👋 Test interrupted by user');
+  process.exit(0);
+});
+
+main().catch(console.error);

package/lib/assistants/BaseAssistant.js

@@ -9,7 +9,6 @@ const { createProvider } = require('../providers/createProvider');
  */
 class BaseAssistant {
   constructor(options = {}) {
-    console.log('options', options);
     this.assistantId = options.assistantId || null;
     this.thread = options.thread || null;
     this.status = options.status || 'idle';

package/lib/core/NexusMessaging.js

@@ -338,8 +338,6 @@ class NexusMessaging {
     // Handle immediate response messages (interactive, flows, etc.)
     if (messageData.interactive) {
       return await this.handleInteractive(messageData);
-    } else if (messageData.media) {
-      return await this.handleMedia(messageData);
     } else if (messageData.command) {
       return await this.handleCommand(messageData);
     } else if (messageData.keyword) {
@@ -347,12 +345,17 @@ class NexusMessaging {
     } else if (messageData.flow) {
       return await this.handleFlow(messageData);
     } else {
-      // For regular messages, use batching if enabled
+      // For regular messages and media, use batching if enabled
       console.log('Batching config:', this.batchingConfig);
       if (this.batchingConfig.enabled && chatId) {
         return await this._handleWithBatching(messageData, chatId);
       } else {
-        return await this.handleMessage(messageData);
+        // Handle media and regular messages without batching
+        if (messageData.media) {
+          return await this.handleMedia(messageData);
+        } else {
+          return await this.handleMessage(messageData);
+        }
       }
     }
   }
@@ -631,7 +634,7 @@ class NexusMessaging {
    */
   async _handleBatchedMessages(chatId) {
     try {
-      console.log(`Processing batched messages from ${chatId}`);
+      console.log(`Processing batched messages from ${chatId} (including media if any)`);
       
       // Get assistant response
       const botResponse = await replyAssistant(chatId);

package/lib/providers/OpenAIAssistantsProvider.js

@@ -155,7 +155,8 @@ class OpenAIAssistantsProvider {
   }
 
   async getRun({ threadId, runId }) {
-    return this.client.beta.threads.runs.retrieve(this._ensureId(threadId), this._ensureId(runId));
+    const run = await this.client.beta.threads.runs.retrieve(this._ensureId(runId), { thread_id: this._ensureId(threadId) });
+    return run;
   }
 
   async listRuns({ threadId, limit, order = 'desc', activeOnly = false } = {}) {
@@ -177,16 +178,18 @@ class OpenAIAssistantsProvider {
 
   async submitToolOutputs({ threadId, runId, toolOutputs }) {
     return this.client.beta.threads.runs.submitToolOutputs(
-      this._ensureId(threadId),
       this._ensureId(runId),
-      { tool_outputs: toolOutputs }
+      { 
+        thread_id: this._ensureId(threadId) ,
+        tool_outputs: toolOutputs
+      }
     );
   }
 
   async cancelRun({ threadId, runId }) {
     return this.client.beta.threads.runs.cancel(
-      this._ensureId(threadId),
-      this._ensureId(runId)
+      this._ensureId(runId),
+      { thread_id: this._ensureId(threadId) }
     );
   }
 

package/lib/services/assistantService.js

@@ -60,7 +60,8 @@ const runAssistantAndWait = async ({
     let completed = false;
 
     try {
-      completed = await checkRunStatus(assistant, run.thread_id, run.id, 0, maxRetries);
+      console.log('RUN ID', run.id, thread.thread_id);
+      completed = await checkRunStatus(assistant, thread.thread_id, run.id, 0, maxRetries);
     } finally {
       if (filter) {
         await Thread.updateOne(filter, { $set: { run_id: null } });
@@ -395,7 +396,7 @@ const replyAssistant = async function (code, message_ = null, thread_ = null, ru
 
     if (!patientMsg) return null;
 
-    const assistant = getAssistantById(thread?.prompt_id || thread?.assistant_id, thread);
+    const assistant = getAssistantById(process.env.VARIANT === 'responses' ? thread?.prompt_id : thread?.assistant_id, thread);
     assistant.setReplies(patientReply);
 
     const { run, output, completed } = await runAssistantAndWait({

package/package.json

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