@digipair/skill-llm 0.8.37 → 0.8.40

package/chat.esm.js

@@ -1,4 +1,144 @@
-import { c as checkValidTemplate, a as BasePromptTemplate, I as ImagePromptValue, b as BaseMessage, P as PromptTemplate, H as HumanMessage, S as SystemMessage, i as isBaseMessage, d as coerceMessageLikeToMessage, C as ChatMessage, e as ChatPromptValue, p as parseFString, B as BaseStringPromptTemplate, A as AIMessage, R as Runnable } from './index.esm2.js';
+import { a as BaseMessage, c as checkValidTemplate, b as BasePromptTemplate, I as ImagePromptValue, P as PromptTemplate, H as HumanMessage, S as SystemMessage, i as isBaseMessage, d as coerceMessageLikeToMessage, C as ChatPromptValue, p as parseMustache, e as parseFString, B as BaseStringPromptTemplate, A as AIMessage, R as Runnable } from './index.esm2.js';
+
+function _assert_this_initialized$2(self) {
+    if (self === void 0) {
+        throw new ReferenceError("this hasn't been initialised - super() hasn't been called");
+    }
+    return self;
+}
+function _class_call_check$2(instance, Constructor) {
+    if (!(instance instanceof Constructor)) {
+        throw new TypeError("Cannot call a class as a function");
+    }
+}
+function _defineProperties$2(target, props) {
+    for(var i = 0; i < props.length; i++){
+        var descriptor = props[i];
+        descriptor.enumerable = descriptor.enumerable || false;
+        descriptor.configurable = true;
+        if ("value" in descriptor) descriptor.writable = true;
+        Object.defineProperty(target, descriptor.key, descriptor);
+    }
+}
+function _create_class$2(Constructor, protoProps, staticProps) {
+    if (protoProps) _defineProperties$2(Constructor.prototype, protoProps);
+    if (staticProps) _defineProperties$2(Constructor, staticProps);
+    return Constructor;
+}
+function _get_prototype_of$2(o) {
+    _get_prototype_of$2 = Object.setPrototypeOf ? Object.getPrototypeOf : function getPrototypeOf(o) {
+        return o.__proto__ || Object.getPrototypeOf(o);
+    };
+    return _get_prototype_of$2(o);
+}
+function _inherits$2(subClass, superClass) {
+    if (typeof superClass !== "function" && superClass !== null) {
+        throw new TypeError("Super expression must either be null or a function");
+    }
+    subClass.prototype = Object.create(superClass && superClass.prototype, {
+        constructor: {
+            value: subClass,
+            writable: true,
+            configurable: true
+        }
+    });
+    if (superClass) _set_prototype_of$2(subClass, superClass);
+}
+function _possible_constructor_return$2(self, call) {
+    if (call && (_type_of$2(call) === "object" || typeof call === "function")) {
+        return call;
+    }
+    return _assert_this_initialized$2(self);
+}
+function _set_prototype_of$2(o, p) {
+    _set_prototype_of$2 = Object.setPrototypeOf || function setPrototypeOf(o, p) {
+        o.__proto__ = p;
+        return o;
+    };
+    return _set_prototype_of$2(o, p);
+}
+function _type_of$2(obj) {
+    "@swc/helpers - typeof";
+    return obj && typeof Symbol !== "undefined" && obj.constructor === Symbol ? "symbol" : typeof obj;
+}
+function _is_native_reflect_construct$2() {
+    if (typeof Reflect === "undefined" || !Reflect.construct) return false;
+    if (Reflect.construct.sham) return false;
+    if (typeof Proxy === "function") return true;
+    try {
+        Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function() {}));
+        return true;
+    } catch (e) {
+        return false;
+    }
+}
+function _create_super$2(Derived) {
+    var hasNativeReflectConstruct = _is_native_reflect_construct$2();
+    return function _createSuperInternal() {
+        var Super = _get_prototype_of$2(Derived), result;
+        if (hasNativeReflectConstruct) {
+            var NewTarget = _get_prototype_of$2(this).constructor;
+            result = Reflect.construct(Super, arguments, NewTarget);
+        } else {
+            result = Super.apply(this, arguments);
+        }
+        return _possible_constructor_return$2(this, result);
+    };
+}
+/**
+ * Represents a chat message in a conversation.
+ */ var ChatMessage = /*#__PURE__*/ function(BaseMessage) {
+    _inherits$2(ChatMessage, BaseMessage);
+    var _super = _create_super$2(ChatMessage);
+    function ChatMessage(fields, role) {
+        _class_call_check$2(this, ChatMessage);
+        var _this;
+        if (typeof fields === "string") {
+            // eslint-disable-next-line no-param-reassign, @typescript-eslint/no-non-null-assertion
+            fields = {
+                content: fields,
+                role: role
+            };
+        }
+        _this = _super.call(this, fields);
+        Object.defineProperty(_assert_this_initialized$2(_this), "role", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        _this.role = fields.role;
+        return _this;
+    }
+    _create_class$2(ChatMessage, [
+        {
+            key: "_getType",
+            value: function _getType() {
+                return "generic";
+            }
+        }
+    ], [
+        {
+            key: "lc_name",
+            value: function lc_name() {
+                return "ChatMessage";
+            }
+        },
+        {
+            key: "_chatMessageClass",
+            value: function _chatMessageClass() {
+                return ChatMessage;
+            }
+        },
+        {
+            key: "isInstance",
+            value: function isInstance(message) {
+                return message._getType() === "generic";
+            }
+        }
+    ]);
+    return ChatMessage;
+}(BaseMessage);
 
 function _array_like_to_array$1(arr, len) {
     if (len == null || len > arr.length) len = arr.length;
@@ -866,6 +1006,101 @@ function _ts_values(o) {
     ]);
     return BaseMessagePromptTemplate;
 }(Runnable);
+/**
+ * Class that represents a placeholder for messages in a chat prompt. It
+ * extends the BaseMessagePromptTemplate.
+ */ var MessagesPlaceholder = /*#__PURE__*/ function(BaseMessagePromptTemplate) {
+    _inherits(MessagesPlaceholder, BaseMessagePromptTemplate);
+    var _super = _create_super(MessagesPlaceholder);
+    function MessagesPlaceholder(fields) {
+        _class_call_check(this, MessagesPlaceholder);
+        var _this;
+        if (typeof fields === "string") {
+            // eslint-disable-next-line no-param-reassign
+            fields = {
+                variableName: fields
+            };
+        }
+        _this = _super.call(this, fields);
+        Object.defineProperty(_assert_this_initialized(_this), "variableName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(_assert_this_initialized(_this), "optional", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        _this.variableName = fields.variableName;
+        var _fields_optional;
+        _this.optional = (_fields_optional = fields.optional) !== null && _fields_optional !== void 0 ? _fields_optional : false;
+        return _this;
+    }
+    _create_class(MessagesPlaceholder, [
+        {
+            key: "inputVariables",
+            get: function get() {
+                return [
+                    this.variableName
+                ];
+            }
+        },
+        {
+            key: "validateInputOrThrow",
+            value: function validateInputOrThrow(input, variableName) {
+                if (this.optional && !input) {
+                    return false;
+                } else if (!input) {
+                    var error = new Error('Error: Field "'.concat(variableName, '" in prompt uses a MessagesPlaceholder, which expects an array of BaseMessages as an input value. Received: undefined'));
+                    error.name = "InputFormatError";
+                    throw error;
+                }
+                var isInputBaseMessage = false;
+                if (Array.isArray(input)) {
+                    isInputBaseMessage = input.every(function(message) {
+                        return isBaseMessage(message);
+                    });
+                } else {
+                    isInputBaseMessage = isBaseMessage(input);
+                }
+                if (!isInputBaseMessage) {
+                    var readableInput = typeof input === "string" ? input : JSON.stringify(input, null, 2);
+                    var error1 = new Error('Error: Field "'.concat(variableName, '" in prompt uses a MessagesPlaceholder, which expects an array of BaseMessages as an input value. Received: ').concat(readableInput));
+                    error1.name = "InputFormatError";
+                    throw error1;
+                }
+                return true;
+            }
+        },
+        {
+            key: "formatMessages",
+            value: function formatMessages(values) {
+                var _this = this;
+                return _async_to_generator(function() {
+                    var _values_this_variableName;
+                    return _ts_generator(this, function(_state) {
+                        _this.validateInputOrThrow(values[_this.variableName], _this.variableName);
+                        return [
+                            2,
+                            (_values_this_variableName = values[_this.variableName]) !== null && _values_this_variableName !== void 0 ? _values_this_variableName : []
+                        ];
+                    });
+                })();
+            }
+        }
+    ], [
+        {
+            key: "lc_name",
+            value: function lc_name() {
+                return "MessagesPlaceholder";
+            }
+        }
+    ]);
+    return MessagesPlaceholder;
+}(BaseMessagePromptTemplate);
 /**
  * Abstract class that serves as a base for creating message string prompt
  * templates. It extends the BaseMessagePromptTemplate.
@@ -1049,8 +1284,10 @@ function _ts_values(o) {
         },
         {
             key: "fromTemplate",
-            value: function fromTemplate(template, role) {
-                return new this(PromptTemplate.fromTemplate(template), role);
+            value: function fromTemplate(template, role, options) {
+                return new this(PromptTemplate.fromTemplate(template, {
+                    templateFormat: options === null || options === void 0 ? void 0 : options.templateFormat
+                }), role);
             }
         }
     ]);
@@ -1358,7 +1595,7 @@ var _StringImageMessagePromptTemplate = /*#__PURE__*/ function(BaseMessagePrompt
             key: "fromTemplate",
             value: function fromTemplate(template, additionalOptions) {
                 if (typeof template === "string") {
-                    return new this(PromptTemplate.fromTemplate(template));
+                    return new this(PromptTemplate.fromTemplate(template, additionalOptions));
                 }
                 var prompt = [];
                 var _iteratorNormalCompletion = true, _didIteratorError = false, _iteratorError = undefined;
@@ -1373,14 +1610,19 @@ var _StringImageMessagePromptTemplate = /*#__PURE__*/ function(BaseMessagePrompt
                                 var _item_text;
                                 text = (_item_text = item.text) !== null && _item_text !== void 0 ? _item_text : "";
                             }
-                            prompt.push(PromptTemplate.fromTemplate(text));
+                            prompt.push(PromptTemplate.fromTemplate(text, additionalOptions));
                         } else if (typeof item === "object" && "image_url" in item) {
                             var _item_image_url;
                             var imgTemplate = (_item_image_url = item.image_url) !== null && _item_image_url !== void 0 ? _item_image_url : "";
                             var imgTemplateObject = void 0;
                             var inputVariables = [];
                             if (typeof imgTemplate === "string") {
-                                var parsedTemplate = parseFString(imgTemplate);
+                                var parsedTemplate = void 0;
+                                if ((additionalOptions === null || additionalOptions === void 0 ? void 0 : additionalOptions.templateFormat) === "mustache") {
+                                    parsedTemplate = parseMustache(imgTemplate);
+                                } else {
+                                    parsedTemplate = parseFString(imgTemplate);
+                                }
                                 var variables = parsedTemplate.flatMap(function(item) {
                                     return item.type === "variable" ? [
                                         item.name
@@ -1406,7 +1648,12 @@ var _StringImageMessagePromptTemplate = /*#__PURE__*/ function(BaseMessagePrompt
                                 });
                             } else if (typeof imgTemplate === "object") {
                                 if ("url" in imgTemplate) {
-                                    var parsedTemplate1 = parseFString(imgTemplate.url);
+                                    var parsedTemplate1 = void 0;
+                                    if ((additionalOptions === null || additionalOptions === void 0 ? void 0 : additionalOptions.templateFormat) === "mustache") {
+                                        parsedTemplate1 = parseMustache(imgTemplate.url);
+                                    } else {
+                                        parsedTemplate1 = parseFString(imgTemplate.url);
+                                    }
                                     inputVariables = parsedTemplate1.flatMap(function(item) {
                                         return item.type === "variable" ? [
                                             item.name
@@ -1549,19 +1796,49 @@ var _StringImageMessagePromptTemplate = /*#__PURE__*/ function(BaseMessagePrompt
 function _isBaseMessagePromptTemplate(baseMessagePromptTemplateLike) {
     return typeof baseMessagePromptTemplateLike.formatMessages === "function";
 }
-function _coerceMessagePromptTemplateLike(messagePromptTemplateLike) {
+function _coerceMessagePromptTemplateLike(messagePromptTemplateLike, extra) {
     if (_isBaseMessagePromptTemplate(messagePromptTemplateLike) || isBaseMessage(messagePromptTemplateLike)) {
         return messagePromptTemplateLike;
     }
+    if (Array.isArray(messagePromptTemplateLike) && messagePromptTemplateLike[0] === "placeholder") {
+        var messageContent = messagePromptTemplateLike[1];
+        if (typeof messageContent !== "string" || messageContent[0] !== "{" || messageContent[messageContent.length - 1] !== "}") {
+            throw new Error('Invalid placeholder template: "'.concat(messagePromptTemplateLike[1], '". Expected a variable name surrounded by curly braces.'));
+        }
+        var variableName = messageContent.slice(1, -1);
+        return new MessagesPlaceholder({
+            variableName: variableName,
+            optional: true
+        });
+    }
     var message = coerceMessageLikeToMessage(messagePromptTemplateLike);
+    var templateData;
+    if (typeof message.content === "string") {
+        templateData = message.content;
+    } else {
+        // Assuming message.content is an array of complex objects, transform it.
+        templateData = message.content.map(function(item) {
+            if ("text" in item) {
+                return {
+                    text: item.text
+                };
+            } else if ("image_url" in item) {
+                return {
+                    image_url: item.image_url
+                };
+            } else {
+                return item;
+            }
+        });
+    }
     if (message._getType() === "human") {
-        return HumanMessagePromptTemplate.fromTemplate(message.content);
+        return HumanMessagePromptTemplate.fromTemplate(templateData, extra);
     } else if (message._getType() === "ai") {
-        return AIMessagePromptTemplate.fromTemplate(message.content);
+        return AIMessagePromptTemplate.fromTemplate(templateData, extra);
     } else if (message._getType() === "system") {
-        return SystemMessagePromptTemplate.fromTemplate(message.content);
+        return SystemMessagePromptTemplate.fromTemplate(templateData, extra);
     } else if (ChatMessage.isInstance(message)) {
-        return ChatMessagePromptTemplate.fromTemplate(message.content, message.role);
+        return ChatMessagePromptTemplate.fromTemplate(message.content, message.role, extra);
     } else {
         throw new Error('Could not coerce message prompt template from input. Received message type: "'.concat(message._getType(), '".'));
     }
@@ -1604,6 +1881,16 @@ function isMessagesPlaceholder(x) {
             writable: true,
             value: true
         });
+        Object.defineProperty(_assert_this_initialized(_this), "templateFormat", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "f-string"
+        });
+        // If input is mustache and validateTemplate is not defined, set it to false
+        if (input.templateFormat === "mustache" && input.validateTemplate === undefined) {
+            _this.validateTemplate = false;
+        }
         Object.assign(_assert_this_initialized(_this), input);
         if (_this.validateTemplate) {
             var inputVariablesMessages = new Set();
@@ -1683,6 +1970,7 @@ function isMessagesPlaceholder(x) {
         {
             key: "_parseImagePrompts",
             value: function _parseImagePrompts(message, inputValues) {
+                var _this = this;
                 return _async_to_generator(function() {
                     var formattedMessageContent;
                     return _ts_generator(this, function(_state) {
@@ -1714,7 +2002,9 @@ function isMessagesPlaceholder(x) {
                                                         } else {
                                                             imageUrl = item.image_url.url;
                                                         }
-                                                        promptTemplatePlaceholder = PromptTemplate.fromTemplate(imageUrl);
+                                                        promptTemplatePlaceholder = PromptTemplate.fromTemplate(imageUrl, {
+                                                            templateFormat: _this.templateFormat
+                                                        });
                                                         return [
                                                             4,
                                                             promptTemplatePlaceholder.format(inputValues)
@@ -1914,10 +2204,8 @@ function isMessagesPlaceholder(x) {
         },
         {
             key: "fromTemplate",
-            value: /**
-     * Load prompt template from a template f-string
-     */ function fromTemplate(template) {
-                var prompt = PromptTemplate.fromTemplate(template);
+            value: function fromTemplate(template, options) {
+                var prompt = PromptTemplate.fromTemplate(template, options);
                 var humanTemplate = new HumanMessagePromptTemplate({
                     prompt: prompt
                 });
@@ -1937,7 +2225,7 @@ function isMessagesPlaceholder(x) {
                 var flattenedMessages = promptMessages.reduce(function(acc, promptMessage) {
                     return acc.concat(// eslint-disable-next-line no-instanceof/no-instanceof
                     _instanceof(promptMessage, ChatPromptTemplate) ? promptMessage.promptMessages : [
-                        _coerceMessagePromptTemplateLike(promptMessage)
+                        _coerceMessagePromptTemplateLike(promptMessage, extra)
                     ]);
                 }, []);
                 var flattenedPartialVariables = promptMessages.reduce(function(acc, promptMessage) {
@@ -1992,7 +2280,8 @@ function isMessagesPlaceholder(x) {
                 return new this(_object_spread_props(_object_spread({}, extra), {
                     inputVariables: _to_consumable_array(inputVariables),
                     promptMessages: flattenedMessages,
-                    partialVariables: flattenedPartialVariables
+                    partialVariables: flattenedPartialVariables,
+                    templateFormat: extra === null || extra === void 0 ? void 0 : extra.templateFormat
                 }));
             }
         },