@assistant-ui/react 0.12.14 → 0.12.15

package/dist/primitives/actionBarMore/ActionBarMoreRoot.js

@@ -1,10 +1,43 @@
 "use client";
 import { jsx as _jsx } from "react/jsx-runtime";
+import { useCallback, useEffect, useRef } from "react";
 import { DropdownMenu as DropdownMenuPrimitive } from "radix-ui";
 import { useDropdownMenuScope } from "./scope.js";
-export const ActionBarMorePrimitiveRoot = ({ __scopeActionBarMore, ...rest }) => {
+import { useActionBarInteractionContext } from "../actionBar/ActionBarInteractionContext.js";
+export const ActionBarMorePrimitiveRoot = ({ __scopeActionBarMore, open, onOpenChange, ...rest }) => {
     const scope = useDropdownMenuScope(__scopeActionBarMore);
-    return _jsx(DropdownMenuPrimitive.Root, { ...scope, ...rest });
+    const actionBarInteraction = useActionBarInteractionContext();
+    const releaseInteractionLockRef = useRef(null);
+    const isControlled = open !== undefined;
+    const setInteractionOpen = useCallback((nextOpen) => {
+        if (nextOpen) {
+            if (releaseInteractionLockRef.current)
+                return;
+            releaseInteractionLockRef.current =
+                actionBarInteraction?.acquireInteractionLock() ?? null;
+            return;
+        }
+        releaseInteractionLockRef.current?.();
+        releaseInteractionLockRef.current = null;
+    }, [actionBarInteraction]);
+    const handleOpenChange = useCallback((nextOpen) => {
+        if (!isControlled) {
+            setInteractionOpen(nextOpen);
+        }
+        onOpenChange?.(nextOpen);
+    }, [isControlled, setInteractionOpen, onOpenChange]);
+    useEffect(() => {
+        if (!isControlled)
+            return;
+        setInteractionOpen(Boolean(open));
+    }, [isControlled, open, setInteractionOpen]);
+    useEffect(() => {
+        return () => {
+            releaseInteractionLockRef.current?.();
+            releaseInteractionLockRef.current = null;
+        };
+    }, []);
+    return (_jsx(DropdownMenuPrimitive.Root, { ...scope, ...rest, ...(open !== undefined ? { open } : null), onOpenChange: handleOpenChange }));
 };
 ActionBarMorePrimitiveRoot.displayName = "ActionBarMorePrimitive.Root";
 //# sourceMappingURL=ActionBarMoreRoot.js.map

package/dist/primitives/actionBarMore/ActionBarMoreRoot.js.map

@@ -1 +1 @@
-{"version":3,"file":"ActionBarMoreRoot.js","sourceRoot":"","sources":["../../../src/primitives/actionBarMore/ActionBarMoreRoot.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAGb,OAAO,EAAE,YAAY,IAAI,qBAAqB,EAAE,MAAM,UAAU,CAAC;AACjE,OAAO,EAAe,oBAAoB,EAAE,mBAAgB;AAM5D,MAAM,CAAC,MAAM,0BAA0B,GAEnC,CAAC,EACH,oBAAoB,EACpB,GAAG,IAAI,EACuC,EAAE,EAAE;IAClD,MAAM,KAAK,GAAG,oBAAoB,CAAC,oBAAoB,CAAC,CAAC;IAEzD,OAAO,KAAC,qBAAqB,CAAC,IAAI,OAAK,KAAK,KAAM,IAAI,GAAI,CAAC;AAC7D,CAAC,CAAC;AAEF,0BAA0B,CAAC,WAAW,GAAG,6BAA6B,CAAC"}
+{"version":3,"file":"ActionBarMoreRoot.js","sourceRoot":"","sources":["../../../src/primitives/actionBarMore/ActionBarMoreRoot.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAAM,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAC3D,OAAO,EAAE,YAAY,IAAI,qBAAqB,EAAE,MAAM,UAAU,CAAC;AACjE,OAAO,EAAe,oBAAoB,EAAE,mBAAgB;AAC5D,OAAO,EAAE,8BAA8B,EAAE,oDAAiD;AAM1F,MAAM,CAAC,MAAM,0BAA0B,GAEnC,CAAC,EACH,oBAAoB,EACpB,IAAI,EACJ,YAAY,EACZ,GAAG,IAAI,EACuC,EAAE,EAAE;IAClD,MAAM,KAAK,GAAG,oBAAoB,CAAC,oBAAoB,CAAC,CAAC;IACzD,MAAM,oBAAoB,GAAG,8BAA8B,EAAE,CAAC;IAC9D,MAAM,yBAAyB,GAAG,MAAM,CAAsB,IAAI,CAAC,CAAC;IACpE,MAAM,YAAY,GAAG,IAAI,KAAK,SAAS,CAAC;IAExC,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,QAAiB,EAAE,EAAE;QACpB,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,yBAAyB,CAAC,OAAO;gBAAE,OAAO;YAC9C,yBAAyB,CAAC,OAAO;gBAC/B,oBAAoB,EAAE,sBAAsB,EAAE,IAAI,IAAI,CAAC;YACzD,OAAO;QACT,CAAC;QAED,yBAAyB,CAAC,OAAO,EAAE,EAAE,CAAC;QACtC,yBAAyB,CAAC,OAAO,GAAG,IAAI,CAAC;IAC3C,CAAC,EACD,CAAC,oBAAoB,CAAC,CACvB,CAAC;IAEF,MAAM,gBAAgB,GAAG,WAAW,CAClC,CAAC,QAAiB,EAAE,EAAE;QACpB,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QACD,YAAY,EAAE,CAAC,QAAQ,CAAC,CAAC;IAC3B,CAAC,EACD,CAAC,YAAY,EAAE,kBAAkB,EAAE,YAAY,CAAC,CACjD,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,YAAY;YAAE,OAAO;QAC1B,kBAAkB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACpC,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,EAAE,kBAAkB,CAAC,CAAC,CAAC;IAE7C,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,GAAG,EAAE;YACV,yBAAyB,CAAC,OAAO,EAAE,EAAE,CAAC;YACtC,yBAAyB,CAAC,OAAO,GAAG,IAAI,CAAC;QAC3C,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CACL,KAAC,qBAAqB,CAAC,IAAI,OACrB,KAAK,KACL,IAAI,KACJ,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAC1C,YAAY,EAAE,gBAAgB,GAC9B,CACH,CAAC;AACJ,CAAC,CAAC;AAEF,0BAA0B,CAAC,WAAW,GAAG,6BAA6B,CAAC"}

package/dist/utils/json/is-json-equal.d.ts

@@ -0,0 +1,2 @@
+export declare const isJSONValueEqual: (a: unknown, b: unknown) => boolean;
+//# sourceMappingURL=is-json-equal.d.ts.map

package/dist/utils/json/is-json-equal.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"is-json-equal.d.ts","sourceRoot":"","sources":["../../../src/utils/json/is-json-equal.ts"],"names":[],"mappings":"AA4CA,eAAO,MAAM,gBAAgB,GAAI,GAAG,OAAO,EAAE,GAAG,OAAO,KAAG,OAGzD,CAAC"}

package/dist/utils/json/is-json-equal.js

@@ -0,0 +1,31 @@
+import { isJSONValue, isRecord } from "./is-json.js";
+const MAX_JSON_DEPTH = 100;
+const isJSONValueEqualAtDepth = (a, b, currentDepth) => {
+    if (a === b)
+        return true;
+    if (currentDepth > MAX_JSON_DEPTH)
+        return false;
+    if (a == null || b == null)
+        return false;
+    if (Array.isArray(a)) {
+        if (!Array.isArray(b) || a.length !== b.length)
+            return false;
+        return a.every((item, index) => isJSONValueEqualAtDepth(item, b[index], currentDepth + 1));
+    }
+    if (Array.isArray(b))
+        return false;
+    if (!isRecord(a) || !isRecord(b))
+        return false;
+    const aKeys = Object.keys(a);
+    const bKeys = Object.keys(b);
+    if (aKeys.length !== bKeys.length)
+        return false;
+    return aKeys.every((key) => Object.hasOwn(b, key) &&
+        isJSONValueEqualAtDepth(a[key], b[key], currentDepth + 1));
+};
+export const isJSONValueEqual = (a, b) => {
+    if (!isJSONValue(a) || !isJSONValue(b))
+        return false;
+    return isJSONValueEqualAtDepth(a, b, 0);
+};
+//# sourceMappingURL=is-json-equal.js.map

package/dist/utils/json/is-json-equal.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"is-json-equal.js","sourceRoot":"","sources":["../../../src/utils/json/is-json-equal.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,qBAAkB;AAElD,MAAM,cAAc,GAAG,GAAG,CAAC;AAE3B,MAAM,uBAAuB,GAAG,CAC9B,CAAoB,EACpB,CAAoB,EACpB,YAAoB,EACX,EAAE;IACX,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACzB,IAAI,YAAY,GAAG,cAAc;QAAE,OAAO,KAAK,CAAC;IAEhD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI;QAAE,OAAO,KAAK,CAAC;IAEzC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QACrB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC7D,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAC7B,uBAAuB,CACrB,IAAI,EACJ,CAAC,CAAC,KAAK,CAAsB,EAC7B,YAAY,GAAG,CAAC,CACjB,CACF,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QAAE,OAAO,KAAK,CAAC;IACnC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAAE,OAAO,KAAK,CAAC;IAE/C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7B,IAAI,KAAK,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IAEhD,OAAO,KAAK,CAAC,KAAK,CAChB,CAAC,GAAG,EAAE,EAAE,CACN,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,CAAC;QACrB,uBAAuB,CACrB,CAAC,CAAC,GAAG,CAAsB,EAC3B,CAAC,CAAC,GAAG,CAAsB,EAC3B,YAAY,GAAG,CAAC,CACjB,CACJ,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAU,EAAE,CAAU,EAAW,EAAE;IAClE,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;QAAE,OAAO,KAAK,CAAC;IACrD,OAAO,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;AAC1C,CAAC,CAAC"}

package/dist/utils/json/is-json.d.ts

@@ -1,4 +1,5 @@
 import { ReadonlyJSONArray, ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
+export declare function isRecord(value: unknown): value is Record<string, unknown>;
 export declare function isJSONValue(value: unknown, currentDepth?: number): value is ReadonlyJSONValue;
 export declare function isJSONArray(value: unknown): value is ReadonlyJSONArray;
 export declare function isJSONObject(value: unknown): value is ReadonlyJSONObject;

package/dist/utils/json/is-json.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"is-json.d.ts","sourceRoot":"","sources":["../../../src/utils/json/is-json.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,wBAAwB,CAAC;AAEhC,wBAAgB,WAAW,CACzB,KAAK,EAAE,OAAO,EACd,YAAY,GAAE,MAAU,GACvB,KAAK,IAAI,iBAAiB,CA+B5B;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,iBAAiB,CAEtE;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,kBAAkB,CAQxE"}
+{"version":3,"file":"is-json.d.ts","sourceRoot":"","sources":["../../../src/utils/json/is-json.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,wBAAwB,CAAC;AAEhC,wBAAgB,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAEzE;AAED,wBAAgB,WAAW,CACzB,KAAK,EAAE,OAAO,EACd,YAAY,GAAE,MAAU,GACvB,KAAK,IAAI,iBAAiB,CA+B5B;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,iBAAiB,CAEtE;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,kBAAkB,CAOxE"}

package/dist/utils/json/is-json.js

@@ -1,3 +1,6 @@
+export function isRecord(value) {
+    return value != null && typeof value === "object" && !Array.isArray(value);
+}
 export function isJSONValue(value, currentDepth = 0) {
     // Protect against too deep recursion
     if (currentDepth > 100) {
@@ -15,7 +18,7 @@ export function isJSONValue(value, currentDepth = 0) {
     if (Array.isArray(value)) {
         return value.every((item) => isJSONValue(item, currentDepth + 1));
     }
-    if (typeof value === "object") {
+    if (isRecord(value)) {
         return Object.entries(value).every(([key, val]) => typeof key === "string" && isJSONValue(val, currentDepth + 1));
     }
     return false;
@@ -24,8 +27,7 @@ export function isJSONArray(value) {
     return Array.isArray(value) && value.every(isJSONValue);
 }
 export function isJSONObject(value) {
-    return (value != null &&
-        typeof value === "object" &&
+    return (isRecord(value) &&
         Object.entries(value).every(([key, val]) => typeof key === "string" && isJSONValue(val)));
 }
 //# sourceMappingURL=is-json.js.map

package/dist/utils/json/is-json.js.map

@@ -1 +1 @@
-{"version":3,"file":"is-json.js","sourceRoot":"","sources":["../../../src/utils/json/is-json.ts"],"names":[],"mappings":"AAMA,MAAM,UAAU,WAAW,CACzB,KAAc,EACd,eAAuB,CAAC;IAExB,qCAAqC;IACrC,IAAI,YAAY,GAAG,GAAG,EAAE,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IACE,KAAK,KAAK,IAAI;QACd,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS,EAC1B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,8BAA8B;IAC9B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CAChC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CACb,OAAO,GAAG,KAAK,QAAQ,IAAI,WAAW,CAAC,GAAG,EAAE,YAAY,GAAG,CAAC,CAAC,CAChE,CAAC;IACJ,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAc;IACzC,OAAO,CACL,KAAK,IAAI,IAAI;QACb,OAAO,KAAK,KAAK,QAAQ;QACzB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CACzB,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,IAAI,WAAW,CAAC,GAAG,CAAC,CAC5D,CACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"is-json.js","sourceRoot":"","sources":["../../../src/utils/json/is-json.ts"],"names":[],"mappings":"AAMA,MAAM,UAAU,QAAQ,CAAC,KAAc;IACrC,OAAO,KAAK,IAAI,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC7E,CAAC;AAED,MAAM,UAAU,WAAW,CACzB,KAAc,EACd,eAAuB,CAAC;IAExB,qCAAqC;IACrC,IAAI,YAAY,GAAG,GAAG,EAAE,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IACE,KAAK,KAAK,IAAI;QACd,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS,EAC1B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,8BAA8B;IAC9B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACpB,OAAO,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CAChC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CACb,OAAO,GAAG,KAAK,QAAQ,IAAI,WAAW,CAAC,GAAG,EAAE,YAAY,GAAG,CAAC,CAAC,CAChE,CAAC;IACJ,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAc;IACzC,OAAO,CACL,QAAQ,CAAC,KAAK,CAAC;QACf,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CACzB,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,IAAI,WAAW,CAAC,GAAG,CAAC,CAC5D,CACF,CAAC;AACJ,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react",
-  "version": "0.12.14",
+  "version": "0.12.15",
   "description": "TypeScript/React library for AI Chat",
   "keywords": [
     "radix-ui",
@@ -49,19 +49,19 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-cloud": "^0.1.20",
     "@assistant-ui/core": "^0.1.2",
-    "@assistant-ui/tap": "^0.5.1",
-    "@assistant-ui/store": "^0.2.1",
+    "@assistant-ui/store": "^0.2.2",
+    "@assistant-ui/tap": "^0.5.2",
     "@radix-ui/primitive": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.2",
     "@radix-ui/react-context": "^1.1.3",
     "@radix-ui/react-primitive": "^2.1.4",
     "@radix-ui/react-use-callback-ref": "^1.1.1",
     "@radix-ui/react-use-escape-keydown": "^1.1.1",
-    "radix-ui": "^1.4.3",
+    "assistant-cloud": "^0.1.21",
     "assistant-stream": "^0.3.4",
     "nanoid": "^5.1.6",
+    "radix-ui": "^1.4.3",
     "react-textarea-autosize": "^8.5.9",
     "zod": "^4.3.6",
     "zustand": "^5.0.11"
@@ -83,7 +83,7 @@
   "devDependencies": {
     "@testing-library/react": "^16.3.2",
     "@types/json-schema": "^7.0.15",
-    "@types/node": "^25.3.0",
+    "@types/node": "^25.3.3",
     "@types/react": "^19.2.14",
     "@types/react-dom": "^19.2.3",
     "jsdom": "^28.1.0",

package/src/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.test.ts

@@ -6,18 +6,20 @@ import { act, renderHook, waitFor } from "@testing-library/react";
 import { describe, expect, it, vi } from "vitest";
 import type { AssistantTransportState } from "./types";
 import { ToolExecutionStatus, useToolInvocations } from "./useToolInvocations";
-import { ReadonlyJSONObject } from "assistant-stream/utils";
+import { ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
 
 const createState = (
   messages: ThreadAssistantMessage[],
+  isRunning: boolean = true,
 ): AssistantTransportState => ({
   messages,
-  isRunning: true,
+  isRunning,
 });
 
 const createAssistantMessage = (
   argsText: string,
   args: Record<string, unknown>,
+  options?: { result?: ReadonlyJSONValue; isError?: boolean },
 ): ThreadAssistantMessage => ({
   id: "m-1",
   role: "assistant",
@@ -37,6 +39,8 @@ const createAssistantMessage = (
       toolName: "weatherSearch",
       args: args as ReadonlyJSONObject,
       argsText,
+      ...(options?.result !== undefined && { result: options.result }),
+      ...(options?.isError !== undefined && { isError: options.isError }),
     },
   ],
 });
@@ -223,4 +227,223 @@ describe("useToolInvocations", () => {
     });
     expect(Object.keys(statuses)).not.toContain("tool-1:rewrite:0");
   });
+
+  it("does not close args stream early for non-executable tool snapshots", () => {
+    const getTools = () => ({
+      weatherSearch: {
+        parameters: { type: "object", properties: {} },
+      } satisfies Tool,
+    });
+    const onResult = vi.fn();
+    const setToolStatuses = vi.fn();
+    const warnSpy = vi.spyOn(console, "warn").mockImplementation(() => {});
+
+    try {
+      const { rerender } = renderHook(
+        ({ state }: { state: AssistantTransportState }) =>
+          useToolInvocations({
+            state,
+            getTools,
+            onResult,
+            setToolStatuses,
+          }),
+        {
+          initialProps: {
+            state: createState([]),
+          },
+        },
+      );
+
+      act(() => {
+        rerender({
+          state: createState([createAssistantMessage("{}", {})]),
+        });
+      });
+
+      act(() => {
+        rerender({
+          state: createState([
+            createAssistantMessage('{"title":"Weekly"', {
+              title: "Weekly",
+            }),
+          ]),
+        });
+      });
+
+      act(() => {
+        rerender({
+          state: createState([
+            createAssistantMessage('{"title":"Weekly","columns":["name"]}', {
+              title: "Weekly",
+              columns: ["name"],
+            }),
+          ]),
+        });
+      });
+
+      expect(warnSpy).not.toHaveBeenCalledWith(
+        "argsText updated after controller was closed:",
+        expect.anything(),
+      );
+      expect(warnSpy).not.toHaveBeenCalledWith(
+        "argsText updated after controller was closed, restarting tool args stream:",
+        expect.anything(),
+      );
+      expect(onResult).not.toHaveBeenCalled();
+    } finally {
+      warnSpy.mockRestore();
+    }
+  });
+
+  it("closes non-executable complete args stream after run settles", () => {
+    const getTools = () => ({
+      weatherSearch: {
+        parameters: { type: "object", properties: {} },
+      } satisfies Tool,
+    });
+    const onResult = vi.fn();
+    const setToolStatuses = vi.fn();
+    const warnSpy = vi.spyOn(console, "warn").mockImplementation(() => {});
+
+    try {
+      const { rerender } = renderHook(
+        ({ state }: { state: AssistantTransportState }) =>
+          useToolInvocations({
+            state,
+            getTools,
+            onResult,
+            setToolStatuses,
+          }),
+        {
+          initialProps: {
+            state: createState([]),
+          },
+        },
+      );
+
+      act(() => {
+        rerender({
+          state: createState(
+            [
+              createAssistantMessage('{"title":"Weekly"}', {
+                title: "Weekly",
+              }),
+            ],
+            true,
+          ),
+        });
+      });
+
+      act(() => {
+        rerender({
+          state: createState(
+            [
+              createAssistantMessage('{"title":"Weekly"}', {
+                title: "Weekly",
+              }),
+            ],
+            false,
+          ),
+        });
+      });
+
+      act(() => {
+        rerender({
+          state: createState(
+            [
+              createAssistantMessage('{"title":"Weekly","columns":["name"]}', {
+                title: "Weekly",
+                columns: ["name"],
+              }),
+            ],
+            false,
+          ),
+        });
+      });
+
+      expect(warnSpy).toHaveBeenCalledWith(
+        "argsText updated after controller was closed, restarting tool args stream:",
+        expect.objectContaining({
+          previous: '{"title":"Weekly"}',
+          next: '{"title":"Weekly","columns":["name"]}',
+        }),
+      );
+      expect(onResult).not.toHaveBeenCalled();
+    } finally {
+      warnSpy.mockRestore();
+    }
+  });
+
+  it("handles backend result when equivalent complete argsText reorders keys", async () => {
+    let resolveExecute: ((value: unknown) => void) | undefined;
+    const execute = vi.fn(
+      () =>
+        new Promise<unknown>((resolve) => {
+          resolveExecute = resolve;
+        }),
+    );
+    const getTools = () => ({
+      weatherSearch: {
+        parameters: { type: "object", properties: {} },
+        execute,
+      } satisfies Tool,
+    });
+    const onResult = vi.fn();
+    const setToolStatuses = vi.fn();
+
+    const { rerender } = renderHook(
+      ({ state }: { state: AssistantTransportState }) =>
+        useToolInvocations({
+          state,
+          getTools,
+          onResult,
+          setToolStatuses,
+        }),
+      {
+        initialProps: {
+          state: createState([]),
+        },
+      },
+    );
+
+    act(() => {
+      rerender({
+        state: createState([
+          createAssistantMessage('{"a":1,"b":2}', {
+            a: 1,
+            b: 2,
+          }),
+        ]),
+      });
+    });
+
+    await waitFor(() => {
+      expect(execute).toHaveBeenCalledTimes(1);
+    });
+
+    act(() => {
+      rerender({
+        state: createState([
+          createAssistantMessage(
+            '{"b":2,"a":1}',
+            {
+              a: 1,
+              b: 2,
+            },
+            {
+              result: { source: "backend" },
+            },
+          ),
+        ]),
+      });
+    });
+
+    await act(async () => {
+      resolveExecute?.({ source: "client" });
+    });
+
+    await waitFor(() => {
+      expect(onResult).not.toHaveBeenCalled();
+    });
+  });
 });