@peopl-health/nexus 1.1.4 → 1.1.6

package/lib/services/assistantService.js

@@ -1,6 +1,12 @@
 const { Historial_Clinico_ID } = require('../config/airtableConfig.js');
-const { SALES_ASST, QUMIO_REMINDERS_ASST, DOCTOR_SCHEDULE_ASST, PATIENT_SCHEDULE_ASST } = require('../config/assistantConfig.js');
-const { openaiClient } = require('../config/llmConfig.js');
+
+let llmProvider = null;
+const configureLLMProvider = (provider) => {
+  llmProvider = provider;
+};
+
+let assistantConfig = null;
+let assistantRegistry = {};
 
 const { Message, formatTimestamp } = require('../models/messageModel.js');
 const { Thread } = require('../models/threadModel.js');
@@ -9,26 +15,28 @@ const { checkRunStatus, getCurRow } = require('../helpers/assistantHelper.js');
 const { processMessage, getLastMessages } = require('../helpers/assistantHelper.js');
 const { delay } = require('../helpers/whatsappHelper.js');
 
-const { GeneralAssistant } = require('../assistants/generalAssistant.js');
-const { SalesAssistant } = require('../assistants/salesAssistant.js');
-const { QumioRemindersAssistant } = require('../assistants/qumioRemindersAssistant.js');
-const { DoctorScheduleAssistant } = require('../assistants/doctorScheduleAssistant.js');
-const { PatientScheduleAssistant } = require('../assistants/patientScheduleAssistant.js');
+const configureAssistants = (config) => {
+  if (!config) {
+    throw new Error('Assistant configuration is required');
+  }
+  assistantConfig = config;
+};
 
+const registerAssistant = (assistantId, AssistantClass) => {
+  assistantRegistry[assistantId] = AssistantClass;
+};
 
 const getAssistantById = (assistant_id, thread) => {
-  switch (assistant_id) {
-  case SALES_ASST:
-    return new SalesAssistant(thread);
-  case QUMIO_REMINDERS_ASST:
-    return new QumioRemindersAssistant(thread);
-  case DOCTOR_SCHEDULE_ASST:
-    return new DoctorScheduleAssistant(thread);
-  case PATIENT_SCHEDULE_ASST:
-    return new PatientScheduleAssistant(thread);
-  default:
-    return new GeneralAssistant(thread);
+  if (!assistantConfig) {
+    throw new Error('Assistants not configured. Call configureAssistants() first.');
+  }
+  
+  const AssistantClass = assistantRegistry[assistant_id];
+  if (!AssistantClass) {
+    throw new Error(`Assistant '${assistant_id}' not found. Available assistants: ${Object.keys(assistantRegistry).join(', ')}`);
   }
+  
+  return new AssistantClass(thread);
 };
 
 
@@ -50,7 +58,7 @@ const createAssistant = async (code, assistant_id, messages=[], prevThread=null)
 
   // Add new messages to memory
   for (const message of messages) {
-    await openaiClient.beta.threads.messages.create(
+    await llmProvider.beta.threads.messages.create(
       initialThread.id, { role: 'assistant', content: message }
     );
   }
@@ -73,7 +81,7 @@ const createAssistant = async (code, assistant_id, messages=[], prevThread=null)
 
   // Delete previous thread
   if (prevThread) {
-    await openaiClient.beta.threads.del(prevThread.thread_id);
+    await llmProvider.beta.threads.del(prevThread.thread_id);
   }
 
   return thread;
@@ -87,7 +95,7 @@ const addMsgAssistant = async (code, inMessages, reply = false) => {
 
     for (const message of inMessages) {
       console.log(message);
-      await openaiClient.beta.threads.messages.create(
+      await llmProvider.beta.threads.messages.create(
         thread.thread_id, { role: 'assistant', content: message }
       );
     }
@@ -95,7 +103,7 @@ const addMsgAssistant = async (code, inMessages, reply = false) => {
     if (!reply) return null;
 
     const assistant = getAssistantById(thread.assistant_id, thread);
-    const run = await openaiClient.beta.threads.runs.create(
+    const run = await llmProvider.beta.threads.runs.create(
       thread.thread_id, 
       { 
         assistant_id: thread.assistant_id
@@ -106,7 +114,7 @@ const addMsgAssistant = async (code, inMessages, reply = false) => {
     await checkRunStatus(assistant, run.thread_id, run.id);
     await Thread.updateOne({ code: thread.code, active: true }, { $set: { run_id: null } });  
 
-    const messages = await openaiClient.beta.threads.messages.list(run.thread_id, { run_id: run.id });
+    const messages = await llmProvider.beta.threads.messages.list(run.thread_id, { run_id: run.id });
     const ans = messages.data[0].content[0].text.value;
     console.log('THE ANS IS', ans);
         
@@ -124,7 +132,7 @@ const addInsAssistant = async (code, instruction) => {
     if (thread === null) return null;
 
     const assistant = getAssistantById(thread.assistant_id, thread);
-    const run = await openaiClient.beta.threads.runs.create(
+    const run = await llmProvider.beta.threads.runs.create(
       thread.thread_id, { 
         assistant_id: thread.assistant_id,
         additional_instructions: instruction,
@@ -138,7 +146,7 @@ const addInsAssistant = async (code, instruction) => {
     await checkRunStatus(assistant, run.thread_id, run.id);
     await Thread.updateOne({ code: thread.code, active: true }, { $set: { run_id: null } });  
 
-    const messages = await openaiClient.beta.threads.messages.list(run.thread_id, { run_id: run.id });
+    const messages = await llmProvider.beta.threads.messages.list(run.thread_id, { run_id: run.id });
     console.log(messages.data[0].content);
     const ans = messages.data[0].content[0].text.value;
         
@@ -166,7 +174,7 @@ const getThread = async (code, message = null) => {
 
     while (thread && thread.run_id) {
       console.log(`Wait for ${thread.run_id} to be executed`);
-      const run = await openaiClient.beta.threads.runs.retrieve(thread.thread_id, thread.run_id);
+      const run = await llmProvider.beta.threads.runs.retrieve(thread.thread_id, thread.run_id);
       if (run.status === 'cancelled' || run.status === 'expired' || run.status === 'completed') {
         await Thread.updateOne({ code: code }, { $set: { run_id: null } });
       }
@@ -204,11 +212,11 @@ const replyAssistant = async function (code, message_ = null, thread_ = null, ru
       return null;
     }
 
-    let activeRuns = await openaiClient.beta.threads.runs.list(thread.thread_id);
+    let activeRuns = await llmProvider.beta.threads.runs.list(thread.thread_id);
     console.log('ACTIVE RUNS:', activeRuns.length);
     while (activeRuns.length > 0) {
       console.log(`ACTIVE RUNS ${thread.thread_id}`);
-      activeRuns = await openaiClient.beta.threads.runs.list(thread.thread_id);
+      activeRuns = await llmProvider.beta.threads.runs.list(thread.thread_id);
       await delay(5000);
     }
 
@@ -223,21 +231,6 @@ const replyAssistant = async function (code, message_ = null, thread_ = null, ru
        
     if (urls.length > 0) {
       console.log('urls', urls);
-      /*for (const url of urls) {
-        console.log("url", url);
-        await addRecord(Monitoreo_ID, 'estudios', [{"fields": {"estudios": urls, 
-          "combined_estudios": [ { "url": url} ], "patient_id": [thread.patient_id]}}]);
-      }
-      const { pdfBuffer, processedFiles } = await combineImagesToPDF(code);
-      console.log("AFTER COMBINED IN BUFFER", processedFiles);
-      const key = `${code}-${Date.now()}-combined.pdf`;
-      await AWS.uploadBufferToS3(pdfBuffer, bucketName, key, "application/pdf");
-      const url = await AWS.generatePresignedUrl(bucketName, key);
-      console.log("New record", {"estudios": urls, "combined_estudios": 
-        [ { "url": url} ], "patient_id": [thread.patient_id]});
-      await addRecord(Monitoreo_ID, 'estudios', [{"fields": {"estudios": urls, 
-        "combined_estudios": [ { "url": url} ], "patient_id": [thread.patient_id]}}]);
-      await cleanupFiles(processedFiles);*/
     }
 
     thread = await getThread(code);
@@ -245,7 +238,7 @@ const replyAssistant = async function (code, message_ = null, thread_ = null, ru
     if (!patientMsg || !thread || thread?.stopped) return null;
 
     const assistant = getAssistantById(thread.assistant_id, thread);
-    const run = await openaiClient.beta.threads.runs.create(
+    const run = await llmProvider.beta.threads.runs.create(
       thread.thread_id, 
       { 
         assistant_id: thread.assistant_id,
@@ -261,7 +254,7 @@ const replyAssistant = async function (code, message_ = null, thread_ = null, ru
     console.log('RUN STATUS', runStatus);
     await Thread.updateOne({ code: thread.code, active: true }, { $set: { run_id: null } });  
 
-    const messages = await openaiClient.beta.threads.messages.list(run.thread_id, { run_id: run.id });
+    const messages = await llmProvider.beta.threads.messages.list(run.thread_id, { run_id: run.id });
     const reply = messages.data?.[0]?.content?.[0]?.text?.value || '';
     console.log(reply);
 
@@ -292,5 +285,8 @@ module.exports = {
   replyAssistant,
   addMsgAssistant,
   addInsAssistant,
-  switchAssistant
+  switchAssistant,
+  configureAssistants,
+  registerAssistant,
+  configureLLMProvider
 };

package/lib/storage/MongoStorage.js

@@ -54,9 +54,9 @@ class MongoStorage {
     });
 
     return {
-      Message: mongoose.model('Message', messageSchema),
-      Interaction: mongoose.model('Interaction', interactionSchema),
-      Thread: mongoose.model('Thread', threadSchema)
+      Message: mongoose.models.Message || mongoose.model('Message', messageSchema),
+      Interaction: mongoose.models.Interaction || mongoose.model('Interaction', interactionSchema),
+      Thread: mongoose.models.Thread || mongoose.model('Thread', threadSchema)
     };
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@peopl-health/nexus",
-  "version": "1.1.4",
+  "version": "1.1.6",
   "description": "Core messaging and assistant library for WhatsApp communication platforms",
   "publishConfig": {
     "access": "public"
@@ -51,10 +51,12 @@
   "license": "MIT",
   "dependencies": {
     "airtable": "^0.12.2",
+    "aws-sdk": "2.1674.0",
     "axios": "^1.5.0",
     "dotenv": "^16.4.7",
     "moment-timezone": "^0.5.43",
     "mongoose": "^7.5.0",
+    "pdf-lib": "1.17.1",
     "pino": "^8.15.0",
     "pino-pretty": "^10.2.0",
     "uuid": "^9.0.0"
@@ -69,6 +71,7 @@
     "@types/node": "^20.5.0",
     "eslint": "^8.47.0",
     "jest": "^29.6.2",
+    "sharp": "0.32.6",
     "typescript": "^5.1.6"
   },
   "engines": {

package/lib/services/whatsappService.js

@@ -1,23 +0,0 @@
-let sendMessage;
-let sendScheduledMessage;
-if (process.env.MESSAGING_PROVIDER === 'baileys') {
-  const baileyAdapter = require('../adapters/baileys');
-  console.log('Baileys adapter:', baileyAdapter);
-  console.log('Has sendMessage?', 'sendMessage' in baileyAdapter);
-  sendMessage = baileyAdapter.sendMessage;
-  sendScheduledMessage = baileyAdapter.sendScheduledMessage;
-} else if (process.env.MESSAGING_PROVIDER === 'twilio') {
-  const { sendMessage: twilioSendMessage } = require('../messaging/messageService');
-  const { sendScheduledMessage: twilioSendScheduledMessage } = require('../messaging/scheduledMessageService');
-  const twilioAdapter = require('../adapters/twilio');
-  const twilioClient = twilioAdapter.twilioClient;
-  sendMessage = (messageData) => twilioSendMessage(twilioClient, messageData);
-  sendScheduledMessage = (messageData) => twilioSendScheduledMessage(twilioClient, messageData);
-} else {
-  throw new Error('Unsupported MESSAGING_PROVIDER specified');
-}
-
-module.exports = {
-  sendMessage,
-  sendScheduledMessage
-};