llm-exe 0.0.6 → 1.0.0

package/dist/prompt/chat.js

@@ -52,6 +52,14 @@ class ChatPrompt extends _base_1.BasePrompt {
                 case "assistant":
                     this.addAssistantMessage(content);
                     break;
+                case "function":
+                    (0, utils_1.assert)(name, "Function message requires name");
+                    this.addFunctionMessage(content, name);
+                    break;
+                case "function_call":
+                    (0, utils_1.assert)(name, "Function message requires name");
+                    this.addFunctionCallMessage({ name: name, arguments: content });
+                    break;
             }
         }
         return this;
@@ -85,6 +93,37 @@ class ChatPrompt extends _base_1.BasePrompt {
         });
         return this;
     }
+    /**
+     * addFunctionMessage Helper to add an assistant message to the prompt.
+     * @param content The message content.
+     * @return ChatPrompt so it can be chained.
+     */
+    addFunctionMessage(content, name) {
+        this.messages.push({
+            role: "function",
+            name,
+            content,
+        });
+        return this;
+    }
+    /**
+     * addFunctionCallMessage Helper to add an assistant message to the prompt.
+     * @param content The message content.
+     * @return ChatPrompt so it can be chained.
+     */
+    addFunctionCallMessage(function_call) {
+        if (function_call) {
+            this.messages.push({
+                role: "assistant",
+                function_call: {
+                    name: function_call.name,
+                    arguments: (0, utils_1.maybeStringifyJSON)(function_call.arguments),
+                },
+                content: null,
+            });
+        }
+        return this;
+    }
     /**
      * addFromHistory Adds multiple messages at one time.
      * @param history History of chat messages.
@@ -98,11 +137,19 @@ class ChatPrompt extends _base_1.BasePrompt {
                         this.addUserMessage(message.content, message?.name);
                         break;
                     case "assistant":
-                        this.addAssistantMessage(message.content);
+                        if (message.function_call) {
+                            this.addFunctionCallMessage(message.function_call);
+                        }
+                        else if (message?.content) {
+                            this.addAssistantMessage(message?.content);
+                        }
                         break;
                     case "system":
                         this.addSystemMessage(message.content);
                         break;
+                    case "function":
+                        this.addFunctionMessage(message.content, message.name);
+                        break;
                 }
             }
         }
@@ -113,10 +160,16 @@ class ChatPrompt extends _base_1.BasePrompt {
      * @param content The message content
      * @return returns ChatPrompt so it can be chained.
      */
-    addChatHistoryPlaceholder(key) {
+    addChatHistoryPlaceholder(key, options) {
         const start = `{{> DialogueHistory `;
         const params = [`key='${String(key)}'`];
         const end = `}}`;
+        if (options?.assistant) {
+            params.push(`assistant='${options.assistant}'`);
+        }
+        if (options?.user) {
+            params.push(`user='${options.user}'`);
+        }
         this.messages.push({
             role: "placeholder",
             content: `${start}${params.join(" ")}${end}`,
@@ -163,17 +216,39 @@ class ChatPrompt extends _base_1.BasePrompt {
                 switch (token) {
                     case ">DialogueHistory": {
                         /* istanbul ignore next */
-                        const { key = "" } = data;
+                        const { key = "", user } = data;
                         const history = (0, utils_1.get)(replacements, key, []);
                         if (history && Array.isArray(history)) {
                             for (const message of history) {
                                 switch (message.role) {
-                                    case "user":
-                                        messagesOut.push((0, utils_1.pick)(message, ["role", "content", "name"]));
+                                    case "user": {
+                                        const m = (0, utils_1.pick)(message, ["role", "content", "name"]);
+                                        if (user) {
+                                            m["name"] = user;
+                                        }
+                                        messagesOut.push(m);
+                                        break;
+                                    }
+                                    case "assistant": {
+                                        if (message.function_call) {
+                                            messagesOut.push({
+                                                role: "assistant",
+                                                content: null,
+                                                function_call: message.function_call
+                                            });
+                                        }
+                                        else if (message?.content) {
+                                            messagesOut.push({
+                                                role: "assistant",
+                                                content: message.content,
+                                            });
+                                        }
                                         break;
-                                    case "assistant":
+                                    }
+                                    case "function":
                                         messagesOut.push({
-                                            role: "assistant",
+                                            role: "function",
+                                            name: message.name,
                                             content: message.content,
                                         });
                                         break;
@@ -197,7 +272,7 @@ class ChatPrompt extends _base_1.BasePrompt {
                                 content: (0, utils_1.replaceTemplateString)(content, replacements, {
                                     partials: this.partials,
                                     helpers: this.helpers,
-                                })
+                                }),
                             };
                             if (!name || role !== "user") {
                                 delete message.name;
@@ -208,13 +283,23 @@ class ChatPrompt extends _base_1.BasePrompt {
                     }
                 }
             }
+            else if (message.role === "function") {
+                messagesOut.push(Object.assign({}, message, {
+                    content: (0, utils_1.replaceTemplateString)(message.content, replacements, {
+                        partials: this.partials,
+                        helpers: this.helpers,
+                    }),
+                }));
+            }
             else {
                 if (safeToParseTemplate.includes(message.role)) {
                     messagesOut.push(Object.assign({}, message, {
-                        content: (0, utils_1.replaceTemplateString)(message.content, replacements, {
-                            partials: this.partials,
-                            helpers: this.helpers,
-                        }),
+                        content: message.content
+                            ? (0, utils_1.replaceTemplateString)(message.content, replacements, {
+                                partials: this.partials,
+                                helpers: this.helpers,
+                            })
+                            : null,
                     }));
                 }
                 else {

package/dist/prompt/chat.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat.js","sourceRoot":"src/","sources":["prompt/chat.ts"],"names":[],"mappings":";;;AAAA,mCAMiB;AACjB,mCAAqC;AAgBrC;;;;GAIG;AACH,MAAa,UAA0C,SAAQ,kBAAa;IAa1E;;;;OAIG;IACH,YACE,0BAAmC,EACnC,OAA2B;QAE3B,KAAK,CAAC,0BAA0B,CAAC,CAAC;QArBpC;;WAEG;QACH;;;;mBAAgC,MAAM;WAAC;QAEvC;;;;WAIG;QACH;;;;mBAA6B,KAAK;WAAC;QAYjC,IAAI,OAAO,EAAE,uBAAuB,EAAE;YACpC,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;SAChC;IACH,CAAC;IAwBD,WAAW,CACT,OAAe,EACf,IAAsB,EACtB,IAAa;QAEb,IAAI,OAAO,EAAE;YACX,QAAQ,IAAI,EAAE;gBACZ,KAAK,QAAQ;oBACX,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBAC/B,MAAM;gBACR,KAAK,MAAM;oBACT,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;oBACnC,MAAM;gBACR,KAAK,WAAW;oBACd,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;oBAClC,MAAM;aACT;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACH,cAAc,CAAC,OAAe,EAAE,IAAa;QAC3C,MAAM,OAAO,GAAqB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO;SACR,CAAC;QACF,IAAI,IAAI,EAAE;YACR,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;SACrB;QACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,mBAAmB,CAAC,OAAe;QACjC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,WAAW;YACjB,OAAO;SACR,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,cAAc,CAAC,OAAsB;QACnC,IAAI,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YACrC,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE;gBAC7B,QAAQ,OAAO,CAAC,IAAI,EAAE;oBACpB,KAAK,MAAM;wBACT,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;wBACpD,MAAM;oBACR,KAAK,WAAW;wBACd,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;wBAC1C,MAAM;oBACR,KAAK,QAAQ;wBACX,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;wBACvC,MAAM;iBACT;aACF;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,yBAAyB,CAAC,GAAY;QACpC,MAAM,KAAK,GAAG,sBAAsB,CAAC;QACrC,MAAM,MAAM,GAAG,CAAC,QAAQ,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,aAAa;YACnB,OAAO,EAAE,GAAG,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,EAAE;SAC7C,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,qBAAqB,CACnB,OAAe,EACf,OAAyB,MAAM,EAC/B,IAAa;QAEb,IAAI,OAAO,EAAE;YACX,MAAM,KAAK,GAAG,wBAAwB,CAAC;YACvC,MAAM,MAAM,GAAG,CAAC,SAAS,IAAI,GAAG,EAAE,YAAY,IAAA,cAAM,EAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClE,MAAM,GAAG,GAAG,IAAI,CAAC;YACjB,IAAG,IAAI,EAAC;gBACN,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,GAAG,CAAC,CAAA;aAC9B;YACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,GAAG,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,EAAE;aAC7C,CAAC,CAAC;SACJ;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,MAAS;QACd,MAAM,WAAW,GAAkB,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAClD,MAAM,mBAAmB,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAEpD,IAAI,IAAI,CAAC,kBAAkB,EAAE;YAC3B,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;SAClC;QAED,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE;YACnC,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,EAAE;gBAClC,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,IAAA,qCAA6B,EACtD,OAAO,CAAC,OAAO,CAChB,CAAC;gBACF,QAAQ,KAAK,EAAE;oBACb,KAAK,kBAAkB,CAAC,CAAC;wBACvB,0BAA0B;wBAC1B,MAAM,EAAE,GAAG,GAAG,EAAE,EAAE,GAAG,IAAI,CAAC;wBAC1B,MAAM,OAAO,GAAkB,IAAA,WAAG,EAAC,YAAY,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBAC1D,IAAI,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;4BACrC,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE;gCAC7B,QAAQ,OAAO,CAAC,IAAI,EAAE;oCACpB,KAAK,MAAM;wCACT,WAAW,CAAC,IAAI,CACd,IAAA,YAAI,EAAC,OAAO,EAAE,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAC3C,CAAC;wCACF,MAAM;oCACR,KAAK,WAAW;wCACd,WAAW,CAAC,IAAI,CAAC;4CACf,IAAI,EAAE,WAAW;4CACjB,OAAO,EAAE,OAAO,CAAC,OAAO;yCACzB,CAAC,CAAC;wCACH,MAAM;oCACR,KAAK,QAAQ;wCACX,WAAW,CAAC,IAAI,CAAC;4CACf,IAAI,EAAE,QAAQ;4CACd,OAAO,EAAE,OAAO,CAAC,OAAO;yCACzB,CAAC,CAAC;wCACH,MAAM;iCACT;6BACF;yBACF;wBACD,MAAM;qBACP;oBACD,KAAK,oBAAoB,CAAC,CAAC;wBACzB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;wBACrC,IAAG,IAAI,IAAI,OAAO,EAAC;4BACjB,MAAM,OAAO,GAAG;gCACd,IAAI;gCACJ,IAAI;gCACJ,OAAO,EAAE,IAAA,6BAAqB,EAAC,OAAO,EAAE,YAAY,EAAE;oCACpD,QAAQ,EAAE,IAAI,CAAC,QAAQ;oCACvB,OAAO,EAAE,IAAI,CAAC,OAAO;iCACtB,CAAC;6BACH,CAAA;4BAED,IAAG,CAAC,IAAI,IAAI,IAAI,KAAK,MAAM,EAAC;gCAC1B,OAAO,OAAO,CAAC,IAAI,CAAC;6BACrB;4BACD,WAAW,CAAC,IAAI,CAAC,OAAuB,CAAC,CAAC;yBAC3C;wBACD,MAAM;qBACP;iBACF;aACF;iBAAM;gBACL,IAAI,mBAAmB,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;oBAC9C,WAAW,CAAC,IAAI,CACd,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE;wBACzB,OAAO,EAAE,IAAA,6BAAqB,EAAC,OAAO,CAAC,OAAO,EAAE,YAAY,EAAE;4BAC5D,QAAQ,EAAE,IAAI,CAAC,QAAQ;4BACvB,OAAO,EAAE,IAAI,CAAC,OAAO;yBACtB,CAAC;qBACH,CAAC,CACH,CAAC;iBACH;qBAAM;oBACL,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;iBAC3B;aACF;SACF;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;;;OAIG;IACH,QAAQ;QACN,0CAA0C;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAzQD,gCAyQC"}
+{"version":3,"file":"chat.js","sourceRoot":"src/","sources":["prompt/chat.ts"],"names":[],"mappings":";;;AAAA,mCAQiB;AACjB,mCAAqC;AAgBrC;;;;GAIG;AACH,MAAa,UAA0C,SAAQ,kBAAa;IAa1E;;;;OAIG;IACH,YACE,0BAAmC,EACnC,OAA2B;QAE3B,KAAK,CAAC,0BAA0B,CAAC,CAAC;QArBpC;;WAEG;QACH;;;;mBAAgC,MAAM;WAAC;QAEvC;;;;WAIG;QACH;;;;mBAA6B,KAAK;WAAC;QAYjC,IAAI,OAAO,EAAE,uBAAuB,EAAE;YACpC,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;SAChC;IACH,CAAC;IAkCD,WAAW,CACT,OAAe,EACf,IAAsB,EACtB,IAAa;QAEb,IAAI,OAAO,EAAE;YACX,QAAQ,IAAI,EAAE;gBACZ,KAAK,QAAQ;oBACX,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBAC/B,MAAM;gBACR,KAAK,MAAM;oBACT,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;oBACnC,MAAM;gBACR,KAAK,WAAW;oBACd,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;oBAClC,MAAM;gBACR,KAAK,UAAU;oBACb,IAAA,cAAM,EAAC,IAAI,EAAE,gCAAgC,CAAC,CAAC;oBAC/C,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;oBACvC,MAAM;gBACR,KAAK,eAAe;oBAClB,IAAA,cAAM,EAAC,IAAI,EAAE,gCAAgC,CAAC,CAAC;oBAC/C,IAAI,CAAC,sBAAsB,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;oBAChE,MAAM;aACT;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACH,cAAc,CAAC,OAAe,EAAE,IAAa;QAC3C,MAAM,OAAO,GAAqB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO;SACR,CAAC;QACF,IAAI,IAAI,EAAE;YACR,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;SACrB;QACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,mBAAmB,CAAC,OAAe;QACjC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,WAAW;YACjB,OAAO;SACR,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,kBAAkB,CAAC,OAAe,EAAE,IAAY;QAC9C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,UAAU;YAChB,IAAI;YACJ,OAAO;SACR,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IACD;;;;OAIG;IAEH,sBAAsB,CAAC,aAAmD;QACxE,IAAI,aAAa,EAAE;YACjB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,WAAW;gBACjB,aAAa,EAAE;oBACb,IAAI,EAAE,aAAa,CAAC,IAAI;oBACxB,SAAS,EAAE,IAAA,0BAAkB,EAAC,aAAa,CAAC,SAAS,CAAC;iBACvD;gBACD,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;SACJ;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IACD;;;;OAIG;IACH,cAAc,CAAC,OAAsB;QACnC,IAAI,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YACrC,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE;gBAC7B,QAAQ,OAAO,CAAC,IAAI,EAAE;oBACpB,KAAK,MAAM;wBACT,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;wBACpD,MAAM;oBACR,KAAK,WAAW;wBACd,IAAI,OAAO,CAAC,aAAa,EAAE;4BACzB,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;yBACpD;6BAAM,IAAI,OAAO,EAAE,OAAO,EAAE;4BAC3B,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;yBAC5C;wBACD,MAAM;oBACR,KAAK,QAAQ;wBACX,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;wBACvC,MAAM;oBACR,KAAK,UAAU;wBACb,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;wBACvD,MAAM;iBACT;aACF;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,yBAAyB,CACvB,GAAY,EACZ,OAA+C;QAE/C,MAAM,KAAK,GAAG,sBAAsB,CAAC;QACrC,MAAM,MAAM,GAAG,CAAC,QAAQ,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC;QAEjB,IAAI,OAAO,EAAE,SAAS,EAAE;YACtB,MAAM,CAAC,IAAI,CAAC,cAAc,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC;SACjD;QACD,IAAI,OAAO,EAAE,IAAI,EAAE;YACjB,MAAM,CAAC,IAAI,CAAC,SAAS,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC;SACvC;QACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,aAAa;YACnB,OAAO,EAAE,GAAG,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,EAAE;SAC7C,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,qBAAqB,CACnB,OAAe,EACf,OAAyB,MAAM,EAC/B,IAAa;QAEb,IAAI,OAAO,EAAE;YACX,MAAM,KAAK,GAAG,wBAAwB,CAAC;YACvC,MAAM,MAAM,GAAG,CAAC,SAAS,IAAI,GAAG,EAAE,YAAY,IAAA,cAAM,EAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClE,MAAM,GAAG,GAAG,IAAI,CAAC;YACjB,IAAI,IAAI,EAAE;gBACR,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,GAAG,CAAC,CAAC;aAC/B;YACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,GAAG,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,EAAE;aAC7C,CAAC,CAAC;SACJ;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,MAAS;QACd,MAAM,WAAW,GAAkB,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAClD,MAAM,mBAAmB,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAEpD,IAAI,IAAI,CAAC,kBAAkB,EAAE;YAC3B,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;SAClC;QAED,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE;YACnC,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,EAAE;gBAClC,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,IAAA,qCAA6B,EACtD,OAAO,CAAC,OAAO,CAChB,CAAC;gBACF,QAAQ,KAAK,EAAE;oBACb,KAAK,kBAAkB,CAAC,CAAC;wBACvB,0BAA0B;wBAC1B,MAAM,EAAE,GAAG,GAAG,EAAE,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;wBAChC,MAAM,OAAO,GAAkB,IAAA,WAAG,EAAC,YAAY,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBAC1D,IAAI,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;4BACrC,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE;gCAC7B,QAAQ,OAAO,CAAC,IAAI,EAAE;oCACpB,KAAK,MAAM,CAAC,CAAC;wCACX,MAAM,CAAC,GAAG,IAAA,YAAI,EAAC,OAAO,EAAE,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;wCACrD,IAAI,IAAI,EAAE;4CACR,CAAC,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;yCAClB;wCACD,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;wCACpB,MAAM;qCACP;oCAED,KAAK,WAAW,CAAC,CAAA;wCACf,IAAG,OAAO,CAAC,aAAa,EAAC;4CACvB,WAAW,CAAC,IAAI,CAAC;gDACf,IAAI,EAAE,WAAW;gDACjB,OAAO,EAAE,IAAI;gDACb,aAAa,EAAE,OAAO,CAAC,aAAa;6CACrC,CAAC,CAAC;yCACJ;6CAAK,IAAG,OAAO,EAAE,OAAO,EAAC;4CACxB,WAAW,CAAC,IAAI,CAAC;gDACf,IAAI,EAAE,WAAW;gDACjB,OAAO,EAAE,OAAO,CAAC,OAAO;6CACzB,CAAC,CAAC;yCACJ;wCACD,MAAM;qCACP;oCACD,KAAK,UAAU;wCACb,WAAW,CAAC,IAAI,CAAC;4CACf,IAAI,EAAE,UAAU;4CAChB,IAAI,EAAE,OAAO,CAAC,IAAI;4CAClB,OAAO,EAAE,OAAO,CAAC,OAAO;yCACzB,CAAC,CAAC;wCACH,MAAM;oCACR,KAAK,QAAQ;wCACX,WAAW,CAAC,IAAI,CAAC;4CACf,IAAI,EAAE,QAAQ;4CACd,OAAO,EAAE,OAAO,CAAC,OAAO;yCACzB,CAAC,CAAC;wCACH,MAAM;iCACT;6BACF;yBACF;wBACD,MAAM;qBACP;oBACD,KAAK,oBAAoB,CAAC,CAAC;wBACzB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;wBACrC,IAAI,IAAI,IAAI,OAAO,EAAE;4BACnB,MAAM,OAAO,GAAG;gCACd,IAAI;gCACJ,IAAI;gCACJ,OAAO,EAAE,IAAA,6BAAqB,EAAC,OAAO,EAAE,YAAY,EAAE;oCACpD,QAAQ,EAAE,IAAI,CAAC,QAAQ;oCACvB,OAAO,EAAE,IAAI,CAAC,OAAO;iCACtB,CAAC;6BACH,CAAC;4BAEF,IAAI,CAAC,IAAI,IAAI,IAAI,KAAK,MAAM,EAAE;gCAC5B,OAAO,OAAO,CAAC,IAAI,CAAC;6BACrB;4BACD,WAAW,CAAC,IAAI,CAAC,OAAuB,CAAC,CAAC;yBAC3C;wBACD,MAAM;qBACP;iBACF;aACF;iBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,EAAE;gBACtC,WAAW,CAAC,IAAI,CACd,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE;oBACzB,OAAO,EAAE,IAAA,6BAAqB,EAAC,OAAO,CAAC,OAAO,EAAE,YAAY,EAAE;wBAC5D,QAAQ,EAAE,IAAI,CAAC,QAAQ;wBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;qBACtB,CAAC;iBACH,CAAC,CACH,CAAC;aACH;iBAAM;gBACL,IAAI,mBAAmB,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;oBAC9C,WAAW,CAAC,IAAI,CACd,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE;wBACzB,OAAO,EAAE,OAAO,CAAC,OAAO;4BACtB,CAAC,CAAC,IAAA,6BAAqB,EAAC,OAAO,CAAC,OAAO,EAAE,YAAY,EAAE;gCACnD,QAAQ,EAAE,IAAI,CAAC,QAAQ;gCACvB,OAAO,EAAE,IAAI,CAAC,OAAO;6BACtB,CAAC;4BACJ,CAAC,CAAC,IAAI;qBACT,CAAC,CACH,CAAC;iBACH;qBAAM;oBACL,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;iBAC3B;aACF;SACF;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;;;OAIG;IACH,QAAQ;QACN,0CAA0C;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AA5WD,gCA4WC"}

package/dist/state/dialogue.d.ts

@@ -1,4 +1,4 @@
-import { IChatMessages, IChatUserMessage } from "../types";
+import { IChatMessages } from "../types";
 import { BaseStateItem } from "./item";
 export declare class Dialogue extends BaseStateItem<IChatMessages> {
     name: string;
@@ -6,13 +6,19 @@ export declare class Dialogue extends BaseStateItem<IChatMessages> {
     setUserMessage(content: string, name?: string): this;
     setAssistantMessage(content: string): this;
     setSystemMessage(content: string): this;
+    setFunctionMessage(content: string, name: string): this;
+    setFunctionCallMessage(input: {
+        function_call: {
+            name: string;
+            arguments: string;
+        };
+    }): this;
     setMessageTurn(userMessage: string, assistantMessage: string, systemMessage?: string): this;
     setHistory(messages: IChatMessages): this;
     getHistory(): IChatMessages;
     serialize(): {
         class: string;
         name: string;
-        value: (IChatUserMessage | import("../types").IChatAssistantMessage | import("../types").IChatSystemMessage)[];
+        value: import("../types").IChatMessage[];
     };
-    deserialize(): void;
 }

package/dist/state/dialogue.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Dialogue = void 0;
 const item_1 = require("./item");
+const utils_1 = require("../utils");
 class Dialogue extends item_1.BaseStateItem {
     constructor(name) {
         super(name, []);
@@ -44,6 +45,27 @@ class Dialogue extends item_1.BaseStateItem {
         }
         return this;
     }
+    setFunctionMessage(content, name) {
+        if (content) {
+            this.value.push({
+                role: "function",
+                name,
+                content,
+            });
+        }
+        return this;
+    }
+    setFunctionCallMessage(input) {
+        this.value.push({
+            role: "assistant",
+            function_call: {
+                name: input?.function_call.name,
+                arguments: (0, utils_1.maybeStringifyJSON)(input?.function_call.arguments),
+            },
+            content: null,
+        });
+        return this;
+    }
     setMessageTurn(userMessage, assistantMessage, systemMessage = "") {
         this.setUserMessage(userMessage);
         this.setAssistantMessage(assistantMessage);
@@ -57,11 +79,19 @@ class Dialogue extends item_1.BaseStateItem {
                     this.setUserMessage(message?.content, message?.name);
                     break;
                 case "assistant":
-                    this.setAssistantMessage(message?.content);
+                    if (message.function_call) {
+                        this.setFunctionCallMessage({ function_call: message.function_call });
+                    }
+                    else if (message?.content) {
+                        this.setAssistantMessage(message?.content);
+                    }
                     break;
                 case "system":
                     this.setSystemMessage(message?.content);
                     break;
+                case "function":
+                    this.setFunctionMessage(message?.content, message.name);
+                    break;
             }
         }
         return this;
@@ -76,7 +106,6 @@ class Dialogue extends item_1.BaseStateItem {
             value: [...this.value],
         };
     }
-    deserialize() { }
 }
 exports.Dialogue = Dialogue;
 //# sourceMappingURL=dialogue.js.map

package/dist/state/dialogue.js.map

@@ -1 +1 @@
-{"version":3,"file":"dialogue.js","sourceRoot":"src/","sources":["state/dialogue.ts"],"names":[],"mappings":";;;AACA,iCAAuC;AAEvC,MAAa,QAAS,SAAQ,oBAA4B;IAGxD,YAAY,IAAY;QACtB,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAHlB;;;;;WAAoB;QAIlB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,cAAc,CAAC,OAAe,EAAE,IAAa;QAC3C,IAAI,OAAO,EAAE;YACX,MAAM,GAAG,GAAqB;gBAC5B,IAAI,EAAE,MAAM;gBACZ,OAAO;aACR,CAAC;YACF,IAAI,IAAI,EAAE;gBACR,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;aACjB;YACD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;SACtB;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,mBAAmB,CAAC,OAAe;QACjC,IAAI,OAAO,EAAE;YACX,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,WAAW;gBACjB,OAAO;aACR,CAAC,CAAC;SACJ;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB,CAAC,OAAe;QAC9B,IAAI,OAAO,EAAE;YACX,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,QAAQ;gBACd,OAAO;aACR,CAAC,CAAC;SACJ;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,cAAc,CACZ,WAAmB,EACnB,gBAAwB,EACxB,gBAAwB,EAAE;QAE1B,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QACjC,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;QAC3C,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,UAAU,CAAC,QAAuB;QAChC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;YAC9B,QAAQ,OAAO,EAAE,IAAI,EAAE;gBACrB,KAAK,MAAM;oBACT,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;oBACrD,MAAM;gBACR,KAAK,WAAW;oBACd,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBAC3C,MAAM;gBACR,KAAK,QAAQ;oBACX,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBACxC,MAAM;aACT;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,UAAU;YACjB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;SACvB,CAAC;IACJ,CAAC;IACD,WAAW,KAAI,CAAC;CACjB;AAlFD,4BAkFC"}
+{"version":3,"file":"dialogue.js","sourceRoot":"src/","sources":["state/dialogue.ts"],"names":[],"mappings":";;;AACA,iCAAuC;AACvC,mCAA6C;AAE7C,MAAa,QAAS,SAAQ,oBAA4B;IAGxD,YAAY,IAAY;QACtB,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAHlB;;;;;WAAoB;QAIlB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,cAAc,CAAC,OAAe,EAAE,IAAa;QAC3C,IAAI,OAAO,EAAE;YACX,MAAM,GAAG,GAAqB;gBAC5B,IAAI,EAAE,MAAM;gBACZ,OAAO;aACR,CAAC;YACF,IAAI,IAAI,EAAE;gBACR,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;aACjB;YACD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;SACtB;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,mBAAmB,CAAC,OAAe;QACjC,IAAI,OAAO,EAAE;YACX,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,WAAW;gBACjB,OAAO;aACR,CAAC,CAAC;SACJ;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB,CAAC,OAAe;QAC9B,IAAI,OAAO,EAAE;YACX,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,QAAQ;gBACd,OAAO;aACR,CAAC,CAAC;SACJ;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,kBAAkB,CAAC,OAAe,EAAE,IAAY;QAC9C,IAAI,OAAO,EAAE;YACX,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,UAAU;gBAChB,IAAI;gBACJ,OAAO;aACR,CAAC,CAAC;SACJ;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IACD,sBAAsB,CAAC,KAA2D;QAChF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,WAAW;YACjB,aAAa,EAAE;gBACb,IAAI,EAAE,KAAK,EAAE,aAAa,CAAC,IAAI;gBAC/B,SAAS,EAAE,IAAA,0BAAkB,EAAC,KAAK,EAAE,aAAa,CAAC,SAAS,CAAC;aAC9D;YACD,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IACD,cAAc,CACZ,WAAmB,EACnB,gBAAwB,EACxB,gBAAwB,EAAE;QAE1B,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QACjC,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;QAC3C,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,UAAU,CAAC,QAAuB;QAChC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;YAC9B,QAAQ,OAAO,EAAE,IAAI,EAAE;gBACrB,KAAK,MAAM;oBACT,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;oBACrD,MAAM;gBACR,KAAK,WAAW;oBACd,IAAG,OAAO,CAAC,aAAa,EAAC;wBACvB,IAAI,CAAC,sBAAsB,CAAC,EAAE,aAAa,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;qBACvE;yBAAK,IAAG,OAAO,EAAE,OAAO,EAAC;wBACxB,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;qBAC5C;oBACD,MAAM;gBACR,KAAK,QAAQ;oBACX,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBACxC,MAAM;gBACR,KAAK,UAAU;oBACb,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;oBACxD,MAAM;aACT;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAA;IACxB,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,UAAU;YACjB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;SACvB,CAAC;IACJ,CAAC;CAEF;AA9GD,4BA8GC"}

package/dist/state/item.d.ts

@@ -16,7 +16,6 @@ export declare abstract class BaseStateItem<T> implements Serializable {
         name: string;
         value: any;
     };
-    deserialize(): void;
 }
 export declare class DefaultStateItem extends BaseStateItem<any> {
     constructor(name: string, defaultValue: any);

package/dist/state/item.js

@@ -52,7 +52,6 @@ class BaseStateItem {
             value: this.serializeValue(),
         };
     }
-    deserialize() { }
 }
 exports.BaseStateItem = BaseStateItem;
 class DefaultStateItem extends BaseStateItem {

package/dist/state/item.js.map

@@ -1 +1 @@
-{"version":3,"file":"item.js","sourceRoot":"src/","sources":["state/item.ts"],"names":[],"mappings":";;;AACA,mCAAiC;AAEjC,MAAsB,aAAa;IAKjC,YAAY,GAAW,EAAE,YAAe;QAJxC;;;;;WAAsB;QACtB;;;;;WAAmB;QACnB;;;;;WAA0B;QAGxB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IACD,QAAQ,CAAC,KAAQ;QACf,IAAA,cAAM,EACJ,OAAO,KAAK,KAAK,OAAO,IAAI,CAAC,KAAK,EAClC,gCAAgC,OAAO,IAAI;aACxC,KAAK,cAAc,OAAO,KAAK,EAAE,CACrC,CAAC;QACF,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IACD,MAAM;QACJ,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;IACD,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IACD,UAAU;QACR,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC;IACjC,CAAC;IACD,cAAc;QACZ,OAAO;YACL,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,IAAI,CAAC,QAAQ,EAAE;SACjC,CAAC;IACJ,CAAC;IACD,SAAS;QACP,OAAO;YACL,KAAK,EAAE,eAAe;YACtB,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;YACnB,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE;SAC7B,CAAC;IACJ,CAAC;IACD,WAAW,KAAI,CAAC;CACjB;AAxCD,sCAwCC;AAED,MAAa,gBAAiB,SAAQ,aAAkB;IACtD,YAAY,IAAY,EAAE,YAAiB;QACzC,KAAK,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IAC5B,CAAC;CAGF;AAND,4CAMC"}
+{"version":3,"file":"item.js","sourceRoot":"src/","sources":["state/item.ts"],"names":[],"mappings":";;;AACA,mCAAiC;AAEjC,MAAsB,aAAa;IAKjC,YAAY,GAAW,EAAE,YAAe;QAJxC;;;;;WAAsB;QACtB;;;;;WAAmB;QACnB;;;;;WAA0B;QAGxB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IACD,QAAQ,CAAC,KAAQ;QACf,IAAA,cAAM,EACJ,OAAO,KAAK,KAAK,OAAO,IAAI,CAAC,KAAK,EAClC,gCAAgC,OAAO,IAAI;aACxC,KAAK,cAAc,OAAO,KAAK,EAAE,CACrC,CAAC;QACF,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IACD,MAAM;QACJ,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;IACD,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IACD,UAAU;QACR,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC;IACjC,CAAC;IACD,cAAc;QACZ,OAAO;YACL,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,IAAI,CAAC,QAAQ,EAAE;SACjC,CAAC;IACJ,CAAC;IACD,SAAS;QACP,OAAO;YACL,KAAK,EAAE,eAAe;YACtB,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;YACnB,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE;SAC7B,CAAC;IACJ,CAAC;CAEF;AAxCD,sCAwCC;AAED,MAAa,gBAAiB,SAAQ,aAAkB;IACtD,YAAY,IAAY,EAAE,YAAiB;QACzC,KAAK,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IAC5B,CAAC;CAGF;AAND,4CAMC"}

package/dist/utils/const.d.ts

@@ -1,3 +1,16 @@
 export declare const hookOnComplete = "onComplete";
 export declare const hookOnError = "onError";
 export declare const hookOnSuccess = "onSuccess";
+export declare const OpenAiPricing: {
+    "gpt-3.5-turbo": number[];
+    "gpt-3.5-turbo-0613": number[];
+    "gpt-3.5-turbo-16k": number[];
+    "gpt-4": number[];
+    "gpt-4-0613": number[];
+    "gpt-4-32k-0613": number[];
+    davinci: number[];
+    "text-curie-001": number[];
+    "text-babbage-001": number[];
+    "text-ada-001": number[];
+    "text-embedding-ada-002": number[];
+};

package/dist/utils/const.js

@@ -1,7 +1,20 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.hookOnSuccess = exports.hookOnError = exports.hookOnComplete = void 0;
+exports.OpenAiPricing = exports.hookOnSuccess = exports.hookOnError = exports.hookOnComplete = void 0;
 exports.hookOnComplete = `onComplete`;
 exports.hookOnError = `onError`;
 exports.hookOnSuccess = `onSuccess`;
+exports.OpenAiPricing = {
+    "gpt-3.5-turbo": [1000, 0.0015, 0.002],
+    "gpt-3.5-turbo-0613": [1000, 0.0015, 0.002],
+    "gpt-3.5-turbo-16k": [1000, 0.003, 0.004],
+    "gpt-4": [1000, 0.03, 0.06],
+    "gpt-4-0613": [1000, 0.03, 0.06],
+    "gpt-4-32k-0613": [1000, 0.03, 0.06],
+    davinci: [1000, 0.02, 0.02],
+    "text-curie-001": [1000, 0.002, 0.002],
+    "text-babbage-001": [1000, 0.0005, 0.0005],
+    "text-ada-001": [1000, 0.0004, 0.0004],
+    "text-embedding-ada-002": [1000, 0.0004, 0.0004],
+};
 //# sourceMappingURL=const.js.map

package/dist/utils/const.js.map

@@ -1 +1 @@
-{"version":3,"file":"const.js","sourceRoot":"src/","sources":["utils/const.ts"],"names":[],"mappings":";;;AAAa,QAAA,cAAc,GAAG,YAAY,CAAC;AAC9B,QAAA,WAAW,GAAG,SAAS,CAAC;AACxB,QAAA,aAAa,GAAG,WAAW,CAAC"}
+{"version":3,"file":"const.js","sourceRoot":"src/","sources":["utils/const.ts"],"names":[],"mappings":";;;AAAa,QAAA,cAAc,GAAG,YAAY,CAAC;AAC9B,QAAA,WAAW,GAAG,SAAS,CAAC;AACxB,QAAA,aAAa,GAAG,WAAW,CAAC;AAG5B,QAAA,aAAa,GAAG;IACzB,eAAe,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC;IACtC,oBAAoB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC;IAC3C,mBAAmB,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC;IACzC,OAAO,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC;IAC3B,YAAY,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC;IAChC,gBAAgB,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC;IACpC,OAAO,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC;IAC3B,gBAAgB,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC;IACtC,kBAAkB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC;IAC1C,cAAc,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC;IACtC,wBAAwB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC;CACjD,CAAC"}

package/dist/utils/modules/index.d.ts

@@ -18,7 +18,7 @@ export declare const chunkArray: (arr: any[], chunkSize: number) => any;
 export declare function defineSchema<T>(obj: Narrow<T>): Narrow<T>;
 export declare function enforceResultAttributes<O>(input: any): {
     result: O;
-    attributes: {};
+    attributes: Record<string, any>;
 };
 export declare function ensureInputIsObject<T extends ObjectArray>(input: T): {
     input: T;
@@ -46,12 +46,16 @@ export declare function toNumber(value: any): number;
 export declare function extractPromptPlaceholderToken(tok: string): {
     token: string;
     key?: undefined;
+    assistant?: undefined;
+    user?: undefined;
     name?: undefined;
     content?: undefined;
     role?: undefined;
 } | {
     token: string;
     key: string;
+    assistant: string;
+    user: string;
     name?: undefined;
     content?: undefined;
     role?: undefined;
@@ -61,5 +65,7 @@ export declare function extractPromptPlaceholderToken(tok: string): {
     content: string;
     role: string;
     key?: undefined;
+    assistant?: undefined;
+    user?: undefined;
 };
 export declare function getEnvironmentVariable(name: string): string | undefined;

package/dist/utils/modules/index.js

@@ -143,10 +143,14 @@ function extractPromptPlaceholderToken(tok) {
     const token = tok.replace(/ /g, "");
     if (token.substring(2, 18) === ">DialogueHistory") {
         const matchKey = tok.match(/key=(['"`])((?:(?!\1).)*)\1/);
+        const matchAssistant = tok.match(/assistant=(['"`])((?:(?!\1).)*)\1/);
+        const matchUser = tok.match(/user=(['"`])((?:(?!\1).)*)\1/);
         if (matchKey) {
             return {
                 token: ">DialogueHistory",
                 key: matchKey[2],
+                assistant: (0, lodash_get_1.default)(matchAssistant, '[2]', ""),
+                user: (0, lodash_get_1.default)(matchUser, '[2]', ""),
             };
         }
     }

package/dist/utils/modules/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["utils/modules/index.ts"],"names":[],"mappings":";;;;;;AACA,yDAA4C;AAG5C,4DAA6B;AASpB,cATF,oBAAG,CASE;AARZ,4DAA6B;AAQf,cARP,oBAAG,CAQO;AAPjB,8DAA+B;AAOZ,eAPZ,qBAAI,CAOY;AANvB,wEAAyC;AAMhB,oBANlB,0BAAS,CAMkB;AALlC,sEAAuC;AAKH,mBAL7B,yBAAQ,CAK6B;AAJ5C,kEAAmC;AAIW,iBAJvC,uBAAM,CAIuC;AAFpD,+BAAoC;AACjB,qFADJ,SAAM,OACE;AAGvB,2DAA4D;AAAnD,0HAAA,oBAAoB,OAAA;AAC7B,iEAAgE;AAAvD,8HAAA,qBAAqB,OAAA;AAC9B,+DAA8D;AAArD,4HAAA,oBAAoB,OAAA;AAC7B,+BAIgB;AAHd,0GAAA,kBAAkB,OAAA;AAClB,sGAAA,cAAc,OAAA;AACd,2GAAA,mBAAmB,OAAA;AAGrB,SAAgB,MAAM,CACpB,SAAc,EACd,OAAoC;IAEpC,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,KAAK,EAAE;QACxE,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE;YAC/B,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;SAC1B;aAAM,IAAI,OAAO,YAAY,KAAK,EAAE;YACnC,MAAM,OAAO,CAAC;SACf;aAAM;YACL,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;SACpC;KACF;AACH,CAAC;AAbD,wBAaC;AAEM,MAAM,UAAU,GAAG,CAAC,GAAU,EAAE,SAAiB,EAAE,EAAE,CAC1D,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE;IACjC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;IACvC,MAAM,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1C,OAAO,MAAM,CAAC;AAChB,CAAC,EAAE,EAAE,CAAC,CAAC;AANI,QAAA,UAAU,cAMd;AAET,SAAgB,YAAY,CAAI,GAAc;IAC3C,GAAW,CAAC,oBAAoB,GAAG,KAAK,CAAC;IAC1C,OAAO,IAAA,2BAAO,EAAC,GAAG,CAAC,CAAC;AACtB,CAAC;AAHD,oCAGC;AAED,SAAgB,uBAAuB,CAAI,KAAU;IAInD,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;KAC1C;IACD,IACE,OAAO,KAAK,KAAK,QAAQ;QACzB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC;YAC/B,QAAQ,IAAI,KAAK;YACjB,YAAY,IAAI,KAAK,CAAC;YACtB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC;gBAC9B,CAAC,QAAQ,IAAI,KAAK,IAAI,YAAY,IAAI,KAAK,CAAC,CAAC,CAAC,EAClD;QACA,OAAO,KAAK,CAAC;KACd;IACD,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;AAC3C,CAAC;AAlBD,0DAkBC;AAUD,SAAgB,mBAAmB,CAAI,KAAU;IAC/C,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,WAAW,EAAE;QAClD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;KACzB;IACD,QAAQ,OAAO,KAAK,EAAE;QACpB,KAAK,QAAQ,CAAC,CAAC;YACb,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;gBACxB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;aACzB;YACD,OAAO,KAAK,CAAC;SACd;QACD;YACE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;KAC3B;AACH,CAAC;AAdD,kDAcC;AACD,SAAgB,iBAAiB,CAAC,IAAS,EAAE,WAAmB;IAC9D,MAAM,IAAI,GAAG,IAAI,EAAE,IAAI,CAAC;IACxB,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;QACpC,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,QAAQ,EAAE;YACrC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;SACnC;aAAM;YACL,OAAO,IAAI,CAAC;SACb;KACF;IACD,IAAI,MAAM,GAAG,4BAA4B,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAChE,0BAA0B;IAC1B,OAAO,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;AAC1C,CAAC;AAZD,8CAYC;AAED,SAAgB,qBAAqB,CACnC,GAAM;IAEN,MAAM,CAAC,OAAO,GAAG,KAAK,QAAQ,EAAE,gBAAgB,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CACjE,CAAC;AACT,CAAC;AAPD,sDAOC;AAED,SAAgB,oBAAoB,CAAC,MAAM,GAAG,EAAE,EAAE,MAAM,GAAG,EAAE;IAC3D,yFAAyF;IACzF,MAAM,UAAU,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/C,MAAM,WAAW,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,CAAC,KAAK,GAAG,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9D,MAAM,UAAU,GAAG,CAAC,KAAK,GAAG,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAChE,OAAO,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU,GAAG,MAAM,EAAE,CAAC;AACvD,CAAC;AAPD,oDAOC;AAED,SAAgB,MAAM,CAAC,KAAU;IAC/B,OAAO,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;AAChC,CAAC;AAFD,wBAEC;AAED,SAAgB,WAAW,CAAC,KAAU;IACpC,OAAO,KAAK,KAAK,SAAS,IAAI,OAAO,KAAK,KAAK,SAAS,CAAC;AAC3D,CAAC;AAFD,kCAEC;AAED,SAAgB,QAAQ,CAAC,KAAU;IACjC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AAC7D,CAAC;AAFD,4BAEC;AAED,SAAgB,QAAQ,CAAC,KAAU;IACjC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;QAC7B,OAAO,KAAK,CAAC;KACd;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACpD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;KACtB;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AARD,4BAQC;AAED,SAAgB,6BAA6B,CAAC,GAAW;IACvD,IAAI,CAAC,GAAG;QAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IAC/B,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACpC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,kBAAkB,EAAE;QACjD,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;QAC1D,IAAI,QAAQ,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,kBAAkB;gBACzB,GAAG,EAAE,QAAQ,CAAC,CAAC,CAAC;aACjB,CAAC;SACH;KACF;SACI,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,oBAAoB,EAAE;QACxD,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC5D,MAAM,YAAY,GAAG,GAAG,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;QAClE,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC5D,IAAI,SAAS,EAAE;YACb,OAAO;gBACL,KAAK,EAAE,oBAAoB;gBAC3B,IAAI,EAAE,IAAA,oBAAG,EAAC,SAAS,EAAE,KAAK,CAAC;gBAC3B,OAAO,EAAC,IAAA,yBAAQ,EAAC,IAAA,oBAAG,EAAC,YAAY,EAAE,KAAK,EAAE,EAAE,CAAC,CAAE;gBAC/C,IAAI,EAAE,IAAA,oBAAG,EAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;aAChC,CAAC;SACH;KACF;IACD,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;AACvB,CAAC;AA1BD,sEA0BC;AAED,SAAgB,sBAAsB,CAAC,IAAY;IACjD,IAAG,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,EAAE,GAAG,EAAC;QAC7C,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;KACzB;SAAI;QACH,OAAO,SAAS,CAAA;KACjB;AACH,CAAC;AAND,wDAMC"}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["utils/modules/index.ts"],"names":[],"mappings":";;;;;;AACA,yDAA4C;AAG5C,4DAA6B;AASpB,cATF,oBAAG,CASE;AARZ,4DAA6B;AAQf,cARP,oBAAG,CAQO;AAPjB,8DAA+B;AAOZ,eAPZ,qBAAI,CAOY;AANvB,wEAAyC;AAMhB,oBANlB,0BAAS,CAMkB;AALlC,sEAAuC;AAKH,mBAL7B,yBAAQ,CAK6B;AAJ5C,kEAAmC;AAIW,iBAJvC,uBAAM,CAIuC;AAFpD,+BAAoC;AACjB,qFADJ,SAAM,OACE;AAGvB,2DAA4D;AAAnD,0HAAA,oBAAoB,OAAA;AAC7B,iEAAgE;AAAvD,8HAAA,qBAAqB,OAAA;AAC9B,+DAA8D;AAArD,4HAAA,oBAAoB,OAAA;AAC7B,+BAIgB;AAHd,0GAAA,kBAAkB,OAAA;AAClB,sGAAA,cAAc,OAAA;AACd,2GAAA,mBAAmB,OAAA;AAGrB,SAAgB,MAAM,CACpB,SAAc,EACd,OAAoC;IAEpC,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,KAAK,EAAE;QACxE,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE;YAC/B,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;SAC1B;aAAM,IAAI,OAAO,YAAY,KAAK,EAAE;YACnC,MAAM,OAAO,CAAC;SACf;aAAM;YACL,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;SACpC;KACF;AACH,CAAC;AAbD,wBAaC;AAEM,MAAM,UAAU,GAAG,CAAC,GAAU,EAAE,SAAiB,EAAE,EAAE,CAC1D,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE;IACjC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;IACvC,MAAM,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1C,OAAO,MAAM,CAAC;AAChB,CAAC,EAAE,EAAE,CAAC,CAAC;AANI,QAAA,UAAU,cAMd;AAET,SAAgB,YAAY,CAAI,GAAc;IAC3C,GAAW,CAAC,oBAAoB,GAAG,KAAK,CAAC;IAC1C,OAAO,IAAA,2BAAO,EAAC,GAAG,CAAC,CAAC;AACtB,CAAC;AAHD,oCAGC;AAED,SAAgB,uBAAuB,CAAI,KAAU;IAInD,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;KAC1C;IACD,IACE,OAAO,KAAK,KAAK,QAAQ;QACzB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC;YAC/B,QAAQ,IAAI,KAAK;YACjB,YAAY,IAAI,KAAK,CAAC;YACtB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC;gBAC9B,CAAC,QAAQ,IAAI,KAAK,IAAI,YAAY,IAAI,KAAK,CAAC,CAAC,CAAC,EAClD;QACA,OAAO,KAAK,CAAC;KACd;IACD,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;AAC3C,CAAC;AAlBD,0DAkBC;AAUD,SAAgB,mBAAmB,CAAI,KAAU;IAC/C,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,WAAW,EAAE;QAClD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;KACzB;IACD,QAAQ,OAAO,KAAK,EAAE;QACpB,KAAK,QAAQ,CAAC,CAAC;YACb,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;gBACxB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;aACzB;YACD,OAAO,KAAK,CAAC;SACd;QACD;YACE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;KAC3B;AACH,CAAC;AAdD,kDAcC;AACD,SAAgB,iBAAiB,CAAC,IAAS,EAAE,WAAmB;IAC9D,MAAM,IAAI,GAAG,IAAI,EAAE,IAAI,CAAC;IACxB,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;QACpC,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,QAAQ,EAAE;YACrC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;SACnC;aAAM;YACL,OAAO,IAAI,CAAC;SACb;KACF;IACD,IAAI,MAAM,GAAG,4BAA4B,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAChE,0BAA0B;IAC1B,OAAO,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;AAC1C,CAAC;AAZD,8CAYC;AAED,SAAgB,qBAAqB,CACnC,GAAM;IAEN,MAAM,CAAC,OAAO,GAAG,KAAK,QAAQ,EAAE,gBAAgB,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CACjE,CAAC;AACT,CAAC;AAPD,sDAOC;AAED,SAAgB,oBAAoB,CAAC,MAAM,GAAG,EAAE,EAAE,MAAM,GAAG,EAAE;IAC3D,yFAAyF;IACzF,MAAM,UAAU,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/C,MAAM,WAAW,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,CAAC,KAAK,GAAG,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9D,MAAM,UAAU,GAAG,CAAC,KAAK,GAAG,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAChE,OAAO,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU,GAAG,MAAM,EAAE,CAAC;AACvD,CAAC;AAPD,oDAOC;AAED,SAAgB,MAAM,CAAC,KAAU;IAC/B,OAAO,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;AAChC,CAAC;AAFD,wBAEC;AAED,SAAgB,WAAW,CAAC,KAAU;IACpC,OAAO,KAAK,KAAK,SAAS,IAAI,OAAO,KAAK,KAAK,SAAS,CAAC;AAC3D,CAAC;AAFD,kCAEC;AAED,SAAgB,QAAQ,CAAC,KAAU;IACjC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AAC7D,CAAC;AAFD,4BAEC;AAED,SAAgB,QAAQ,CAAC,KAAU;IACjC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;QAC7B,OAAO,KAAK,CAAC;KACd;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACpD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;KACtB;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AARD,4BAQC;AAED,SAAgB,6BAA6B,CAAC,GAAW;IACvD,IAAI,CAAC,GAAG;QAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IAC/B,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACpC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,kBAAkB,EAAE;QACjD,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;QAC1D,MAAM,cAAc,GAAG,GAAG,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACtE,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC5D,IAAI,QAAQ,EAAE;YACZ,OAAO;gBACL,KAAK,EAAE,kBAAkB;gBACzB,GAAG,EAAE,QAAQ,CAAC,CAAC,CAAC;gBAChB,SAAS,EAAE,IAAA,oBAAG,EAAC,cAAc,EAAE,KAAK,EAAE,EAAE,CAAC;gBACzC,IAAI,EAAC,IAAA,oBAAG,EAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;aAC/B,CAAC;SACH;KACF;SACI,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,oBAAoB,EAAE;QACxD,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC5D,MAAM,YAAY,GAAG,GAAG,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;QAClE,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC5D,IAAI,SAAS,EAAE;YACb,OAAO;gBACL,KAAK,EAAE,oBAAoB;gBAC3B,IAAI,EAAE,IAAA,oBAAG,EAAC,SAAS,EAAE,KAAK,CAAC;gBAC3B,OAAO,EAAC,IAAA,yBAAQ,EAAC,IAAA,oBAAG,EAAC,YAAY,EAAE,KAAK,EAAE,EAAE,CAAC,CAAE;gBAC/C,IAAI,EAAE,IAAA,oBAAG,EAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;aAChC,CAAC;SACH;KACF;IACD,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;AACvB,CAAC;AA9BD,sEA8BC;AAED,SAAgB,sBAAsB,CAAC,IAAY;IACjD,IAAG,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,EAAE,GAAG,EAAC;QAC7C,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;KACzB;SAAI;QACH,OAAO,SAAS,CAAA;KACjB;AACH,CAAC;AAND,wDAMC"}

package/dist/utils/modules/json-schema-filter.js

@@ -50,7 +50,15 @@ function filterObjectOnSchema(schema, doc, detach, property) {
                 }
                 else {
                     // keep the child if it's defined properly or null
-                    result[key] = filteredChild;
+                    if (sp.type === "integer" || sp.type === "number") {
+                        result[key] = (0, _1.toNumber)(filteredChild);
+                    }
+                    else if (sp.type === "boolean") {
+                        result[key] = !!filteredChild;
+                    }
+                    else {
+                        result[key] = filteredChild;
+                    }
                 }
             }
         });

package/dist/utils/modules/json-schema-filter.js.map

@@ -1 +1 @@
-{"version":3,"file":"json-schema-filter.js","sourceRoot":"src/","sources":["utils/modules/json-schema-filter.ts"],"names":[],"mappings":";;;AAAA,yBAAyB;AAEzB,SAAS,QAAQ,CAAC,GAAQ;IACxB,OAAO,GAAG,KAAK,MAAM,CAAC,GAAG,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,OAAO,CAAC,UAAe;IAC9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE;QAC9B,OAAO,UAAU,CAAC;KACnB;IAED,wCAAwC;IACxC,sEAAsE;IACtE,IAAI,SAAS,GAAG,UAAU,CAAC;IAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACzC,IAAI,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,IAAI,KAAK,MAAM,EAAE;YACnB,OAAO,IAAI,CAAC;SACb;KACF;AACH,CAAC;AAED,SAAgB,oBAAoB,CAAC,MAAW,EAAE,GAAQ,EAAE,MAAY,EAAE,QAAiB;IACzF,IAAI,MAAW,CAAC,CAAC,2FAA2F;IAE5G,iEAAiE;IACjE,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,SAAS,EAAE;QACrC,OAAO,GAAG,CAAC;KACZ;IAED,sDAAsD;IACtD,IAAI,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAChC,IAAI,IAAI,KAAK,QAAQ,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,UAAU,EAAE;QAC3D,MAAM,GAAG,EAAE,CAAC,CAAC,0DAA0D;QAEvE,iCAAiC;QACjC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,UAAU,GAAG;YAClD,IAAI,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;YACrB,IAAI,EAAE,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAEhC,IAAI,aAAa,GAAG,oBAAoB,CAAC,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;YAEtE,IAAG,QAAQ,EAAC;gBACV,MAAM,CAAC,GAAG,CAAC,GAAG,IAAA,MAAG,EAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAA;aACpC;iBAAI;gBACH,IAAI,aAAa,KAAK,SAAS,EAAE;oBAC/B,IAAI,OAAO,EAAE,EAAE,OAAO,KAAK,WAAW,EAAE;wBACtC,iDAAiD;wBACjD,MAAM,CAAC,GAAG,CAAC,GAAG,IAAA,MAAG,EAAC,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,CAAA;qBACrC;yBAAM;wBACL,sDAAsD;wBACtD,OAAO;qBACR;iBACF;qBAAM;oBACL,kDAAkD;oBAClD,MAAM,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC;iBAC7B;aACF;QAEH,CAAC,CAAC,CAAC;KACJ;SAAM,IAAI,IAAI,KAAK,QAAQ,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,MAAM,EAAE;QACvD,OAAO,EAAE,CAAC;KACX;SAAM,IAAI,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,KAAK,EAAE;QACjE,sCAAsC;QACtC,MAAM,GAAG,EAAE,CAAC;QACZ,GAAG,CAAC,OAAO,CAAC,UAAU,IAAI;YACxB,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;KACJ;SAAM;QACL,uEAAuE;QACvE,MAAM,GAAG,GAAG,CAAC;KACd;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AApDD,oDAoDC;AAED,SAAgB,iBAAiB,CAAC,MAAW,EAAE,QAAgB;IAC7D,IAAG,MAAM,CAAC,IAAI,KAAK,OAAO,EAAC;QACzB,OAAO,oBAAoB,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAA;KAC/D;IACD,OAAO,oBAAoB,CAAC,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAA;AAC9D,CAAC;AALD,8CAKC"}
+{"version":3,"file":"json-schema-filter.js","sourceRoot":"src/","sources":["utils/modules/json-schema-filter.ts"],"names":[],"mappings":";;;AAAA,yBAAmC;AAEnC,SAAS,QAAQ,CAAC,GAAQ;IACxB,OAAO,GAAG,KAAK,MAAM,CAAC,GAAG,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,OAAO,CAAC,UAAe;IAC9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE;QAC9B,OAAO,UAAU,CAAC;KACnB;IAED,wCAAwC;IACxC,sEAAsE;IACtE,IAAI,SAAS,GAAG,UAAU,CAAC;IAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACzC,IAAI,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,IAAI,KAAK,MAAM,EAAE;YACnB,OAAO,IAAI,CAAC;SACb;KACF;AACH,CAAC;AAED,SAAgB,oBAAoB,CAAC,MAAW,EAAE,GAAQ,EAAE,MAAY,EAAE,QAAiB;IACzF,IAAI,MAAW,CAAC,CAAC,2FAA2F;IAE5G,iEAAiE;IACjE,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,SAAS,EAAE;QACrC,OAAO,GAAG,CAAC;KACZ;IAED,sDAAsD;IACtD,IAAI,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAChC,IAAI,IAAI,KAAK,QAAQ,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,UAAU,EAAE;QAC3D,MAAM,GAAG,EAAE,CAAC,CAAC,0DAA0D;QAEvE,iCAAiC;QACjC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,UAAU,GAAG;YAClD,IAAI,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;YACrB,IAAI,EAAE,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAEhC,IAAI,aAAa,GAAG,oBAAoB,CAAC,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;YAEtE,IAAG,QAAQ,EAAC;gBACV,MAAM,CAAC,GAAG,CAAC,GAAG,IAAA,MAAG,EAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAA;aACpC;iBAAI;gBACH,IAAI,aAAa,KAAK,SAAS,EAAE;oBAC/B,IAAI,OAAO,EAAE,EAAE,OAAO,KAAK,WAAW,EAAE;wBACtC,iDAAiD;wBACjD,MAAM,CAAC,GAAG,CAAC,GAAG,IAAA,MAAG,EAAC,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,CAAA;qBACrC;yBAAM;wBACL,sDAAsD;wBACtD,OAAO;qBACR;iBACF;qBAAM;oBACL,kDAAkD;oBAClD,IAAG,EAAE,CAAC,IAAI,KAAK,SAAS,IAAI,EAAE,CAAC,IAAI,KAAK,QAAQ,EAAC;wBAC/C,MAAM,CAAC,GAAG,CAAC,GAAG,IAAA,WAAQ,EAAC,aAAa,CAAC,CAAC;qBACvC;yBAAK,IAAG,EAAE,CAAC,IAAI,KAAK,SAAS,EAAC;wBAC7B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,aAAa,CAAA;qBAC9B;yBAAI;wBACH,MAAM,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC;qBAC7B;iBACF;aACF;QAEH,CAAC,CAAC,CAAC;KACJ;SAAM,IAAI,IAAI,KAAK,QAAQ,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,MAAM,EAAE;QACvD,OAAO,EAAE,CAAC;KACX;SAAM,IAAI,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,KAAK,EAAE;QACjE,sCAAsC;QACtC,MAAM,GAAG,EAAE,CAAC;QACZ,GAAG,CAAC,OAAO,CAAC,UAAU,IAAI;YACxB,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;KACJ;SAAM;QACL,uEAAuE;QACvE,MAAM,GAAG,GAAG,CAAC;KACd;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AA1DD,oDA0DC;AAED,SAAgB,iBAAiB,CAAC,MAAW,EAAE,QAAgB;IAC7D,IAAG,MAAM,CAAC,IAAI,KAAK,OAAO,EAAC;QACzB,OAAO,oBAAoB,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAA;KAC/D;IACD,OAAO,oBAAoB,CAAC,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAA;AAC9D,CAAC;AALD,8CAKC"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "llm-exe",
-  "version": "0.0.6",
-  "description": "A package that provides simplified base components to make building and maintaining LLM-powered applications easier.",
+  "version": "1.0.0",
+  "description": "Simplify building LLM-powered apps with easy-to-use base components, supporting text and chat-based prompts with handlebars template engine, output parsers, and flexible function calling capabilities.",
   "keywords": [
     "ai",
     "llm",
@@ -9,7 +9,8 @@
     "gpt-3.5-turbo",
     "gpt-4",
     "chain",
-    "prompt"
+    "prompt",
+    "agent"
   ],
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -77,5 +78,6 @@
   "repository": {
     "type": "git",
     "url": "https://github.com/gregreindel/llm-exe"
-  }
+  },
+  "homepage": "https:/llm-exe.com"
 }