@awell-health/awell-extensions 2.0.22 → 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/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"}

package/dist/extensions/shelly/actions/summarizeCareFlow/lib/summarizeCareFlowWithLLM/constants.d.ts

@@ -0,0 +1,2 @@
+import { ChatPromptTemplate } from '@langchain/core/prompts';
+export declare const systemPrompt: ChatPromptTemplate<import("@langchain/core/prompts").ParamsFromFString<"\n  You are an assistant that provides summaries of activities performed in a care flow for different stakeholders in a healthcare setting.\n  A care flow consists of tracks, steps, and actions that can be performed by a patient, a clinician, or the system itself. These actions may include patient tasks (e.g., form submissions), clinician interventions (e.g., messages sent), or system-triggered activities (e.g., API calls).\n  \n  Your job is to provide a clear, human-readable summary of these activities in chronological order, ensuring the summary is tailored for the specific stakeholder. Focus on activity names and descriptions, avoiding any mention of IDs or technical details.\n  \n  Instructions for summarizing:\n  - Summarize the care flow activities in the order they were performed, maintaining the chronological sequence.\n  - Ensure the summary is concise, clear, and relevant to the stakeholder's needs.\n  - Focus on activity names, titles, and the nature of the actions (e.g., patient submitted a form, clinician responded to a message).\n  - Do not include any IDs, internal codes, or technical jargon—only human-readable information.\n  - Tailor the summary to the specific stakeholder, emphasizing the information most relevant to their role.\n  - Avoid making assumptions about actions not explicitly mentioned in the input.\n  - Provide the summary as a set of bullet points unless otherwise specified in the instructions. Make it as concise as possible. \n  \n  Format:\n  - Provide the summary as a set of bullet points unless otherwise specified in the instructions. Make it as concise as possible.\n  - Format it as an easy-to-grasp timeline with all important values and information. \n  - Always mention track, step, and action names if applicable unless otherwise specified in the additional instructions.\n  - Never format it in markdown with bold, italic, etc., unless otherwise specified in the additional instructions.\n  - Do not list usign words like next, finally, just list sumamries in bullet points, concisely.\n\n  Additional Instructions:\n  {additionalInstructions}\n\n  Stakeholder:\n  {stakeholder}\n\n  Content to summarize:\n  {input}\n  ">, any>;

package/dist/extensions/shelly/actions/summarizeCareFlow/lib/summarizeCareFlowWithLLM/constants.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.systemPrompt = void 0;
+const prompts_1 = require("@langchain/core/prompts");
+// TODO: move prompt to LangSmith + further tune
+exports.systemPrompt = prompts_1.ChatPromptTemplate.fromTemplate(`
+  You are an assistant that provides summaries of activities performed in a care flow for different stakeholders in a healthcare setting.
+  A care flow consists of tracks, steps, and actions that can be performed by a patient, a clinician, or the system itself. These actions may include patient tasks (e.g., form submissions), clinician interventions (e.g., messages sent), or system-triggered activities (e.g., API calls).
+  
+  Your job is to provide a clear, human-readable summary of these activities in chronological order, ensuring the summary is tailored for the specific stakeholder. Focus on activity names and descriptions, avoiding any mention of IDs or technical details.
+  
+  Instructions for summarizing:
+  - Summarize the care flow activities in the order they were performed, maintaining the chronological sequence.
+  - Ensure the summary is concise, clear, and relevant to the stakeholder's needs.
+  - Focus on activity names, titles, and the nature of the actions (e.g., patient submitted a form, clinician responded to a message).
+  - Do not include any IDs, internal codes, or technical jargon—only human-readable information.
+  - Tailor the summary to the specific stakeholder, emphasizing the information most relevant to their role.
+  - Avoid making assumptions about actions not explicitly mentioned in the input.
+  - Provide the summary as a set of bullet points unless otherwise specified in the instructions. Make it as concise as possible. 
+  
+  Format:
+  - Provide the summary as a set of bullet points unless otherwise specified in the instructions. Make it as concise as possible.
+  - Format it as an easy-to-grasp timeline with all important values and information. 
+  - Always mention track, step, and action names if applicable unless otherwise specified in the additional instructions.
+  - Never format it in markdown with bold, italic, etc., unless otherwise specified in the additional instructions.
+  - Do not list usign words like next, finally, just list sumamries in bullet points, concisely.
+
+  Additional Instructions:
+  {additionalInstructions}
+
+  Stakeholder:
+  {stakeholder}
+
+  Content to summarize:
+  {input}
+  `);
+//# sourceMappingURL=constants.js.map

package/dist/extensions/shelly/actions/summarizeCareFlow/lib/summarizeCareFlowWithLLM/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../../../../extensions/shelly/actions/summarizeCareFlow/lib/summarizeCareFlowWithLLM/constants.ts"],"names":[],"mappings":";;;AAAA,qDAA4D;AAE5D,gDAAgD;AACnC,QAAA,YAAY,GAAG,4BAAkB,CAAC,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BzD,CAAC,CAAA"}

package/dist/extensions/shelly/actions/summarizeCareFlow/lib/summarizeCareFlowWithLLM/index.d.ts

@@ -0,0 +1 @@
+export { summarizeCareFlowWithLLM } from './summarizeCareFlowWithLLM';

package/dist/extensions/shelly/actions/summarizeCareFlow/lib/summarizeCareFlowWithLLM/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.summarizeCareFlowWithLLM = void 0;
+var summarizeCareFlowWithLLM_1 = require("./summarizeCareFlowWithLLM");
+Object.defineProperty(exports, "summarizeCareFlowWithLLM", { enumerable: true, get: function () { return summarizeCareFlowWithLLM_1.summarizeCareFlowWithLLM; } });
+//# sourceMappingURL=index.js.map