@botpress/cognitive 0.1.38 → 0.1.40

package/dist/index.mjs

@@ -32,9 +32,8 @@ var require_options = __commonJS({
       __assign = Object.assign || function(t) {
         for (var s, i = 1, n = arguments.length; i < n; i++) {
           s = arguments[i];
-          for (var p in s)
-            if (Object.prototype.hasOwnProperty.call(s, p))
-              t[p] = s[p];
+          for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
+            t[p] = s[p];
         }
         return t;
       };
@@ -116,7 +115,7 @@ var require_delay_base = __commonJS({
     var jitter_factory_1 = require_jitter_factory();
     var Delay = (
       /** @class */
-      function() {
+      (function() {
         function Delay2(options) {
           this.options = options;
           this.attempt = 0;
@@ -157,7 +156,7 @@ var require_delay_base = __commonJS({
           configurable: true
         });
         return Delay2;
-      }()
+      })()
     );
     exports.Delay = Delay;
   }
@@ -167,14 +166,12 @@ var require_delay_base = __commonJS({
 var require_skip_first_delay = __commonJS({
   "../../node_modules/.pnpm/exponential-backoff@3.1.1/node_modules/exponential-backoff/dist/delay/skip-first/skip-first.delay.js"(exports) {
     "use strict";
-    var __extends = exports && exports.__extends || function() {
+    var __extends = exports && exports.__extends || /* @__PURE__ */ (function() {
       var extendStatics = function(d, b) {
         extendStatics = Object.setPrototypeOf || { __proto__: [] } instanceof Array && function(d2, b2) {
           d2.__proto__ = b2;
         } || function(d2, b2) {
-          for (var p in b2)
-            if (b2.hasOwnProperty(p))
-              d2[p] = b2[p];
+          for (var p in b2) if (b2.hasOwnProperty(p)) d2[p] = b2[p];
         };
         return extendStatics(d, b);
       };
@@ -185,7 +182,7 @@ var require_skip_first_delay = __commonJS({
         }
         d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
       };
-    }();
+    })();
     var __awaiter = exports && exports.__awaiter || function(thisArg, _arguments, P, generator) {
       function adopt(value) {
         return value instanceof P ? value : new P(function(resolve) {
@@ -215,8 +212,7 @@ var require_skip_first_delay = __commonJS({
     };
     var __generator = exports && exports.__generator || function(thisArg, body) {
       var _ = { label: 0, sent: function() {
-        if (t[0] & 1)
-          throw t[1];
+        if (t[0] & 1) throw t[1];
         return t[1];
       }, trys: [], ops: [] }, f, y, t, g;
       return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() {
@@ -228,64 +224,58 @@ var require_skip_first_delay = __commonJS({
         };
       }
       function step(op) {
-        if (f)
-          throw new TypeError("Generator is already executing.");
-        while (_)
-          try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done)
-              return t;
-            if (y = 0, t)
-              op = [op[0] & 2, t.value];
-            switch (op[0]) {
-              case 0:
-              case 1:
+        if (f) throw new TypeError("Generator is already executing.");
+        while (_) try {
+          if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+          if (y = 0, t) op = [op[0] & 2, t.value];
+          switch (op[0]) {
+            case 0:
+            case 1:
+              t = op;
+              break;
+            case 4:
+              _.label++;
+              return { value: op[1], done: false };
+            case 5:
+              _.label++;
+              y = op[1];
+              op = [0];
+              continue;
+            case 7:
+              op = _.ops.pop();
+              _.trys.pop();
+              continue;
+            default:
+              if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
+                _ = 0;
+                continue;
+              }
+              if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
+                _.label = op[1];
+                break;
+              }
+              if (op[0] === 6 && _.label < t[1]) {
+                _.label = t[1];
                 t = op;
                 break;
-              case 4:
-                _.label++;
-                return { value: op[1], done: false };
-              case 5:
-                _.label++;
-                y = op[1];
-                op = [0];
-                continue;
-              case 7:
-                op = _.ops.pop();
-                _.trys.pop();
-                continue;
-              default:
-                if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
-                  _ = 0;
-                  continue;
-                }
-                if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
-                  _.label = op[1];
-                  break;
-                }
-                if (op[0] === 6 && _.label < t[1]) {
-                  _.label = t[1];
-                  t = op;
-                  break;
-                }
-                if (t && _.label < t[2]) {
-                  _.label = t[2];
-                  _.ops.push(op);
-                  break;
-                }
-                if (t[2])
-                  _.ops.pop();
-                _.trys.pop();
-                continue;
-            }
-            op = body.call(thisArg, _);
-          } catch (e) {
-            op = [6, e];
-            y = 0;
-          } finally {
-            f = t = 0;
+              }
+              if (t && _.label < t[2]) {
+                _.label = t[2];
+                _.ops.push(op);
+                break;
+              }
+              if (t[2]) _.ops.pop();
+              _.trys.pop();
+              continue;
           }
-        if (op[0] & 5)
-          throw op[1];
+          op = body.call(thisArg, _);
+        } catch (e) {
+          op = [6, e];
+          y = 0;
+        } finally {
+          f = t = 0;
+        }
+        if (op[0] & 5) throw op[1];
         return { value: op[0] ? op[1] : void 0, done: true };
       }
     };
@@ -293,7 +283,7 @@ var require_skip_first_delay = __commonJS({
     var delay_base_1 = require_delay_base();
     var SkipFirstDelay = (
       /** @class */
-      function(_super) {
+      (function(_super) {
         __extends(SkipFirstDelay2, _super);
         function SkipFirstDelay2() {
           return _super !== null && _super.apply(this, arguments) || this;
@@ -320,7 +310,7 @@ var require_skip_first_delay = __commonJS({
           configurable: true
         });
         return SkipFirstDelay2;
-      }(delay_base_1.Delay)
+      })(delay_base_1.Delay)
     );
     exports.SkipFirstDelay = SkipFirstDelay;
   }
@@ -330,14 +320,12 @@ var require_skip_first_delay = __commonJS({
 var require_always_delay = __commonJS({
   "../../node_modules/.pnpm/exponential-backoff@3.1.1/node_modules/exponential-backoff/dist/delay/always/always.delay.js"(exports) {
     "use strict";
-    var __extends = exports && exports.__extends || function() {
+    var __extends = exports && exports.__extends || /* @__PURE__ */ (function() {
       var extendStatics = function(d, b) {
         extendStatics = Object.setPrototypeOf || { __proto__: [] } instanceof Array && function(d2, b2) {
           d2.__proto__ = b2;
         } || function(d2, b2) {
-          for (var p in b2)
-            if (b2.hasOwnProperty(p))
-              d2[p] = b2[p];
+          for (var p in b2) if (b2.hasOwnProperty(p)) d2[p] = b2[p];
         };
         return extendStatics(d, b);
       };
@@ -348,18 +336,18 @@ var require_always_delay = __commonJS({
         }
         d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
       };
-    }();
+    })();
     Object.defineProperty(exports, "__esModule", { value: true });
     var delay_base_1 = require_delay_base();
     var AlwaysDelay = (
       /** @class */
-      function(_super) {
+      (function(_super) {
         __extends(AlwaysDelay2, _super);
         function AlwaysDelay2() {
           return _super !== null && _super.apply(this, arguments) || this;
         }
         return AlwaysDelay2;
-      }(delay_base_1.Delay)
+      })(delay_base_1.Delay)
     );
     exports.AlwaysDelay = AlwaysDelay;
   }
@@ -420,8 +408,7 @@ var require_backoff = __commonJS({
     };
     var __generator = exports && exports.__generator || function(thisArg, body) {
       var _ = { label: 0, sent: function() {
-        if (t[0] & 1)
-          throw t[1];
+        if (t[0] & 1) throw t[1];
         return t[1];
       }, trys: [], ops: [] }, f, y, t, g;
       return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() {
@@ -433,64 +420,58 @@ var require_backoff = __commonJS({
         };
       }
       function step(op) {
-        if (f)
-          throw new TypeError("Generator is already executing.");
-        while (_)
-          try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done)
-              return t;
-            if (y = 0, t)
-              op = [op[0] & 2, t.value];
-            switch (op[0]) {
-              case 0:
-              case 1:
+        if (f) throw new TypeError("Generator is already executing.");
+        while (_) try {
+          if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+          if (y = 0, t) op = [op[0] & 2, t.value];
+          switch (op[0]) {
+            case 0:
+            case 1:
+              t = op;
+              break;
+            case 4:
+              _.label++;
+              return { value: op[1], done: false };
+            case 5:
+              _.label++;
+              y = op[1];
+              op = [0];
+              continue;
+            case 7:
+              op = _.ops.pop();
+              _.trys.pop();
+              continue;
+            default:
+              if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
+                _ = 0;
+                continue;
+              }
+              if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
+                _.label = op[1];
+                break;
+              }
+              if (op[0] === 6 && _.label < t[1]) {
+                _.label = t[1];
                 t = op;
                 break;
-              case 4:
-                _.label++;
-                return { value: op[1], done: false };
-              case 5:
-                _.label++;
-                y = op[1];
-                op = [0];
-                continue;
-              case 7:
-                op = _.ops.pop();
-                _.trys.pop();
-                continue;
-              default:
-                if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
-                  _ = 0;
-                  continue;
-                }
-                if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
-                  _.label = op[1];
-                  break;
-                }
-                if (op[0] === 6 && _.label < t[1]) {
-                  _.label = t[1];
-                  t = op;
-                  break;
-                }
-                if (t && _.label < t[2]) {
-                  _.label = t[2];
-                  _.ops.push(op);
-                  break;
-                }
-                if (t[2])
-                  _.ops.pop();
-                _.trys.pop();
-                continue;
-            }
-            op = body.call(thisArg, _);
-          } catch (e) {
-            op = [6, e];
-            y = 0;
-          } finally {
-            f = t = 0;
+              }
+              if (t && _.label < t[2]) {
+                _.label = t[2];
+                _.ops.push(op);
+                break;
+              }
+              if (t[2]) _.ops.pop();
+              _.trys.pop();
+              continue;
           }
-        if (op[0] & 5)
-          throw op[1];
+          op = body.call(thisArg, _);
+        } catch (e) {
+          op = [6, e];
+          y = 0;
+        } finally {
+          f = t = 0;
+        }
+        if (op[0] & 5) throw op[1];
         return { value: op[0] ? op[1] : void 0, done: true };
       }
     };
@@ -518,7 +499,7 @@ var require_backoff = __commonJS({
     exports.backOff = backOff3;
     var BackOff = (
       /** @class */
-      function() {
+      (function() {
         function BackOff2(request, options) {
           this.request = request;
           this.options = options;
@@ -530,8 +511,7 @@ var require_backoff = __commonJS({
             return __generator(this, function(_a) {
               switch (_a.label) {
                 case 0:
-                  if (!!this.attemptLimitReached)
-                    return [3, 7];
+                  if (!!this.attemptLimitReached) return [3, 7];
                   _a.label = 1;
                 case 1:
                   _a.trys.push([1, 4, , 6]);
@@ -585,7 +565,7 @@ var require_backoff = __commonJS({
           });
         };
         return BackOff2;
-      }()
+      })()
     );
   }
 });
@@ -656,54 +636,57 @@ import axios from "axios";
 var isBrowser = () => typeof window !== "undefined" && typeof window.fetch === "function";
 var CognitiveBeta = class {
   _axiosClient;
-  _config;
+  _apiUrl;
+  _timeout;
+  _withCredentials;
+  _headers;
   constructor(props) {
-    this._config = {
-      baseUrl: props.baseUrl || "https://cognitive.botpress.cloud",
-      timeout: props.timeout || 60001,
-      token: props.token || "",
-      botId: props.botId || "",
-      headers: props.headers || {}
-    };
+    this._apiUrl = props.apiUrl || "https://api.botpress.cloud";
+    this._timeout = props.timeout || 60001;
+    this._withCredentials = props.withCredentials || false;
+    this._headers = { ...props.headers };
+    if (props.botId) {
+      this._headers["X-Bot-Id"] = props.botId;
+    }
+    if (props.token) {
+      this._headers["Authorization"] = `Bearer ${props.token}`;
+    }
     this._axiosClient = axios.create({
-      headers: {
-        Authorization: `Bearer ${this._config.token}`,
-        "X-Bot-Id": this._config.botId,
-        ...this._config.headers
-      },
-      baseURL: this._config.baseUrl
+      headers: this._headers,
+      withCredentials: this._withCredentials,
+      baseURL: this._apiUrl
     });
   }
   async generateText(input, options = {}) {
-    const signal = options.signal ?? AbortSignal.timeout(this._config.timeout);
+    const signal = options.signal ?? AbortSignal.timeout(this._timeout);
     const { data } = await this._withServerRetry(
-      () => this._axiosClient.post("/v1/generate-text", input, {
+      () => this._axiosClient.post("/v2/cognitive/generate-text", input, {
         signal,
-        timeout: options.timeout ?? this._config.timeout
+        timeout: options.timeout ?? this._timeout
       })
     );
     return data;
   }
   async listModels(input, options = {}) {
-    const signal = options.signal ?? AbortSignal.timeout(this._config.timeout);
+    const signal = options.signal ?? AbortSignal.timeout(this._timeout);
     const { data } = await this._withServerRetry(
-      () => this._axiosClient.post("/v1/models", input, {
+      () => this._axiosClient.post("/v2/cognitive/models", input, {
         signal,
-        timeout: options.timeout ?? this._config.timeout
+        timeout: options.timeout ?? this._timeout
       })
     );
     return data;
   }
   async *generateTextStream(request, options = {}) {
-    const signal = options.signal ?? AbortSignal.timeout(this._config.timeout);
+    const signal = options.signal ?? AbortSignal.timeout(this._timeout);
     if (isBrowser()) {
-      const res2 = await fetch(`${this._config.baseUrl}/v1/generate-text-stream`, {
+      const res2 = await fetch(`${this._apiUrl}/v2/cognitive/generate-text-stream`, {
         method: "POST",
         headers: {
-          Authorization: `Bearer ${this._config.token}`,
-          "X-Bot-Id": this._config.botId,
+          ...this._headers,
           "Content-Type": "application/json"
         },
+        credentials: this._withCredentials ? "include" : "omit",
         body: JSON.stringify({ ...request, stream: true }),
         signal
       });
@@ -718,7 +701,7 @@ var CognitiveBeta = class {
         throw new Error("No response body received for streaming request");
       }
       const reader = body.getReader();
-      const iterable = async function* () {
+      const iterable = (async function* () {
         for (; ; ) {
           const { value, done } = await reader.read();
           if (done) {
@@ -728,7 +711,7 @@ var CognitiveBeta = class {
             yield value;
           }
         }
-      }();
+      })();
       for await (const obj of this._ndjson(iterable)) {
         yield obj;
       }
@@ -741,7 +724,7 @@ var CognitiveBeta = class {
         {
           responseType: "stream",
           signal,
-          timeout: options.timeout ?? this._config.timeout
+          timeout: options.timeout ?? this._timeout
         }
       )
     );
@@ -923,7 +906,7 @@ var getBestModels = (models, boosts = {}) => models.sort((a, b) => scoreModel(b,
 var getFastModels = (models, boosts = {}) => models.sort((a, b) => scoreModel(b, "fast", boosts) - scoreModel(a, "fast", boosts));
 var pickModel = (models, downtimes = []) => {
   const copy = [...models];
-  const elasped = (date) => new Date().getTime() - new Date(date).getTime();
+  const elasped = (date) => (/* @__PURE__ */ new Date()).getTime() - new Date(date).getTime();
   const DOWNTIME_THRESHOLD = 1e3 * 60 * DOWNTIME_THRESHOLD_MINUTES;
   if (!copy.length) {
     throw new Error("At least one model is required");
@@ -1037,7 +1020,7 @@ var RemoteModelProvider = class extends ModelProvider {
 };
 
 // src/client.ts
-var Cognitive = class {
+var Cognitive = class _Cognitive {
   ["$$IS_COGNITIVE"] = true;
   static isCognitiveClient(obj) {
     return obj?.$$IS_COGNITIVE === true;
@@ -1068,7 +1051,7 @@ var Cognitive = class {
     return this._client;
   }
   clone() {
-    const copy = new Cognitive({
+    const copy = new _Cognitive({
       client: this._client.clone(),
       provider: this._provider,
       timeout: this._timeoutMs,
@@ -1152,10 +1135,7 @@ var Cognitive = class {
     if (!this._useBeta) {
       return this._generateContent(input);
     }
-    const betaClient = new CognitiveBeta({
-      headers: this._client.config.headers,
-      baseUrl: this._client.config.apiUrl.includes(".cloud") ? "https://cognitive.botpress.cloud" : "https://cognitive.botpress.dev"
-    });
+    const betaClient = new CognitiveBeta(this._client.config);
     const response = await betaClient.generateText(input);
     return {
       output: {
@@ -1237,7 +1217,7 @@ var Cognitive = class {
           if (action === "fallback") {
             this._downtimes.push({
               ref: `${integration}:${model}`,
-              startedAt: new Date().toISOString(),
+              startedAt: (/* @__PURE__ */ new Date()).toISOString(),
               reason: "Model is down"
             });
             this._cleanupOldDowntimes();