@peopl-health/nexus 6.0.0-dev.702 → 6.0.0-dev.718

package/lib/clinical/AssistantProcessor.js

@@ -2,6 +2,7 @@ const { sanitizeOutput } = require('../utils/formatUtils');
 
 const { getThread } = require('../helpers/threadHelper');
 const { runAssistantWithRetries } = require('./helpers/assistantHelper');
+const { overlayCohortPreset } = require('./helpers/cohortHelper');
 
 const { forkShadowTurn } = require('./services/shadowService');
 const { diverge } = require('./services/divergenceService');
@@ -18,8 +19,9 @@ class AssistantProcessor {
   setSendMessage(fn) { this.sendMessage = fn; }
 
   async resolveThread(code) {
-    const thread = await getThread(code);
-    if (!thread) return null;
+    const stored = await getThread(code);
+    if (!stored) return null;
+    const thread = await overlayCohortPreset(stored);
     const assistant = getAssistantById(thread.getAssistantId(), thread);
     return { thread, assistant };
   }

package/lib/clinical/config/cohortConfig.js

@@ -0,0 +1,68 @@
+const { Config_ID } = require('../../config/airtableConfig');
+const runtimeConfig = require('../../config/runtimeConfig');
+
+const { stripWhatsAppFormat } = require('../../helpers/twilioHelper');
+const { getRecordByFilter } = require('../../services/airtableService');
+
+const { logger } = require('../../utils/logger');
+const MapCache = require('../../utils/MapCache');
+
+const COHORT_TABLE = 'clinical_cohort';
+const CACHE_TTL = 5 * 60 * 1000;
+const CACHE_KEY = 'cohort';
+const LANES = ['preset', 'bridge', 'fhir_write', 'fhir_read'];
+
+const cache = new MapCache({ maxSize: 1, ttl: CACHE_TTL });
+let inFlightRead = null;
+let generation = 0;
+
+function isCohortEnabled() {
+  return runtimeConfig.get('COHORT_ENABLED', 'false') === 'true';
+}
+
+async function readRegistry() {
+  const records = await getRecordByFilter(Config_ID, COHORT_TABLE, 'TRUE()');
+  if (!records) throw new Error('cohort registry could not be read from the Config base');
+
+  const lanesByCode = new Map();
+  for (const row of records) {
+    if (!row.whatsapp_id || row.status !== 'active') continue;
+    lanesByCode.set(stripWhatsAppFormat(row.whatsapp_id), new Set(row.lanes || []));
+  }
+  return lanesByCode;
+}
+
+async function load() {
+  const cached = cache.get(CACHE_KEY);
+  if (cached) return cached;
+  if (!inFlightRead) {
+    const readGeneration = generation;
+    inFlightRead = readRegistry()
+      .then((lanesByCode) => {
+        if (readGeneration === generation) cache.set(CACHE_KEY, lanesByCode);
+        return lanesByCode;
+      })
+      .finally(() => { inFlightRead = null; });
+  }
+  return inFlightRead;
+}
+
+async function isInCohort(code, lane) {
+  if (!LANES.includes(lane)) throw new Error(`unknown cohort lane: ${lane}`);
+  if (!code || !isCohortEnabled()) return false;
+  try {
+    const lanes = (await load()).get(stripWhatsAppFormat(code));
+    return Boolean(lanes && lanes.has(lane));
+  } catch (error) {
+    logger.error('[cohortConfig] registry read failed; gate fails closed', { code, lane, error: error.message });
+    return false;
+  }
+}
+
+function clearCohortCache() {
+  generation += 1;
+  cache.clear();
+  inFlightRead = null;
+}
+
+module.exports = { isInCohort, clearCohortCache };

package/lib/clinical/helpers/cohortHelper.js

@@ -0,0 +1,20 @@
+const { isInCohort } = require('../config/cohortConfig');
+const { Thread } = require('../../models/threadModel');
+
+const runtimeConfig = require('../../config/runtimeConfig');
+const { logger } = require('../../utils/logger');
+
+async function overlayCohortPreset(thread) {
+  const enrolled = await isInCohort(thread.code, 'preset');
+  if (!enrolled) return thread;
+  try {
+    const base = thread.toObject();
+    delete base._id;
+    return new Thread({ ...base, preset_id: runtimeConfig.get('COHORT_PRESET_ID', 'clinical_agent'), preset_version: null });
+  } catch (error) {
+    logger.error('[cohortHelper] overlay failed; patient stays on the control path', { code: thread.code, error: error.message });
+    return thread;
+  }
+}
+
+module.exports = { overlayCohortPreset };

package/lib/controllers/cohortController.js

@@ -0,0 +1,16 @@
+const { logger } = require('../utils/logger');
+
+const { clearCohortCache } = require('../clinical/config/cohortConfig');
+
+const clearCohortCacheController = async (req, res) => {
+  try {
+    clearCohortCache();
+    logger.info('[CohortController] Cohort cache cleared', { by: req.ip });
+    return res.status(200).json({ success: true, message: 'Cohort cache cleared' });
+  } catch (error) {
+    logger.error('[CohortController] Failed to clear cohort cache', { error: error.message });
+    return res.status(500).json({ success: false, error: 'Failed to clear cohort cache' });
+  }
+};
+
+module.exports = { clearCohortCacheController };

package/lib/helpers/twilioHelper.js

@@ -121,10 +121,18 @@ const ensureWhatsAppFormat = (code) => {
   return trimmed;
 };
 
+const stripWhatsAppFormat = (code) => {
+  const raw = String(code).trim();
+  const unprefixed = raw.startsWith('whatsapp:') ? raw.slice('whatsapp:'.length) : raw;
+  const beforeAt = unprefixed.includes('@') ? unprefixed.slice(0, unprefixed.indexOf('@')) : unprefixed;
+  return beforeAt.replace(/^\+/, '');
+};
+
 module.exports = {
   convertTwilioToInternalFormat,
   downloadMediaFromTwilio,
   getMediaTypeFromContentType,
   extractTitle,
-  ensureWhatsAppFormat
+  ensureWhatsAppFormat,
+  stripWhatsAppFormat
 };

package/lib/routes/index.js

@@ -90,6 +90,10 @@ const presetRouteDefinitions = {
   'POST /cache/clear': 'clearPresetCacheController'
 };
 
+const cohortRouteDefinitions = {
+  'POST /cache/clear': 'clearCohortCacheController'
+};
+
 const createRouter = (routeDefinitions, controllers) => {
   const router = express.Router();
   
@@ -115,6 +119,7 @@ const messageController = require('../controllers/messageController');
 const messageStatusController = require('../controllers/messageStatusController');
 const patientController = require('../controllers/patientController');
 const presetController = require('../controllers/presetController');
+const cohortController = require('../controllers/cohortController');
 const qualityMessageController = require('../controllers/qualityMessageController');
 const templateController = require('../controllers/templateController');
 const templateFlowController = require('../controllers/templateFlowController');
@@ -181,6 +186,9 @@ const builtInControllers = {
   // Preset controllers
   clearPresetCacheController: presetController.clearPresetCacheController,
 
+  // Cohort controllers
+  clearCohortCacheController: cohortController.clearCohortCacheController,
+
   // Template controllers
   createTemplate: templateController.createTemplate,
   listTemplates: templateController.listTemplates,
@@ -223,6 +231,7 @@ const setupDefaultRoutes = (app) => {
   app.use('/api/message', createRouter(messageRouteDefinitions, builtInControllers));
   app.use('/api/patient', createRouter(patientRouteDefinitions, builtInControllers));
   app.use('/api/preset', createRouter(presetRouteDefinitions, builtInControllers));
+  app.use('/api/cohort', createRouter(cohortRouteDefinitions, builtInControllers));
   app.use('/api/template', createRouter(templateRouteDefinitions, builtInControllers));
   app.use('/api/prescription', createRouter(prescriptionRouteDefinitions, builtInControllers));
   app.use('/api/dashboard', createRouter(dashboardRouteDefinitions, builtInControllers));
@@ -235,6 +244,7 @@ module.exports = {
   messageRoutes: messageRouteDefinitions,
   patientRoutes: patientRouteDefinitions,
   presetRoutes: presetRouteDefinitions,
+  cohortRoutes: cohortRouteDefinitions,
   templateRoutes: templateRouteDefinitions,
   prescriptionRoutes: prescriptionRouteDefinitions,
   dashboardRoutes: dashboardRouteDefinitions,

package/package.json

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