@scout9/app 1.0.0-alpha.0.6.4 → 1.0.0-alpha.0.6.6

package/dist/{dev-fee40f1c.cjs → dev-df88036b.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var spirits = require("./spirits-e626085b.cjs");
+var spirits = require("./spirits-9719ae4f.cjs");
 var require$$0$4 = require('util');
 var require$$0$3 = require('stream');
 var require$$1 = require('path');
@@ -21,7 +21,7 @@ var node_url = require('node:url');
 var node_events = require('node:events');
 var Stream = require('node:stream');
 var node_string_decoder = require('node:string_decoder');
-var macros = require("./macros-a37c4a77.cjs");
+var macros = require("./macros-1a4fd407.cjs");
 var readline = require('node:readline');
 var process$2 = require('node:process');
 var node_os = require('node:os');
@@ -35241,7 +35241,7 @@ function _loadUserPackageJson() {
           targetPkgUrl = isTest ? packageTestJsonUrl : packageJsonUrl;
           _context2.t0 = JSON;
           _context2.next = 10;
-          return fs__default["default"].readFile(new URL(targetPkgUrl, (typeof document === 'undefined' ? new (require('u' + 'rl').URL)('file:' + __filename).href : (document.currentScript && document.currentScript.src || new URL('dev-fee40f1c.js', document.baseURI).href))), 'utf-8');
+          return fs__default["default"].readFile(new URL(targetPkgUrl, (typeof document === 'undefined' ? new (require('u' + 'rl').URL)('file:' + __filename).href : (document.currentScript && document.currentScript.src || new URL('dev-df88036b.js', document.baseURI).href))), 'utf-8');
         case 10:
           _context2.t1 = _context2.sent;
           pkg = _context2.t0.parse.call(_context2.t0, _context2.t1);

package/dist/{index-b59a82b2.cjs → index-33d77bc4.cjs}

@@ -1,8 +1,8 @@
 'use strict';
 
-var spirits = require("./spirits-e626085b.cjs");
-var dev = require("./dev-fee40f1c.cjs");
-var macros = require("./macros-a37c4a77.cjs");
+var spirits = require("./spirits-9719ae4f.cjs");
+var dev = require("./dev-df88036b.cjs");
+var macros = require("./macros-1a4fd407.cjs");
 var require$$0 = require('fs');
 var require$$2$1 = require('events');
 var require$$1 = require('path');
@@ -29483,7 +29483,7 @@ class Body {
     }
     const {
       toFormData
-    } = await Promise.resolve().then(function () { return require("./multipart-parser-06b501ee.cjs"); });
+    } = await Promise.resolve().then(function () { return require("./multipart-parser-17ab0a54.cjs"); });
     return toFormData(this.body, ct);
   }
 
@@ -41884,7 +41884,7 @@ function _loadUserPackageJson() {
           targetPkgUrl = isTest ? packageTestJsonUrl : packageJsonUrl;
           _context.t0 = JSON;
           _context.next = 10;
-          return fs__default["default"].readFile(new URL(targetPkgUrl, (typeof document === 'undefined' ? new (require('u' + 'rl').URL)('file:' + __filename).href : (document.currentScript && document.currentScript.src || new URL('index-b59a82b2.js', document.baseURI).href))), 'utf-8');
+          return fs__default["default"].readFile(new URL(targetPkgUrl, (typeof document === 'undefined' ? new (require('u' + 'rl').URL)('file:' + __filename).href : (document.currentScript && document.currentScript.src || new URL('index-33d77bc4.js', document.baseURI).href))), 'utf-8');
         case 10:
           _context.t1 = _context.sent;
           pkg = _context.t0.parse.call(_context.t0, _context.t1);
@@ -43176,7 +43176,7 @@ var ProjectFiles = /*#__PURE__*/function () {
   return ProjectFiles;
 }();
 
-var __filename$1 = node_url.fileURLToPath((typeof document === 'undefined' ? new (require('u' + 'rl').URL)('file:' + __filename).href : (document.currentScript && document.currentScript.src || new URL('index-b59a82b2.js', document.baseURI).href)));
+var __filename$1 = node_url.fileURLToPath((typeof document === 'undefined' ? new (require('u' + 'rl').URL)('file:' + __filename).href : (document.currentScript && document.currentScript.src || new URL('index-33d77bc4.js', document.baseURI).href)));
 var __dirname$1 = path__default["default"].dirname(__filename$1);
 function zipDirectory(source, out) {
   var archive = archiver$1('tar', {
@@ -43391,7 +43391,7 @@ function _buildApp() {
         case 11:
           _context4.t0 = JSON;
           _context4.next = 14;
-          return fs__default["default"].readFile(new URL(templatePackagePath, (typeof document === 'undefined' ? new (require('u' + 'rl').URL)('file:' + __filename).href : (document.currentScript && document.currentScript.src || new URL('index-b59a82b2.js', document.baseURI).href))), 'utf-8');
+          return fs__default["default"].readFile(new URL(templatePackagePath, (typeof document === 'undefined' ? new (require('u' + 'rl').URL)('file:' + __filename).href : (document.currentScript && document.currentScript.src || new URL('index-33d77bc4.js', document.baseURI).href))), 'utf-8');
         case 14:
           _context4.t1 = _context4.sent;
           packageTemplate = _context4.t0.parse.call(_context4.t0, _context4.t1);

package/dist/index.cjs

@@ -2,10 +2,10 @@
 
 Object.defineProperty(exports, '__esModule', { value: true });
 
-var index = require("./index-b59a82b2.cjs");
-var dev = require("./dev-fee40f1c.cjs");
-require("./spirits-e626085b.cjs");
-require("./macros-a37c4a77.cjs");
+var index = require("./index-33d77bc4.cjs");
+var dev = require("./dev-df88036b.cjs");
+require("./spirits-9719ae4f.cjs");
+require("./macros-1a4fd407.cjs");
 require('fs');
 require('events');
 require('path');

package/dist/{macros-a37c4a77.cjs → macros-1a4fd407.cjs}

@@ -4047,7 +4047,13 @@ var MessageSchema = z.object({
   delayInSeconds: z.number({
     description: 'How long to delay the message manually in seconds'
   }).nullable().optional(),
-  entities: z.array(MessageEntitySchema).optional().nullable()
+  entities: z.array(MessageEntitySchema).optional().nullable(),
+  ignoreTransform: z["boolean"]({
+    description: 'If set to true, the PMT will not transform, message will be sent as is'
+  }).optional(),
+  mediaUrls: z.array(z.string(), {
+    description: 'Media urls to attach to the transported message'
+  }).optional().nullable()
 });
 
 var CustomerValueSchema = z.union([z["boolean"](), z.number(), z.string()]);

package/dist/{multipart-parser-06b501ee.cjs → multipart-parser-17ab0a54.cjs}

@@ -2,9 +2,9 @@
 
 require('node:fs');
 require('node:path');
-var index = require("./index-b59a82b2.cjs");
-require("./spirits-e626085b.cjs");
-require("./dev-fee40f1c.cjs");
+var index = require("./index-33d77bc4.cjs");
+require("./spirits-9719ae4f.cjs");
+require("./dev-df88036b.cjs");
 require('util');
 require('stream');
 require('path');
@@ -23,7 +23,7 @@ require('node:url');
 require('node:events');
 require('node:stream');
 require('node:string_decoder');
-require("./macros-a37c4a77.cjs");
+require("./macros-1a4fd407.cjs");
 require('node:readline');
 require('node:process');
 require('node:os');

package/dist/schemas.cjs

@@ -2,7 +2,7 @@
 
 Object.defineProperty(exports, '__esModule', { value: true });
 
-var macros = require("./macros-a37c4a77.cjs");
+var macros = require("./macros-1a4fd407.cjs");
 
 
 

package/dist/{spirits-e626085b.cjs → spirits-9719ae4f.cjs}

@@ -730,6 +730,7 @@ var _excluded = ["keywords"];
  * @property {Change<any>} context
  * @property {Change<import('@scout9/app').Message>} message
  * @property {Array<import('@scout9/app').Followup>} followup
+ * @property {Array<import('@scout9/app').EntityContextUpsert>} entityContextUpsert
  */
 var Spirits = {
   /**
@@ -740,13 +741,14 @@ var Spirits = {
   customer: function () {
     var _customer = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(input) {
       var _recentUserMessage;
-      var customer, config, parser, workflow, generator, transformer, idGenerator, _input$progress, progress, messageBefore, contextBefore, messagesBefore, conversationBefore, conversation, messages, context, message, followup, updateConversation, updateContext, userMessages, recentUserMessage, lockConversation, incrementLockAttempt, _addInstruction, addInstruction, persona, invalidRoles, parsePayload, index, _message, previousUserMessages, oldKeyCount, newKeyCount, noNewContext, slots, hasNoInstructions, hasNoCustomMessage, messagesToTransform, previousLockAttempt, resettedIntent, _forward, _forwardNote, _iterator, _step, _step$value, forward, forwardNote, instructions, removeInstructions, manualMessage, scheduled, resetIntent, secondsDelay, contextUpsert, anticipate, slotFollowup, _slots, map, i, _anticipate$i, keywords, _slot, slotId, _iterator3, _step3, instruction, _iterator4, _step4, _loop, manualMessageObj, now, generatorPayload, agentMessages, lastAgentMessage, _generatorPayload$ent, _generatorPayload$con, _iterator2, _step2, transformResponse, _agentMessages, _lastAgentMessage;
+      var customer, config, parser, workflow, generator, transformer, idGenerator, _input$progress, progress, messageBefore, contextBefore, messagesBefore, conversationBefore, conversation, messages, context, message, followup, entityContextUpsert, updateConversation, updateContext, userMessages, recentUserMessage, lockConversation, incrementLockAttempt, _addInstruction, addInstruction, persona, invalidRoles, parsePayload, index, _message, previousUserMessages, oldKeyCount, newKeyCount, noNewContext, slots, hasNoInstructions, hasNoCustomMessage, messagesToTransform, previousLockAttempt, resettedIntent, _forward, _forwardNote, _iterator, _step, _step$value, forward, forwardNote, instructions, removeInstructions, manualMessage, scheduled, resetIntent, secondsDelay, contextUpsert, anticipate, slotFollowup, slotEntityContextUpsert, _slots, map, i, _anticipate$i, keywords, _slot, slotId, _iterator3, _step3, instruction, _iterator4, _step4, _loop, manualMessageObj, now, generatorPayload, agentMessages, lastAgentMessage, _generatorPayload$ent, _generatorPayload$con, _iterator2, _step2, transformResponse, _agentMessages, _lastAgentMessage;
       return _regeneratorRuntime().wrap(function _callee$(_context2) {
         while (1) switch (_context2.prev = _context2.next) {
           case 0:
             customer = input.customer, config = input.config, parser = input.parser, workflow = input.workflow, generator = input.generator, transformer = input.transformer, idGenerator = input.idGenerator, _input$progress = input.progress, progress = _input$progress === void 0 ? function (message, level, type, payload) {} : _input$progress, messageBefore = input.message, contextBefore = input.context, messagesBefore = input.messages, conversationBefore = input.conversation;
-            conversation = input.conversation, messages = input.messages, context = input.context, message = input.message;
-            followup = []; // 0. Setup Helpers
+            conversation = input.conversation, messages = input.messages, context = input.context, message = input.message; // Storing post process events here
+            followup = [];
+            entityContextUpsert = []; // 0. Setup Helpers
             updateConversation = function updateConversation(previousConversation, conversationUpdates) {
               progress('Update conversation', 'info', 'UPDATE_CONVERSATION', conversationUpdates);
               return _objectSpread2(_objectSpread2({}, previousConversation), conversationUpdates);
@@ -835,40 +837,40 @@ var Spirits = {
               }
             }; // 1. Check inputs
             if (conversation.$agent) {
-              _context2.next = 13;
+              _context2.next = 14;
               break;
             }
             throw new Error("SpiritsError: No agent found in conversation, must define \".$agent\" in the conversation");
-          case 13:
+          case 14:
             persona = (config.persona || config.personas || config.agents).find(function (p) {
               return p.id === conversation.$agent;
             });
             if (persona) {
-              _context2.next = 18;
+              _context2.next = 19;
               break;
             }
             if (!(config.persona || config.personas || config.agents).some(function (a) {
               return !a.id;
             })) {
-              _context2.next = 17;
+              _context2.next = 18;
               break;
             }
             throw new Error("SpiritsError: No persona found (\"".concat(conversation.$agent, "\") in provided config, some persona's did not contain an \"id\" (Internal Mapping Error)"));
-          case 17:
-            throw new Error("SpiritsError: No persona found (\"".concat(conversation.$agent, "\") in provided config"));
           case 18:
+            throw new Error("SpiritsError: No persona found (\"".concat(conversation.$agent, "\") in provided config"));
+          case 19:
             if (messages.every(function (m) {
               return !!m.id;
             })) {
-              _context2.next = 20;
+              _context2.next = 21;
               break;
             }
             throw new Error("SpiritsError: Every message must have an \".id\", ensure all messages have an id assigned before running");
-          case 20:
+          case 21:
             if (messages.every(function (m) {
               return m.role === 'customer' || m.role === 'agent' || m.role === 'system';
             })) {
-              _context2.next = 23;
+              _context2.next = 24;
               break;
             }
             invalidRoles = messages.filter(function (m) {
@@ -877,7 +879,7 @@ var Spirits = {
             throw new Error("SpiritsError: Every message must have a role of \"customer\", \"agent\", or \"system\". Got invalid roles: ".concat(invalidRoles.map(function (m) {
               return m.role;
             }).join(', ')));
-          case 23:
+          case 24:
             // if message is not in messages, then add it
             if (!messages.find(function (m) {
               return m.id === input.message.id;
@@ -887,9 +889,9 @@ var Spirits = {
 
             // 2. Parse the message
             progress('Parsing message', 'info', 'SET_PROCESSING', 'user');
-            _context2.next = 27;
+            _context2.next = 28;
             return parser(message.content, 'en');
-          case 27:
+          case 28:
             parsePayload = _context2.sent;
             if (parsePayload.intent) {
               message.intent = parsePayload.intent;
@@ -960,7 +962,7 @@ var Spirits = {
             }
             noNewContext = Object.keys(parsePayload.context).length === 0; // 3. Run the workflow
             progress('Running workflow', 'info', 'SET_PROCESSING', 'system');
-            _context2.next = 44;
+            _context2.next = 45;
             return workflow({
               messages: messages,
               conversation: conversation,
@@ -989,7 +991,7 @@ var Spirits = {
                 return accumulator;
               }, []);
             });
-          case 44:
+          case 45:
             slots = _context2.sent;
             hasNoInstructions = slots.every(function (s) {
               return !s.instructions || Array.isArray(s.instructions) && s.instructions.length === 0;
@@ -1015,20 +1017,20 @@ var Spirits = {
             }
             resettedIntent = false;
             _iterator = _createForOfIteratorHelper(slots);
-            _context2.prev = 52;
+            _context2.prev = 53;
             _iterator.s();
-          case 54:
+          case 55:
             if ((_step = _iterator.n()).done) {
-              _context2.next = 114;
+              _context2.next = 125;
               break;
             }
-            _step$value = _step.value, forward = _step$value.forward, forwardNote = _step$value.forwardNote, instructions = _step$value.instructions, removeInstructions = _step$value.removeInstructions, manualMessage = _step$value.message, scheduled = _step$value.scheduled, resetIntent = _step$value.resetIntent, secondsDelay = _step$value.secondsDelay, contextUpsert = _step$value.contextUpsert, anticipate = _step$value.anticipate, slotFollowup = _step$value.followup;
+            _step$value = _step.value, forward = _step$value.forward, forwardNote = _step$value.forwardNote, instructions = _step$value.instructions, removeInstructions = _step$value.removeInstructions, manualMessage = _step$value.message, scheduled = _step$value.scheduled, resetIntent = _step$value.resetIntent, secondsDelay = _step$value.secondsDelay, contextUpsert = _step$value.contextUpsert, anticipate = _step$value.anticipate, slotFollowup = _step$value.followup, slotEntityContextUpsert = _step$value.entityContextUpsert;
             if (!anticipate) {
-              _context2.next = 69;
+              _context2.next = 70;
               break;
             }
             if (!Array.isArray(anticipate)) {
-              _context2.next = 64;
+              _context2.next = 65;
               break;
             }
             // 'literal' anticipation
@@ -1048,11 +1050,11 @@ var Spirits = {
               slots: _slots,
               map: map
             });
-            _context2.next = 69;
+            _context2.next = 70;
             break;
-          case 64:
+          case 65:
             if (!('yes' in anticipate && 'no' in anticipate && 'did' in anticipate)) {
-              _context2.next = 68;
+              _context2.next = 69;
               break;
             }
             // "did" anticipation
@@ -1064,14 +1066,17 @@ var Spirits = {
               },
               did: anticipate.did
             });
-            _context2.next = 69;
+            _context2.next = 70;
             break;
-          case 68:
-            throw new Error("Invalid anticipate payload \"".concat(JSON.stringify(anticipate), "\""));
           case 69:
+            throw new Error("Invalid anticipate payload \"".concat(JSON.stringify(anticipate), "\""));
+          case 70:
             if (slotFollowup) {
               followup.push(slotFollowup);
             }
+            if (slotEntityContextUpsert && slotEntityContextUpsert.length) {
+              entityContextUpsert.push.apply(entityContextUpsert, _toConsumableArray(slotEntityContextUpsert));
+            }
 
             // Forward to agent or other agent
             if (forward) {
@@ -1116,19 +1121,19 @@ var Spirits = {
 
             // Insert instructions context
             if (!instructions) {
-              _context2.next = 86;
+              _context2.next = 88;
               break;
             }
             if (!(typeof instructions === 'string')) {
-              _context2.next = 76;
+              _context2.next = 78;
               break;
             }
             addInstruction(instructions, previousLockAttempt);
-            _context2.next = 86;
+            _context2.next = 88;
             break;
-          case 76:
+          case 78:
             if (!Array.isArray(instructions)) {
-              _context2.next = 81;
+              _context2.next = 83;
               break;
             }
             _iterator3 = _createForOfIteratorHelper(instructions);
@@ -1146,25 +1151,25 @@ var Spirits = {
             } finally {
               _iterator3.f();
             }
-            _context2.next = 86;
+            _context2.next = 88;
             break;
-          case 81:
+          case 83:
             if (!(_typeof(instructions) === 'object' && 'content' in instructions)) {
-              _context2.next = 85;
+              _context2.next = 87;
               break;
             }
             addInstruction(instructions.content, previousLockAttempt, instructions.id);
-            _context2.next = 86;
+            _context2.next = 88;
             break;
-          case 85:
+          case 87:
             throw new Error("SpiritsError: instructions must be a string or array or {content: \"<instruction>\"}, got: ".concat(JSON.stringify(instructions)));
-          case 86:
+          case 88:
             if (!removeInstructions) {
-              _context2.next = 103;
+              _context2.next = 105;
               break;
             }
             _iterator4 = _createForOfIteratorHelper(removeInstructions);
-            _context2.prev = 88;
+            _context2.prev = 90;
             _loop = /*#__PURE__*/_regeneratorRuntime().mark(function _loop() {
               var instructionId, index;
               return _regeneratorRuntime().wrap(function _loop$(_context) {
@@ -1189,43 +1194,59 @@ var Spirits = {
               }, _loop);
             });
             _iterator4.s();
-          case 91:
+          case 93:
             if ((_step4 = _iterator4.n()).done) {
-              _context2.next = 95;
+              _context2.next = 97;
               break;
             }
-            return _context2.delegateYield(_loop(), "t0", 93);
-          case 93:
-            _context2.next = 91;
-            break;
+            return _context2.delegateYield(_loop(), "t0", 95);
           case 95:
-            _context2.next = 100;
+            _context2.next = 93;
             break;
           case 97:
-            _context2.prev = 97;
-            _context2.t1 = _context2["catch"](88);
+            _context2.next = 102;
+            break;
+          case 99:
+            _context2.prev = 99;
+            _context2.t1 = _context2["catch"](90);
             _iterator4.e(_context2.t1);
-          case 100:
-            _context2.prev = 100;
+          case 102:
+            _context2.prev = 102;
             _iterator4.f();
-            return _context2.finish(100);
-          case 103:
+            return _context2.finish(102);
+          case 105:
             if (!manualMessage) {
-              _context2.next = 110;
+              _context2.next = 121;
               break;
             }
+            /** @type {import('@scout9/app').Message} */
             manualMessageObj = {
-              id: idGenerator('agent'),
+              id: idGenerator('persona'),
               role: 'agent',
-              content: manualMessage,
+              // @TODO switch role to persona
+              content: '',
               time: new Date().toISOString()
             };
-            if (!(typeof manualMessage !== 'string')) {
-              _context2.next = 107;
+            if (!(_typeof(manualMessage) === 'object')) {
+              _context2.next = 113;
               break;
             }
-            throw new Error('Manual message must be of type "string"');
-          case 107:
+            Object.assign(manualMessageObj, manualMessage);
+            manualMessageObj.role = 'agent';
+            manualMessageObj.time = new Date().toISOString();
+            _context2.next = 118;
+            break;
+          case 113:
+            if (!(typeof manualMessage === 'string')) {
+              _context2.next = 117;
+              break;
+            }
+            manualMessageObj.content = manualMessage;
+            _context2.next = 118;
+            break;
+          case 117:
+            throw new Error('Manual message must be of type "string" or "DirectMessage"');
+          case 118:
             if (scheduled) {
               manualMessageObj.time = new Date(scheduled * 1000).toISOString();
               manualMessageObj.scheduled = manualMessageObj.time;
@@ -1237,7 +1258,7 @@ var Spirits = {
             }
             messages.push(manualMessageObj);
             progress('Added manual message', 'info', 'ADD_MESSAGE', messages[messages.length - 1]);
-          case 110:
+          case 121:
             if (contextUpsert) {
               context = updateContext(context, contextUpsert);
               progress('Upserted context', 'info', 'UPDATE_CONTEXT', contextUpsert);
@@ -1245,21 +1266,21 @@ var Spirits = {
             if (resetIntent) {
               resettedIntent = true;
             }
-          case 112:
-            _context2.next = 54;
+          case 123:
+            _context2.next = 55;
             break;
-          case 114:
-            _context2.next = 119;
+          case 125:
+            _context2.next = 130;
             break;
-          case 116:
-            _context2.prev = 116;
-            _context2.t2 = _context2["catch"](52);
+          case 127:
+            _context2.prev = 127;
+            _context2.t2 = _context2["catch"](53);
             _iterator.e(_context2.t2);
-          case 119:
-            _context2.prev = 119;
+          case 130:
+            _context2.prev = 130;
             _iterator.f();
-            return _context2.finish(119);
-          case 122:
+            return _context2.finish(130);
+          case 133:
             if (resettedIntent && !_forward) {
               conversation.intent = null;
               conversation.intentScore = null;
@@ -1278,16 +1299,16 @@ var Spirits = {
             // 4. Generate response
             // If conversation previously locked, don't generate
             if (input.conversation.locked) {
-              _context2.next = 170;
+              _context2.next = 181;
               break;
             }
             if (!((!conversation.locked || !hasNoInstructions) && !!hasNoCustomMessage)) {
-              _context2.next = 137;
+              _context2.next = 148;
               break;
             }
-            _context2.prev = 125;
+            _context2.prev = 136;
             progress('Generating message', 'info', 'SET_PROCESSING', 'system');
-            _context2.next = 129;
+            _context2.next = 140;
             return generator({
               messages: messages,
               persona: persona,
@@ -1295,7 +1316,7 @@ var Spirits = {
               llm: config.llm,
               pmt: config.pmt
             });
-          case 129:
+          case 140:
             generatorPayload = _context2.sent;
             if (!generatorPayload.send) {
               progress('Generated response', 'failed', undefined, {
@@ -1334,29 +1355,29 @@ var Spirits = {
                 }
               }
             }
-            _context2.next = 137;
+            _context2.next = 148;
             break;
-          case 133:
-            _context2.prev = 133;
-            _context2.t3 = _context2["catch"](125);
+          case 144:
+            _context2.prev = 144;
+            _context2.t3 = _context2["catch"](136);
             console.error("Locking conversation, error generating response: ".concat(_context2.t3.message));
             conversation = lockConversation(conversation, 'API: ' + _context2.t3.message);
-          case 137:
+          case 148:
             if (!(messagesToTransform.length && transformer)) {
-              _context2.next = 169;
+              _context2.next = 180;
               break;
             }
-            _context2.prev = 138;
+            _context2.prev = 149;
             _iterator2 = _createForOfIteratorHelper(messagesToTransform);
-            _context2.prev = 140;
+            _context2.prev = 151;
             _iterator2.s();
-          case 142:
+          case 153:
             if ((_step2 = _iterator2.n()).done) {
-              _context2.next = 153;
+              _context2.next = 164;
               break;
             }
             _step2.value;
-            _context2.next = 146;
+            _context2.next = 157;
             return transformer({
               message: messagesToTransform,
               persona: persona,
@@ -1364,7 +1385,7 @@ var Spirits = {
               messages: messages,
               context: context
             });
-          case 146:
+          case 157:
             transformResponse = _context2.sent;
             progress('Generated response', 'success', undefined, undefined);
             // Check if already had message
@@ -1384,36 +1405,36 @@ var Spirits = {
               });
               progress('Added agent message', 'info', 'ADD_MESSAGE', messages[messages.length - 1]);
             }
-          case 151:
-            _context2.next = 142;
+          case 162:
+            _context2.next = 153;
             break;
-          case 153:
-            _context2.next = 158;
+          case 164:
+            _context2.next = 169;
             break;
-          case 155:
-            _context2.prev = 155;
-            _context2.t4 = _context2["catch"](140);
+          case 166:
+            _context2.prev = 166;
+            _context2.t4 = _context2["catch"](151);
             _iterator2.e(_context2.t4);
-          case 158:
-            _context2.prev = 158;
+          case 169:
+            _context2.prev = 169;
             _iterator2.f();
-            return _context2.finish(158);
-          case 161:
-            _context2.next = 167;
+            return _context2.finish(169);
+          case 172:
+            _context2.next = 178;
             break;
-          case 163:
-            _context2.prev = 163;
-            _context2.t5 = _context2["catch"](138);
+          case 174:
+            _context2.prev = 174;
+            _context2.t5 = _context2["catch"](149);
             console.error("Locking conversation, error transforming response: ".concat(_context2.t5.message));
             conversation = lockConversation(conversation, 'API: ' + _context2.t5.message);
-          case 167:
-            _context2.next = 170;
+          case 178:
+            _context2.next = 181;
             break;
-          case 169:
+          case 180:
             if (messagesToTransform.length) {
               console.warn("No transformer provided");
             }
-          case 170:
+          case 181:
             progress('Parsing message', 'info', 'SET_PROCESSING', null);
             return _context2.abrupt("return", {
               conversation: {
@@ -1434,13 +1455,14 @@ var Spirits = {
                 before: contextBefore,
                 after: context
               },
-              followup: followup
+              followup: followup,
+              entityContextUpsert: entityContextUpsert
             });
-          case 172:
+          case 183:
           case "end":
             return _context2.stop();
         }
-      }, _callee, null, [[52, 116, 119, 122], [88, 97, 100, 103], [125, 133], [138, 163], [140, 155, 158, 161]]);
+      }, _callee, null, [[53, 127, 130, 133], [90, 99, 102, 105], [136, 144], [149, 174], [151, 166, 169, 172]]);
     }));
     function customer(_x) {
       return _customer.apply(this, arguments);

package/dist/spirits.cjs

@@ -2,7 +2,7 @@
 
 Object.defineProperty(exports, '__esModule', { value: true });
 
-var spirits = require("./spirits-e626085b.cjs");
+var spirits = require("./spirits-9719ae4f.cjs");