assistant-stream 0.1.2 → 0.1.3

package/dist/ai-sdk.d.mts

@@ -1,5 +1,6 @@
 import { TextStreamPart, Tool, ObjectStreamPart, LanguageModelV1StreamPart } from 'ai';
-import { A as AssistantStream, a as AssistantStreamChunk, b as AssistantStreamController } from './assistant-stream-54RSz6y3.mjs';
+import { A as AssistantStream, a as AssistantStreamChunk, b as AssistantStreamController } from './assistant-stream-ujRn7KXt.mjs';
+import './json-value-DvDjYV7y.mjs';
 
 declare const fromStreamText: (stream: ReadableStream<TextStreamPart<Record<string, Tool>>>) => AssistantStream;
 declare const fromStreamObject: (stream: ReadableStream<ObjectStreamPart<unknown>>, toolName: string) => AssistantStream;

package/dist/ai-sdk.d.ts

@@ -1,5 +1,6 @@
 import { TextStreamPart, Tool, ObjectStreamPart, LanguageModelV1StreamPart } from 'ai';
-import { A as AssistantStream, a as AssistantStreamChunk, b as AssistantStreamController } from './assistant-stream-54RSz6y3.js';
+import { A as AssistantStream, a as AssistantStreamChunk, b as AssistantStreamController } from './assistant-stream-CgeUyrON.js';
+import './json-value-DvDjYV7y.js';
 
 declare const fromStreamText: (stream: ReadableStream<TextStreamPart<Record<string, Tool>>>) => AssistantStream;
 declare const fromStreamObject: (stream: ReadableStream<ObjectStreamPart<unknown>>, toolName: string) => AssistantStream;

package/dist/{assistant-stream-54RSz6y3.d.mts → assistant-stream-CgeUyrON.d.ts}

@@ -1,8 +1,4 @@
-type ReadonlyJSONValue = null | string | number | boolean | ReadonlyJSONObject | ReadonlyJSONArray;
-type ReadonlyJSONObject = {
-    readonly [key: string]: ReadonlyJSONValue;
-};
-type ReadonlyJSONArray = readonly ReadonlyJSONValue[];
+import { R as ReadonlyJSONValue, a as ReadonlyJSONObject } from './json-value-DvDjYV7y.js';
 
 type PartInit = {
     readonly type: "text" | "reasoning";
@@ -226,4 +222,4 @@ type AssistantStreamController = {
 declare function createAssistantStream(callback: (controller: AssistantStreamController) => PromiseLike<void> | void): AssistantStream;
 declare function createAssistantStreamResponse(callback: (controller: AssistantStreamController) => PromiseLike<void> | void): Response;
 
-export { AssistantStream as A, type ReadonlyJSONValue as R, ToolResponse as T, type AssistantStreamChunk as a, type AssistantStreamController as b, type AssistantMessage as c, type AssistantStreamEncoder as d, createAssistantStream as e, createAssistantStreamResponse as f };
+export { AssistantStream as A, ToolResponse as T, type AssistantStreamChunk as a, type AssistantStreamController as b, type AssistantMessage as c, type AssistantStreamEncoder as d, createAssistantStream as e, createAssistantStreamResponse as f };

package/dist/{assistant-stream-54RSz6y3.d.ts → assistant-stream-ujRn7KXt.d.mts}

@@ -1,8 +1,4 @@
-type ReadonlyJSONValue = null | string | number | boolean | ReadonlyJSONObject | ReadonlyJSONArray;
-type ReadonlyJSONObject = {
-    readonly [key: string]: ReadonlyJSONValue;
-};
-type ReadonlyJSONArray = readonly ReadonlyJSONValue[];
+import { R as ReadonlyJSONValue, a as ReadonlyJSONObject } from './json-value-DvDjYV7y.mjs';
 
 type PartInit = {
     readonly type: "text" | "reasoning";
@@ -226,4 +222,4 @@ type AssistantStreamController = {
 declare function createAssistantStream(callback: (controller: AssistantStreamController) => PromiseLike<void> | void): AssistantStream;
 declare function createAssistantStreamResponse(callback: (controller: AssistantStreamController) => PromiseLike<void> | void): Response;
 
-export { AssistantStream as A, type ReadonlyJSONValue as R, ToolResponse as T, type AssistantStreamChunk as a, type AssistantStreamController as b, type AssistantMessage as c, type AssistantStreamEncoder as d, createAssistantStream as e, createAssistantStreamResponse as f };
+export { AssistantStream as A, ToolResponse as T, type AssistantStreamChunk as a, type AssistantStreamController as b, type AssistantMessage as c, type AssistantStreamEncoder as d, createAssistantStream as e, createAssistantStreamResponse as f };

package/dist/{chunk-C7JS3OBQ.mjs → chunk-LHNPFNMI.mjs}

@@ -1,4 +1,4 @@
-// src/core/utils/json/parse-partial-json.ts
+// src/core/utils/json/parse-partial-json-object.ts
 import sjson from "secure-json-parse";
 
 // src/core/utils/json/fix-json.ts
@@ -6,6 +6,14 @@ function fixJson(input) {
   const stack = ["ROOT"];
   let lastValidIndex = -1;
   let literalStart = null;
+  const path = [];
+  let currentKey;
+  function pushCurrentKeyToPath() {
+    if (currentKey !== void 0) {
+      path.push(JSON.parse('"' + currentKey + '"'));
+      currentKey = void 0;
+    }
+  }
   function processValueStart(char, i, swapState) {
     {
       switch (char) {
@@ -14,6 +22,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_STRING");
+          pushCurrentKeyToPath();
           break;
         }
         case "f":
@@ -30,6 +39,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_NUMBER");
+          pushCurrentKeyToPath();
           break;
         }
         case "0":
@@ -46,6 +56,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_NUMBER");
+          pushCurrentKeyToPath();
           break;
         }
         case "{": {
@@ -53,6 +64,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_OBJECT_START");
+          pushCurrentKeyToPath();
           break;
         }
         case "[": {
@@ -60,6 +72,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_ARRAY_START");
+          pushCurrentKeyToPath();
           break;
         }
       }
@@ -75,6 +88,7 @@ function fixJson(input) {
       case "}": {
         lastValidIndex = i;
         stack.pop();
+        currentKey = path.pop();
         break;
       }
     }
@@ -84,11 +98,13 @@ function fixJson(input) {
       case ",": {
         stack.pop();
         stack.push("INSIDE_ARRAY_AFTER_COMMA");
+        currentKey = (Number(currentKey) + 1).toString();
         break;
       }
       case "]": {
         lastValidIndex = i;
         stack.pop();
+        currentKey = path.pop();
         break;
       }
     }
@@ -105,11 +121,13 @@ function fixJson(input) {
           case '"': {
             stack.pop();
             stack.push("INSIDE_OBJECT_KEY");
+            currentKey = "";
             break;
           }
           case "}": {
             lastValidIndex = i;
             stack.pop();
+            currentKey = path.pop();
             break;
           }
         }
@@ -120,6 +138,7 @@ function fixJson(input) {
           case '"': {
             stack.pop();
             stack.push("INSIDE_OBJECT_KEY");
+            currentKey = "";
             break;
           }
         }
@@ -132,6 +151,15 @@ function fixJson(input) {
             stack.push("INSIDE_OBJECT_AFTER_KEY");
             break;
           }
+          case "\\": {
+            stack.push("INSIDE_STRING_ESCAPE");
+            currentKey += char;
+            break;
+          }
+          default: {
+            currentKey += char;
+            break;
+          }
         }
         break;
       }
@@ -158,6 +186,7 @@ function fixJson(input) {
           case '"': {
             stack.pop();
             lastValidIndex = i;
+            currentKey = path.pop();
             break;
           }
           case "\\": {
@@ -175,10 +204,12 @@ function fixJson(input) {
           case "]": {
             lastValidIndex = i;
             stack.pop();
+            currentKey = path.pop();
             break;
           }
           default: {
             lastValidIndex = i;
+            currentKey = "0";
             processValueStart(char, i, "INSIDE_ARRAY_AFTER_VALUE");
             break;
           }
@@ -190,11 +221,13 @@ function fixJson(input) {
           case ",": {
             stack.pop();
             stack.push("INSIDE_ARRAY_AFTER_COMMA");
+            currentKey = (Number(currentKey) + 1).toString();
             break;
           }
           case "]": {
             lastValidIndex = i;
             stack.pop();
+            currentKey = path.pop();
             break;
           }
           default: {
@@ -210,7 +243,11 @@ function fixJson(input) {
       }
       case "INSIDE_STRING_ESCAPE": {
         stack.pop();
-        lastValidIndex = i;
+        if (stack[stack.length - 1] === "INSIDE_STRING") {
+          lastValidIndex = i;
+        } else if (stack[stack.length - 1] === "INSIDE_OBJECT_KEY") {
+          currentKey += char;
+        }
         break;
       }
       case "INSIDE_NUMBER": {
@@ -236,6 +273,7 @@ function fixJson(input) {
           }
           case ",": {
             stack.pop();
+            currentKey = path.pop();
             if (stack[stack.length - 1] === "INSIDE_ARRAY_AFTER_VALUE") {
               processAfterArrayValue(char, i);
             }
@@ -246,6 +284,7 @@ function fixJson(input) {
           }
           case "}": {
             stack.pop();
+            currentKey = path.pop();
             if (stack[stack.length - 1] === "INSIDE_OBJECT_AFTER_VALUE") {
               processAfterObjectValue(char, i);
             }
@@ -253,6 +292,7 @@ function fixJson(input) {
           }
           case "]": {
             stack.pop();
+            currentKey = path.pop();
             if (stack[stack.length - 1] === "INSIDE_ARRAY_AFTER_VALUE") {
               processAfterArrayValue(char, i);
             }
@@ -260,6 +300,7 @@ function fixJson(input) {
           }
           default: {
             stack.pop();
+            currentKey = path.pop();
             break;
           }
         }
@@ -316,23 +357,69 @@ function fixJson(input) {
       }
     }
   }
-  return result;
+  return [result, path];
 }
 
-// src/core/utils/json/parse-partial-json.ts
-var parsePartialJson = (json) => {
+// src/core/utils/json/parse-partial-json-object.ts
+var PARTIAL_JSON_OBJECT_META_SYMBOL = Symbol(
+  "aui.parse-partial-json-object.meta"
+);
+var getPartialJsonObjectMeta = (obj) => {
+  return obj?.[PARTIAL_JSON_OBJECT_META_SYMBOL];
+};
+var parsePartialJsonObject = (json) => {
+  if (json.length === 0)
+    return {
+      [PARTIAL_JSON_OBJECT_META_SYMBOL]: { state: "partial", partialPath: [] }
+    };
   try {
-    return sjson.parse(json);
+    const res = sjson.parse(json);
+    if (typeof res !== "object" || res === null)
+      throw new Error("argsText is expected to be an object");
+    res[PARTIAL_JSON_OBJECT_META_SYMBOL] = {
+      state: "complete",
+      partialPath: []
+    };
+    return res;
   } catch {
     try {
-      return sjson.parse(fixJson(json));
+      const [fixedJson, partialPath] = fixJson(json);
+      const res = sjson.parse(fixedJson);
+      if (typeof res !== "object" || res === null)
+        throw new Error("argsText is expected to be an object");
+      res[PARTIAL_JSON_OBJECT_META_SYMBOL] = {
+        state: "partial",
+        partialPath
+      };
+      return res;
     } catch {
       return void 0;
     }
   }
 };
+var getFieldState = (parent, parentMeta, fieldPath) => {
+  if (typeof parent !== "object" || parent === null) return parentMeta.state;
+  if (parentMeta.state === "complete") return "complete";
+  if (fieldPath.length === 0) return parentMeta.state;
+  const [field, ...restPath] = fieldPath;
+  if (!Object.prototype.hasOwnProperty.call(parent, field)) return "partial";
+  const [partialField, ...restPartialPath] = parentMeta.partialPath;
+  if (field !== partialField) return "complete";
+  const child = parent[field];
+  const childMeta = {
+    state: "partial",
+    partialPath: restPartialPath
+  };
+  return getFieldState(child, childMeta, restPath);
+};
+var getPartialJsonObjectFieldState = (args, fieldPath) => {
+  const meta = getPartialJsonObjectMeta(args);
+  if (!meta) throw new Error("unable to determine args state");
+  return getFieldState(args, meta, fieldPath.map(String));
+};
 
 export {
-  parsePartialJson
+  parsePartialJsonObject,
+  getPartialJsonObjectFieldState
 };
-//# sourceMappingURL=chunk-C7JS3OBQ.mjs.map
+//# sourceMappingURL=chunk-LHNPFNMI.mjs.map

package/dist/chunk-LHNPFNMI.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/core/utils/json/parse-partial-json-object.ts","../src/core/utils/json/fix-json.ts"],"sourcesContent":["import sjson from \"secure-json-parse\";\nimport { fixJson } from \"./fix-json\";\nimport { ReadonlyJSONObject } from \"./json-value\";\n\nconst PARTIAL_JSON_OBJECT_META_SYMBOL = Symbol(\n  \"aui.parse-partial-json-object.meta\",\n);\n\ntype FieldState = \"complete\" | \"partial\";\n\ntype PartialJsonObjectMeta = {\n  state: \"complete\" | \"partial\";\n  partialPath: string[];\n};\n\nexport const getPartialJsonObjectMeta = (\n  obj: Record<symbol, unknown>,\n): PartialJsonObjectMeta | undefined => {\n  return obj?.[PARTIAL_JSON_OBJECT_META_SYMBOL] as PartialJsonObjectMeta;\n};\n\nexport const parsePartialJsonObject = (\n  json: string,\n):\n  | (ReadonlyJSONObject & {\n      [PARTIAL_JSON_OBJECT_META_SYMBOL]: PartialJsonObjectMeta;\n    })\n  | undefined => {\n  if (json.length === 0)\n    return {\n      [PARTIAL_JSON_OBJECT_META_SYMBOL]: { state: \"partial\", partialPath: [] },\n    };\n\n  try {\n    const res = sjson.parse(json);\n    if (typeof res !== \"object\" || res === null)\n      throw new Error(\"argsText is expected to be an object\");\n\n    res[PARTIAL_JSON_OBJECT_META_SYMBOL] = {\n      state: \"complete\",\n      partialPath: [],\n    };\n    return res;\n  } catch {\n    try {\n      const [fixedJson, partialPath] = fixJson(json);\n      const res = sjson.parse(fixedJson);\n      if (typeof res !== \"object\" || res === null)\n        throw new Error(\"argsText is expected to be an object\");\n\n      res[PARTIAL_JSON_OBJECT_META_SYMBOL] = {\n        state: \"partial\",\n        partialPath,\n      };\n      return res;\n    } catch {\n      return undefined;\n    }\n  }\n};\n\nconst getFieldState = (\n  parent: unknown,\n  parentMeta: PartialJsonObjectMeta,\n  fieldPath: string[],\n): FieldState => {\n  if (typeof parent !== \"object\" || parent === null) return parentMeta.state;\n\n  // 1) parent is complete: return \"complete\"\n  if (parentMeta.state === \"complete\") return \"complete\";\n\n  // 2) we finished traversing: return parent state\n  if (fieldPath.length === 0) return parentMeta.state;\n\n  const [field, ...restPath] = fieldPath as [string, ...string[]];\n\n  // 3) field doesn't yet exist in parent: return \"partial\"\n  if (!Object.prototype.hasOwnProperty.call(parent, field)) return \"partial\";\n\n  const [partialField, ...restPartialPath] = parentMeta.partialPath;\n\n  // 4) field exists but is not partial: return \"complete\"\n  if (field !== partialField) return \"complete\";\n\n  // 5) field exists and is partial: return child state\n  const child = (parent as Record<string, unknown>)[field];\n  const childMeta: PartialJsonObjectMeta = {\n    state: \"partial\",\n    partialPath: restPartialPath,\n  };\n\n  return getFieldState(child, childMeta, restPath);\n};\n\nexport const getPartialJsonObjectFieldState = (\n  args: Record<string, unknown>,\n  fieldPath: (string | number)[],\n): FieldState => {\n  const meta = getPartialJsonObjectMeta(args);\n  if (!meta) throw new Error(\"unable to determine args state\");\n\n  return getFieldState(args, meta, fieldPath.map(String));\n};\n","// LICENSE for this file only\n\n// MIT License\n\n// Copyright (c) 2025 AgentbaseAI Inc.\n// Copyright (c) 2023 Lars Grammel\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\ntype State =\n  | \"ROOT\"\n  | \"FINISH\"\n  | \"INSIDE_STRING\"\n  | \"INSIDE_STRING_ESCAPE\"\n  | \"INSIDE_LITERAL\"\n  | \"INSIDE_NUMBER\"\n  | \"INSIDE_OBJECT_START\"\n  | \"INSIDE_OBJECT_KEY\"\n  | \"INSIDE_OBJECT_AFTER_KEY\"\n  | \"INSIDE_OBJECT_BEFORE_VALUE\"\n  | \"INSIDE_OBJECT_AFTER_VALUE\"\n  | \"INSIDE_OBJECT_AFTER_COMMA\"\n  | \"INSIDE_ARRAY_START\"\n  | \"INSIDE_ARRAY_AFTER_VALUE\"\n  | \"INSIDE_ARRAY_AFTER_COMMA\";\n\n// Implemented as a scanner with additional fixing\n// that performs a single linear time scan pass over the partial JSON.\n//\n// The states should ideally match relevant states from the JSON spec:\n// https://www.json.org/json-en.html\n//\n// Please note that invalid JSON is not considered/covered, because it\n// is assumed that the resulting JSON will be processed by a standard\n// JSON parser that will detect any invalid JSON.\n\n// Returns a tuple of [fixedJson, partialPath]\n// partialPath is an array of object/array keys that represent\n// the currently partial values. An object is considered partial\n// if through appending extra characters to the JSON string, its\n// value could change.\n\n// Example input: '{\"foo\":[{\"a\":f'\n// Example output: ['{\"foo\":[{\"a\":false}]}', ['foo', '0']]\n// Example input: '{\"foo\":'\n// Example output: ['{}', []]\n\nexport function fixJson(input: string): [string, string[]] {\n  const stack: State[] = [\"ROOT\"];\n  let lastValidIndex = -1;\n  let literalStart: number | null = null;\n  const path: string[] = [];\n  let currentKey: string | undefined;\n\n  function pushCurrentKeyToPath(): void {\n    if (currentKey !== undefined) {\n      path.push(JSON.parse('\"' + currentKey + '\"'));\n      currentKey = undefined;\n    }\n  }\n\n  function processValueStart(char: string, i: number, swapState: State) {\n    {\n      switch (char) {\n        case '\"': {\n          lastValidIndex = i;\n          stack.pop();\n          stack.push(swapState);\n          stack.push(\"INSIDE_STRING\");\n\n          pushCurrentKeyToPath();\n          break;\n        }\n\n        case \"f\":\n        case \"t\":\n        case \"n\": {\n          lastValidIndex = i;\n          literalStart = i;\n          stack.pop();\n          stack.push(swapState);\n          stack.push(\"INSIDE_LITERAL\");\n          break;\n        }\n\n        case \"-\": {\n          stack.pop();\n          stack.push(swapState);\n          stack.push(\"INSIDE_NUMBER\");\n\n          pushCurrentKeyToPath();\n          break;\n        }\n        case \"0\":\n        case \"1\":\n        case \"2\":\n        case \"3\":\n        case \"4\":\n        case \"5\":\n        case \"6\":\n        case \"7\":\n        case \"8\":\n        case \"9\": {\n          lastValidIndex = i;\n          stack.pop();\n          stack.push(swapState);\n          stack.push(\"INSIDE_NUMBER\");\n\n          pushCurrentKeyToPath();\n          break;\n        }\n\n        case \"{\": {\n          lastValidIndex = i;\n          stack.pop();\n          stack.push(swapState);\n          stack.push(\"INSIDE_OBJECT_START\");\n\n          pushCurrentKeyToPath();\n          break;\n        }\n\n        case \"[\": {\n          lastValidIndex = i;\n          stack.pop();\n          stack.push(swapState);\n          stack.push(\"INSIDE_ARRAY_START\");\n\n          pushCurrentKeyToPath();\n          break;\n        }\n      }\n    }\n  }\n\n  function processAfterObjectValue(char: string, i: number) {\n    switch (char) {\n      case \",\": {\n        stack.pop();\n        stack.push(\"INSIDE_OBJECT_AFTER_COMMA\");\n        break;\n      }\n      case \"}\": {\n        lastValidIndex = i;\n        stack.pop();\n        currentKey = path.pop();\n        break;\n      }\n    }\n  }\n\n  function processAfterArrayValue(char: string, i: number) {\n    switch (char) {\n      case \",\": {\n        stack.pop();\n        stack.push(\"INSIDE_ARRAY_AFTER_COMMA\");\n        currentKey = (Number(currentKey) + 1).toString();\n        break;\n      }\n      case \"]\": {\n        lastValidIndex = i;\n        stack.pop();\n        currentKey = path.pop();\n        break;\n      }\n    }\n  }\n\n  for (let i = 0; i < input.length; i++) {\n    const char = input[i]!;\n    const currentState = stack[stack.length - 1];\n\n    switch (currentState) {\n      case \"ROOT\":\n        processValueStart(char, i, \"FINISH\");\n        break;\n\n      case \"INSIDE_OBJECT_START\": {\n        switch (char) {\n          case '\"': {\n            stack.pop();\n            stack.push(\"INSIDE_OBJECT_KEY\");\n            currentKey = \"\";\n            break;\n          }\n          case \"}\": {\n            lastValidIndex = i;\n            stack.pop();\n            currentKey = path.pop();\n            break;\n          }\n        }\n        break;\n      }\n\n      case \"INSIDE_OBJECT_AFTER_COMMA\": {\n        switch (char) {\n          case '\"': {\n            stack.pop();\n            stack.push(\"INSIDE_OBJECT_KEY\");\n            currentKey = \"\";\n            break;\n          }\n        }\n        break;\n      }\n\n      case \"INSIDE_OBJECT_KEY\": {\n        switch (char) {\n          case '\"': {\n            stack.pop();\n            stack.push(\"INSIDE_OBJECT_AFTER_KEY\");\n            break;\n          }\n          case \"\\\\\": {\n            stack.push(\"INSIDE_STRING_ESCAPE\");\n            currentKey += char;\n            break;\n          }\n          default: {\n            currentKey += char;\n            break;\n          }\n        }\n        break;\n      }\n\n      case \"INSIDE_OBJECT_AFTER_KEY\": {\n        switch (char) {\n          case \":\": {\n            stack.pop();\n            stack.push(\"INSIDE_OBJECT_BEFORE_VALUE\");\n\n            break;\n          }\n        }\n        break;\n      }\n\n      case \"INSIDE_OBJECT_BEFORE_VALUE\": {\n        processValueStart(char, i, \"INSIDE_OBJECT_AFTER_VALUE\");\n        break;\n      }\n\n      case \"INSIDE_OBJECT_AFTER_VALUE\": {\n        processAfterObjectValue(char, i);\n        break;\n      }\n\n      case \"INSIDE_STRING\": {\n        switch (char) {\n          case '\"': {\n            stack.pop();\n            lastValidIndex = i;\n\n            currentKey = path.pop();\n            break;\n          }\n\n          case \"\\\\\": {\n            stack.push(\"INSIDE_STRING_ESCAPE\");\n            break;\n          }\n\n          default: {\n            lastValidIndex = i;\n          }\n        }\n\n        break;\n      }\n\n      case \"INSIDE_ARRAY_START\": {\n        switch (char) {\n          case \"]\": {\n            lastValidIndex = i;\n            stack.pop();\n            currentKey = path.pop();\n            break;\n          }\n\n          default: {\n            lastValidIndex = i;\n            currentKey = \"0\";\n            processValueStart(char, i, \"INSIDE_ARRAY_AFTER_VALUE\");\n            break;\n          }\n        }\n        break;\n      }\n\n      case \"INSIDE_ARRAY_AFTER_VALUE\": {\n        switch (char) {\n          case \",\": {\n            stack.pop();\n            stack.push(\"INSIDE_ARRAY_AFTER_COMMA\");\n\n            currentKey = (Number(currentKey) + 1).toString();\n            break;\n          }\n\n          case \"]\": {\n            lastValidIndex = i;\n            stack.pop();\n            currentKey = path.pop();\n            break;\n          }\n\n          default: {\n            lastValidIndex = i;\n            break;\n          }\n        }\n\n        break;\n      }\n\n      case \"INSIDE_ARRAY_AFTER_COMMA\": {\n        processValueStart(char, i, \"INSIDE_ARRAY_AFTER_VALUE\");\n        break;\n      }\n\n      case \"INSIDE_STRING_ESCAPE\": {\n        stack.pop();\n\n        if (stack[stack.length - 1] === \"INSIDE_STRING\") {\n          lastValidIndex = i;\n        } else if (stack[stack.length - 1] === \"INSIDE_OBJECT_KEY\") {\n          currentKey += char;\n        }\n\n        break;\n      }\n\n      case \"INSIDE_NUMBER\": {\n        switch (char) {\n          case \"0\":\n          case \"1\":\n          case \"2\":\n          case \"3\":\n          case \"4\":\n          case \"5\":\n          case \"6\":\n          case \"7\":\n          case \"8\":\n          case \"9\": {\n            lastValidIndex = i;\n            break;\n          }\n\n          case \"e\":\n          case \"E\":\n          case \"-\":\n          case \".\": {\n            break;\n          }\n\n          case \",\": {\n            stack.pop();\n            currentKey = path.pop();\n\n            if (stack[stack.length - 1] === \"INSIDE_ARRAY_AFTER_VALUE\") {\n              processAfterArrayValue(char, i);\n            }\n\n            if (stack[stack.length - 1] === \"INSIDE_OBJECT_AFTER_VALUE\") {\n              processAfterObjectValue(char, i);\n            }\n\n            break;\n          }\n\n          case \"}\": {\n            stack.pop();\n            currentKey = path.pop();\n\n            if (stack[stack.length - 1] === \"INSIDE_OBJECT_AFTER_VALUE\") {\n              processAfterObjectValue(char, i);\n            }\n\n            break;\n          }\n\n          case \"]\": {\n            stack.pop();\n            currentKey = path.pop();\n\n            if (stack[stack.length - 1] === \"INSIDE_ARRAY_AFTER_VALUE\") {\n              processAfterArrayValue(char, i);\n            }\n\n            break;\n          }\n\n          default: {\n            stack.pop();\n            currentKey = path.pop();\n            break;\n          }\n        }\n\n        break;\n      }\n\n      case \"INSIDE_LITERAL\": {\n        const partialLiteral = input.substring(literalStart!, i + 1);\n\n        if (\n          !\"false\".startsWith(partialLiteral) &&\n          !\"true\".startsWith(partialLiteral) &&\n          !\"null\".startsWith(partialLiteral)\n        ) {\n          stack.pop();\n\n          if (stack[stack.length - 1] === \"INSIDE_OBJECT_AFTER_VALUE\") {\n            processAfterObjectValue(char, i);\n          } else if (stack[stack.length - 1] === \"INSIDE_ARRAY_AFTER_VALUE\") {\n            processAfterArrayValue(char, i);\n          }\n        } else {\n          lastValidIndex = i;\n        }\n\n        break;\n      }\n    }\n  }\n\n  let result = input.slice(0, lastValidIndex + 1);\n\n  for (let i = stack.length - 1; i >= 0; i--) {\n    const state = stack[i];\n\n    switch (state) {\n      case \"INSIDE_STRING\": {\n        result += '\"';\n        break;\n      }\n\n      case \"INSIDE_OBJECT_KEY\":\n      case \"INSIDE_OBJECT_AFTER_KEY\":\n      case \"INSIDE_OBJECT_AFTER_COMMA\":\n      case \"INSIDE_OBJECT_START\":\n      case \"INSIDE_OBJECT_BEFORE_VALUE\":\n      case \"INSIDE_OBJECT_AFTER_VALUE\": {\n        result += \"}\";\n        break;\n      }\n\n      case \"INSIDE_ARRAY_START\":\n      case \"INSIDE_ARRAY_AFTER_COMMA\":\n      case \"INSIDE_ARRAY_AFTER_VALUE\": {\n        result += \"]\";\n        break;\n      }\n\n      case \"INSIDE_LITERAL\": {\n        const partialLiteral = input.substring(literalStart!, input.length);\n\n        if (\"true\".startsWith(partialLiteral)) {\n          result += \"true\".slice(partialLiteral.length);\n        } else if (\"false\".startsWith(partialLiteral)) {\n          result += \"false\".slice(partialLiteral.length);\n        } else if (\"null\".startsWith(partialLiteral)) {\n          result += \"null\".slice(partialLiteral.length);\n        }\n      }\n    }\n  }\n\n  return [result, path];\n}\n"],"mappings":";AAAA,OAAO,WAAW;;;AC+DX,SAAS,QAAQ,OAAmC;AACzD,QAAM,QAAiB,CAAC,MAAM;AAC9B,MAAI,iBAAiB;AACrB,MAAI,eAA8B;AAClC,QAAM,OAAiB,CAAC;AACxB,MAAI;AAEJ,WAAS,uBAA6B;AACpC,QAAI,eAAe,QAAW;AAC5B,WAAK,KAAK,KAAK,MAAM,MAAM,aAAa,GAAG,CAAC;AAC5C,mBAAa;AAAA,IACf;AAAA,EACF;AAEA,WAAS,kBAAkB,MAAc,GAAW,WAAkB;AACpE;AACE,cAAQ,MAAM;AAAA,QACZ,KAAK,KAAK;AACR,2BAAiB;AACjB,gBAAM,IAAI;AACV,gBAAM,KAAK,SAAS;AACpB,gBAAM,KAAK,eAAe;AAE1B,+BAAqB;AACrB;AAAA,QACF;AAAA,QAEA,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK,KAAK;AACR,2BAAiB;AACjB,yBAAe;AACf,gBAAM,IAAI;AACV,gBAAM,KAAK,SAAS;AACpB,gBAAM,KAAK,gBAAgB;AAC3B;AAAA,QACF;AAAA,QAEA,KAAK,KAAK;AACR,gBAAM,IAAI;AACV,gBAAM,KAAK,SAAS;AACpB,gBAAM,KAAK,eAAe;AAE1B,+BAAqB;AACrB;AAAA,QACF;AAAA,QACA,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK,KAAK;AACR,2BAAiB;AACjB,gBAAM,IAAI;AACV,gBAAM,KAAK,SAAS;AACpB,gBAAM,KAAK,eAAe;AAE1B,+BAAqB;AACrB;AAAA,QACF;AAAA,QAEA,KAAK,KAAK;AACR,2BAAiB;AACjB,gBAAM,IAAI;AACV,gBAAM,KAAK,SAAS;AACpB,gBAAM,KAAK,qBAAqB;AAEhC,+BAAqB;AACrB;AAAA,QACF;AAAA,QAEA,KAAK,KAAK;AACR,2BAAiB;AACjB,gBAAM,IAAI;AACV,gBAAM,KAAK,SAAS;AACpB,gBAAM,KAAK,oBAAoB;AAE/B,+BAAqB;AACrB;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,WAAS,wBAAwB,MAAc,GAAW;AACxD,YAAQ,MAAM;AAAA,MACZ,KAAK,KAAK;AACR,cAAM,IAAI;AACV,cAAM,KAAK,2BAA2B;AACtC;AAAA,MACF;AAAA,MACA,KAAK,KAAK;AACR,yBAAiB;AACjB,cAAM,IAAI;AACV,qBAAa,KAAK,IAAI;AACtB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,WAAS,uBAAuB,MAAc,GAAW;AACvD,YAAQ,MAAM;AAAA,MACZ,KAAK,KAAK;AACR,cAAM,IAAI;AACV,cAAM,KAAK,0BAA0B;AACrC,sBAAc,OAAO,UAAU,IAAI,GAAG,SAAS;AAC/C;AAAA,MACF;AAAA,MACA,KAAK,KAAK;AACR,yBAAiB;AACjB,cAAM,IAAI;AACV,qBAAa,KAAK,IAAI;AACtB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,WAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACrC,UAAM,OAAO,MAAM,CAAC;AACpB,UAAM,eAAe,MAAM,MAAM,SAAS,CAAC;AAE3C,YAAQ,cAAc;AAAA,MACpB,KAAK;AACH,0BAAkB,MAAM,GAAG,QAAQ;AACnC;AAAA,MAEF,KAAK,uBAAuB;AAC1B,gBAAQ,MAAM;AAAA,UACZ,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,kBAAM,KAAK,mBAAmB;AAC9B,yBAAa;AACb;AAAA,UACF;AAAA,UACA,KAAK,KAAK;AACR,6BAAiB;AACjB,kBAAM,IAAI;AACV,yBAAa,KAAK,IAAI;AACtB;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF;AAAA,MAEA,KAAK,6BAA6B;AAChC,gBAAQ,MAAM;AAAA,UACZ,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,kBAAM,KAAK,mBAAmB;AAC9B,yBAAa;AACb;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF;AAAA,MAEA,KAAK,qBAAqB;AACxB,gBAAQ,MAAM;AAAA,UACZ,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,kBAAM,KAAK,yBAAyB;AACpC;AAAA,UACF;AAAA,UACA,KAAK,MAAM;AACT,kBAAM,KAAK,sBAAsB;AACjC,0BAAc;AACd;AAAA,UACF;AAAA,UACA,SAAS;AACP,0BAAc;AACd;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF;AAAA,MAEA,KAAK,2BAA2B;AAC9B,gBAAQ,MAAM;AAAA,UACZ,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,kBAAM,KAAK,4BAA4B;AAEvC;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF;AAAA,MAEA,KAAK,8BAA8B;AACjC,0BAAkB,MAAM,GAAG,2BAA2B;AACtD;AAAA,MACF;AAAA,MAEA,KAAK,6BAA6B;AAChC,gCAAwB,MAAM,CAAC;AAC/B;AAAA,MACF;AAAA,MAEA,KAAK,iBAAiB;AACpB,gBAAQ,MAAM;AAAA,UACZ,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,6BAAiB;AAEjB,yBAAa,KAAK,IAAI;AACtB;AAAA,UACF;AAAA,UAEA,KAAK,MAAM;AACT,kBAAM,KAAK,sBAAsB;AACjC;AAAA,UACF;AAAA,UAEA,SAAS;AACP,6BAAiB;AAAA,UACnB;AAAA,QACF;AAEA;AAAA,MACF;AAAA,MAEA,KAAK,sBAAsB;AACzB,gBAAQ,MAAM;AAAA,UACZ,KAAK,KAAK;AACR,6BAAiB;AACjB,kBAAM,IAAI;AACV,yBAAa,KAAK,IAAI;AACtB;AAAA,UACF;AAAA,UAEA,SAAS;AACP,6BAAiB;AACjB,yBAAa;AACb,8BAAkB,MAAM,GAAG,0BAA0B;AACrD;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF;AAAA,MAEA,KAAK,4BAA4B;AAC/B,gBAAQ,MAAM;AAAA,UACZ,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,kBAAM,KAAK,0BAA0B;AAErC,0BAAc,OAAO,UAAU,IAAI,GAAG,SAAS;AAC/C;AAAA,UACF;AAAA,UAEA,KAAK,KAAK;AACR,6BAAiB;AACjB,kBAAM,IAAI;AACV,yBAAa,KAAK,IAAI;AACtB;AAAA,UACF;AAAA,UAEA,SAAS;AACP,6BAAiB;AACjB;AAAA,UACF;AAAA,QACF;AAEA;AAAA,MACF;AAAA,MAEA,KAAK,4BAA4B;AAC/B,0BAAkB,MAAM,GAAG,0BAA0B;AACrD;AAAA,MACF;AAAA,MAEA,KAAK,wBAAwB;AAC3B,cAAM,IAAI;AAEV,YAAI,MAAM,MAAM,SAAS,CAAC,MAAM,iBAAiB;AAC/C,2BAAiB;AAAA,QACnB,WAAW,MAAM,MAAM,SAAS,CAAC,MAAM,qBAAqB;AAC1D,wBAAc;AAAA,QAChB;AAEA;AAAA,MACF;AAAA,MAEA,KAAK,iBAAiB;AACpB,gBAAQ,MAAM;AAAA,UACZ,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK,KAAK;AACR,6BAAiB;AACjB;AAAA,UACF;AAAA,UAEA,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK,KAAK;AACR;AAAA,UACF;AAAA,UAEA,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,yBAAa,KAAK,IAAI;AAEtB,gBAAI,MAAM,MAAM,SAAS,CAAC,MAAM,4BAA4B;AAC1D,qCAAuB,MAAM,CAAC;AAAA,YAChC;AAEA,gBAAI,MAAM,MAAM,SAAS,CAAC,MAAM,6BAA6B;AAC3D,sCAAwB,MAAM,CAAC;AAAA,YACjC;AAEA;AAAA,UACF;AAAA,UAEA,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,yBAAa,KAAK,IAAI;AAEtB,gBAAI,MAAM,MAAM,SAAS,CAAC,MAAM,6BAA6B;AAC3D,sCAAwB,MAAM,CAAC;AAAA,YACjC;AAEA;AAAA,UACF;AAAA,UAEA,KAAK,KAAK;AACR,kBAAM,IAAI;AACV,yBAAa,KAAK,IAAI;AAEtB,gBAAI,MAAM,MAAM,SAAS,CAAC,MAAM,4BAA4B;AAC1D,qCAAuB,MAAM,CAAC;AAAA,YAChC;AAEA;AAAA,UACF;AAAA,UAEA,SAAS;AACP,kBAAM,IAAI;AACV,yBAAa,KAAK,IAAI;AACtB;AAAA,UACF;AAAA,QACF;AAEA;AAAA,MACF;AAAA,MAEA,KAAK,kBAAkB;AACrB,cAAM,iBAAiB,MAAM,UAAU,cAAe,IAAI,CAAC;AAE3D,YACE,CAAC,QAAQ,WAAW,cAAc,KAClC,CAAC,OAAO,WAAW,cAAc,KACjC,CAAC,OAAO,WAAW,cAAc,GACjC;AACA,gBAAM,IAAI;AAEV,cAAI,MAAM,MAAM,SAAS,CAAC,MAAM,6BAA6B;AAC3D,oCAAwB,MAAM,CAAC;AAAA,UACjC,WAAW,MAAM,MAAM,SAAS,CAAC,MAAM,4BAA4B;AACjE,mCAAuB,MAAM,CAAC;AAAA,UAChC;AAAA,QACF,OAAO;AACL,2BAAiB;AAAA,QACnB;AAEA;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,MAAI,SAAS,MAAM,MAAM,GAAG,iBAAiB,CAAC;AAE9C,WAAS,IAAI,MAAM,SAAS,GAAG,KAAK,GAAG,KAAK;AAC1C,UAAM,QAAQ,MAAM,CAAC;AAErB,YAAQ,OAAO;AAAA,MACb,KAAK,iBAAiB;AACpB,kBAAU;AACV;AAAA,MACF;AAAA,MAEA,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK,6BAA6B;AAChC,kBAAU;AACV;AAAA,MACF;AAAA,MAEA,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK,4BAA4B;AAC/B,kBAAU;AACV;AAAA,MACF;AAAA,MAEA,KAAK,kBAAkB;AACrB,cAAM,iBAAiB,MAAM,UAAU,cAAe,MAAM,MAAM;AAElE,YAAI,OAAO,WAAW,cAAc,GAAG;AACrC,oBAAU,OAAO,MAAM,eAAe,MAAM;AAAA,QAC9C,WAAW,QAAQ,WAAW,cAAc,GAAG;AAC7C,oBAAU,QAAQ,MAAM,eAAe,MAAM;AAAA,QAC/C,WAAW,OAAO,WAAW,cAAc,GAAG;AAC5C,oBAAU,OAAO,MAAM,eAAe,MAAM;AAAA,QAC9C;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,SAAO,CAAC,QAAQ,IAAI;AACtB;;;ADneA,IAAM,kCAAkC;AAAA,EACtC;AACF;AASO,IAAM,2BAA2B,CACtC,QACsC;AACtC,SAAO,MAAM,+BAA+B;AAC9C;AAEO,IAAM,yBAAyB,CACpC,SAKe;AACf,MAAI,KAAK,WAAW;AAClB,WAAO;AAAA,MACL,CAAC,+BAA+B,GAAG,EAAE,OAAO,WAAW,aAAa,CAAC,EAAE;AAAA,IACzE;AAEF,MAAI;AACF,UAAM,MAAM,MAAM,MAAM,IAAI;AAC5B,QAAI,OAAO,QAAQ,YAAY,QAAQ;AACrC,YAAM,IAAI,MAAM,sCAAsC;AAExD,QAAI,+BAA+B,IAAI;AAAA,MACrC,OAAO;AAAA,MACP,aAAa,CAAC;AAAA,IAChB;AACA,WAAO;AAAA,EACT,QAAQ;AACN,QAAI;AACF,YAAM,CAAC,WAAW,WAAW,IAAI,QAAQ,IAAI;AAC7C,YAAM,MAAM,MAAM,MAAM,SAAS;AACjC,UAAI,OAAO,QAAQ,YAAY,QAAQ;AACrC,cAAM,IAAI,MAAM,sCAAsC;AAExD,UAAI,+BAA+B,IAAI;AAAA,QACrC,OAAO;AAAA,QACP;AAAA,MACF;AACA,aAAO;AAAA,IACT,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AACF;AAEA,IAAM,gBAAgB,CACpB,QACA,YACA,cACe;AACf,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO,WAAW;AAGrE,MAAI,WAAW,UAAU,WAAY,QAAO;AAG5C,MAAI,UAAU,WAAW,EAAG,QAAO,WAAW;AAE9C,QAAM,CAAC,OAAO,GAAG,QAAQ,IAAI;AAG7B,MAAI,CAAC,OAAO,UAAU,eAAe,KAAK,QAAQ,KAAK,EAAG,QAAO;AAEjE,QAAM,CAAC,cAAc,GAAG,eAAe,IAAI,WAAW;AAGtD,MAAI,UAAU,aAAc,QAAO;AAGnC,QAAM,QAAS,OAAmC,KAAK;AACvD,QAAM,YAAmC;AAAA,IACvC,OAAO;AAAA,IACP,aAAa;AAAA,EACf;AAEA,SAAO,cAAc,OAAO,WAAW,QAAQ;AACjD;AAEO,IAAM,iCAAiC,CAC5C,MACA,cACe;AACf,QAAM,OAAO,yBAAyB,IAAI;AAC1C,MAAI,CAAC,KAAM,OAAM,IAAI,MAAM,gCAAgC;AAE3D,SAAO,cAAc,MAAM,MAAM,UAAU,IAAI,MAAM,CAAC;AACxD;","names":[]}

package/dist/index.d.mts

@@ -1,5 +1,6 @@
-import { a as AssistantStreamChunk, T as ToolResponse, R as ReadonlyJSONValue, c as AssistantMessage, d as AssistantStreamEncoder, A as AssistantStream } from './assistant-stream-54RSz6y3.mjs';
-export { e as createAssistantStream, f as createAssistantStreamResponse } from './assistant-stream-54RSz6y3.mjs';
+import { a as AssistantStreamChunk, T as ToolResponse, c as AssistantMessage, d as AssistantStreamEncoder, A as AssistantStream } from './assistant-stream-ujRn7KXt.mjs';
+export { e as createAssistantStream, f as createAssistantStreamResponse } from './assistant-stream-ujRn7KXt.mjs';
+import { R as ReadonlyJSONValue } from './json-value-DvDjYV7y.mjs';
 
 declare class PipeableTransformStream<I, O> extends TransformStream<I, O> {
     constructor(transform: (readable: ReadableStream<I>) => ReadableStream<O>);

package/dist/index.d.ts

@@ -1,5 +1,6 @@
-import { a as AssistantStreamChunk, T as ToolResponse, R as ReadonlyJSONValue, c as AssistantMessage, d as AssistantStreamEncoder, A as AssistantStream } from './assistant-stream-54RSz6y3.js';
-export { e as createAssistantStream, f as createAssistantStreamResponse } from './assistant-stream-54RSz6y3.js';
+import { a as AssistantStreamChunk, T as ToolResponse, c as AssistantMessage, d as AssistantStreamEncoder, A as AssistantStream } from './assistant-stream-CgeUyrON.js';
+export { e as createAssistantStream, f as createAssistantStreamResponse } from './assistant-stream-CgeUyrON.js';
+import { R as ReadonlyJSONValue } from './json-value-DvDjYV7y.js';
 
 declare class PipeableTransformStream<I, O> extends TransformStream<I, O> {
     constructor(transform: (readable: ReadableStream<I>) => ReadableStream<O>);

package/dist/index.js

@@ -1097,7 +1097,7 @@ var ToolExecutionStream = class extends PipeableTransformStream {
   }
 };
 
-// src/core/utils/json/parse-partial-json.ts
+// src/core/utils/json/parse-partial-json-object.ts
 var import_secure_json_parse2 = __toESM(require("secure-json-parse"));
 
 // src/core/utils/json/fix-json.ts
@@ -1105,6 +1105,14 @@ function fixJson(input) {
   const stack = ["ROOT"];
   let lastValidIndex = -1;
   let literalStart = null;
+  const path = [];
+  let currentKey;
+  function pushCurrentKeyToPath() {
+    if (currentKey !== void 0) {
+      path.push(JSON.parse('"' + currentKey + '"'));
+      currentKey = void 0;
+    }
+  }
   function processValueStart(char, i, swapState) {
     {
       switch (char) {
@@ -1113,6 +1121,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_STRING");
+          pushCurrentKeyToPath();
           break;
         }
         case "f":
@@ -1129,6 +1138,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_NUMBER");
+          pushCurrentKeyToPath();
           break;
         }
         case "0":
@@ -1145,6 +1155,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_NUMBER");
+          pushCurrentKeyToPath();
           break;
         }
         case "{": {
@@ -1152,6 +1163,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_OBJECT_START");
+          pushCurrentKeyToPath();
           break;
         }
         case "[": {
@@ -1159,6 +1171,7 @@ function fixJson(input) {
           stack.pop();
           stack.push(swapState);
           stack.push("INSIDE_ARRAY_START");
+          pushCurrentKeyToPath();
           break;
         }
       }
@@ -1174,6 +1187,7 @@ function fixJson(input) {
       case "}": {
         lastValidIndex = i;
         stack.pop();
+        currentKey = path.pop();
         break;
       }
     }
@@ -1183,11 +1197,13 @@ function fixJson(input) {
       case ",": {
         stack.pop();
         stack.push("INSIDE_ARRAY_AFTER_COMMA");
+        currentKey = (Number(currentKey) + 1).toString();
         break;
       }
       case "]": {
         lastValidIndex = i;
         stack.pop();
+        currentKey = path.pop();
         break;
       }
     }
@@ -1204,11 +1220,13 @@ function fixJson(input) {
           case '"': {
             stack.pop();
             stack.push("INSIDE_OBJECT_KEY");
+            currentKey = "";
             break;
           }
           case "}": {
             lastValidIndex = i;
             stack.pop();
+            currentKey = path.pop();
             break;
           }
         }
@@ -1219,6 +1237,7 @@ function fixJson(input) {
           case '"': {
             stack.pop();
             stack.push("INSIDE_OBJECT_KEY");
+            currentKey = "";
             break;
           }
         }
@@ -1231,6 +1250,15 @@ function fixJson(input) {
             stack.push("INSIDE_OBJECT_AFTER_KEY");
             break;
           }
+          case "\\": {
+            stack.push("INSIDE_STRING_ESCAPE");
+            currentKey += char;
+            break;
+          }
+          default: {
+            currentKey += char;
+            break;
+          }
         }
         break;
       }
@@ -1257,6 +1285,7 @@ function fixJson(input) {
           case '"': {
             stack.pop();
             lastValidIndex = i;
+            currentKey = path.pop();
             break;
           }
           case "\\": {
@@ -1274,10 +1303,12 @@ function fixJson(input) {
           case "]": {
             lastValidIndex = i;
             stack.pop();
+            currentKey = path.pop();
             break;
           }
           default: {
             lastValidIndex = i;
+            currentKey = "0";
             processValueStart(char, i, "INSIDE_ARRAY_AFTER_VALUE");
             break;
           }
@@ -1289,11 +1320,13 @@ function fixJson(input) {
           case ",": {
             stack.pop();
             stack.push("INSIDE_ARRAY_AFTER_COMMA");
+            currentKey = (Number(currentKey) + 1).toString();
             break;
           }
           case "]": {
             lastValidIndex = i;
             stack.pop();
+            currentKey = path.pop();
             break;
           }
           default: {
@@ -1309,7 +1342,11 @@ function fixJson(input) {
       }
       case "INSIDE_STRING_ESCAPE": {
         stack.pop();
-        lastValidIndex = i;
+        if (stack[stack.length - 1] === "INSIDE_STRING") {
+          lastValidIndex = i;
+        } else if (stack[stack.length - 1] === "INSIDE_OBJECT_KEY") {
+          currentKey += char;
+        }
         break;
       }
       case "INSIDE_NUMBER": {
@@ -1335,6 +1372,7 @@ function fixJson(input) {
           }
           case ",": {
             stack.pop();
+            currentKey = path.pop();
             if (stack[stack.length - 1] === "INSIDE_ARRAY_AFTER_VALUE") {
               processAfterArrayValue(char, i);
             }
@@ -1345,6 +1383,7 @@ function fixJson(input) {
           }
           case "}": {
             stack.pop();
+            currentKey = path.pop();
             if (stack[stack.length - 1] === "INSIDE_OBJECT_AFTER_VALUE") {
               processAfterObjectValue(char, i);
             }
@@ -1352,6 +1391,7 @@ function fixJson(input) {
           }
           case "]": {
             stack.pop();
+            currentKey = path.pop();
             if (stack[stack.length - 1] === "INSIDE_ARRAY_AFTER_VALUE") {
               processAfterArrayValue(char, i);
             }
@@ -1359,6 +1399,7 @@ function fixJson(input) {
           }
           default: {
             stack.pop();
+            currentKey = path.pop();
             break;
           }
         }
@@ -1415,16 +1456,38 @@ function fixJson(input) {
       }
     }
   }
-  return result;
+  return [result, path];
 }
 
-// src/core/utils/json/parse-partial-json.ts
-var parsePartialJson = (json) => {
+// src/core/utils/json/parse-partial-json-object.ts
+var PARTIAL_JSON_OBJECT_META_SYMBOL = Symbol(
+  "aui.parse-partial-json-object.meta"
+);
+var parsePartialJsonObject = (json) => {
+  if (json.length === 0)
+    return {
+      [PARTIAL_JSON_OBJECT_META_SYMBOL]: { state: "partial", partialPath: [] }
+    };
   try {
-    return import_secure_json_parse2.default.parse(json);
+    const res = import_secure_json_parse2.default.parse(json);
+    if (typeof res !== "object" || res === null)
+      throw new Error("argsText is expected to be an object");
+    res[PARTIAL_JSON_OBJECT_META_SYMBOL] = {
+      state: "complete",
+      partialPath: []
+    };
+    return res;
   } catch {
     try {
-      return import_secure_json_parse2.default.parse(fixJson(json));
+      const [fixedJson, partialPath] = fixJson(json);
+      const res = import_secure_json_parse2.default.parse(fixedJson);
+      if (typeof res !== "object" || res === null)
+        throw new Error("argsText is expected to be an object");
+      res[PARTIAL_JSON_OBJECT_META_SYMBOL] = {
+        state: "partial",
+        partialPath
+      };
+      return res;
     } catch {
       return void 0;
     }
@@ -1553,12 +1616,7 @@ var handleTextDelta = (message, chunk) => {
       return { ...part, text: part.text + chunk.textDelta };
     } else if (part.type === "tool-call") {
       const newArgsText = part.argsText + chunk.textDelta;
-      let newArgs;
-      try {
-        newArgs = parsePartialJson(newArgsText);
-      } catch (err) {
-        newArgs = part.args;
-      }
+      const newArgs = parsePartialJsonObject(newArgsText) ?? part.args;
       return { ...part, argsText: newArgsText, args: newArgs };
     } else {
       throw new Error(