@browser-ai/core 2.1.1 → 2.1.3

package/dist/index.mjs

@@ -454,6 +454,91 @@ var ToolCallFenceDetector = class {
   }
 };
 
+// ../shared/src/streaming/tool-call-stream-utils.ts
+function extractToolName(content) {
+  const jsonMatch = content.match(/\{\s*"name"\s*:\s*"([^"]+)"/);
+  if (jsonMatch) {
+    return jsonMatch[1];
+  }
+  return null;
+}
+var ARGUMENTS_FIELD_REGEX = /"arguments"\s*:\s*/g;
+var ARGUMENTS_SEARCH_OVERLAP = 32;
+function createArgumentsStreamState() {
+  return {
+    searchFrom: 0,
+    valueStartIndex: null,
+    parseIndex: 0,
+    started: false,
+    depth: 0,
+    inString: false,
+    escaped: false,
+    complete: false
+  };
+}
+function extractArgumentsDelta(content, state) {
+  if (state.complete) {
+    return "";
+  }
+  if (state.valueStartIndex === null) {
+    ARGUMENTS_FIELD_REGEX.lastIndex = state.searchFrom;
+    const match = ARGUMENTS_FIELD_REGEX.exec(content);
+    ARGUMENTS_FIELD_REGEX.lastIndex = 0;
+    if (!match || match.index === void 0) {
+      state.searchFrom = Math.max(0, content.length - ARGUMENTS_SEARCH_OVERLAP);
+      return "";
+    }
+    state.valueStartIndex = match.index + match[0].length;
+    state.parseIndex = state.valueStartIndex;
+    state.searchFrom = state.valueStartIndex;
+  }
+  if (state.parseIndex >= content.length) {
+    return "";
+  }
+  let delta = "";
+  for (let i = state.parseIndex; i < content.length; i++) {
+    const char = content[i];
+    delta += char;
+    if (!state.started) {
+      if (!/\s/.test(char)) {
+        state.started = true;
+        if (char === "{" || char === "[") {
+          state.depth = 1;
+        }
+      }
+      continue;
+    }
+    if (state.escaped) {
+      state.escaped = false;
+      continue;
+    }
+    if (char === "\\") {
+      state.escaped = true;
+      continue;
+    }
+    if (char === '"') {
+      state.inString = !state.inString;
+      continue;
+    }
+    if (!state.inString) {
+      if (char === "{" || char === "[") {
+        state.depth += 1;
+      } else if (char === "}" || char === "]") {
+        if (state.depth > 0) {
+          state.depth -= 1;
+          if (state.depth === 0) {
+            state.parseIndex = i + 1;
+            state.complete = true;
+            return delta;
+          }
+        }
+      }
+    }
+  }
+  state.parseIndex = content.length;
+  return delta;
+}
+
 // src/convert-to-browser-ai-messages.ts
 import {
   UnsupportedFunctionalityError
@@ -884,6 +969,20 @@ 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
+   */
+  getInputQuota() {
+    return this.getCurrentSession()?.inputQuota;
+  }
+  /**
+   * Gets the input usage for the current session, if available
+   * @returns The input usage or undefined if not available
+   */
+  getInputUsage() {
+    return this.getCurrentSession()?.inputUsage;
+  }
   /**
    * Prepares merged session options from base config and request options
    *
@@ -935,63 +1034,6 @@ var SessionManager = class {
 function doesBrowserSupportBrowserAI() {
   return typeof LanguageModel !== "undefined";
 }
-function extractToolName(content) {
-  const jsonMatch = content.match(/\{\s*"name"\s*:\s*"([^"]+)"/);
-  if (jsonMatch) {
-    return jsonMatch[1];
-  }
-  return null;
-}
-function extractArgumentsContent(content) {
-  const match = content.match(/"arguments"\s*:\s*/);
-  if (!match || match.index === void 0) {
-    return "";
-  }
-  const startIndex = match.index + match[0].length;
-  let result = "";
-  let depth = 0;
-  let inString = false;
-  let escaped = false;
-  let started = false;
-  for (let i = startIndex; i < content.length; i++) {
-    const char = content[i];
-    result += char;
-    if (!started) {
-      if (!/\s/.test(char)) {
-        started = true;
-        if (char === "{" || char === "[") {
-          depth = 1;
-        }
-      }
-      continue;
-    }
-    if (escaped) {
-      escaped = false;
-      continue;
-    }
-    if (char === "\\") {
-      escaped = true;
-      continue;
-    }
-    if (char === '"') {
-      inString = !inString;
-      continue;
-    }
-    if (!inString) {
-      if (char === "{" || char === "[") {
-        depth += 1;
-      } else if (char === "}" || char === "]") {
-        if (depth > 0) {
-          depth -= 1;
-          if (depth === 0) {
-            break;
-          }
-        }
-      }
-    }
-  }
-  return result;
-}
 var BrowserAIChatLanguageModel = class {
   constructor(modelId, options = {}) {
     this.specificationVersion = "v3";
@@ -1177,6 +1219,20 @@ var BrowserAIChatLanguageModel = class {
       warnings
     };
   }
+  /**
+   * Gets the input usage for the current session, if available
+   * @returns The input usage or undefined if not available
+   */
+  getInputUsage() {
+    return this.sessionManager.getInputUsage();
+  }
+  /**
+   * Gets the input quota for the current session, if available
+   * @returns The input quota or undefined if not available
+   */
+  getInputQuota() {
+    return this.sessionManager.getInputQuota();
+  }
   /**
    * Check the availability of the browser AI model
    * @returns Promise resolving to "unavailable", "available", or "available-after-download"
@@ -1328,7 +1384,7 @@ var BrowserAIChatLanguageModel = class {
             let currentToolCallId = null;
             let toolInputStartEmitted = false;
             let accumulatedFenceContent = "";
-            let streamedArgumentsLength = 0;
+            let argumentsStreamState = createArgumentsStreamState();
             let insideFence = false;
             while (!aborted) {
               const { done, value } = await currentReader.read();
@@ -1349,7 +1405,7 @@ var BrowserAIChatLanguageModel = class {
                   currentToolCallId = `call_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;
                   toolInputStartEmitted = false;
                   accumulatedFenceContent = "";
-                  streamedArgumentsLength = 0;
+                  argumentsStreamState = createArgumentsStreamState();
                   insideFence = true;
                   continue;
                 }
@@ -1359,19 +1415,16 @@ var BrowserAIChatLanguageModel = class {
                     accumulatedFenceContent += result.safeContent;
                   }
                   if (toolInputStartEmitted && currentToolCallId) {
-                    const argsContent = extractArgumentsContent(
-                      accumulatedFenceContent
+                    const delta = extractArgumentsDelta(
+                      accumulatedFenceContent,
+                      argumentsStreamState
                     );
-                    if (argsContent.length > streamedArgumentsLength) {
-                      const delta = argsContent.slice(streamedArgumentsLength);
-                      streamedArgumentsLength = argsContent.length;
-                      if (delta.length > 0) {
-                        controller.enqueue({
-                          type: "tool-input-delta",
-                          id: currentToolCallId,
-                          delta
-                        });
-                      }
+                    if (delta.length > 0) {
+                      controller.enqueue({
+                        type: "tool-input-delta",
+                        id: currentToolCallId,
+                        delta
+                      });
                     }
                   }
                   const parsed = parseJsonFunctionCalls(result.completeFence);
@@ -1387,7 +1440,7 @@ var BrowserAIChatLanguageModel = class {
                     currentToolCallId = null;
                     toolInputStartEmitted = false;
                     accumulatedFenceContent = "";
-                    streamedArgumentsLength = 0;
+                    argumentsStreamState = createArgumentsStreamState();
                     insideFence = false;
                     continue;
                   }
@@ -1409,21 +1462,16 @@ var BrowserAIChatLanguageModel = class {
                         });
                         toolInputStartEmitted = true;
                       }
-                      const argsContent = extractArgumentsContent(
-                        accumulatedFenceContent
+                      const delta = extractArgumentsDelta(
+                        accumulatedFenceContent,
+                        argumentsStreamState
                       );
-                      if (argsContent.length > streamedArgumentsLength) {
-                        const delta = argsContent.slice(
-                          streamedArgumentsLength
-                        );
-                        streamedArgumentsLength = argsContent.length;
-                        if (delta.length > 0) {
-                          controller.enqueue({
-                            type: "tool-input-delta",
-                            id: toolCallId,
-                            delta
-                          });
-                        }
+                      if (delta.length > 0) {
+                        controller.enqueue({
+                          type: "tool-input-delta",
+                          id: toolCallId,
+                          delta
+                        });
                       }
                     } else {
                       controller.enqueue({
@@ -1460,7 +1508,7 @@ var BrowserAIChatLanguageModel = class {
                   currentToolCallId = null;
                   toolInputStartEmitted = false;
                   accumulatedFenceContent = "";
-                  streamedArgumentsLength = 0;
+                  argumentsStreamState = createArgumentsStreamState();
                   insideFence = false;
                   continue;
                 }
@@ -1478,21 +1526,16 @@ var BrowserAIChatLanguageModel = class {
                       toolInputStartEmitted = true;
                     }
                     if (toolInputStartEmitted && currentToolCallId) {
-                      const argsContent = extractArgumentsContent(
-                        accumulatedFenceContent
+                      const delta = extractArgumentsDelta(
+                        accumulatedFenceContent,
+                        argumentsStreamState
                       );
-                      if (argsContent.length > streamedArgumentsLength) {
-                        const delta = argsContent.slice(
-                          streamedArgumentsLength
-                        );
-                        streamedArgumentsLength = argsContent.length;
-                        if (delta.length > 0) {
-                          controller.enqueue({
-                            type: "tool-input-delta",
-                            id: currentToolCallId,
-                            delta
-                          });
-                        }
+                      if (delta.length > 0) {
+                        controller.enqueue({
+                          type: "tool-input-delta",
+                          id: currentToolCallId,
+                          delta
+                        });
                       }
                     }
                   }