@assistant-ui/react-ink 0.0.2 → 0.0.4

package/dist/primitives/toolCall/ToolFallback.js

@@ -0,0 +1,103 @@
+import { jsx as _jsx, Fragment as _Fragment, jsxs as _jsxs } from "react/jsx-runtime";
+import { useMemo } from "react";
+import { Box, Text } from "ink";
+import Spinner from "ink-spinner";
+const STATUS_ICONS = {
+    complete: "+",
+    error: "x",
+    "requires-action": "?",
+};
+const STATUS_COLORS = {
+    running: "yellow",
+    complete: "green",
+    error: "red",
+    "requires-action": "cyan",
+};
+const prettyPrintArgs = (argsText) => {
+    if (!argsText)
+        return "";
+    try {
+        return JSON.stringify(JSON.parse(argsText), null, 2);
+    }
+    catch {
+        return argsText;
+    }
+};
+const formatResult = (result) => {
+    if (result === undefined || result === null)
+        return "";
+    if (typeof result === "string")
+        return result;
+    try {
+        return JSON.stringify(result, null, 2);
+    }
+    catch {
+        return String(result);
+    }
+};
+const truncate = (text, maxLines) => {
+    const lines = text.split("\n");
+    if (lines.length <= maxLines)
+        return text;
+    return (lines.slice(0, maxLines).join("\n") +
+        `\n... (${lines.length - maxLines} more lines)`);
+};
+const resolveStatus = (part, status) => {
+    if (status?.type === "requires-action")
+        return "requires-action";
+    if (status?.type === "incomplete")
+        return "error";
+    if (status?.type === "complete")
+        return "complete";
+    if (part.isError)
+        return "error";
+    if (part.result !== undefined)
+        return "complete";
+    if (part.interrupt)
+        return "requires-action";
+    return "running";
+};
+const getDisplayResult = (result, status) => {
+    if (result !== undefined)
+        return result;
+    if (status?.type === "incomplete")
+        return status.error;
+    return undefined;
+};
+const getFallbackErrorText = (status) => {
+    if (status?.type === "incomplete" && status.reason === "cancelled") {
+        return "Tool call cancelled";
+    }
+    return "Tool call failed";
+};
+const StatusIcon = ({ status }) => {
+    if (status === "running") {
+        return (_jsx(Text, { color: STATUS_COLORS.running, children: _jsx(Spinner, { type: "line" }) }));
+    }
+    return _jsx(Text, { color: STATUS_COLORS[status], children: STATUS_ICONS[status] });
+};
+export const ToolFallback = ({ toolName, args, argsText, result: partResult, isError, interrupt, status, expanded: expandedProp, maxArgLines = 20, maxResultLines = 20, maxResultPreviewLines = 1, renderHeader, renderArgs, renderResult, }) => {
+    const displayStatus = resolveStatus({
+        interrupt,
+        isError,
+        result: partResult,
+    }, status);
+    const expanded = expandedProp ??
+        (displayStatus === "running" ||
+            displayStatus === "requires-action" ||
+            displayStatus === "error");
+    const result = getDisplayResult(partResult, status);
+    const resultStr = result !== undefined || displayStatus === "error"
+        ? formatResult(result)
+        : "";
+    const argsDisplay = useMemo(() => prettyPrintArgs(argsText), [argsText]);
+    return (_jsxs(Box, { flexDirection: "column", children: [_jsx(Box, { gap: 1, children: renderHeader ? (renderHeader({
+                    toolName,
+                    status: displayStatus,
+                    expanded,
+                })) : (_jsxs(_Fragment, { children: [_jsx(StatusIcon, { status: displayStatus }), _jsx(Text, { bold: true, children: toolName }), !expanded && resultStr ? (_jsx(Text, { dimColor: true, children: truncate(resultStr, maxResultPreviewLines) })) : null] })) }), expanded && (_jsxs(Box, { flexDirection: "column", marginLeft: 2, children: [argsText ? (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { dimColor: true, children: "Args:" }), renderArgs ? (renderArgs({ args, argsText })) : (_jsx(Text, { children: truncate(argsDisplay, maxArgLines) }))] })) : null, partResult !== undefined || displayStatus === "error" ? (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { dimColor: true, children: displayStatus === "error" ? "Error:" : "Result:" }), renderResult ? (renderResult({
+                                result,
+                                isError: displayStatus === "error",
+                            })) : displayStatus === "error" ? (_jsx(Text, { color: "red", children: truncate(resultStr || getFallbackErrorText(status), maxResultLines) })) : (_jsx(Text, { children: truncate(resultStr, maxResultLines) }))] })) : null, displayStatus === "running" && _jsx(Text, { dimColor: true, children: "Running..." }), displayStatus === "requires-action" && (_jsx(Text, { color: "cyan", children: "Waiting for approval..." }))] }))] }));
+};
+//# sourceMappingURL=ToolFallback.js.map

package/dist/primitives/toolCall/ToolFallback.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolFallback.js","sourceRoot":"","sources":["../../../src/primitives/toolCall/ToolFallback.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAkB,OAAO,EAAE,MAAM,OAAO,CAAC;AAChD,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,OAAO,MAAM,aAAa,CAAC;AAgBlC,MAAM,YAAY,GAAuD;IACvE,QAAQ,EAAE,GAAG;IACb,KAAK,EAAE,GAAG;IACV,iBAAiB,EAAE,GAAG;CACvB,CAAC;AAEF,MAAM,aAAa,GAAmC;IACpD,OAAO,EAAE,QAAQ;IACjB,QAAQ,EAAE,OAAO;IACjB,KAAK,EAAE,KAAK;IACZ,iBAAiB,EAAE,MAAM;CAC1B,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC,QAAgB,EAAU,EAAE;IACnD,IAAI,CAAC,QAAQ;QAAE,OAAO,EAAE,CAAC;IACzB,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACvD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,QAAQ,CAAC;IAClB,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,YAAY,GAAG,CAAC,MAAe,EAAU,EAAE;IAC/C,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,IAAI;QAAE,OAAO,EAAE,CAAC;IACvD,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC9C,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACzC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;IACxB,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAE,QAAgB,EAAU,EAAE;IAC1D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/B,IAAI,KAAK,CAAC,MAAM,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC1C,OAAO,CACL,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACnC,UAAU,KAAK,CAAC,MAAM,GAAG,QAAQ,cAAc,CAChD,CAAC;AACJ,CAAC,CAAC;AAuBF,MAAM,aAAa,GAAG,CACpB,IAIC,EACD,MAAkC,EAClB,EAAE;IAClB,IAAI,MAAM,EAAE,IAAI,KAAK,iBAAiB;QAAE,OAAO,iBAAiB,CAAC;IACjE,IAAI,MAAM,EAAE,IAAI,KAAK,YAAY;QAAE,OAAO,OAAO,CAAC;IAClD,IAAI,MAAM,EAAE,IAAI,KAAK,UAAU;QAAE,OAAO,UAAU,CAAC;IACnD,IAAI,IAAI,CAAC,OAAO;QAAE,OAAO,OAAO,CAAC;IACjC,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;QAAE,OAAO,UAAU,CAAC;IACjD,IAAI,IAAI,CAAC,SAAS;QAAE,OAAO,iBAAiB,CAAC;IAC7C,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC;AAEF,MAAM,gBAAgB,GAAG,CACvB,MAAe,EACf,MAAkC,EAClC,EAAE;IACF,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO,MAAM,CAAC;IACxC,IAAI,MAAM,EAAE,IAAI,KAAK,YAAY;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC;IACvD,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAAC,MAAkC,EAAE,EAAE;IAClE,IAAI,MAAM,EAAE,IAAI,KAAK,YAAY,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;QACnE,OAAO,qBAAqB,CAAC;IAC/B,CAAC;IACD,OAAO,kBAAkB,CAAC;AAC5B,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,CAAC,EAAE,MAAM,EAA8B,EAAE,EAAE;IAC5D,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,OAAO,CACL,KAAC,IAAI,IAAC,KAAK,EAAE,aAAa,CAAC,OAAO,YAChC,KAAC,OAAO,IAAC,IAAI,EAAC,MAAM,GAAG,GAClB,CACR,CAAC;IACJ,CAAC;IACD,OAAO,KAAC,IAAI,IAAC,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC,YAAG,YAAY,CAAC,MAAM,CAAC,GAAQ,CAAC;AAC3E,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,EAC3B,QAAQ,EACR,IAAI,EACJ,QAAQ,EACR,MAAM,EAAE,UAAU,EAClB,OAAO,EACP,SAAS,EACT,MAAM,EACN,QAAQ,EAAE,YAAY,EACtB,WAAW,GAAG,EAAE,EAChB,cAAc,GAAG,EAAE,EACnB,qBAAqB,GAAG,CAAC,EACzB,YAAY,EACZ,UAAU,EACV,YAAY,GACM,EAAE,EAAE;IACtB,MAAM,aAAa,GAAG,aAAa,CACjC;QACE,SAAS;QACT,OAAO;QACP,MAAM,EAAE,UAAU;KACnB,EACD,MAAM,CACP,CAAC;IACF,MAAM,QAAQ,GACZ,YAAY;QACZ,CAAC,aAAa,KAAK,SAAS;YAC1B,aAAa,KAAK,iBAAiB;YACnC,aAAa,KAAK,OAAO,CAAC,CAAC;IAE/B,MAAM,MAAM,GAAG,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IACpD,MAAM,SAAS,GACb,MAAM,KAAK,SAAS,IAAI,aAAa,KAAK,OAAO;QAC/C,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC;QACtB,CAAC,CAAC,EAAE,CAAC;IACT,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEzE,OAAO,CACL,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,aACzB,KAAC,GAAG,IAAC,GAAG,EAAE,CAAC,YACR,YAAY,CAAC,CAAC,CAAC,CACd,YAAY,CAAC;oBACX,QAAQ;oBACR,MAAM,EAAE,aAAa;oBACrB,QAAQ;iBACT,CAAC,CACH,CAAC,CAAC,CAAC,CACF,8BACE,KAAC,UAAU,IAAC,MAAM,EAAE,aAAa,GAAI,EACrC,KAAC,IAAI,IAAC,IAAI,kBAAE,QAAQ,GAAQ,EAC3B,CAAC,QAAQ,IAAI,SAAS,CAAC,CAAC,CAAC,CACxB,KAAC,IAAI,IAAC,QAAQ,kBAAE,QAAQ,CAAC,SAAS,EAAE,qBAAqB,CAAC,GAAQ,CACnE,CAAC,CAAC,CAAC,IAAI,IACP,CACJ,GACG,EAEL,QAAQ,IAAI,CACX,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,EAAC,UAAU,EAAE,CAAC,aACtC,QAAQ,CAAC,CAAC,CAAC,CACV,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,aACzB,KAAC,IAAI,IAAC,QAAQ,4BAAa,EAC1B,UAAU,CAAC,CAAC,CAAC,CACZ,UAAU,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAC/B,CAAC,CAAC,CAAC,CACF,KAAC,IAAI,cAAE,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC,GAAQ,CAClD,IACG,CACP,CAAC,CAAC,CAAC,IAAI,EAEP,UAAU,KAAK,SAAS,IAAI,aAAa,KAAK,OAAO,CAAC,CAAC,CAAC,CACvD,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,aACzB,KAAC,IAAI,IAAC,QAAQ,kBACX,aAAa,KAAK,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,GAC5C,EACN,YAAY,CAAC,CAAC,CAAC,CACd,YAAY,CAAC;gCACX,MAAM;gCACN,OAAO,EAAE,aAAa,KAAK,OAAO;6BACnC,CAAC,CACH,CAAC,CAAC,CAAC,aAAa,KAAK,OAAO,CAAC,CAAC,CAAC,CAC9B,KAAC,IAAI,IAAC,KAAK,EAAC,KAAK,YACd,QAAQ,CACP,SAAS,IAAI,oBAAoB,CAAC,MAAM,CAAC,EACzC,cAAc,CACf,GACI,CACR,CAAC,CAAC,CAAC,CACF,KAAC,IAAI,cAAE,QAAQ,CAAC,SAAS,EAAE,cAAc,CAAC,GAAQ,CACnD,IACG,CACP,CAAC,CAAC,CAAC,IAAI,EAEP,aAAa,KAAK,SAAS,IAAI,KAAC,IAAI,IAAC,QAAQ,iCAAkB,EAE/D,aAAa,KAAK,iBAAiB,IAAI,CACtC,KAAC,IAAI,IAAC,KAAK,EAAC,MAAM,wCAA+B,CAClD,IACG,CACP,IACG,CACP,CAAC;AACJ,CAAC,CAAC"}

package/dist/primitives/toolCall/index.d.ts

@@ -0,0 +1,2 @@
+export { ToolFallback as Fallback, type ToolFallbackProps as FallbackProps, type ToolCallStatus, } from "./ToolFallback.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/primitives/toolCall/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/primitives/toolCall/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,YAAY,IAAI,QAAQ,EACxB,KAAK,iBAAiB,IAAI,aAAa,EACvC,KAAK,cAAc,GACpB,0BAAuB"}

package/dist/primitives/toolCall/index.js

@@ -0,0 +1,2 @@
+export { ToolFallback as Fallback, } from "./ToolFallback.js";
+//# sourceMappingURL=index.js.map

package/dist/primitives/toolCall/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/primitives/toolCall/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,YAAY,IAAI,QAAQ,GAGzB,0BAAuB"}

package/dist/runtimes/useLocalRuntime.d.ts

@@ -1,18 +1,2 @@
-import type { AssistantRuntime, ChatModelAdapter, ThreadMessageLike } from "@assistant-ui/core";
-import type { LocalRuntimeOptionsBase } from "@assistant-ui/core/internal";
-import type { TitleGenerationAdapter } from "../adapters/TitleGenerationAdapter.js";
-type AsyncStorageLike = {
-    getItem(key: string): Promise<string | null>;
-    setItem(key: string, value: string): Promise<void>;
-    removeItem(key: string): Promise<void>;
-};
-export type LocalRuntimeOptions = Omit<LocalRuntimeOptionsBase, "adapters"> & {
-    adapters?: Omit<LocalRuntimeOptionsBase["adapters"], "chatModel"> | undefined;
-    initialMessages?: readonly ThreadMessageLike[] | undefined;
-    titleGenerator?: TitleGenerationAdapter | undefined;
-    storage?: AsyncStorageLike | undefined;
-    storagePrefix?: string | undefined;
-};
-export declare const useLocalRuntime: (chatModel: ChatModelAdapter, options?: LocalRuntimeOptions) => AssistantRuntime;
-export {};
+export { useLocalRuntime, type LocalRuntimeOptions, } from "@assistant-ui/core/react";
 //# sourceMappingURL=useLocalRuntime.d.ts.map

package/dist/runtimes/useLocalRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useLocalRuntime.d.ts","sourceRoot":"","sources":["../../src/runtimes/useLocalRuntime.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,gBAAgB,EAChB,gBAAgB,EAChB,iBAAiB,EAElB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,6BAA6B,CAAC;AAM3E,OAAO,KAAK,EAAE,sBAAsB,EAAE,8CAA2C;AAKjF,KAAK,gBAAgB,GAAG;IACtB,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACxC,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG,IAAI,CAAC,uBAAuB,EAAE,UAAU,CAAC,GAAG;IAC5E,QAAQ,CAAC,EAAE,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC,GAAG,SAAS,CAAC;IAC9E,eAAe,CAAC,EAAE,SAAS,iBAAiB,EAAE,GAAG,SAAS,CAAC;IAC3D,cAAc,CAAC,EAAE,sBAAsB,GAAG,SAAS,CAAC;IACpD,OAAO,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAAC;IACvC,aAAa,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACpC,CAAC;AAmDF,eAAO,MAAM,eAAe,GAC1B,WAAW,gBAAgB,EAC3B,UAAS,mBAAwB,KAChC,gBAmBF,CAAC"}
+{"version":3,"file":"useLocalRuntime.d.ts","sourceRoot":"","sources":["../../src/runtimes/useLocalRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,EACf,KAAK,mBAAmB,GACzB,MAAM,0BAA0B,CAAC"}

package/dist/runtimes/useLocalRuntime.js

@@ -1,62 +1,2 @@
-import { useEffect, useMemo, useRef, useState } from "react";
-import { InMemoryThreadListAdapter } from "@assistant-ui/core";
-import { AssistantRuntimeImpl, LocalRuntimeCore, } from "@assistant-ui/core/internal";
-import { useAuiState } from "@assistant-ui/store";
-import { useRemoteThreadListRuntime } from "./useRemoteThreadListRuntime.js";
-import { createLocalStorageAdapter } from "../adapters/LocalStorageThreadListAdapter.js";
-import { useRuntimeAdapters } from "../context/providers/RuntimeAdapterProvider.js";
-const useInnerLocalRuntime = (chatModel, options) => {
-    const { initialMessages, ...restOptions } = options;
-    const { modelContext, ...threadListAdapters } = useRuntimeAdapters() ?? {};
-    const opt = {
-        ...restOptions,
-        adapters: {
-            ...threadListAdapters,
-            ...restOptions.adapters,
-            chatModel,
-        },
-    };
-    const [runtime] = useState(() => new LocalRuntimeCore(opt, initialMessages));
-    const threadIdRef = useRef(undefined);
-    threadIdRef.current = useAuiState((s) => s.threadListItem.remoteId);
-    useEffect(() => {
-        runtime.threads
-            .getMainThreadRuntimeCore()
-            .__internal_setGetThreadId(() => threadIdRef.current);
-    }, [runtime]);
-    useEffect(() => {
-        return () => {
-            runtime.threads.getMainThreadRuntimeCore().detach();
-        };
-    }, [runtime]);
-    useEffect(() => {
-        runtime.threads.getMainThreadRuntimeCore().__internal_setOptions(opt);
-        runtime.threads.getMainThreadRuntimeCore().__internal_load();
-    });
-    useEffect(() => {
-        if (!modelContext)
-            return undefined;
-        return runtime.registerModelContextProvider(modelContext);
-    }, [modelContext, runtime]);
-    return useMemo(() => new AssistantRuntimeImpl(runtime), [runtime]);
-};
-export const useLocalRuntime = (chatModel, options = {}) => {
-    const { titleGenerator, storage, storagePrefix, ...innerOptions } = options;
-    const adapter = useMemo(() => {
-        if (!storage)
-            return new InMemoryThreadListAdapter();
-        return createLocalStorageAdapter({
-            storage,
-            prefix: storagePrefix,
-            titleGenerator,
-        });
-    }, [storage, storagePrefix, titleGenerator]);
-    return useRemoteThreadListRuntime({
-        runtimeHook: function RuntimeHook() {
-            return useInnerLocalRuntime(chatModel, innerOptions);
-        },
-        adapter,
-        allowNesting: true,
-    });
-};
+export { useLocalRuntime, } from "@assistant-ui/core/react";
 //# sourceMappingURL=useLocalRuntime.js.map

package/dist/runtimes/useLocalRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useLocalRuntime.js","sourceRoot":"","sources":["../../src/runtimes/useLocalRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAO7D,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAE/D,OAAO,EACL,oBAAoB,EACpB,gBAAgB,GACjB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,OAAO,EAAE,0BAA0B,EAAE,wCAAqC;AAC1E,OAAO,EAAE,yBAAyB,EAAE,qDAAkD;AACtF,OAAO,EAAE,kBAAkB,EAAE,uDAAoD;AAgBjF,MAAM,oBAAoB,GAAG,CAC3B,SAA2B,EAC3B,OAGC,EACiB,EAAE;IACpB,MAAM,EAAE,eAAe,EAAE,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC;IACpD,MAAM,EAAE,YAAY,EAAE,GAAG,kBAAkB,EAAE,GAAG,kBAAkB,EAAE,IAAI,EAAE,CAAC;IAE3E,MAAM,GAAG,GAA4B;QACnC,GAAG,WAAW;QACd,QAAQ,EAAE;YACR,GAAG,kBAAkB;YACrB,GAAG,WAAW,CAAC,QAAQ;YACvB,SAAS;SACV;KACF,CAAC;IAEF,MAAM,CAAC,OAAO,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,IAAI,gBAAgB,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC,CAAC;IAE7E,MAAM,WAAW,GAAG,MAAM,CAAqB,SAAS,CAAC,CAAC;IAC1D,WAAW,CAAC,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAEpE,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,OAAO;aACZ,wBAAwB,EAAE;aAC1B,yBAAyB,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC1D,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAEd,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,GAAG,EAAE;YACV,OAAO,CAAC,OAAO,CAAC,wBAAwB,EAAE,CAAC,MAAM,EAAE,CAAC;QACtD,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAEd,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,OAAO,CAAC,wBAAwB,EAAE,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC;QACtE,OAAO,CAAC,OAAO,CAAC,wBAAwB,EAAE,CAAC,eAAe,EAAE,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,YAAY;YAAE,OAAO,SAAS,CAAC;QACpC,OAAO,OAAO,CAAC,4BAA4B,CAAC,YAAY,CAAC,CAAC;IAC5D,CAAC,EAAE,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC;IAE5B,OAAO,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,oBAAoB,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;AACrE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,CAC7B,SAA2B,EAC3B,UAA+B,EAAE,EACf,EAAE;IACpB,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC;IAE5E,MAAM,OAAO,GAA4B,OAAO,CAAC,GAAG,EAAE;QACpD,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,yBAAyB,EAAE,CAAC;QACrD,OAAO,yBAAyB,CAAC;YAC/B,OAAO;YACP,MAAM,EAAE,aAAa;YACrB,cAAc;SACf,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,OAAO,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC,CAAC;IAE7C,OAAO,0BAA0B,CAAC;QAChC,WAAW,EAAE,SAAS,WAAW;YAC/B,OAAO,oBAAoB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;QACvD,CAAC;QACD,OAAO;QACP,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;AACL,CAAC,CAAC"}
+{"version":3,"file":"useLocalRuntime.js","sourceRoot":"","sources":["../../src/runtimes/useLocalRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,GAEhB,MAAM,0BAA0B,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ink",
-  "version": "0.0.2",
+  "version": "0.0.4",
   "description": "React Ink (terminal UI) bindings for assistant-ui",
   "keywords": [
     "assistant",
@@ -34,10 +34,11 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.1.3",
+    "@assistant-ui/core": "^0.1.5",
     "@assistant-ui/store": "^0.2.2",
     "@assistant-ui/tap": "^0.5.2",
-    "assistant-stream": "^0.3.4",
+    "assistant-stream": "^0.3.5",
+    "ink-spinner": "^5.0.0",
     "zustand": "^5.0.11"
   },
   "peerDependencies": {
@@ -53,6 +54,7 @@
   "devDependencies": {
     "@types/react": "^19.2.14",
     "ink": "^5.2.0",
+    "ink-testing-library": "^4.0.0",
     "react": "18.3.1",
     "vitest": "^4.0.18",
     "@assistant-ui/x-buildutils": "0.0.2"

package/src/context/AssistantContext.tsx

@@ -1,31 +1,4 @@
-import { type ReactNode, memo } from "react";
-import { useAui } from "@assistant-ui/store";
-import type { AssistantRuntime } from "@assistant-ui/core";
-import { AssistantProviderBase } from "@assistant-ui/core/react";
-
-export const useAssistantRuntime = (): AssistantRuntime => {
-  const aui = useAui();
-  const runtime = aui.threads().__internal_getAssistantRuntime?.();
-  if (!runtime) {
-    throw new Error(
-      "useAssistantRuntime must be used within an AssistantProvider",
-    );
-  }
-  return runtime;
-};
-
-export const AssistantProvider = memo(
-  ({
-    runtime,
-    children,
-  }: {
-    runtime: AssistantRuntime;
-    children: ReactNode;
-  }) => {
-    return (
-      <AssistantProviderBase runtime={runtime}>
-        {children}
-      </AssistantProviderBase>
-    );
-  },
-);
+export {
+  AssistantRuntimeProvider,
+  useAssistantRuntime,
+} from "@assistant-ui/core/react";

package/src/context/index.ts

@@ -1 +1,4 @@
-export { AssistantProvider, useAssistantRuntime } from "./AssistantContext";
+export {
+  AssistantRuntimeProvider,
+  useAssistantRuntime,
+} from "./AssistantContext";

package/src/index.ts

@@ -1,4 +1,4 @@
-import "@assistant-ui/core/react"; // store-augmentation side-effect (tools, dataRenderers scopes)
+/// <reference types="@assistant-ui/core/react" />
 
 // Re-export core types
 export type {
@@ -92,27 +92,7 @@ export {
 } from "@assistant-ui/store";
 
 // Context providers and hooks
-export { AssistantProvider, useAssistantRuntime } from "./context";
-
-// Primitive hooks
-export {
-  useThreadMessages,
-  useThreadIsRunning,
-  useThreadIsEmpty,
-  useComposerSend,
-  useComposerCancel,
-  useMessageReload,
-  useMessageBranching,
-  useActionBarCopy,
-  type UseActionBarCopyOptions,
-  useActionBarEdit,
-  useActionBarReload,
-  useActionBarFeedbackPositive,
-  useActionBarFeedbackNegative,
-  useEditComposerSend,
-  useEditComposerCancel,
-  useComposerAddAttachment,
-} from "./primitive-hooks";
+export { AssistantRuntimeProvider, useAssistantRuntime } from "./context";
 
 // Runtime
 export {
@@ -122,16 +102,17 @@ export {
 } from "./runtimes";
 
 // Primitives
-export * from "./primitives/thread";
-export * from "./primitives/composer";
-export * from "./primitives/message";
-export * from "./primitives/threadList";
-export * from "./primitives/actionBar";
-export * from "./primitives/branchPicker";
-export * from "./primitives/attachment";
-export * from "./primitives/threadListItem";
-export * from "./primitives/chainOfThought";
-export * from "./primitives/suggestion";
+export * as ThreadPrimitive from "./primitives/thread";
+export * as ComposerPrimitive from "./primitives/composer";
+export * as MessagePrimitive from "./primitives/message";
+export * as ThreadListPrimitive from "./primitives/threadList";
+export * as ActionBarPrimitive from "./primitives/actionBar";
+export * as BranchPickerPrimitive from "./primitives/branchPicker";
+export * as AttachmentPrimitive from "./primitives/attachment";
+export * as ThreadListItemPrimitive from "./primitives/threadListItem";
+export * as ChainOfThoughtPrimitive from "./primitives/chainOfThought";
+export * as SuggestionPrimitive from "./primitives/suggestion";
+export * as ToolCallPrimitive from "./primitives/toolCall";
 
 // Re-export shared providers from core/react
 export {
@@ -148,10 +129,3 @@ export {
 export * from "./model-context";
 export * from "./client";
 export * from "./types";
-
-// Adapters
-export {
-  type TitleGenerationAdapter,
-  createSimpleTitleAdapter,
-  createLocalStorageAdapter,
-} from "./adapters";

package/src/primitives/actionBar/ActionBarCopy.tsx

@@ -2,7 +2,7 @@ import type { ReactNode } from "react";
 import {
   useActionBarCopy,
   type UseActionBarCopyOptions,
-} from "../../primitive-hooks/useActionBarCopy";
+} from "@assistant-ui/core/react";
 import { Pressable, type PressableProps } from "../internal/Pressable";
 
 export type ActionBarCopyProps = Omit<PressableProps, "onPress" | "children"> &

package/src/primitives/actionBar/ActionBarEdit.tsx

@@ -1,5 +1,5 @@
 import type { ReactNode } from "react";
-import { useActionBarEdit } from "../../primitive-hooks/useActionBarEdit";
+import { useActionBarEdit } from "@assistant-ui/core/react";
 import { Pressable, type PressableProps } from "../internal/Pressable";
 
 export type ActionBarEditProps = Omit<PressableProps, "onPress"> & {

package/src/primitives/actionBar/ActionBarFeedbackNegative.tsx

@@ -1,5 +1,5 @@
 import type { ReactNode } from "react";
-import { useActionBarFeedbackNegative } from "../../primitive-hooks/useActionBarFeedback";
+import { useActionBarFeedbackNegative } from "@assistant-ui/core/react";
 import { Pressable, type PressableProps } from "../internal/Pressable";
 
 export type ActionBarFeedbackNegativeProps = Omit<

package/src/primitives/actionBar/ActionBarFeedbackPositive.tsx

@@ -1,5 +1,5 @@
 import type { ReactNode } from "react";
-import { useActionBarFeedbackPositive } from "../../primitive-hooks/useActionBarFeedback";
+import { useActionBarFeedbackPositive } from "@assistant-ui/core/react";
 import { Pressable, type PressableProps } from "../internal/Pressable";
 
 export type ActionBarFeedbackPositiveProps = Omit<

package/src/primitives/actionBar/ActionBarReload.tsx

@@ -1,5 +1,5 @@
 import type { ReactNode } from "react";
-import { useActionBarReload } from "../../primitive-hooks/useActionBarReload";
+import { useActionBarReload } from "@assistant-ui/core/react";
 import { Pressable, type PressableProps } from "../internal/Pressable";
 
 export type ActionBarReloadProps = Omit<PressableProps, "onPress"> & {

package/src/primitives/actionBar/index.ts

@@ -1,11 +1,20 @@
-export { ActionBarCopy, type ActionBarCopyProps } from "./ActionBarCopy";
-export { ActionBarEdit, type ActionBarEditProps } from "./ActionBarEdit";
-export { ActionBarReload, type ActionBarReloadProps } from "./ActionBarReload";
 export {
-  ActionBarFeedbackPositive,
-  type ActionBarFeedbackPositiveProps,
+  ActionBarCopy as Copy,
+  type ActionBarCopyProps as CopyProps,
+} from "./ActionBarCopy";
+export {
+  ActionBarEdit as Edit,
+  type ActionBarEditProps as EditProps,
+} from "./ActionBarEdit";
+export {
+  ActionBarReload as Reload,
+  type ActionBarReloadProps as ReloadProps,
+} from "./ActionBarReload";
+export {
+  ActionBarFeedbackPositive as FeedbackPositive,
+  type ActionBarFeedbackPositiveProps as FeedbackPositiveProps,
 } from "./ActionBarFeedbackPositive";
 export {
-  ActionBarFeedbackNegative,
-  type ActionBarFeedbackNegativeProps,
+  ActionBarFeedbackNegative as FeedbackNegative,
+  type ActionBarFeedbackNegativeProps as FeedbackNegativeProps,
 } from "./ActionBarFeedbackNegative";

package/src/primitives/attachment/index.ts

@@ -1,7 +1,16 @@
-export { AttachmentRoot, type AttachmentRootProps } from "./AttachmentRoot";
-export { AttachmentName, type AttachmentNameProps } from "./AttachmentName";
-export { AttachmentThumb, type AttachmentThumbProps } from "./AttachmentThumb";
 export {
-  AttachmentRemove,
-  type AttachmentRemoveProps,
+  AttachmentRoot as Root,
+  type AttachmentRootProps as RootProps,
+} from "./AttachmentRoot";
+export {
+  AttachmentName as Name,
+  type AttachmentNameProps as NameProps,
+} from "./AttachmentName";
+export {
+  AttachmentThumb as Thumb,
+  type AttachmentThumbProps as ThumbProps,
+} from "./AttachmentThumb";
+export {
+  AttachmentRemove as Remove,
+  type AttachmentRemoveProps as RemoveProps,
 } from "./AttachmentRemove";

package/src/primitives/branchPicker/index.ts

@@ -1,16 +1,16 @@
 export {
-  BranchPickerPrevious,
-  type BranchPickerPreviousProps,
+  BranchPickerPrevious as Previous,
+  type BranchPickerPreviousProps as PreviousProps,
 } from "./BranchPickerPrevious";
 export {
-  BranchPickerNext,
-  type BranchPickerNextProps,
+  BranchPickerNext as Next,
+  type BranchPickerNextProps as NextProps,
 } from "./BranchPickerNext";
 export {
-  BranchPickerNumber,
-  type BranchPickerNumberProps,
+  BranchPickerNumber as Number,
+  type BranchPickerNumberProps as NumberProps,
 } from "./BranchPickerNumber";
 export {
-  BranchPickerCount,
-  type BranchPickerCountProps,
+  BranchPickerCount as Count,
+  type BranchPickerCountProps as CountProps,
 } from "./BranchPickerCount";

package/src/primitives/chainOfThought/index.ts

@@ -1,9 +1,9 @@
 export {
-  ChainOfThoughtRoot,
-  type ChainOfThoughtRootProps,
+  ChainOfThoughtRoot as Root,
+  type ChainOfThoughtRootProps as RootProps,
 } from "./ChainOfThoughtRoot";
 export {
-  ChainOfThoughtAccordionTrigger,
-  type ChainOfThoughtAccordionTriggerProps,
+  ChainOfThoughtAccordionTrigger as AccordionTrigger,
+  type ChainOfThoughtAccordionTriggerProps as AccordionTriggerProps,
 } from "./ChainOfThoughtAccordionTrigger";
-export { ChainOfThoughtPrimitiveParts } from "@assistant-ui/core/react";
+export { ChainOfThoughtPrimitiveParts as Parts } from "@assistant-ui/core/react";

package/src/primitives/composer/ComposerAddAttachment.tsx

@@ -1,6 +1,5 @@
 import type { ReactNode } from "react";
-import { useAuiState } from "@assistant-ui/store";
-import { useComposerAddAttachment } from "../../primitive-hooks/useComposerAddAttachment";
+import { useComposerAddAttachment } from "@assistant-ui/core/react";
 import { Pressable, type PressableProps } from "../internal/Pressable";
 
 export type ComposerAddAttachmentProps = Omit<PressableProps, "onPress"> & {
@@ -12,11 +11,10 @@ export const ComposerAddAttachment = ({
   disabled: disabledProp,
   ...pressableProps
 }: ComposerAddAttachmentProps) => {
-  const isDisabled = useAuiState((s) => !s.composer.isEditing);
-  const { addAttachment: _addAttachment } = useComposerAddAttachment();
+  const { disabled } = useComposerAddAttachment();
 
   return (
-    <Pressable disabled={disabledProp ?? isDisabled} {...pressableProps}>
+    <Pressable disabled={disabledProp ?? disabled} {...pressableProps}>
       {children}
     </Pressable>
   );

package/src/primitives/composer/ComposerCancel.tsx

@@ -1,5 +1,5 @@
 import type { ReactNode } from "react";
-import { useComposerCancel } from "../../primitive-hooks/useComposerCancel";
+import { useComposerCancel } from "@assistant-ui/core/react";
 import { Pressable, type PressableProps } from "../internal/Pressable";
 
 export type ComposerCancelProps = Omit<PressableProps, "onPress"> & {
@@ -11,12 +11,12 @@ export const ComposerCancel = ({
   disabled,
   ...pressableProps
 }: ComposerCancelProps) => {
-  const { cancel, canCancel } = useComposerCancel();
+  const { cancel, disabled: hookDisabled } = useComposerCancel();
 
   return (
     <Pressable
       onPress={cancel}
-      disabled={disabled ?? !canCancel}
+      disabled={disabled ?? hookDisabled}
       {...pressableProps}
     >
       {children}

package/src/primitives/composer/ComposerSend.tsx

@@ -1,5 +1,5 @@
 import type { ReactNode } from "react";
-import { useComposerSend } from "../../primitive-hooks/useComposerSend";
+import { useComposerSend } from "@assistant-ui/core/react";
 import { Pressable, type PressableProps } from "../internal/Pressable";
 
 export type ComposerSendProps = Omit<PressableProps, "onPress"> & {
@@ -11,12 +11,12 @@ export const ComposerSend = ({
   disabled,
   ...pressableProps
 }: ComposerSendProps) => {
-  const { send, canSend } = useComposerSend();
+  const { send, disabled: hookDisabled } = useComposerSend();
 
   return (
     <Pressable
       onPress={send}
-      disabled={disabled ?? !canSend}
+      disabled={disabled ?? hookDisabled}
       {...pressableProps}
     >
       {children}

package/src/primitives/composer/index.ts

@@ -1,25 +1,25 @@
-export { ComposerRoot, type ComposerRootProps } from "./ComposerRoot";
 export {
-  ComposerAttachments,
-  ComposerAttachmentByIndex,
+  ComposerRoot as Root,
+  type ComposerRootProps as RootProps,
+} from "./ComposerRoot";
+export {
+  ComposerAttachments as Attachments,
+  ComposerAttachmentByIndex as AttachmentByIndex,
 } from "./ComposerAttachments";
-export { ComposerInput, type ComposerInputProps } from "./ComposerInput";
-export { ComposerSend, type ComposerSendProps } from "./ComposerSend";
-export { ComposerCancel, type ComposerCancelProps } from "./ComposerCancel";
 export {
-  ComposerAddAttachment,
-  type ComposerAddAttachmentProps,
-} from "./ComposerAddAttachment";
-export { ComposerIf } from "./ComposerIf";
+  ComposerInput as Input,
+  type ComposerInputProps as InputProps,
+} from "./ComposerInput";
 export {
-  EditComposerInput,
-  type EditComposerInputProps,
-} from "./EditComposerInput";
+  ComposerSend as Send,
+  type ComposerSendProps as SendProps,
+} from "./ComposerSend";
 export {
-  EditComposerSend,
-  type EditComposerSendProps,
-} from "./EditComposerSend";
+  ComposerCancel as Cancel,
+  type ComposerCancelProps as CancelProps,
+} from "./ComposerCancel";
 export {
-  EditComposerCancel,
-  type EditComposerCancelProps,
-} from "./EditComposerCancel";
+  ComposerAddAttachment as AddAttachment,
+  type ComposerAddAttachmentProps as AddAttachmentProps,
+} from "./ComposerAddAttachment";
+export { ComposerIf as If } from "./ComposerIf";