prompt-api-polyfill 1.10.0 → 1.12.0

package/README.md

@@ -7,7 +7,7 @@ supporting dynamic backends:
 - **Firebase AI Logic** (cloud)
 - **Google Gemini API** (cloud)
 - **OpenAI API** (cloud)
-- **Transformers.js** (local after initial model download)
+- **Transformers.js** (local after initial model download, **default backend**)
 
 When loaded in the browser, it defines a global:
 
@@ -20,6 +20,9 @@ natively available.
 
 ## Supported Backends
 
+You can choose between multiple cloud and local backends. If no backend is
+configured, the polyfill will use Transformers.js with the default model.
+
 ### Firebase AI Logic (cloud)
 
 - **Uses**: `firebase/ai` SDK.
@@ -49,6 +52,7 @@ natively available.
 
 - **Uses**: `@huggingface/transformers` SDK.
 - **Select by setting**: `window.TRANSFORMERS_CONFIG`.
+- **Note**: This is the **default backend** if no other configuration is provided.
 - **Model**: Uses default if not specified (see
   [`backends/defaults.js`](backends/defaults.js)).
 

package/dist/backends/firebase.js

@@ -1,6 +1,6 @@
 import { initializeApp as Fe } from "firebase/app";
 import { VertexAIBackend as ze, GoogleAIBackend as Ue, getAI as je, getGenerativeModel as Ve, InferenceMode as We } from "firebase/ai";
-import { P as Ke, D as Ge } from "../chunks/defaults-_qJIFiOb.js";
+import { P as Ke, D as Ge } from "../chunks/defaults-j68Er1oH.js";
 const qe = () => {
 };
 const be = function(t) {
@@ -89,9 +89,9 @@ const be = function(t) {
     this.init_();
     const r = e ? this.byteToCharMapWebSafe_ : this.byteToCharMap_, n = [];
     for (let s = 0; s < t.length; s += 3) {
-      const o = t[s], i = s + 1 < t.length, c = i ? t[s + 1] : 0, a = s + 2 < t.length, l = a ? t[s + 2] : 0, y = o >> 2, f = (o & 3) << 4 | c >> 4;
+      const o = t[s], i = s + 1 < t.length, c = i ? t[s + 1] : 0, a = s + 2 < t.length, l = a ? t[s + 2] : 0, p = o >> 2, f = (o & 3) << 4 | c >> 4;
       let k = (c & 15) << 2 | l >> 6, S = l & 63;
-      a || (S = 64, i || (k = 64)), n.push(r[y], r[f], r[k], r[S]);
+      a || (S = 64, i || (k = 64)), n.push(r[p], r[f], r[k], r[S]);
     }
     return n.join("");
   },
@@ -321,7 +321,7 @@ function dt(t, e = at, r = ct) {
 function ut(t) {
   return t && t._delegate ? t._delegate : t;
 }
-class w {
+class y {
   /**
    *
    * @param name The public service name, e.g. app, auth, firestore, database
@@ -344,7 +344,7 @@ class w {
     return this.onInstanceCreated = e, this;
   }
 }
-const E = "[DEFAULT]";
+const _ = "[DEFAULT]";
 class ft {
   constructor(e, r) {
     this.name = e, this.container = r, this.component = null, this.instances = /* @__PURE__ */ new Map(), this.instancesDeferred = /* @__PURE__ */ new Map(), this.instancesOptions = /* @__PURE__ */ new Map(), this.onInitCallbacks = /* @__PURE__ */ new Map();
@@ -397,7 +397,7 @@ class ft {
     if (this.component = e, !!this.shouldAutoInitialize()) {
       if (gt(e))
         try {
-          this.getOrInitializeService({ instanceIdentifier: E });
+          this.getOrInitializeService({ instanceIdentifier: _ });
         } catch {
         }
       for (const [r, n] of this.instancesDeferred.entries()) {
@@ -412,7 +412,7 @@ class ft {
       }
     }
   }
-  clearInstance(e = E) {
+  clearInstance(e = _) {
     this.instancesDeferred.delete(e), this.instancesOptions.delete(e), this.instances.delete(e);
   }
   // app.delete() will call this method on every provider to delete the services
@@ -427,10 +427,10 @@ class ft {
   isComponentSet() {
     return this.component != null;
   }
-  isInitialized(e = E) {
+  isInitialized(e = _) {
     return this.instances.has(e);
   }
-  getOptions(e = E) {
+  getOptions(e = _) {
     return this.instancesOptions.get(e) || {};
   }
   initialize(e = {}) {
@@ -490,15 +490,15 @@ class ft {
       }
     return n || null;
   }
-  normalizeInstanceIdentifier(e = E) {
-    return this.component ? this.component.multipleInstances ? e : E : e;
+  normalizeInstanceIdentifier(e = _) {
+    return this.component ? this.component.multipleInstances ? e : _ : e;
   }
   shouldAutoInitialize() {
     return !!this.component && this.component.instantiationMode !== "EXPLICIT";
   }
 }
 function pt(t) {
-  return t === E ? void 0 : t;
+  return t === _ ? void 0 : t;
 }
 function gt(t) {
   return t.instantiationMode === "EAGER";
@@ -647,7 +647,7 @@ function Tt(t) {
     const s = () => {
       t.removeEventListener("success", o), t.removeEventListener("error", i);
     }, o = () => {
-      r(m(t.result)), s();
+      r(b(t.result)), s();
     }, i = () => {
       n(t.error), s();
     };
@@ -683,7 +683,7 @@ let U = {
       if (e === "store")
         return r.objectStoreNames[1] ? void 0 : r.objectStore(r.objectStoreNames[0]);
     }
-    return m(t[e]);
+    return b(t[e]);
   },
   set(t, e, r) {
     return t[e] = r, !0;
@@ -698,17 +698,17 @@ function Dt(t) {
 function kt(t) {
   return t === IDBDatabase.prototype.transaction && !("objectStoreNames" in IDBTransaction.prototype) ? function(e, ...r) {
     const n = t.call(N(this), e, ...r);
-    return Ie.set(n, e.sort ? e.sort() : [e]), m(n);
+    return Ie.set(n, e.sort ? e.sort() : [e]), b(n);
   } : It().includes(t) ? function(...e) {
-    return t.apply(N(this), e), m(Ae.get(this));
+    return t.apply(N(this), e), b(Ae.get(this));
   } : function(...e) {
-    return m(t.apply(N(this), e));
+    return b(t.apply(N(this), e));
   };
 }
 function St(t) {
   return typeof t == "function" ? kt(t) : (t instanceof IDBTransaction && Ct(t), yt(t, At()) ? new Proxy(t, U) : t);
 }
-function m(t) {
+function b(t) {
   if (t instanceof IDBRequest)
     return Tt(t);
   if (O.has(t))
@@ -718,9 +718,9 @@ function m(t) {
 }
 const N = (t) => J.get(t);
 function vt(t, e, { blocked: r, upgrade: n, blocking: s, terminated: o } = {}) {
-  const i = indexedDB.open(t, e), c = m(i);
+  const i = indexedDB.open(t, e), c = b(i);
   return n && i.addEventListener("upgradeneeded", (a) => {
-    n(m(i.result), a.oldVersion, a.newVersion, m(i.transaction), a);
+    n(b(i.result), a.oldVersion, a.newVersion, b(i.transaction), a);
   }), r && i.addEventListener("blocked", (a) => r(
     // Casting due to https://github.com/microsoft/TypeScript-DOM-lib-generator/pull/1405
     a.oldVersion,
@@ -777,8 +777,8 @@ class Pt {
 function Mt(t) {
   return t.getComponent()?.type === "VERSION";
 }
-const j = "@firebase/app", ae = "0.14.8";
-const p = new ye("@firebase/app"), Ot = "@firebase/app-compat", Nt = "@firebase/analytics-compat", $t = "@firebase/analytics", xt = "@firebase/app-check-compat", Lt = "@firebase/app-check", Ht = "@firebase/auth", Ft = "@firebase/auth-compat", zt = "@firebase/database", Ut = "@firebase/data-connect", jt = "@firebase/database-compat", Vt = "@firebase/functions", Wt = "@firebase/functions-compat", Kt = "@firebase/installations", Gt = "@firebase/installations-compat", qt = "@firebase/messaging", Xt = "@firebase/messaging-compat", Yt = "@firebase/performance", Jt = "@firebase/performance-compat", Zt = "@firebase/remote-config", Qt = "@firebase/remote-config-compat", er = "@firebase/storage", tr = "@firebase/storage-compat", rr = "@firebase/firestore", nr = "@firebase/ai", sr = "@firebase/firestore-compat", or = "firebase";
+const j = "@firebase/app", ae = "0.14.9";
+const g = new ye("@firebase/app"), Ot = "@firebase/app-compat", Nt = "@firebase/analytics-compat", $t = "@firebase/analytics", xt = "@firebase/app-check-compat", Lt = "@firebase/app-check", Ht = "@firebase/auth", Ft = "@firebase/auth-compat", zt = "@firebase/database", Ut = "@firebase/data-connect", jt = "@firebase/database-compat", Vt = "@firebase/functions", Wt = "@firebase/functions-compat", Kt = "@firebase/installations", Gt = "@firebase/installations-compat", qt = "@firebase/messaging", Xt = "@firebase/messaging-compat", Yt = "@firebase/performance", Jt = "@firebase/performance-compat", Zt = "@firebase/remote-config", Qt = "@firebase/remote-config-compat", er = "@firebase/storage", tr = "@firebase/storage-compat", rr = "@firebase/firestore", nr = "@firebase/ai", sr = "@firebase/firestore-compat", or = "firebase";
 const V = "[DEFAULT]", ir = {
   [j]: "fire-core",
   [Ot]: "fire-core-compat",
@@ -815,13 +815,13 @@ function ce(t, e) {
   try {
     t.container.addComponent(e);
   } catch (r) {
-    p.debug(`Component ${e.name} failed to register with FirebaseApp ${t.name}`, r);
+    g.debug(`Component ${e.name} failed to register with FirebaseApp ${t.name}`, r);
   }
 }
 function I(t) {
   const e = t.name;
   if (W.has(e))
-    return p.debug(`There were multiple attempts to register component ${e}.`), !1;
+    return g.debug(`There were multiple attempts to register component ${e}.`), !1;
   W.set(e, t);
   for (const r of P.values())
     ce(r, t);
@@ -848,10 +848,10 @@ const cr = {
   "idb-delete": "Error thrown when deleting from IndexedDB. Original error: {$originalErrorMessage}.",
   "finalization-registry-not-supported": "FirebaseServerApp deleteOnDeref field defined but the JS runtime does not support FinalizationRegistry.",
   "invalid-server-app-environment": "FirebaseServerApp is not for use in browser environments."
-}, b = new Y("app", "Firebase", cr);
+}, E = new Y("app", "Firebase", cr);
 class lr {
   constructor(e, r, n) {
-    this._isDeleted = !1, this._options = { ...e }, this._config = { ...r }, this._name = r.name, this._automaticDataCollectionEnabled = r.automaticDataCollectionEnabled, this._container = n, this.container.addComponent(new w(
+    this._isDeleted = !1, this._options = { ...e }, this._config = { ...r }, this._name = r.name, this._automaticDataCollectionEnabled = r.automaticDataCollectionEnabled, this._container = n, this.container.addComponent(new y(
       "app",
       () => this,
       "PUBLIC"
@@ -888,7 +888,7 @@ class lr {
    */
   checkDestroyed() {
     if (this.isDeleted)
-      throw b.create("app-deleted", { appName: this._name });
+      throw E.create("app-deleted", { appName: this._name });
   }
 }
 function hr(t, e = {}) {
@@ -900,11 +900,11 @@ function hr(t, e = {}) {
     ...e
   }, s = n.name;
   if (typeof s != "string" || !s)
-    throw b.create("bad-app-name", {
+    throw E.create("bad-app-name", {
       appName: String(s)
     });
   if (r || (r = we()), !r)
-    throw b.create(
+    throw E.create(
       "no-options"
       /* AppError.NO_OPTIONS */
     );
@@ -912,7 +912,7 @@ function hr(t, e = {}) {
   if (o) {
     if (F(r, o.options) && F(n, o.config))
       return o;
-    throw b.create("duplicate-app", { appName: s });
+    throw E.create("duplicate-app", { appName: s });
   }
   const i = new mt(s);
   for (const a of W.values())
@@ -925,7 +925,7 @@ function dr(t = V) {
   if (!e && t === V && we())
     return hr();
   if (!e)
-    throw b.create("no-app", { appName: t });
+    throw E.create("no-app", { appName: t });
   return e;
 }
 function B(t, e, r) {
@@ -936,10 +936,10 @@ function B(t, e, r) {
     const i = [
       `Unable to register library "${n}" with version "${e}":`
     ];
-    s && i.push(`library name "${n}" contains illegal characters (whitespace or "/")`), s && o && i.push("and"), o && i.push(`version name "${e}" contains illegal characters (whitespace or "/")`), p.warn(i.join(" "));
+    s && i.push(`library name "${n}" contains illegal characters (whitespace or "/")`), s && o && i.push("and"), o && i.push(`version name "${e}" contains illegal characters (whitespace or "/")`), g.warn(i.join(" "));
     return;
   }
-  I(new w(
+  I(new y(
     `${n}-version`,
     () => ({ library: n, version: e }),
     "VERSION"
@@ -961,7 +961,7 @@ function Ce() {
       }
     }
   }).catch((t) => {
-    throw b.create("idb-open", {
+    throw E.create("idb-open", {
       originalErrorMessage: t.message
     });
   })), x;
@@ -972,12 +972,12 @@ async function pr(t) {
     return await r.done, n;
   } catch (e) {
     if (e instanceof D)
-      p.warn(e.message);
+      g.warn(e.message);
     else {
-      const r = b.create("idb-get", {
+      const r = E.create("idb-get", {
         originalErrorMessage: e?.message
       });
-      p.warn(r.message);
+      g.warn(r.message);
     }
   }
 }
@@ -987,12 +987,12 @@ async function le(t, e) {
     await n.objectStore(T).put(e, De(t)), await n.done;
   } catch (r) {
     if (r instanceof D)
-      p.warn(r.message);
+      g.warn(r.message);
     else {
-      const n = b.create("idb-set", {
+      const n = E.create("idb-set", {
         originalErrorMessage: r?.message
       });
-      p.warn(n.message);
+      g.warn(n.message);
     }
   }
 }
@@ -1024,7 +1024,7 @@ class br {
       }
       return this._storage.overwrite(this._heartbeatsCache);
     } catch (e) {
-      p.warn(e);
+      g.warn(e);
     }
   }
   /**
@@ -1041,7 +1041,7 @@ class br {
       const e = he(), { heartbeatsToSend: r, unsentEntries: n } = Er(this._heartbeatsCache.heartbeats), s = Ee(JSON.stringify({ version: 2, heartbeats: r }));
       return this._heartbeatsCache.lastSentHeartbeatDate = e, n.length > 0 ? (this._heartbeatsCache.heartbeats = n, await this._storage.overwrite(this._heartbeatsCache)) : (this._heartbeatsCache.heartbeats = [], this._storage.overwrite(this._heartbeatsCache)), s;
     } catch (e) {
-      return p.warn(e), "";
+      return g.warn(e), "";
     }
   }
 }
@@ -1130,12 +1130,12 @@ function wr(t) {
   return e;
 }
 function yr(t) {
-  I(new w(
+  I(new y(
     "platform-logger",
     (e) => new Pt(e),
     "PRIVATE"
     /* ComponentType.PRIVATE */
-  )), I(new w(
+  )), I(new y(
     "heartbeat",
     (e) => new br(e),
     "PRIVATE"
@@ -1281,11 +1281,11 @@ async function Q({ url: t, body: e }, r) {
     throw u.create("fetch-parse-error", {
       originalErrorMessage: `ttl field (timeToLive) is not in standard Protobuf Duration format: ${c.ttl}`
     });
-  const l = Number(a[1]) * 1e3, y = Date.now();
+  const l = Number(a[1]) * 1e3, p = Date.now();
   return {
     token: c.token,
-    expireTimeMillis: y + l,
-    issuedAtTimeMillis: y
+    expireTimeMillis: p + l,
+    issuedAtTimeMillis: p
   };
 }
 function Rr(t, e) {
@@ -1375,21 +1375,21 @@ async function Oe(t) {
 function Ne(t) {
   return `${t.options.appId}-${t.name}`;
 }
-const g = new ye("@firebase/app-check");
+const m = new ye("@firebase/app-check");
 async function xr(t) {
   if (X()) {
     let e;
     try {
       e = await Mr(t);
     } catch (r) {
-      g.warn(`Failed to read token from IndexedDB. Error: ${r}`);
+      m.warn(`Failed to read token from IndexedDB. Error: ${r}`);
     }
     return e;
   }
 }
 function L(t, e) {
   return X() ? Or(t, e).catch((r) => {
-    g.warn(`Failed to write token to IndexedDB. Error: ${r}`);
+    m.warn(`Failed to write token to IndexedDB. Error: ${r}`);
   }) : Promise.resolve();
 }
 async function Lr() {
@@ -1402,7 +1402,7 @@ async function Lr() {
     return t;
   {
     const e = crypto.randomUUID();
-    return Nr(e).catch((r) => g.warn(`Failed to persist debug token to IndexedDB. Error: ${r}`)), e;
+    return Nr(e).catch((r) => m.warn(`Failed to persist debug token to IndexedDB. Error: ${r}`)), e;
   }
 }
 function ee() {
@@ -1437,34 +1437,35 @@ async function G(t, e = !1, r = !1) {
   Z(n);
   const s = d(n);
   let o = s.token, i;
-  if (o && !_(o) && (s.token = void 0, o = void 0), !o) {
+  if (o && !w(o) && (s.token = void 0, o = void 0), !o) {
     const l = await s.cachedTokenPromise;
-    l && (_(l) ? o = l : await L(n, void 0));
+    l && (w(l) ? o = l : await L(n, void 0));
   }
-  if (!e && o && _(o))
+  if (!e && o && w(o))
     return {
       token: o.token
     };
   let c = !1;
   if (ee())
     try {
-      s.exchangeTokenPromise || (s.exchangeTokenPromise = Q(Re(n, await te()), t.heartbeatServiceProvider).finally(() => {
+      const l = await te();
+      s.exchangeTokenPromise || (s.exchangeTokenPromise = Q(Re(n, l), t.heartbeatServiceProvider).finally(() => {
         s.exchangeTokenPromise = void 0;
       }), c = !0);
-      const l = await s.exchangeTokenPromise;
-      return await L(n, l), s.token = l, { token: l.token };
+      const p = await s.exchangeTokenPromise;
+      return await L(n, p), s.token = p, { token: p.token };
     } catch (l) {
-      return l.code === "appCheck/throttled" || l.code === "appCheck/initial-throttle" ? g.warn(l.message) : r && g.error(l), H(l);
+      return l.code === "appCheck/throttled" || l.code === "appCheck/initial-throttle" ? m.warn(l.message) : r && m.error(l), H(l);
     }
   try {
     s.exchangeTokenPromise || (s.exchangeTokenPromise = s.provider.getToken().finally(() => {
       s.exchangeTokenPromise = void 0;
     }), c = !0), o = await d(n).exchangeTokenPromise;
   } catch (l) {
-    l.code === "appCheck/throttled" || l.code === "appCheck/initial-throttle" ? g.warn(l.message) : r && g.error(l), i = l;
+    l.code === "appCheck/throttled" || l.code === "appCheck/initial-throttle" ? m.warn(l.message) : r && m.error(l), i = l;
   }
   let a;
-  return o ? i ? _(o) ? a = {
+  return o ? i ? w(o) ? a = {
     token: o.token,
     internalError: i
   } : a = H(i) : (a = {
@@ -1489,7 +1490,7 @@ function $e(t, e, r, n) {
     error: n,
     type: e
   };
-  if (o.tokenObservers = [...o.tokenObservers, i], o.token && _(o.token)) {
+  if (o.tokenObservers = [...o.tokenObservers, i], o.token && w(o.token)) {
     const c = o.token;
     Promise.resolve().then(() => {
       r({ token: c.token }), pe(t);
@@ -1542,7 +1543,7 @@ function Le(t, e) {
     } catch {
     }
 }
-function _(t) {
+function w(t) {
   return t.expireTimeMillis - Date.now() > 0;
 }
 function H(t) {
@@ -1573,7 +1574,7 @@ function Kr(t) {
     removeTokenListener: (e) => xe(t.app, e)
   };
 }
-const Gr = "@firebase/app-check", qr = "0.11.0", Xr = "https://www.google.com/recaptcha/enterprise.js";
+const Gr = "@firebase/app-check", qr = "0.11.1", Xr = "https://www.google.com/recaptcha/enterprise.js";
 function Yr(t, e) {
   const r = new A(), n = d(t);
   n.reCAPTCHAState = { initialized: r };
@@ -1728,11 +1729,11 @@ function nn(t = dr(), e) {
 }
 function sn(t, e, r = !1) {
   const n = Ir(t, { ...ke });
-  n.activated = !0, n.provider = e, n.cachedTokenPromise = xr(t).then((s) => (s && _(s) && (n.token = s, Le(t, { token: s.token })), s)), n.isTokenAutoRefreshEnabled = r && t.automaticDataCollectionEnabled, !t.automaticDataCollectionEnabled && r && g.warn("`isTokenAutoRefreshEnabled` is true but `automaticDataCollectionEnabled` was set to false during `initializeApp()`. This blocks automatic token refresh."), n.provider.initialize(t);
+  n.activated = !0, n.provider = e, n.cachedTokenPromise = xr(t).then((s) => (s && w(s) && (n.token = s, Le(t, { token: s.token })), s)), n.isTokenAutoRefreshEnabled = r && t.automaticDataCollectionEnabled, !t.automaticDataCollectionEnabled && r && m.warn("`isTokenAutoRefreshEnabled` is true but `automaticDataCollectionEnabled` was set to false during `initializeApp()`. This blocks automatic token refresh."), n.provider.initialize(t);
 }
 const on = "app-check", me = "app-check-internal";
 function an() {
-  I(new w(
+  I(new y(
     on,
     (t) => {
       const e = t.getProvider("app").getImmediate(), r = t.getProvider("heartbeat");
@@ -1745,7 +1746,7 @@ function an() {
     /* InstantiationMode.EXPLICIT */
   ).setInstanceCreatedCallback((t, e, r) => {
     t.getProvider(me).initialize();
-  })), I(new w(
+  })), I(new y(
     me,
     (t) => {
       const e = t.getProvider("app-check").getImmediate();
@@ -1761,7 +1762,6 @@ function an() {
 an();
 class dn extends Ke {
   #e;
-  #t;
   constructor(e) {
     const {
       geminiApiProvider: r,
@@ -1783,19 +1783,40 @@ class dn extends Ke {
       useLimitedUseAppCheckTokens: i || !0
     });
   }
+  /**
+   * Creates a model session and stores it.
+   * @param {Object} _options - LanguageModel options.
+   * @param {Object} sessionParams - Parameters for the cloud or local model.
+   * @returns {any} The created session object.
+   */
   createSession(e, r) {
-    return this.#t = r, this.#e = Ve(this.ai, {
+    return this.#e = Ve(this.ai, {
       mode: We.ONLY_IN_CLOUD,
       inCloudParams: r
     }), this.#e;
   }
+  /**
+   * Generates content (non-streaming).
+   * @param {Array} contents - The history + new message content.
+   * @returns {Promise<{text: string, usage: number}>}
+   */
   async generateContent(e) {
     const r = await this.#e.generateContent({ contents: e }), n = r.response.usageMetadata?.promptTokenCount || 0;
     return { text: r.response.text(), usage: n };
   }
+  /**
+   * Generates content stream.
+   * @param {Array} contents - The history + new content.
+   * @returns {Promise<AsyncIterable>} Stream of chunks.
+   */
   async generateContentStream(e) {
     return (await this.#e.generateContentStream({ contents: e })).stream;
   }
+  /**
+   * Counts tokens.
+   * @param {Array} contents - The content to count.
+   * @returns {Promise<number>} Total tokens.
+   */
   async countTokens(e) {
     const { totalTokens: r } = await this.#e.countTokens({
       contents: e

package/dist/backends/gemini.js

@@ -1,5 +1,5 @@
 import { GoogleGenAI as s } from "@google/genai";
-import { P as a, D as i } from "../chunks/defaults-_qJIFiOb.js";
+import { P as a, D as i } from "../chunks/defaults-j68Er1oH.js";
 class u extends a {
   #n;
   #e;
@@ -7,9 +7,20 @@ class u extends a {
   constructor(t) {
     super(t.modelName || i.gemini.modelName), this.#n = new s({ apiKey: t.apiKey });
   }
+  /**
+   * Creates a model session.
+   * @param {Object} options - LanguageModel options.
+   * @param {Object} sessionParams - Session parameters.
+   * @returns {Object} The session object.
+   */
   createSession(t, e) {
     return this.#t = e, this.#e = t.modelName || this.modelName, { model: this.#e, params: e };
   }
+  /**
+   * Generates content (non-streaming).
+   * @param {Array} contents - The history + new message content.
+   * @returns {Promise<{text: string, usage: number}>}
+   */
   async generateContent(t) {
     const e = {
       systemInstruction: this.#t.systemInstruction,
@@ -22,6 +33,11 @@ class u extends a {
     }), o = n.usageMetadata?.promptTokenCount || 0;
     return { text: n.text, usage: o };
   }
+  /**
+   * Generates content stream.
+   * @param {Array} contents - The history + new content.
+   * @returns {Promise<AsyncIterable>} Stream of chunks.
+   */
   async generateContentStream(t) {
     const e = {
       systemInstruction: this.#t.systemInstruction,
@@ -42,6 +58,11 @@ class u extends a {
         };
     })();
   }
+  /**
+   * Counts tokens.
+   * @param {Array} contents - The content to count.
+   * @returns {Promise<number>} Total tokens.
+   */
   async countTokens(t) {
     const { totalTokens: e } = await this.#n.models.countTokens({
       model: this.#e,

package/dist/backends/openai.js

@@ -1,5 +1,5 @@
 import u from "openai";
-import { P as m, D as l } from "../chunks/defaults-_qJIFiOb.js";
+import { P as m, D as l } from "../chunks/defaults-j68Er1oH.js";
 class y extends m {
   #e;
   constructor(t) {
@@ -21,6 +21,12 @@ class y extends m {
     }
     return "available";
   }
+  /**
+   * Creates a model session and stores it.
+   * @param {Object} options - LanguageModel options.
+   * @param {Object} sessionParams - Parameters for the cloud or local model.
+   * @returns {any} The created session object.
+   */
   createSession(t, s) {
     this.#e = {
       model: t.modelName || this.modelName,
@@ -93,6 +99,11 @@ class y extends m {
   #s(t) {
     return this.#e.model !== this.modelName ? this.#e.model : t ? `${this.modelName}-audio-preview` : this.modelName;
   }
+  /**
+   * Generates content (non-streaming).
+   * @param {Array} contents - The history + new message content.
+   * @returns {Promise<{text: string, usage: number}>}
+   */
   async generateContent(t) {
     const { messages: s } = this.#r(
       t,
@@ -117,12 +128,17 @@ class y extends m {
           c && typeof c == "object" && "value" in c && (i = JSON.stringify(c.value));
         } catch {
         }
-      const n = a.usage?.prompt_tokens || 0;
-      return { text: i, usage: n };
+      const p = a.usage?.prompt_tokens || 0;
+      return { text: i, usage: p };
     } catch (a) {
       throw console.error("OpenAI Generate Content Error:", a), a;
     }
   }
+  /**
+   * Generates content stream.
+   * @param {Array} contents - The history + new content.
+   * @returns {Promise<AsyncIterable>} Stream of chunks.
+   */
   async generateContentStream(t) {
     const { messages: s } = this.#r(
       t,
@@ -142,11 +158,10 @@ class y extends m {
     try {
       const a = await this.openai.chat.completions.create(o);
       return (async function* () {
-        let p = !0;
-        for await (const i of a) {
-          let n = i.choices[0]?.delta?.content;
-          n && (yield {
-            text: () => n,
+        for await (const n of a) {
+          let i = n.choices[0]?.delta?.content;
+          i && (yield {
+            text: () => i,
             usageMetadata: { totalTokenCount: 0 }
           });
         }
@@ -155,6 +170,11 @@ class y extends m {
       throw console.error("OpenAI Generate Content Stream Error:", a), a;
     }
   }
+  /**
+   * Counts tokens.
+   * @param {Array} contents - The content to count.
+   * @returns {Promise<number>} Total tokens.
+   */
   async countTokens(t) {
     let s = "";
     if (Array.isArray(t)) {
@@ -173,19 +193,19 @@ class y extends m {
     });
     for (const e of t) {
       const o = e.role === "model" ? "assistant" : "user", a = [];
-      for (const p of e.parts)
-        if (p.text)
-          a.push({ type: "text", text: p.text });
-        else if (p.inlineData) {
-          const { data: i, mimeType: n } = p.inlineData;
-          n.startsWith("image/") ? a.push({
+      for (const n of e.parts)
+        if (n.text)
+          a.push({ type: "text", text: n.text });
+        else if (n.inlineData) {
+          const { data: i, mimeType: p } = n.inlineData;
+          p.startsWith("image/") ? a.push({
             type: "image_url",
-            image_url: { url: `data:${n};base64,${i}` }
-          }) : n.startsWith("audio/") && a.push({
+            image_url: { url: `data:${p};base64,${i}` }
+          }) : p.startsWith("audio/") && a.push({
             type: "input_audio",
             input_audio: {
               data: i,
-              format: n.split("/")[1] === "mpeg" ? "mp3" : "wav"
+              format: p.split("/")[1] === "mpeg" ? "mp3" : "wav"
             }
           });
         }