rita 3.0.23 → 3.0.25

package/dist/rita.cjs

@@ -41933,8 +41933,8 @@ var _RiMarkov = class _RiMarkov {
    * @param {boolean} [options.trace] - output trace info to the console
    * @param {number} [options.maxLengthMatch] - # of words allowed in result to match a sequence in the input, default=∞
    * @param {number} [options.maxAttempts=999] - max attempts before to complete one ore more generations before erroring, default=999
-   * @param {function} [options.tokenize] - custom tokenizer with tokenize() method, defaults to RiTa.tokenize()
-   * @param {function} [options.untokenize] - custom untokenizer with untokenize() method, defaults to RiTa.untokenize()
+   * @param {object} [options.tokenize] - custom tokenize() function, defaults to RiTa.tokenize()
+   * @param {function} [options.untokenize] - custom untokenize() function, defaults to RiTa.untokenize()
    * @param {boolean} [options.disableInputChecks=false] - if true, allow result to be present in the input, default
    * @memberof RiMarkov
    */
@@ -41967,7 +41967,7 @@ var _RiMarkov = class _RiMarkov {
    */
   addText(text, multiplier = 1) {
     let sents = Array.isArray(text) ? text : _RiMarkov.parent.sentences(text);
-    let wrap, allWords = [];
+    let allWords = [];
     for (let k = 0; k < multiplier; k++) {
       for (let i = 0; i < sents.length; i++) {
         let words = this.tokenize(sents[i]);
@@ -41984,19 +41984,34 @@ var _RiMarkov = class _RiMarkov {
     }
     return this;
   }
-  /**
-   * Generates `count` joined sentences from the model.
-   * @param {number} [count=1] - the number of sentences to generate (default=1)
+  /** 
+   * @overload
+   * @param {number} count
    * @param {object} [options={}] - options for the generation
    * @param {number} [options.minLength=5] - minimum length of each sentence
    * @param {number} [options.maxLength=35] - maximum length of each sentence
    * @param {number} [options.temperature=1] - temperature acts as a knob to adjust the probability that input elements will be selected for the output. At higher values, infrequent words are more likely to be chosen, while at lower values the most frequent inputs are more likely to be output. If no value is provided, then tokens are chosen according to their relative frequency in the input.
    * @param {boolean} [options.allowDuplicates=false] - if true, allow duplicate sentences in the output
    * @param {string|string[]} [options.seed] - a seed string or array of tokens to start the generation
-   * @param {boolean} [options.trace] - output trace info to the console
-   * @return {string[]} - the generated sentences
+   * @returns {string[]}
+   *
+   * @overload
+   * @param {object} [options={}] - options for the generation
+   * @param {number} [options.minLength=5] - minimum length of each sentence
+   * @param {number} [options.maxLength=35] - maximum length of each sentence
+   * @param {number} [options.temperature=1] - temperature acts as a knob to adjust the probability that input elements will be selected for the output. At higher values, infrequent words are more likely to be chosen, while at lower values the most frequent inputs are more likely to be output. If no value is provided, then tokens are chosen according to their relative frequency in the input.
+   * @param {boolean} [options.allowDuplicates=false] - if true, allow duplicate sentences in the output
+   * @param {string|string[]} [options.seed] - a seed string or array of tokens to start the generation
+   * @returns {string}
    */
   generate(count, options = {}) {
+    let returnsArray = false;
+    if (typeof count === "number") {
+      if (count === 1) {
+        throw Error("For one result, use generate() with no 'count' argument");
+      }
+      returnsArray = true;
+    }
     if (arguments.length === 1 && typeof count === "object") {
       options = count;
       count = 1;
@@ -42182,7 +42197,7 @@ var _RiMarkov = class _RiMarkov {
     }
     unmarkNodes();
     let str = this.untokenize(tokens.map((t) => t.token)).trim();
-    return num > 1 ? this._splitEnds(str) : str;
+    return returnsArray ? this._splitEnds(str) : str;
   }
   /**
    * Converts the model to a JSON-formatted string for storage or serialization
@@ -43221,7 +43236,7 @@ markov_default.parent = RiTa;
 stemmer_default.tokenizer = RiTa.tokenizer;
 RiTa.SILENT = false;
 RiTa.SILENCE_LTS = false;
-RiTa.VERSION = "3.0.23";
+RiTa.VERSION = "3.0.25";
 RiTa.FIRST = 1;
 RiTa.SECOND = 2;
 RiTa.THIRD = 3;