@browser-ai/core 2.1.4 → 2.1.6

package/dist/index.mjs

@@ -721,11 +721,7 @@ import {
 function convertBase64ToUint8Array(base64) {
   try {
     const binaryString = atob(base64);
-    const bytes = new Uint8Array(binaryString.length);
-    for (let i = 0; i < binaryString.length; i++) {
-      bytes[i] = binaryString.charCodeAt(i);
-    }
-    return bytes;
+    return Uint8Array.from(binaryString, (c) => c.charCodeAt(0));
   } catch (error) {
     throw new Error(`Failed to convert base64 to Uint8Array: ${error}`);
   }
@@ -991,19 +987,7 @@ function gatherUnsupportedSettingWarnings(options) {
 }
 
 // src/utils/prompt-utils.ts
-function hasMultimodalContent(prompt) {
-  for (const message of prompt) {
-    if (message.role === "user") {
-      for (const part of message.content) {
-        if (part.type === "file") {
-          return true;
-        }
-      }
-    }
-  }
-  return false;
-}
-function getExpectedInputs(prompt) {
+function getMultimodalInfo(prompt) {
   const inputs = /* @__PURE__ */ new Set();
   for (const message of prompt) {
     if (message.role === "user") {
@@ -1018,7 +1002,11 @@ function getExpectedInputs(prompt) {
       }
     }
   }
-  return Array.from(inputs).map((type) => ({ type }));
+  const hasMultiModalInput = inputs.size > 0;
+  return {
+    hasMultiModalInput,
+    expectedInputs: hasMultiModalInput ? Array.from(inputs, (type) => ({ type })) : void 0
+  };
 }
 
 // src/chat/session-manager.ts
@@ -1068,15 +1056,17 @@ var SessionManager = class {
     }
     const sessionOptions = this.prepareSessionOptions(options);
     this.session = await LanguageModel.create(sessionOptions);
-    const onQuotaOverflow = options?.onQuotaOverflow || this.baseOptions.onQuotaOverflow;
-    if (onQuotaOverflow) {
-      this.session.addEventListener("quotaoverflow", onQuotaOverflow);
+    const onOverflow = options?.onContextOverflow || this.baseOptions.onContextOverflow || options?.onQuotaOverflow || this.baseOptions.onQuotaOverflow;
+    const overflowHandler = onOverflow ?? (() => {
+      console.warn(
+        "Model context window exceeded. Consider handling the 'contextoverflow' event."
+      );
+    });
+    const session = this.session;
+    if ("oncontextoverflow" in session) {
+      session.addEventListener("contextoverflow", overflowHandler);
     } else {
-      this.session.addEventListener("quotaoverflow", () => {
-        console.warn(
-          "Model quota exceeded. Consider handling 'quotaoverflow' event."
-        );
-      });
+      session.addEventListener("quotaoverflow", overflowHandler);
     }
     return this.session;
   }
@@ -1145,18 +1135,34 @@ var SessionManager = class {
     this.session = null;
   }
   /**
-   * Gets the input quota for the current session, if available
-   * @returns The input quota or undefined if not available
+   * Gets the context window size (token limit) for the current session, if available.
+   * @returns The context window size or undefined if not available
+   */
+  getContextWindow() {
+    const session = this.getCurrentSession();
+    if (!session) return void 0;
+    return session.contextWindow ?? session.inputQuota;
+  }
+  /**
+   * Gets the current context usage (tokens consumed) for the current session, if available.
+   * @returns The context usage or undefined if not available
+   */
+  getContextUsage() {
+    const session = this.getCurrentSession();
+    if (!session) return void 0;
+    return session.contextUsage ?? session.inputUsage;
+  }
+  /**
+   * @deprecated Use {@link getContextWindow} instead.
    */
   getInputQuota() {
-    return this.getCurrentSession()?.inputQuota;
+    return this.getContextWindow();
   }
   /**
-   * Gets the input usage for the current session, if available
-   * @returns The input usage or undefined if not available
+   * @deprecated Use {@link getContextUsage} instead.
    */
   getInputUsage() {
-    return this.getCurrentSession()?.inputUsage;
+    return this.getContextUsage();
   }
   /**
    * Prepares merged session options from base config and request options
@@ -1172,7 +1178,8 @@ var SessionManager = class {
         systemMessage,
         expectedInputs,
         onDownloadProgress,
-        onQuotaOverflow,
+        onContextOverflow: _onContextOverflow,
+        onQuotaOverflow: _onQuotaOverflow,
         ...createOptions
       } = options;
       Object.assign(mergedOptions, createOptions);
@@ -1192,17 +1199,8 @@ var SessionManager = class {
         };
       }
     }
-    this.sanitizeOptions(mergedOptions);
     return mergedOptions;
   }
-  /**
-   * Removes custom options that aren't part of LanguageModel.create API
-   *
-   * @param options - Options object to sanitize in-place
-   * @private
-   */
-  sanitizeOptions(options) {
-  }
 };
 
 // src/chat/browser-ai-language-model.ts
@@ -1278,7 +1276,7 @@ var BrowserAIChatLanguageModel = class {
         )
       );
     }
-    const hasMultiModalInput = hasMultimodalContent(prompt);
+    const { hasMultiModalInput, expectedInputs } = getMultimodalInfo(prompt);
     const { systemMessage, messages } = convertToBrowserAIMessages(prompt);
     const promptOptions = {};
     if (responseFormat?.type === "json") {
@@ -1296,7 +1294,7 @@ var BrowserAIChatLanguageModel = class {
       warnings,
       promptOptions,
       hasMultiModalInput,
-      expectedInputs: hasMultiModalInput ? getExpectedInputs(prompt) : void 0,
+      expectedInputs,
       functionTools
     };
   }
@@ -1395,18 +1393,30 @@ var BrowserAIChatLanguageModel = class {
     };
   }
   /**
-   * Gets the input usage for the current session, if available
-   * @returns The input usage or undefined if not available
+   * Gets the current context usage (tokens consumed) for the current session, if available
+   * @returns The context usage or undefined if not available
+   */
+  getContextUsage() {
+    return this.sessionManager.getContextUsage();
+  }
+  /**
+   * Gets the context window size (token limit) for the current session, if available
+   * @returns The context window size or undefined if not available
+   */
+  getContextWindow() {
+    return this.sessionManager.getContextWindow();
+  }
+  /**
+   * @deprecated Use {@link getContextUsage} instead.
    */
   getInputUsage() {
-    return this.sessionManager.getInputUsage();
+    return this.getContextUsage();
   }
   /**
-   * Gets the input quota for the current session, if available
-   * @returns The input quota or undefined if not available
+   * @deprecated Use {@link getContextWindow} instead.
    */
   getInputQuota() {
-    return this.sessionManager.getInputQuota();
+    return this.getContextWindow();
   }
   /**
    * Check the availability of the browser AI model
@@ -1515,7 +1525,7 @@ var BrowserAIChatLanguageModel = class {
             finishReason,
             usage: {
               inputTokens: {
-                total: session.inputUsage,
+                total: session.contextUsage ?? session.inputUsage,
                 noCache: void 0,
                 cacheRead: void 0,
                 cacheWrite: void 0