@awell-health/awell-extensions 2.0.21 → 2.0.23

package/dist/extensions/shelly/actions/categorizeMessage/{realTest.test.js → callingRealOpenAITest.test.js}

@@ -31,7 +31,8 @@ describe.skip('categorizeMessage - Real LLM calls', () => {
         // Real LangChain function is called
         expect(onComplete).toHaveBeenCalledWith({
             data_points: {
-                category: 'Appointment Scheduling', // Expected category from real LLM
+                category: 'Appointment Scheduling',
+                explanation: 'The message explicitly states a desire to schedule an appointment, which directly aligns with the Appointment Scheduling category.',
             },
         });
         expect(onError).not.toHaveBeenCalled();
@@ -53,10 +54,11 @@ describe.skip('categorizeMessage - Real LLM calls', () => {
         // Real LangChain function is called and returns "None"
         expect(onComplete).toHaveBeenCalledWith({
             data_points: {
-                category: 'None', // Expected no match
+                category: 'None',
+                explanation: 'Categorization was ambiguous; we could not find a proper category.',
             },
         });
         expect(onError).not.toHaveBeenCalled();
     });
 });
-//# sourceMappingURL=realTest.test.js.map
+//# sourceMappingURL=callingRealOpenAITest.test.js.map

package/dist/extensions/shelly/actions/categorizeMessage/callingRealOpenAITest.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"callingRealOpenAITest.test.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/categorizeMessage/callingRealOpenAITest.test.ts"],"names":[],"mappings":";;AAAA,mEAA2D;AAC3D,mCAA6C;AAC7C,wBAAqC;AACrC,yBAAsB;AAEtB,MAAM,QAAQ,GAAG;IACf,YAAY,EAAE,OAAO,CAAC,GAAG,CAAC,eAAe;CAC1C,CAAA;AAED,8BAA8B;AAC9B,QAAQ,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE;IACvD,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,GACjE,6BAAW,CAAC,UAAU,CAAC,oBAAiB,CAAC,CAAA;IAE3C,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,EAAE,CAAA,CAAC,+BAA+B;QAC5C,IAAI,CAAC,aAAa,EAAE,CAAA,CAAC,2BAA2B;IAClD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;QACvG,MAAM,OAAO,GAAG,IAAA,2BAAmB,EAAC;YAClC,MAAM,EAAE;gBACN,OAAO,EAAE,wDAAwD;gBACjE,UAAU,EACR,8FAA8F;aACjG;YACD,QAAQ;SACT,CAAC,CAAA;QAEF,MAAM,eAAe,CAAC,OAAO,CAAC;YAC5B,OAAO;YACP,UAAU;YACV,OAAO;YACP,OAAO;SACR,CAAC,CAAA;QAEF,oCAAoC;QACpC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,WAAW,EAAE;gBACX,QAAQ,EAAE,wBAAwB;gBAClC,WAAW,EACT,oIAAoI;aACvI;SACF,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IACxC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,OAAO,GAAG,IAAA,2BAAmB,EAAC;YAClC,MAAM,EAAE;gBACN,OAAO,EAAE,iCAAiC;gBAC1C,UAAU,EACR,8FAA8F;aACjG;YACD,QAAQ;SACT,CAAC,CAAA;QAEF,MAAM,eAAe,CAAC,OAAO,CAAC;YAC5B,OAAO;YACP,UAAU;YACV,OAAO;YACP,OAAO;SACR,CAAC,CAAA;QAEF,uDAAuD;QACvD,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,WAAW,EAAE;gBACX,QAAQ,EAAE,MAAM;gBAChB,WAAW,EACT,oEAAoE;aACvE;SACF,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IACxC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/extensions/shelly/actions/categorizeMessage/categorizeMessage.js

@@ -9,24 +9,28 @@ exports.categorizeMessage = {
     key: 'categorizeMessage',
     category: extensions_core_1.Category.WORKFLOW,
     title: 'Categorize Message',
-    description: 'Categorize the input message into set of predefined categories',
+    description: 'Categorize the input message into set of predefined categories and provides explanation.',
     fields: config_1.fields,
     previewable: false,
     dataPoints: config_1.dataPoints,
     onEvent: async ({ payload, onComplete, onError, helpers }) => {
-        const { langChainOpenAiSdk, fields: { categories, message }, } = await (0, lib_1.validatePayloadAndCreateSdk)({
+        const { ChatModelGPT4oMini, fields: { categories, message }, } = await (0, lib_1.validatePayloadAndCreateSdk)({
             fieldsSchema: config_1.FieldsValidationSchema,
             payload,
         });
         try {
-            const category = await (0, categorizeMessageWithLLM_1.categorizeMessageWithLLM)({
-                langChainOpenAiSdk,
+            const categorization_result = await (0, categorizeMessageWithLLM_1.categorizeMessageWithLLM)({
+                ChatModelGPT4oMini,
                 message,
                 categories,
             });
+            console.log(categorization_result);
+            const category = categorization_result.category;
+            const explanation = categorization_result.explanation;
             await onComplete({
                 data_points: {
                     category,
+                    explanation,
                 },
             });
         }

package/dist/extensions/shelly/actions/categorizeMessage/categorizeMessage.js.map

@@ -1 +1 @@
-{"version":3,"file":"categorizeMessage.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/categorizeMessage/categorizeMessage.ts"],"names":[],"mappings":";;;AAAA,mEAAqE;AACrE,6EAAyE;AACzE,mCAAuD;AAEvD,qCAAqE;AAExD,QAAA,iBAAiB,GAI1B;IACF,GAAG,EAAE,mBAAmB;IACxB,QAAQ,EAAE,0BAAQ,CAAC,QAAQ;IAC3B,KAAK,EAAE,oBAAoB;IAC3B,WAAW,EAAE,gEAAgE;IAC7E,MAAM,EAAN,eAAM;IACN,WAAW,EAAE,KAAK;IAClB,UAAU,EAAV,mBAAU;IACV,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAiB,EAAE;QAC1E,MAAM,EACJ,kBAAkB,EAClB,MAAM,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,GAChC,GAAG,MAAM,IAAA,iCAA2B,EAAC;YACpC,YAAY,EAAE,+BAAsB;YACpC,OAAO;SACR,CAAC,CAAA;QAEF,IAAI;YACF,MAAM,QAAQ,GAAG,MAAM,IAAA,mDAAwB,EAAC;gBAC9C,kBAAkB;gBAClB,OAAO;gBACP,UAAU;aACX,CAAC,CAAA;YAEF,MAAM,UAAU,CAAC;gBACf,WAAW,EAAE;oBACX,QAAQ;iBACT;aACF,CAAC,CAAA;SACH;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAA;YACnD,4BAA4B;YAC5B,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAA;SAC9C;IACH,CAAC;CACF,CAAA"}
+{"version":3,"file":"categorizeMessage.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/categorizeMessage/categorizeMessage.ts"],"names":[],"mappings":";;;AAAA,mEAAqE;AACrE,6EAAyE;AACzE,mCAAuD;AAEvD,qCAAqE;AAExD,QAAA,iBAAiB,GAI1B;IACF,GAAG,EAAE,mBAAmB;IACxB,QAAQ,EAAE,0BAAQ,CAAC,QAAQ;IAC3B,KAAK,EAAE,oBAAoB;IAC3B,WAAW,EACT,0FAA0F;IAC5F,MAAM,EAAN,eAAM;IACN,WAAW,EAAE,KAAK;IAClB,UAAU,EAAV,mBAAU;IACV,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAiB,EAAE;QAC1E,MAAM,EACJ,kBAAkB,EAClB,MAAM,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,GAChC,GAAG,MAAM,IAAA,iCAA2B,EAAC;YACpC,YAAY,EAAE,+BAAsB;YACpC,OAAO;SACR,CAAC,CAAA;QAEF,IAAI;YACF,MAAM,qBAAqB,GAAG,MAAM,IAAA,mDAAwB,EAAC;gBAC3D,kBAAkB;gBAClB,OAAO;gBACP,UAAU;aACX,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAA;YAClC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAA;YAC/C,MAAM,WAAW,GAAG,qBAAqB,CAAC,WAAW,CAAA;YAErD,MAAM,UAAU,CAAC;gBACf,WAAW,EAAE;oBACX,QAAQ;oBACR,WAAW;iBACZ;aACF,CAAC,CAAA;SACH;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAA;YACnD,4BAA4B;YAC5B,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAA;SAC9C;IACH,CAAC;CACF,CAAA"}

package/dist/extensions/shelly/actions/categorizeMessage/categorizeMessage.test.js

@@ -4,15 +4,23 @@ const extensions_core_1 = require("@awell-health/extensions-core");
 const tests_1 = require("@/tests");
 const _1 = require(".");
 const openai_1 = require("@langchain/openai");
-jest.mock('@langchain/openai', () => ({
-    OpenAI: jest.fn().mockImplementation(() => ({
-        pipe: jest.fn().mockReturnThis(),
-        invoke: jest.fn().mockResolvedValue({
-            matched_entity: 'Appointment Scheduling',
-        }),
-    })),
-}));
-const mockedOpenAiSdk = jest.mocked(openai_1.OpenAI);
+// Mock the module
+jest.mock('@langchain/openai', () => {
+    const mockInvoke = jest.fn().mockResolvedValue({
+        matched_category: 'Appointment Scheduling',
+        match_explanation: 'The message contains a request for scheduling an appointment.',
+    });
+    const mockChain = {
+        invoke: mockInvoke,
+    };
+    const mockPipe = jest.fn().mockReturnValue(mockChain);
+    const mockChatOpenAI = jest.fn().mockImplementation(() => ({
+        pipe: mockPipe,
+    }));
+    return {
+        ChatOpenAI: mockChatOpenAI,
+    };
+});
 describe('categorizeMessage - Mocked LLM calls', () => {
     const { onComplete, onError, helpers, extensionAction, clearMocks } = extensions_core_1.TestHelpers.fromAction(_1.categorizeMessage);
     beforeEach(() => {
@@ -20,11 +28,6 @@ describe('categorizeMessage - Mocked LLM calls', () => {
         jest.clearAllMocks();
     });
     it('should work', async () => {
-        const mockedOpenAiInstance = new openai_1.OpenAI();
-        jest.spyOn(mockedOpenAiInstance, 'invoke').mockResolvedValue({
-            // @ts-expect-error fine, we have a parser
-            matched_entity: 'Appointment Scheduling',
-        });
         const categorizeMessageWithLLMSpy = jest.spyOn(require('./lib/categorizeMessageWithLLM'), 'categorizeMessageWithLLM');
         const payload = (0, tests_1.generateTestPayload)({
             fields: {
@@ -41,9 +44,9 @@ describe('categorizeMessage - Mocked LLM calls', () => {
             onError,
             helpers,
         });
-        expect(mockedOpenAiSdk).toHaveBeenCalled();
+        expect(openai_1.ChatOpenAI).toHaveBeenCalled();
         expect(categorizeMessageWithLLMSpy).toHaveBeenCalledWith({
-            langChainOpenAiSdk: expect.any(Object),
+            ChatModelGPT4oMini: expect.any(Object),
             message: 'I would like to schedule an appointment for next week.',
             categories: [
                 'Appointment Scheduling',
@@ -54,7 +57,8 @@ describe('categorizeMessage - Mocked LLM calls', () => {
         });
         expect(onComplete).toHaveBeenCalledWith({
             data_points: {
-                category: 'Appointment Scheduling', // Expected category
+                category: 'Appointment Scheduling',
+                explanation: 'The message contains a request for scheduling an appointment.',
             },
         });
         expect(onError).not.toHaveBeenCalled();

package/dist/extensions/shelly/actions/categorizeMessage/categorizeMessage.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"categorizeMessage.test.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/categorizeMessage/categorizeMessage.test.ts"],"names":[],"mappings":";;AAAA,mEAA2D;AAC3D,mCAA6C;AAC7C,wBAAqC;AACrC,8CAA0C;AAE1C,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IACpC,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QAC1C,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;QAChC,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAClC,cAAc,EAAE,wBAAwB;SACzC,CAAC;KACH,CAAC,CAAC;CACJ,CAAC,CAAC,CAAA;AAEH,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,eAAM,CAAC,CAAA;AAE3C,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,GACjE,6BAAW,CAAC,UAAU,CAAC,oBAAiB,CAAC,CAAA;IAE3C,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,EAAE,CAAA;QACZ,IAAI,CAAC,aAAa,EAAE,CAAA;IACtB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;QAC3B,MAAM,oBAAoB,GAAG,IAAI,eAAM,EAAE,CAAA;QAEzC,IAAI,CAAC,KAAK,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAC,iBAAiB,CAAC;YAC3D,0CAA0C;YAC1C,cAAc,EAAE,wBAAwB;SACzC,CAAC,CAAA;QAEF,MAAM,2BAA2B,GAAG,IAAI,CAAC,KAAK,CAC5C,OAAO,CAAC,gCAAgC,CAAC,EACzC,0BAA0B,CAC3B,CAAA;QAED,MAAM,OAAO,GAAG,IAAA,2BAAmB,EAAC;YAClC,MAAM,EAAE;gBACN,OAAO,EAAE,wDAAwD;gBACjE,UAAU,EACR,8FAA8F;aACjG;YACD,QAAQ,EAAE;gBACR,YAAY,EAAE,GAAG;aAClB;SACF,CAAC,CAAA;QAEF,MAAM,eAAe,CAAC,OAAO,CAAC;YAC5B,OAAO;YACP,UAAU;YACV,OAAO;YACP,OAAO;SACR,CAAC,CAAA;QAEF,MAAM,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,2BAA2B,CAAC,CAAC,oBAAoB,CAAC;YACvD,kBAAkB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YACtC,OAAO,EAAE,wDAAwD;YACjE,UAAU,EAAE;gBACV,wBAAwB;gBACxB,sBAAsB;gBACtB,2BAA2B;gBAC3B,wBAAwB;aACzB;SACF,CAAC,CAAA;QAEF,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,WAAW,EAAE;gBACX,QAAQ,EAAE,wBAAwB,EAAE,oBAAoB;aACzD;SACF,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IACxC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
+{"version":3,"file":"categorizeMessage.test.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/categorizeMessage/categorizeMessage.test.ts"],"names":[],"mappings":";;AAAA,mEAA2D;AAC3D,mCAA6C;AAC7C,wBAAqC;AACrC,8CAA8C;AAE9C,kBAAkB;AAClB,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE;IAClC,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;QAC7C,gBAAgB,EAAE,wBAAwB;QAC1C,iBAAiB,EACf,+DAA+D;KAClE,CAAC,CAAA;IAEF,MAAM,SAAS,GAAG;QAChB,MAAM,EAAE,UAAU;KACnB,CAAA;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,CAAA;IAErD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACzD,IAAI,EAAE,QAAQ;KACf,CAAC,CAAC,CAAA;IAEH,OAAO;QACL,UAAU,EAAE,cAAc;KAC3B,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,GACjE,6BAAW,CAAC,UAAU,CAAC,oBAAiB,CAAC,CAAA;IAE3C,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,EAAE,CAAA;QACZ,IAAI,CAAC,aAAa,EAAE,CAAA;IACtB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;QAC3B,MAAM,2BAA2B,GAAG,IAAI,CAAC,KAAK,CAC5C,OAAO,CAAC,gCAAgC,CAAC,EACzC,0BAA0B,CAC3B,CAAA;QAED,MAAM,OAAO,GAAG,IAAA,2BAAmB,EAAC;YAClC,MAAM,EAAE;gBACN,OAAO,EAAE,wDAAwD;gBACjE,UAAU,EACR,8FAA8F;aACjG;YACD,QAAQ,EAAE;gBACR,YAAY,EAAE,GAAG;aAClB;SACF,CAAC,CAAA;QAEF,MAAM,eAAe,CAAC,OAAO,CAAC;YAC5B,OAAO;YACP,UAAU;YACV,OAAO;YACP,OAAO;SACR,CAAC,CAAA;QAEF,MAAM,CAAC,mBAAU,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACrC,MAAM,CAAC,2BAA2B,CAAC,CAAC,oBAAoB,CAAC;YACvD,kBAAkB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YACtC,OAAO,EAAE,wDAAwD;YACjE,UAAU,EAAE;gBACV,wBAAwB;gBACxB,sBAAsB;gBACtB,2BAA2B;gBAC3B,wBAAwB;aACzB;SACF,CAAC,CAAA;QAEF,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,WAAW,EAAE;gBACX,QAAQ,EAAE,wBAAwB;gBAClC,WAAW,EACT,+DAA+D;aAClE;SACF,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IACxC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/extensions/shelly/actions/categorizeMessage/config/datapoints.d.ts

@@ -3,4 +3,8 @@ export declare const dataPoints: {
         key: string;
         valueType: "string";
     };
+    explanation: {
+        key: string;
+        valueType: "string";
+    };
 };

package/dist/extensions/shelly/actions/categorizeMessage/config/datapoints.js

@@ -6,5 +6,9 @@ exports.dataPoints = {
         key: 'category',
         valueType: 'string',
     },
+    explanation: {
+        key: 'explanation',
+        valueType: 'string',
+    },
 };
 //# sourceMappingURL=datapoints.js.map

package/dist/extensions/shelly/actions/categorizeMessage/config/datapoints.js.map

@@ -1 +1 @@
-{"version":3,"file":"datapoints.js","sourceRoot":"","sources":["../../../../../../extensions/shelly/actions/categorizeMessage/config/datapoints.ts"],"names":[],"mappings":";;;AAEa,QAAA,UAAU,GAAG;IACxB,QAAQ,EAAE;QACR,GAAG,EAAE,UAAU;QACf,SAAS,EAAE,QAAQ;KACpB;CAC4C,CAAA"}
+{"version":3,"file":"datapoints.js","sourceRoot":"","sources":["../../../../../../extensions/shelly/actions/categorizeMessage/config/datapoints.ts"],"names":[],"mappings":";;;AAEa,QAAA,UAAU,GAAG;IACxB,QAAQ,EAAE;QACR,GAAG,EAAE,UAAU;QACf,SAAS,EAAE,QAAQ;KACpB;IACD,WAAW,EAAE;QACX,GAAG,EAAE,aAAa;QAClB,SAAS,EAAE,QAAQ;KACpB;CAC4C,CAAA"}

package/dist/extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.d.ts

@@ -1,6 +1,9 @@
-import { type OpenAI } from '@langchain/openai';
-export declare const categorizeMessageWithLLM: ({ langChainOpenAiSdk, message, categories, }: {
-    langChainOpenAiSdk: OpenAI;
+import { type ChatOpenAI } from '@langchain/openai';
+export declare const categorizeMessageWithLLM: ({ ChatModelGPT4oMini, message, categories, }: {
+    ChatModelGPT4oMini: ChatOpenAI;
     message: string;
     categories: string[];
-}) => Promise<string>;
+}) => Promise<{
+    category: string;
+    explanation: string;
+}>;

package/dist/extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.js

@@ -2,21 +2,39 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.categorizeMessageWithLLM = void 0;
 const constants_1 = require("./constants");
-const categorizeMessageWithLLM = async ({ langChainOpenAiSdk, message, categories, }) => {
-    var _a;
+// TODO: remove console logs eventually
+const categorizeMessageWithLLM = async ({ ChatModelGPT4oMini, message, categories, }) => {
+    var _a, _b;
     const prompt = await constants_1.systemPrompt.format({
         categories: categories.concat('None').join(', '),
         input: message,
     });
-    const chain = langChainOpenAiSdk.pipe(constants_1.parser);
+    const chain = ChatModelGPT4oMini.pipe(constants_1.parser);
     console.log('Prompt:', prompt);
-    const result = await chain.invoke(prompt);
+    let result;
+    try {
+        result = await chain.invoke(prompt);
+    }
+    catch (error) {
+        console.error('Error invoking the chain:', error);
+        throw new Error('Failed to categorize the message due to an internal error.');
+    }
     console.log('Result', typeof result);
-    const matchedEntity = (_a = result.matched_entity) !== null && _a !== void 0 ? _a : 'None';
-    console.log('I am matching:', matchedEntity);
-    // Ensure that the returned matched entity is valid
-    const category = categories.includes(matchedEntity) ? matchedEntity : 'None';
-    return category;
+    const matchedCategory = (_a = result.matched_category) !== null && _a !== void 0 ? _a : 'None';
+    let category;
+    let explanation = (_b = result.match_explanation) !== null && _b !== void 0 ? _b : '';
+    console.log('Matched Category:', matchedCategory);
+    console.log('Explanation:', explanation);
+    // Check if the matched category is valid
+    if (categories.includes(matchedCategory)) {
+        category = matchedCategory;
+    }
+    else {
+        category = 'None';
+        explanation =
+            'Categorization was ambiguous; we could not find a proper category.';
+    }
+    return { category, explanation };
 };
 exports.categorizeMessageWithLLM = categorizeMessageWithLLM;
 //# sourceMappingURL=categorizeMessageWithLLM.js.map

package/dist/extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.js.map

@@ -1 +1 @@
-{"version":3,"file":"categorizeMessageWithLLM.js","sourceRoot":"","sources":["../../../../../../../extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.ts"],"names":[],"mappings":";;;AAAA,2CAAkD;AAG3C,MAAM,wBAAwB,GAAG,KAAK,EAAE,EAC7C,kBAAkB,EAClB,OAAO,EACP,UAAU,GAKX,EAAmB,EAAE;;IACpB,MAAM,MAAM,GAAG,MAAM,wBAAY,CAAC,MAAM,CAAC;QACvC,UAAU,EAAE,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QAChD,KAAK,EAAE,OAAO;KACf,CAAC,CAAA;IAEF,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,CAAC,kBAAM,CAAC,CAAA;IAC7C,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;IAC9B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IAEzC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,MAAM,CAAC,CAAA;IAEpC,MAAM,aAAa,GAAG,MAAA,MAAM,CAAC,cAAc,mCAAI,MAAM,CAAA;IACrD,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAA;IAE5C,mDAAmD;IACnD,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,CAAA;IAE5E,OAAO,QAAQ,CAAA;AACjB,CAAC,CAAA;AA3BY,QAAA,wBAAwB,4BA2BpC"}
+{"version":3,"file":"categorizeMessageWithLLM.js","sourceRoot":"","sources":["../../../../../../../extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.ts"],"names":[],"mappings":";;;AAAA,2CAAkD;AAGlD,uCAAuC;AAChC,MAAM,wBAAwB,GAAG,KAAK,EAAE,EAC7C,kBAAkB,EAClB,OAAO,EACP,UAAU,GAKX,EAAsD,EAAE;;IACvD,MAAM,MAAM,GAAG,MAAM,wBAAY,CAAC,MAAM,CAAC;QACvC,UAAU,EAAE,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QAChD,KAAK,EAAE,OAAO;KACf,CAAC,CAAA;IAEF,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,CAAC,kBAAM,CAAC,CAAA;IAC7C,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;IAC9B,IAAI,MAAM,CAAA;IACV,IAAI;QACF,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;KACpC;IAAC,OAAO,KAAK,EAAE;QACd,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAA;QACjD,MAAM,IAAI,KAAK,CACb,4DAA4D,CAC7D,CAAA;KACF;IAED,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,MAAM,CAAC,CAAA;IAEpC,MAAM,eAAe,GAAG,MAAA,MAAM,CAAC,gBAAgB,mCAAI,MAAM,CAAA;IACzD,IAAI,QAAgB,CAAA;IACpB,IAAI,WAAW,GAAW,MAAA,MAAM,CAAC,iBAAiB,mCAAI,EAAE,CAAA;IAExD,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,eAAe,CAAC,CAAA;IACjD,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,WAAW,CAAC,CAAA;IACxC,yCAAyC;IACzC,IAAI,UAAU,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE;QACxC,QAAQ,GAAG,eAAe,CAAA;KAC3B;SAAM;QACL,QAAQ,GAAG,MAAM,CAAA;QACjB,WAAW;YACT,oEAAoE,CAAA;KACvE;IAED,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAA;AAClC,CAAC,CAAA;AA5CY,QAAA,wBAAwB,4BA4CpC"}

package/dist/extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.test.js

@@ -3,18 +3,18 @@ Object.defineProperty(exports, "__esModule", { value: true });
 require("dotenv/config");
 const _1 = require("./");
 describe('categorizeMessageWithLLM', () => {
-    let langChainOpenAiSdkMock;
+    let ChatModelGPT4oMiniMock;
     beforeEach(() => {
-        // Create a mock for langChainOpenAiSdk
-        langChainOpenAiSdkMock = {
+        ChatModelGPT4oMiniMock = {
             pipe: jest.fn().mockReturnThis(),
-            invoke: jest.fn(), // Will mock this separately for each test
+            invoke: jest.fn(),
         };
     });
     it('should categorize a message about scheduling an appointment using real LLM', async () => {
-        langChainOpenAiSdkMock.invoke.mockResolvedValue({
+        ChatModelGPT4oMiniMock.invoke.mockResolvedValue({
             // @ts-expect-error it's fine, we have a parser
-            matched_entity: 'Appointment Scheduling',
+            matched_category: 'Appointment Scheduling',
+            match_explanation: 'The message contains a request for scheduling an appointment.',
         });
         const categories = [
             'Appointment Scheduling',
@@ -24,16 +24,17 @@ describe('categorizeMessageWithLLM', () => {
         ];
         const message = 'I would like to schedule an appointment for next week.';
         const result = await (0, _1.categorizeMessageWithLLM)({
-            langChainOpenAiSdk: langChainOpenAiSdkMock,
+            ChatModelGPT4oMini: ChatModelGPT4oMiniMock,
             message,
             categories,
         });
-        expect(result).toBe('Appointment Scheduling'); // Expected category
+        expect(result.category).toBe('Appointment Scheduling');
     });
     it('should categorize a message about medication using real LLM', async () => {
-        langChainOpenAiSdkMock.invoke.mockResolvedValue({
+        ChatModelGPT4oMiniMock.invoke.mockResolvedValue({
             // @ts-expect-error it's fine, we have a parser
-            matched_entity: 'Medication Questions',
+            matched_category: 'Medication Questions',
+            match_explanation: 'The message contains a request for medication information.',
         });
         const categories = [
             'Appointment Scheduling',
@@ -43,16 +44,17 @@ describe('categorizeMessageWithLLM', () => {
         ];
         const message = 'Can you tell me the correct dosage for my medication?';
         const result = await (0, _1.categorizeMessageWithLLM)({
-            langChainOpenAiSdk: langChainOpenAiSdkMock,
+            ChatModelGPT4oMini: ChatModelGPT4oMiniMock,
             message,
             categories,
         });
-        expect(result).toBe('Medication Questions'); // Expected category
+        expect(result.category).toBe('Medication Questions');
     });
     it('should return "None" when the message does not match any category using real LLM', async () => {
-        langChainOpenAiSdkMock.invoke.mockResolvedValue({
+        ChatModelGPT4oMiniMock.invoke.mockResolvedValue({
             // @ts-expect-error it's fine, we have a parser
-            matched_entity: 'None',
+            matched_category: 'None',
+            match_explanation: 'The message does not contain any information about scheduling or medication.',
         });
         const categories = [
             'Appointment Scheduling',
@@ -60,13 +62,13 @@ describe('categorizeMessageWithLLM', () => {
             'Medication Questions',
             'Administrative Assistance',
         ];
-        const message = 'Is it going to rain tomorrow?'; // Non-related medical category
+        const message = 'Is it going to rain tomorrow?';
         const result = await (0, _1.categorizeMessageWithLLM)({
-            langChainOpenAiSdk: langChainOpenAiSdkMock,
+            ChatModelGPT4oMini: ChatModelGPT4oMiniMock,
             message,
             categories,
         });
-        expect(result).toBe('None'); // No match expected
+        expect(result.category).toBe('None');
     });
 });
 //# sourceMappingURL=categorizeMessageWithLLM.test.js.map

package/dist/extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"categorizeMessageWithLLM.test.js","sourceRoot":"","sources":["../../../../../../../extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.test.ts"],"names":[],"mappings":";;AAAA,yBAAsB;AACtB,yBAA6C;AAG7C,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,sBAA2C,CAAA;IAE/C,UAAU,CAAC,GAAG,EAAE;QACd,uCAAuC;QACvC,sBAAsB,GAAG;YACvB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YAChC,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,0CAA0C;SAC5B,CAAA;IACrC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAC9C,+CAA+C;YAC/C,cAAc,EAAE,wBAAwB;SACzC,CAAC,CAAA;QAEF,MAAM,UAAU,GAAG;YACjB,wBAAwB;YACxB,0BAA0B;YAC1B,sBAAsB;YACtB,2BAA2B;SAC5B,CAAA;QACD,MAAM,OAAO,GAAG,wDAAwD,CAAA;QAExE,MAAM,MAAM,GAAG,MAAM,IAAA,2BAAwB,EAAC;YAC5C,kBAAkB,EAAE,sBAAsB;YAC1C,OAAO;YACP,UAAU;SACX,CAAC,CAAA;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAA,CAAC,oBAAoB;IACpE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAC9C,+CAA+C;YAC/C,cAAc,EAAE,sBAAsB;SACvC,CAAC,CAAA;QAEF,MAAM,UAAU,GAAG;YACjB,wBAAwB;YACxB,0BAA0B;YAC1B,sBAAsB;YACtB,2BAA2B;SAC5B,CAAA;QACD,MAAM,OAAO,GAAG,uDAAuD,CAAA;QAEvE,MAAM,MAAM,GAAG,MAAM,IAAA,2BAAwB,EAAC;YAC5C,kBAAkB,EAAE,sBAAsB;YAC1C,OAAO;YACP,UAAU;SACX,CAAC,CAAA;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAA,CAAC,oBAAoB;IAClE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAC9C,+CAA+C;YAC/C,cAAc,EAAE,MAAM;SACvB,CAAC,CAAA;QAEF,MAAM,UAAU,GAAG;YACjB,wBAAwB;YACxB,0BAA0B;YAC1B,sBAAsB;YACtB,2BAA2B;SAC5B,CAAA;QACD,MAAM,OAAO,GAAG,+BAA+B,CAAA,CAAC,+BAA+B;QAE/E,MAAM,MAAM,GAAG,MAAM,IAAA,2BAAwB,EAAC;YAC5C,kBAAkB,EAAE,sBAAsB;YAC1C,OAAO;YACP,UAAU;SACX,CAAC,CAAA;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA,CAAC,oBAAoB;IAClD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
+{"version":3,"file":"categorizeMessageWithLLM.test.js","sourceRoot":"","sources":["../../../../../../../extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/categorizeMessageWithLLM.test.ts"],"names":[],"mappings":";;AAAA,yBAAsB;AACtB,yBAA6C;AAG7C,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,sBAA+C,CAAA;IAEnD,UAAU,CAAC,GAAG,EAAE;QACd,sBAAsB,GAAG;YACvB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YAChC,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACoB,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAC9C,+CAA+C;YAC/C,gBAAgB,EAAE,wBAAwB;YAC1C,iBAAiB,EACf,+DAA+D;SAClE,CAAC,CAAA;QAEF,MAAM,UAAU,GAAG;YACjB,wBAAwB;YACxB,0BAA0B;YAC1B,sBAAsB;YACtB,2BAA2B;SAC5B,CAAA;QACD,MAAM,OAAO,GAAG,wDAAwD,CAAA;QAExE,MAAM,MAAM,GAAG,MAAM,IAAA,2BAAwB,EAAC;YAC5C,kBAAkB,EAAE,sBAAsB;YAC1C,OAAO;YACP,UAAU;SACX,CAAC,CAAA;QAEF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAA;IACxD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAC9C,+CAA+C;YAC/C,gBAAgB,EAAE,sBAAsB;YACxC,iBAAiB,EACf,4DAA4D;SAC/D,CAAC,CAAA;QAEF,MAAM,UAAU,GAAG;YACjB,wBAAwB;YACxB,0BAA0B;YAC1B,sBAAsB;YACtB,2BAA2B;SAC5B,CAAA;QACD,MAAM,OAAO,GAAG,uDAAuD,CAAA;QAEvE,MAAM,MAAM,GAAG,MAAM,IAAA,2BAAwB,EAAC;YAC5C,kBAAkB,EAAE,sBAAsB;YAC1C,OAAO;YACP,UAAU;SACX,CAAC,CAAA;QAEF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAC9C,+CAA+C;YAC/C,gBAAgB,EAAE,MAAM;YACxB,iBAAiB,EACf,8EAA8E;SACjF,CAAC,CAAA;QAEF,MAAM,UAAU,GAAG;YACjB,wBAAwB;YACxB,0BAA0B;YAC1B,sBAAsB;YACtB,2BAA2B;SAC5B,CAAA;QACD,MAAM,OAAO,GAAG,+BAA+B,CAAA;QAE/C,MAAM,MAAM,GAAG,MAAM,IAAA,2BAAwB,EAAC;YAC5C,kBAAkB,EAAE,sBAAsB;YAC1C,OAAO;YACP,UAAU;SACX,CAAC,CAAA;QAEF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/constants.d.ts

@@ -2,17 +2,23 @@ import { ChatPromptTemplate } from '@langchain/core/prompts';
 import { StructuredOutputParser } from 'langchain/output_parsers';
 import { z } from 'zod';
 export declare const messageCategoriesSchema: z.ZodObject<{
-    matched_entity: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    matched_category: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    match_explanation: z.ZodString;
 }, "strip", z.ZodTypeAny, {
-    matched_entity: string;
+    matched_category: string;
+    match_explanation: string;
 }, {
-    matched_entity?: string | undefined;
+    match_explanation: string;
+    matched_category?: string | undefined;
 }>;
 export declare const parser: StructuredOutputParser<z.ZodObject<{
-    matched_entity: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    matched_category: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    match_explanation: z.ZodString;
 }, "strip", z.ZodTypeAny, {
-    matched_entity: string;
+    matched_category: string;
+    match_explanation: string;
 }, {
-    matched_entity?: string | undefined;
+    match_explanation: string;
+    matched_category?: string | undefined;
 }>>;
-export declare const systemPrompt: ChatPromptTemplate<import("@langchain/core/prompts").ParamsFromFString<"\n      You are an expert in categorizing messages. Your task is to match the provided message to one of the following categories: {categories}.\n      If no match is found, return \"None.\"\n      \n      Please respond with a valid JSON object with the following key: matched_entity.\n      \n  \n      Input:\n      {input}\n      ">, any>;
+export declare const systemPrompt: ChatPromptTemplate<import("@langchain/core/prompts").ParamsFromFString<"\n      You are an expert in categorizing messages different messages in a clinical context.\n      Use your expertise to solve the critical task:\n      1. Match the provided message to one of the following categories: {categories}.\n      2. Provide a one-sentence explanation of why the message fits the chosen category.\n      \n      Instructions for categorization:\n      - The message may be in different languages.\n      - It is crucial that you only select a category from the provided list.\n      - If no match is found, or if the match is ambiguous, return \"None\".\n      It is critical to double-check your work before returning the final result.\n      \n      Respond exclusively with a valid JSON object containing the following keys:\n      - matched_category: The selected category or \"None\".\n      - match_explanation: A one-sentence explanation of the match. Make sure to provide clear and concise explanation to justify your decision.\n  \n      Input:\n      {input}\n      ">, any>;

package/dist/extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/constants.js

@@ -6,17 +6,27 @@ const output_parsers_1 = require("langchain/output_parsers");
 const zod_1 = require("zod");
 // Define the Zod schema for the structured response
 exports.messageCategoriesSchema = zod_1.z.object({
-    matched_entity: zod_1.z.string().optional().default('None'), // The matched category
+    matched_category: zod_1.z.string().optional().default('None'),
+    match_explanation: zod_1.z.string(), // One-sentence explanation of the match
 });
 // Create a structured output parser
 exports.parser = output_parsers_1.StructuredOutputParser.fromZodSchema(exports.messageCategoriesSchema);
-// Update the prompt to enforce a structured JSON response
+// TODO: move prompt to LangSmith + further tune
 exports.systemPrompt = prompts_1.ChatPromptTemplate.fromTemplate(`
-      You are an expert in categorizing messages. Your task is to match the provided message to one of the following categories: {categories}.
-      If no match is found, return "None."
+      You are an expert in categorizing messages different messages in a clinical context.
+      Use your expertise to solve the critical task:
+      1. Match the provided message to one of the following categories: {categories}.
+      2. Provide a one-sentence explanation of why the message fits the chosen category.
       
-      Please respond with a valid JSON object with the following key: matched_entity.
+      Instructions for categorization:
+      - The message may be in different languages.
+      - It is crucial that you only select a category from the provided list.
+      - If no match is found, or if the match is ambiguous, return "None".
+      It is critical to double-check your work before returning the final result.
       
+      Respond exclusively with a valid JSON object containing the following keys:
+      - matched_category: The selected category or "None".
+      - match_explanation: A one-sentence explanation of the match. Make sure to provide clear and concise explanation to justify your decision.
   
       Input:
       {input}

package/dist/extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../../../../extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/constants.ts"],"names":[],"mappings":";;;AAAA,qDAA4D;AAC5D,6DAAiE;AACjE,6BAAuB;AAEvB,oDAAoD;AACvC,QAAA,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC9C,cAAc,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,uBAAuB;CAC/E,CAAC,CAAA;AAEF,oCAAoC;AACvB,QAAA,MAAM,GAAG,uCAAsB,CAAC,aAAa,CACxD,+BAAuB,CACxB,CAAA;AAED,0DAA0D;AAC7C,QAAA,YAAY,GAAG,4BAAkB,CAAC,YAAY,CAAC;;;;;;;;;OASrD,CAAC,CAAA"}
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../../../../extensions/shelly/actions/categorizeMessage/lib/categorizeMessageWithLLM/constants.ts"],"names":[],"mappings":";;;AAAA,qDAA4D;AAC5D,6DAAiE;AACjE,6BAAuB;AAEvB,oDAAoD;AACvC,QAAA,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC9C,gBAAgB,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC;IACvD,iBAAiB,EAAE,OAAC,CAAC,MAAM,EAAE,EAAE,wCAAwC;CACxE,CAAC,CAAA;AAEF,oCAAoC;AACvB,QAAA,MAAM,GAAG,uCAAsB,CAAC,aAAa,CACxD,+BAAuB,CACxB,CAAA;AAED,gDAAgD;AACnC,QAAA,YAAY,GAAG,4BAAkB,CAAC,YAAY,CAAC;;;;;;;;;;;;;;;;;;OAkBrD,CAAC,CAAA"}

package/dist/extensions/shelly/actions/index.d.ts

@@ -1,10 +1,17 @@
 declare const actions: {
     summarizeForm: import("@awell-health/extensions-core").Action<{
-        prompt: {
+        additionalInstructions: {
             id: string;
             label: string;
             description: string;
-            type: import("@awell-health/extensions-core").FieldType.TEXT;
+            type: import("@awell-health/extensions-core").FieldType.STRING;
+            required: false;
+        };
+        stakeholder: {
+            id: string;
+            label: string;
+            description: string;
+            type: import("@awell-health/extensions-core").FieldType.STRING;
             required: false;
         };
     }, {
@@ -17,11 +24,18 @@ declare const actions: {
         };
     }, "summary">;
     summarizeCareFlow: import("@awell-health/extensions-core").Action<{
-        prompt: {
+        additionalInstructions: {
             id: string;
             label: string;
             description: string;
-            type: import("@awell-health/extensions-core").FieldType.TEXT;
+            type: import("@awell-health/extensions-core").FieldType.STRING;
+            required: false;
+        };
+        stakeholder: {
+            id: string;
+            label: string;
+            description: string;
+            type: import("@awell-health/extensions-core").FieldType.STRING;
             required: false;
         };
     }, {
@@ -56,7 +70,7 @@ declare const actions: {
             required: true;
             description: string;
         };
-    }, "category">;
+    }, "category" | "explanation">;
     medicationFromImage: import("@awell-health/extensions-core").Action<{
         imageUrl: {
             id: string;

package/dist/extensions/shelly/actions/medicationFromImage/medicationFromImage.js

@@ -19,8 +19,22 @@ exports.medicationFromImage = {
             payload,
         });
         console.log(imageUrl);
-        // Call whatever API with the file URL and get back JSON
-        const data = await Promise.resolve({ dummy: 'data' });
+        // Call whatever API with the file URL and get back JSON (ideally in the following format)
+        const data = await Promise.resolve({
+            medications: [
+                {
+                    product_rxcui: '809854',
+                    product_name: 'hydrochlorothiazide 12.5 MG / quinapril 10 MG Oral Tablet [Accuretic]',
+                    brand_name: 'Accuretic',
+                    dose_form_name: 'Oral Tablet',
+                    prescribable_name: 'Accuretic 10 MG / 12.5 MG Oral Tablet',
+                    extracted_medication_name: 'Accuretic',
+                    extracted_brand_name: 'Accuretic',
+                    extracted_dosage: '10 mg / 12.5 mg',
+                    extracted_ndcg: '0006-0711-31', // National Drug Code (optional)
+                },
+            ],
+        });
         await onComplete({
             data_points: {
                 data: JSON.stringify(data),

package/dist/extensions/shelly/actions/medicationFromImage/medicationFromImage.js.map

@@ -1 +1 @@
-{"version":3,"file":"medicationFromImage.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/medicationFromImage/medicationFromImage.ts"],"names":[],"mappings":";;;AAAA,qDAAqD;AACrD,mEAAqE;AACrE,mCAAuD;AAEvD,qCAAqE;AAExD,QAAA,mBAAmB,GAI5B;IACF,GAAG,EAAE,qBAAqB;IAC1B,QAAQ,EAAE,0BAAQ,CAAC,QAAQ;IAC3B,KAAK,EAAE,+BAA+B;IACtC,WAAW,EAAE,oDAAoD;IACjE,MAAM,EAAN,eAAM;IACN,WAAW,EAAE,KAAK;IAClB,UAAU,EAAV,mBAAU;IACV,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAiB,EAAE;QAC1E,MAAM,EACJ,MAAM,EAAE,EAAE,QAAQ,EAAE,GACrB,GAAG,MAAM,IAAA,iCAA2B,EAAC;YACpC,YAAY,EAAE,+BAAsB;YACpC,OAAO;SACR,CAAC,CAAA;QAEF,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QACrB,wDAAwD;QACxD,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA;QAErD,MAAM,UAAU,CAAC;YACf,WAAW,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B;SACF,CAAC,CAAA;IACJ,CAAC;CACF,CAAA"}
+{"version":3,"file":"medicationFromImage.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/medicationFromImage/medicationFromImage.ts"],"names":[],"mappings":";;;AAAA,qDAAqD;AACrD,mEAAqE;AACrE,mCAAuD;AAEvD,qCAAqE;AAExD,QAAA,mBAAmB,GAI5B;IACF,GAAG,EAAE,qBAAqB;IAC1B,QAAQ,EAAE,0BAAQ,CAAC,QAAQ;IAC3B,KAAK,EAAE,+BAA+B;IACtC,WAAW,EAAE,oDAAoD;IACjE,MAAM,EAAN,eAAM;IACN,WAAW,EAAE,KAAK;IAClB,UAAU,EAAV,mBAAU;IACV,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAiB,EAAE;QAC1E,MAAM,EACJ,MAAM,EAAE,EAAE,QAAQ,EAAE,GACrB,GAAG,MAAM,IAAA,iCAA2B,EAAC;YACpC,YAAY,EAAE,+BAAsB;YACpC,OAAO;SACR,CAAC,CAAA;QAEF,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QAErB,0FAA0F;QAC1F,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC;YACjC,WAAW,EAAE;gBACX;oBACE,aAAa,EAAE,QAAQ;oBACvB,YAAY,EACV,uEAAuE;oBACzE,UAAU,EAAE,WAAW;oBACvB,cAAc,EAAE,aAAa;oBAC7B,iBAAiB,EAAE,uCAAuC;oBAC1D,yBAAyB,EAAE,WAAW;oBACtC,oBAAoB,EAAE,WAAW;oBACjC,gBAAgB,EAAE,iBAAiB;oBACnC,cAAc,EAAE,cAAc,EAAE,gCAAgC;iBACjE;aACF;SACF,CAAC,CAAA;QAEF,MAAM,UAAU,CAAC;YACf,WAAW,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B;SACF,CAAC,CAAA;IACJ,CAAC;CACF,CAAA"}

package/dist/extensions/shelly/actions/medicationFromImage/medicationFromImage.test.js

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const extensions_core_1 = require("@awell-health/extensions-core");
 const tests_1 = require("@/tests");
 const _1 = require(".");
-describe('Shelly - Summarize', () => {
+describe('Shelly - Medication From Image', () => {
     const { onComplete, onError, helpers, extensionAction, clearMocks } = extensions_core_1.TestHelpers.fromAction(_1.medicationFromImage);
     beforeEach(() => {
         clearMocks();
@@ -25,7 +25,7 @@ describe('Shelly - Summarize', () => {
         expect(onError).not.toHaveBeenCalled();
         expect(onComplete).toHaveBeenCalledWith({
             data_points: {
-                data: JSON.stringify({ dummy: 'data' }),
+                data: expect.any(String),
             },
         });
     });

package/dist/extensions/shelly/actions/medicationFromImage/medicationFromImage.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"medicationFromImage.test.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/medicationFromImage/medicationFromImage.test.ts"],"names":[],"mappings":";;AAAA,mEAA2D;AAC3D,mCAA6C;AAC7C,wBAAuC;AAEvC,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,GACjE,6BAAW,CAAC,UAAU,CAAC,sBAAmB,CAAC,CAAA;IAE7C,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,eAAe,CAAC,OAAO,CAAC;YAC5B,OAAO,EAAE,IAAA,2BAAmB,EAAC;gBAC3B,MAAM,EAAE;oBACN,QAAQ,EACN,2GAA2G;iBAC9G;gBACD,QAAQ,EAAE;oBACR,YAAY,EAAE,GAAG;iBAClB;aACF,CAAC;YACF,UAAU;YACV,OAAO;YACP,OAAO;SACR,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QACtC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,WAAW,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;aACxC;SACF,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
+{"version":3,"file":"medicationFromImage.test.js","sourceRoot":"","sources":["../../../../../extensions/shelly/actions/medicationFromImage/medicationFromImage.test.ts"],"names":[],"mappings":";;AAAA,mEAA2D;AAC3D,mCAA6C;AAC7C,wBAAuC;AAEvC,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,GACjE,6BAAW,CAAC,UAAU,CAAC,sBAAmB,CAAC,CAAA;IAE7C,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,eAAe,CAAC,OAAO,CAAC;YAC5B,OAAO,EAAE,IAAA,2BAAmB,EAAC;gBAC3B,MAAM,EAAE;oBACN,QAAQ,EACN,2GAA2G;iBAC9G;gBACD,QAAQ,EAAE;oBACR,YAAY,EAAE,GAAG;iBAClB;aACF,CAAC;YACF,UAAU;YACV,OAAO;YACP,OAAO;SACR,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QACtC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,WAAW,EAAE;gBACX,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aACzB;SACF,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/extensions/shelly/actions/summarizeCareFlow/config/fields.d.ts

@@ -1,18 +1,28 @@
 import { FieldType } from '@awell-health/extensions-core';
 import z from 'zod';
 export declare const fields: {
-    prompt: {
+    additionalInstructions: {
         id: string;
         label: string;
         description: string;
-        type: FieldType.TEXT;
+        type: FieldType.STRING;
+        required: false;
+    };
+    stakeholder: {
+        id: string;
+        label: string;
+        description: string;
+        type: FieldType.STRING;
         required: false;
     };
 };
 export declare const FieldsValidationSchema: z.ZodObject<{
-    prompt: z.ZodOptional<z.ZodString>;
+    additionalInstructions: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    stakeholder: z.ZodEffects<z.ZodOptional<z.ZodString>, string, string | undefined>;
 }, "strip", z.ZodTypeAny, {
-    prompt?: string | undefined;
+    additionalInstructions: string;
+    stakeholder: string;
 }, {
-    prompt?: string | undefined;
+    additionalInstructions?: string | undefined;
+    stakeholder?: string | undefined;
 }>;

package/dist/extensions/shelly/actions/summarizeCareFlow/config/fields.js

@@ -7,15 +7,30 @@ exports.FieldsValidationSchema = exports.fields = void 0;
 const extensions_core_1 = require("@awell-health/extensions-core");
 const zod_1 = __importDefault(require("zod"));
 exports.fields = {
-    prompt: {
-        id: 'prompt',
-        label: 'Prompt',
-        description: 'Leave blank to use the default prompt',
-        type: extensions_core_1.FieldType.TEXT,
+    additionalInstructions: {
+        id: 'additionalInstructions',
+        label: 'Additional instructions',
+        description: 'Specify additional instructions for summarization, for example format, length, what to focus on etc. If not specified, default instructions will be used.',
+        type: extensions_core_1.FieldType.STRING,
+        required: false,
+    },
+    stakeholder: {
+        id: 'stakeholder',
+        label: 'Stakeholder',
+        description: 'Defaults to "Clinician"',
+        type: extensions_core_1.FieldType.STRING,
         required: false,
     },
 };
 exports.FieldsValidationSchema = zod_1.default.object({
-    prompt: zod_1.default.string().optional(),
+    additionalInstructions: zod_1.default.string().optional().default(''),
+    stakeholder: zod_1.default
+        .string()
+        .optional()
+        .transform((val) => {
+        if (val === undefined || val === '')
+            return 'Clinician';
+        return val;
+    }),
 });
 //# sourceMappingURL=fields.js.map

package/dist/extensions/shelly/actions/summarizeCareFlow/config/fields.js.map

@@ -1 +1 @@
-{"version":3,"file":"fields.js","sourceRoot":"","sources":["../../../../../../extensions/shelly/actions/summarizeCareFlow/config/fields.ts"],"names":[],"mappings":";;;;;;AAAA,mEAAqE;AACrE,8CAAwC;AAE3B,QAAA,MAAM,GAAG;IACpB,MAAM,EAAE;QACN,EAAE,EAAE,QAAQ;QACZ,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,uCAAuC;QACpD,IAAI,EAAE,2BAAS,CAAC,IAAI;QACpB,QAAQ,EAAE,KAAK;KAChB;CAC8B,CAAA;AAEpB,QAAA,sBAAsB,GAAG,aAAC,CAAC,MAAM,CAAC;IAC7C,MAAM,EAAE,aAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACoB,CAAC,CAAA"}
+{"version":3,"file":"fields.js","sourceRoot":"","sources":["../../../../../../extensions/shelly/actions/summarizeCareFlow/config/fields.ts"],"names":[],"mappings":";;;;;;AAAA,mEAAqE;AACrE,8CAAwC;AAE3B,QAAA,MAAM,GAAG;IACpB,sBAAsB,EAAE;QACtB,EAAE,EAAE,wBAAwB;QAC5B,KAAK,EAAE,yBAAyB;QAChC,WAAW,EACT,2JAA2J;QAC7J,IAAI,EAAE,2BAAS,CAAC,MAAM;QACtB,QAAQ,EAAE,KAAK;KAChB;IACD,WAAW,EAAE;QACX,EAAE,EAAE,aAAa;QACjB,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE,yBAAyB;QACtC,IAAI,EAAE,2BAAS,CAAC,MAAM;QACtB,QAAQ,EAAE,KAAK;KAChB;CAC8B,CAAA;AAEpB,QAAA,sBAAsB,GAAG,aAAC,CAAC,MAAM,CAAC;IAC7C,sBAAsB,EAAE,aAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IACzD,WAAW,EAAE,aAAC;SACX,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,SAAS,CAAC,CAAC,GAAG,EAAU,EAAE;QACzB,IAAI,GAAG,KAAK,SAAS,IAAI,GAAG,KAAK,EAAE;YAAE,OAAO,WAAW,CAAA;QAEvD,OAAO,GAAG,CAAA;IACZ,CAAC,CAAC;CAC6C,CAAC,CAAA"}