@assistant-ui/react 0.11.58 → 0.12.1

package/dist/{client/types/Thread.d.ts → types/scopes/thread.d.ts}

@@ -1,12 +1,12 @@
-import { ReadonlyJSONValue } from "assistant-stream/utils";
-import { ComposerClientState, ComposerClientApi } from "./Composer.js";
-import { MessageClientApi, MessageClientState } from "./Message.js";
-import { CreateAppendMessage, CreateStartRunConfig } from "../../legacy-runtime/runtime/index.js";
-import { ThreadSuggestion, ExportedMessageRepository, ThreadMessageLike } from "../../legacy-runtime/runtime-cores/index.js";
-import { RuntimeCapabilities, SpeechState } from "../../legacy-runtime/runtime-cores/core/ThreadRuntimeCore.js";
-import { CreateResumeRunConfig } from "../../legacy-runtime/runtime/ThreadRuntime.js";
-import { ModelContext } from "../../model-context/index.js";
-export type ThreadClientState = {
+import type { ReadonlyJSONValue } from "assistant-stream/utils";
+import type { RuntimeCapabilities, SpeechState } from "../../legacy-runtime/runtime-cores/core/ThreadRuntimeCore.js";
+import type { ThreadSuggestion, ExportedMessageRepository, ThreadMessageLike } from "../../legacy-runtime/runtime-cores/index.js";
+import type { CreateAppendMessage, CreateStartRunConfig } from "../../legacy-runtime/runtime/index.js";
+import type { CreateResumeRunConfig } from "../../legacy-runtime/runtime/ThreadRuntime.js";
+import type { ModelContext } from "../../model-context/index.js";
+import type { MessageMethods, MessageState } from "./message.js";
+import type { ComposerMethods, ComposerState } from "./composer.js";
+export type ThreadState = {
     /**
      * Whether the thread is empty. A thread is considered empty when it has no messages and is not loading.
      */
@@ -30,10 +30,9 @@ export type ThreadClientState = {
     /**
      * The messages in the currently selected branch of the thread.
      */
-    readonly messages: readonly MessageClientState[];
+    readonly messages: readonly MessageState[];
     /**
      * The thread state.
-     *
      * @deprecated This feature is experimental
      */
     readonly state: ReadonlyJSONValue;
@@ -45,21 +44,19 @@ export type ThreadClientState = {
      * Custom extra information provided by the runtime.
      */
     readonly extras: unknown;
-    /**
-     * @deprecated This API is still under active development and might change without notice.
-     */
+    /** @deprecated This API is still under active development and might change without notice. */
     readonly speech: SpeechState | undefined;
-    readonly composer: ComposerClientState;
+    readonly composer: ComposerState;
 };
-export type ThreadClientApi = {
+export type ThreadMethods = {
     /**
      * Get the current state of the thread.
      */
-    getState(): ThreadClientState;
+    getState(): ThreadState;
     /**
      * The thread composer runtime.
      */
-    readonly composer: ComposerClientApi;
+    composer: ComposerMethods;
     /**
      * Append a new message to the thread.
      *
@@ -85,7 +82,7 @@ export type ThreadClientApi = {
     /**
      * Resume a run with the given configuration.
      * @param config The configuration for resuming the run
-     **/
+     */
     unstable_resumeRun(config: CreateResumeRunConfig): void;
     cancelRun(): void;
     getModelContext(): ModelContext;
@@ -93,7 +90,6 @@ export type ThreadClientApi = {
     import(repository: ExportedMessageRepository): void;
     /**
      * Reset the thread with optional initial messages.
-     *
      * @param initialMessages - Optional array of initial messages to populate the thread
      */
     reset(initialMessages?: readonly ThreadMessageLike[]): void;
@@ -101,10 +97,8 @@ export type ThreadClientApi = {
         id: string;
     } | {
         index: number;
-    }): MessageClientApi;
-    /**
-     * @deprecated This API is still under active development and might change without notice.
-     */
+    }): MessageMethods;
+    /** @deprecated This API is still under active development and might change without notice. */
     stopSpeaking(): void;
     /**
      * Start the voice session for the thread. Establishes any necessary media connections.
@@ -115,4 +109,30 @@ export type ThreadClientApi = {
      */
     stopVoice(): Promise<void>;
 };
-//# sourceMappingURL=Thread.d.ts.map
+export type ThreadMeta = {
+    source: "threads";
+    query: {
+        type: "main";
+    };
+};
+export type ThreadEvents = {
+    "thread.runStart": {
+        threadId: string;
+    };
+    "thread.runEnd": {
+        threadId: string;
+    };
+    "thread.initialize": {
+        threadId: string;
+    };
+    "thread.modelContextUpdate": {
+        threadId: string;
+    };
+};
+export type ThreadClientSchema = {
+    state: ThreadState;
+    methods: ThreadMethods;
+    meta: ThreadMeta;
+    events: ThreadEvents;
+};
+//# sourceMappingURL=thread.d.ts.map

package/dist/types/scopes/thread.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"thread.d.ts","sourceRoot":"","sources":["../../../src/types/scopes/thread.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,KAAK,EACV,mBAAmB,EACnB,WAAW,EACZ,qEAAkE;AACnE,OAAO,KAAK,EACV,gBAAgB,EAChB,yBAAyB,EACzB,iBAAiB,EAClB,oDAA2C;AAC5C,OAAO,KAAK,EACV,mBAAmB,EACnB,oBAAoB,EAErB,8CAAqC;AACtC,OAAO,KAAK,EAAE,qBAAqB,EAAE,sDAAmD;AACxF,OAAO,KAAK,EAAE,YAAY,EAAE,qCAA4B;AACxD,OAAO,KAAK,EAAE,cAAc,EAAE,YAAY,EAAE,qBAAkB;AAC9D,OAAO,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,sBAAmB;AAEjE,MAAM,MAAM,WAAW,GAAG;IACxB;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B;;OAEG;IACH,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC;IAC7B;;OAEG;IACH,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;IAC5B;;OAEG;IACH,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;IAC5B;;OAEG;IACH,QAAQ,CAAC,YAAY,EAAE,mBAAmB,CAAC;IAC3C;;OAEG;IACH,QAAQ,CAAC,QAAQ,EAAE,SAAS,YAAY,EAAE,CAAC;IAC3C;;;OAGG;IACH,QAAQ,CAAC,KAAK,EAAE,iBAAiB,CAAC;IAClC;;OAEG;IACH,QAAQ,CAAC,WAAW,EAAE,SAAS,gBAAgB,EAAE,CAAC;IAClD;;OAEG;IACH,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;IACzB,8FAA8F;IAC9F,QAAQ,CAAC,MAAM,EAAE,WAAW,GAAG,SAAS,CAAC;IACzC,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC;CAClC,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B;;OAEG;IACH,QAAQ,IAAI,WAAW,CAAC;IACxB;;OAEG;IACH,QAAQ,EAAE,eAAe,CAAC;IAC1B;;;;;;;;;;;;;;;OAeG;IACH,MAAM,CAAC,OAAO,EAAE,mBAAmB,GAAG,IAAI,CAAC;IAC3C;;;OAGG;IACH,QAAQ,CAAC,MAAM,EAAE,oBAAoB,GAAG,IAAI,CAAC;IAC7C;;;OAGG;IACH,kBAAkB,CAAC,MAAM,EAAE,qBAAqB,GAAG,IAAI,CAAC;IACxD,SAAS,IAAI,IAAI,CAAC;IAClB,eAAe,IAAI,YAAY,CAAC;IAChC,MAAM,IAAI,yBAAyB,CAAC;IACpC,MAAM,CAAC,UAAU,EAAE,yBAAyB,GAAG,IAAI,CAAC;IACpD;;;OAGG;IACH,KAAK,CAAC,eAAe,CAAC,EAAE,SAAS,iBAAiB,EAAE,GAAG,IAAI,CAAC;IAC5D,OAAO,CAAC,QAAQ,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,cAAc,CAAC;IACtE,8FAA8F;IAC9F,YAAY,IAAI,IAAI,CAAC;IACrB;;OAEG;IACH,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B;;OAEG;IACH,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAG5B,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG;IACvB,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,iBAAiB,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IACxC,eAAe,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IACtC,mBAAmB,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IAC1C,2BAA2B,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;CACnD,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,KAAK,EAAE,WAAW,CAAC;IACnB,OAAO,EAAE,aAAa,CAAC;IACvB,IAAI,EAAE,UAAU,CAAC;IACjB,MAAM,EAAE,YAAY,CAAC;CACtB,CAAC"}

package/dist/types/scopes/thread.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=thread.js.map

package/dist/types/scopes/thread.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thread.js","sourceRoot":"","sources":["../../../src/types/scopes/thread.ts"],"names":[],"mappings":""}

package/dist/types/scopes/threadListItem.d.ts

@@ -0,0 +1,50 @@
+import type { ThreadListItemStatus } from "../../legacy-runtime/runtime/ThreadListItemRuntime.js";
+export type ThreadListItemState = {
+    readonly id: string;
+    readonly remoteId: string | undefined;
+    readonly externalId: string | undefined;
+    readonly title?: string | undefined;
+    readonly status: ThreadListItemStatus;
+};
+export type ThreadListItemMethods = {
+    getState(): ThreadListItemState;
+    switchTo(): void;
+    rename(newTitle: string): void;
+    archive(): void;
+    unarchive(): void;
+    delete(): void;
+    generateTitle(): void;
+    initialize(): Promise<{
+        remoteId: string;
+        externalId: string | undefined;
+    }>;
+    detach(): void;
+};
+export type ThreadListItemMeta = {
+    source: "threads";
+    query: {
+        type: "main";
+    } | {
+        type: "id";
+        id: string;
+    } | {
+        type: "index";
+        index: number;
+        archived?: boolean;
+    };
+};
+export type ThreadListItemEvents = {
+    "threadListItem.switchedTo": {
+        threadId: string;
+    };
+    "threadListItem.switchedAway": {
+        threadId: string;
+    };
+};
+export type ThreadListItemClientSchema = {
+    state: ThreadListItemState;
+    methods: ThreadListItemMethods;
+    meta: ThreadListItemMeta;
+    events: ThreadListItemEvents;
+};
+//# sourceMappingURL=threadListItem.d.ts.map

package/dist/types/scopes/threadListItem.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"threadListItem.d.ts","sourceRoot":"","sources":["../../../src/types/scopes/threadListItem.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,oBAAoB,EAAE,8DAA2D;AAE/F,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IACtC,QAAQ,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,CAAC;IACxC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACpC,QAAQ,CAAC,MAAM,EAAE,oBAAoB,CAAC;CACvC,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG;IAClC,QAAQ,IAAI,mBAAmB,CAAC;IAChC,QAAQ,IAAI,IAAI,CAAC;IACjB,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,OAAO,IAAI,IAAI,CAAC;IAChB,SAAS,IAAI,IAAI,CAAC;IAClB,MAAM,IAAI,IAAI,CAAC;IACf,aAAa,IAAI,IAAI,CAAC;IACtB,UAAU,IAAI,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE,CAAC,CAAC;IAC5E,MAAM,IAAI,IAAI,CAAC;CAGhB,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EACD;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,GAChB;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,EAAE,EAAE,MAAM,CAAA;KAAE,GAC1B;QAAE,IAAI,EAAE,OAAO,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;CAC1D,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,2BAA2B,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IAClD,6BAA6B,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;CACrD,CAAC;AAEF,MAAM,MAAM,0BAA0B,GAAG;IACvC,KAAK,EAAE,mBAAmB,CAAC;IAC3B,OAAO,EAAE,qBAAqB,CAAC;IAC/B,IAAI,EAAE,kBAAkB,CAAC;IACzB,MAAM,EAAE,oBAAoB,CAAC;CAC9B,CAAC"}

package/dist/types/scopes/threadListItem.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=threadListItem.js.map

package/dist/types/scopes/threadListItem.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"threadListItem.js","sourceRoot":"","sources":["../../../src/types/scopes/threadListItem.ts"],"names":[],"mappings":""}

package/dist/types/scopes/threads.d.ts

@@ -0,0 +1,28 @@
+import type { ThreadListItemMethods, ThreadListItemState } from "./threadListItem.js";
+import type { ThreadMethods, ThreadState } from "./thread.js";
+export type ThreadsState = {
+    readonly mainThreadId: string;
+    readonly newThreadId: string | null;
+    readonly isLoading: boolean;
+    readonly threadIds: readonly string[];
+    readonly archivedThreadIds: readonly string[];
+    readonly threadItems: readonly ThreadListItemState[];
+    readonly main: ThreadState;
+};
+export type ThreadsMethods = {
+    getState(): ThreadsState;
+    switchToThread(threadId: string): void;
+    switchToNewThread(): void;
+    item(threadIdOrOptions: "main" | {
+        id: string;
+    } | {
+        index: number;
+        archived?: boolean;
+    }): ThreadListItemMethods;
+    thread(selector: "main"): ThreadMethods;
+};
+export type ThreadsClientSchema = {
+    state: ThreadsState;
+    methods: ThreadsMethods;
+};
+//# sourceMappingURL=threads.d.ts.map

package/dist/types/scopes/threads.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"threads.d.ts","sourceRoot":"","sources":["../../../src/types/scopes/threads.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,qBAAqB,EACrB,mBAAmB,EACpB,4BAAyB;AAC1B,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,oBAAiB;AAE3D,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;IAC5B,QAAQ,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAAC;IACtC,QAAQ,CAAC,iBAAiB,EAAE,SAAS,MAAM,EAAE,CAAC;IAC9C,QAAQ,CAAC,WAAW,EAAE,SAAS,mBAAmB,EAAE,CAAC;IACrD,QAAQ,CAAC,IAAI,EAAE,WAAW,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,IAAI,YAAY,CAAC;IACzB,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACvC,iBAAiB,IAAI,IAAI,CAAC;IAC1B,IAAI,CACF,iBAAiB,EACb,MAAM,GACN;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GACd;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GACxC,qBAAqB,CAAC;IACzB,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,aAAa,CAAC;CAGzC,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,KAAK,EAAE,YAAY,CAAC;IACpB,OAAO,EAAE,cAAc,CAAC;CACzB,CAAC"}

package/dist/types/scopes/threads.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=threads.js.map

package/dist/types/scopes/threads.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"threads.js","sourceRoot":"","sources":["../../../src/types/scopes/threads.ts"],"names":[],"mappings":""}

package/dist/types/scopes/tools.d.ts

@@ -0,0 +1,14 @@
+import type { ToolCallMessagePartComponent } from "../MessagePartComponentTypes.js";
+import type { Unsubscribe } from "../Unsubscribe.js";
+export type ToolsState = {
+    tools: Record<string, ToolCallMessagePartComponent[]>;
+};
+export type ToolsMethods = {
+    getState(): ToolsState;
+    setToolUI(toolName: string, render: ToolCallMessagePartComponent): Unsubscribe;
+};
+export type ToolsClientSchema = {
+    state: ToolsState;
+    methods: ToolsMethods;
+};
+//# sourceMappingURL=tools.d.ts.map

package/dist/types/scopes/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../../src/types/scopes/tools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,4BAA4B,EAAE,wCAAqC;AACjF,OAAO,KAAK,EAAE,WAAW,EAAE,0BAAuB;AAElD,MAAM,MAAM,UAAU,GAAG;IACvB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,4BAA4B,EAAE,CAAC,CAAC;CACvD,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,IAAI,UAAU,CAAC;IACvB,SAAS,CACP,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,4BAA4B,GACnC,WAAW,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,KAAK,EAAE,UAAU,CAAC;IAClB,OAAO,EAAE,YAAY,CAAC;CACvB,CAAC"}

package/dist/types/scopes/tools.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=tools.js.map

package/dist/types/scopes/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../src/types/scopes/tools.ts"],"names":[],"mappings":""}

package/dist/types/store-augmentation.d.ts

@@ -0,0 +1,24 @@
+import "@assistant-ui/store";
+import type { ThreadsClientSchema } from "./scopes/threads.js";
+import type { ThreadListItemClientSchema } from "./scopes/threadListItem.js";
+import type { ThreadClientSchema } from "./scopes/thread.js";
+import type { MessageClientSchema } from "./scopes/message.js";
+import type { PartClientSchema } from "./scopes/part.js";
+import type { ComposerClientSchema } from "./scopes/composer.js";
+import type { AttachmentClientSchema } from "./scopes/attachment.js";
+import type { ToolsClientSchema } from "./scopes/tools.js";
+import type { ModelContextClientSchema } from "./scopes/modelContext.js";
+declare module "@assistant-ui/store" {
+    interface ClientRegistry {
+        threads: ThreadsClientSchema;
+        threadListItem: ThreadListItemClientSchema;
+        thread: ThreadClientSchema;
+        message: MessageClientSchema;
+        part: PartClientSchema;
+        composer: ComposerClientSchema;
+        attachment: AttachmentClientSchema;
+        tools: ToolsClientSchema;
+        modelContext: ModelContextClientSchema;
+    }
+}
+//# sourceMappingURL=store-augmentation.d.ts.map

package/dist/types/store-augmentation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"store-augmentation.d.ts","sourceRoot":"","sources":["../../src/types/store-augmentation.ts"],"names":[],"mappings":"AAAA,OAAO,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,mBAAmB,EAAE,4BAAyB;AAC5D,OAAO,KAAK,EAAE,0BAA0B,EAAE,mCAAgC;AAC1E,OAAO,KAAK,EAAE,kBAAkB,EAAE,2BAAwB;AAC1D,OAAO,KAAK,EAAE,mBAAmB,EAAE,4BAAyB;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,yBAAsB;AACtD,OAAO,KAAK,EAAE,oBAAoB,EAAE,6BAA0B;AAC9D,OAAO,KAAK,EAAE,sBAAsB,EAAE,+BAA4B;AAClE,OAAO,KAAK,EAAE,iBAAiB,EAAE,0BAAuB;AACxD,OAAO,KAAK,EAAE,wBAAwB,EAAE,iCAA8B;AAEtE,OAAO,QAAQ,qBAAqB,CAAC;IACnC,UAAU,cAAc;QACtB,OAAO,EAAE,mBAAmB,CAAC;QAC7B,cAAc,EAAE,0BAA0B,CAAC;QAC3C,MAAM,EAAE,kBAAkB,CAAC;QAC3B,OAAO,EAAE,mBAAmB,CAAC;QAC7B,IAAI,EAAE,gBAAgB,CAAC;QACvB,QAAQ,EAAE,oBAAoB,CAAC;QAC/B,UAAU,EAAE,sBAAsB,CAAC;QACnC,KAAK,EAAE,iBAAiB,CAAC;QACzB,YAAY,EAAE,wBAAwB,CAAC;KACxC;CACF"}

package/dist/types/store-augmentation.js

@@ -0,0 +1,2 @@
+import "@assistant-ui/store";
+//# sourceMappingURL=store-augmentation.js.map

package/dist/types/store-augmentation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store-augmentation.js","sourceRoot":"","sources":["../../src/types/store-augmentation.ts"],"names":[],"mappings":"AAAA,OAAO,qBAAqB,CAAC"}

package/dist/utils/smooth/SmoothContext.js

@@ -2,7 +2,7 @@
 import { jsx as _jsx } from "react/jsx-runtime";
 import { createContext, forwardRef, useContext, useState, } from "react";
 import { create } from "zustand";
-import { useAssistantApi } from "../../context/index.js";
+import { useAui } from "@assistant-ui/store";
 import { createContextStoreHook } from "../../context/react/utils/createContextStoreHook.js";
 const SmoothContext = createContext(null);
 const makeSmoothContext = (initialState) => {
@@ -11,8 +11,8 @@ const makeSmoothContext = (initialState) => {
 };
 export const SmoothContextProvider = ({ children }) => {
     const outer = useSmoothContext({ optional: true });
-    const api = useAssistantApi();
-    const [context] = useState(() => makeSmoothContext(api.part().getState().status));
+    const aui = useAui();
+    const [context] = useState(() => makeSmoothContext(aui.part().getState().status));
     // do not wrap if there is an outer SmoothContextProvider
     if (outer)
         return children;

package/dist/utils/smooth/SmoothContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"SmoothContext.js","sourceRoot":"","sources":["../../../src/utils/smooth/SmoothContext.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAEL,aAAa,EAEb,UAAU,EAEV,UAAU,EACV,QAAQ,GACT,MAAM,OAAO,CAAC;AAEf,OAAO,EAAE,MAAM,EAAiB,MAAM,SAAS,CAAC;AAKhD,OAAO,EAAE,eAAe,EAAE,+BAAsB;AAChD,OAAO,EAAE,sBAAsB,EAAE,4DAAyD;AAQ1F,MAAM,aAAa,GAAG,aAAa,CAA4B,IAAI,CAAC,CAAC;AAErE,MAAM,iBAAiB,GAAG,CACxB,YAA2D,EAC3D,EAAE;IACF,MAAM,eAAe,GAAG,MAAM,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC;IACnD,OAAO,EAAE,eAAe,EAAE,CAAC;AAC7B,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAA0B,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,eAAe,EAAE,CAAC;IAE9B,MAAM,CAAC,OAAO,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAC9B,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAChD,CAAC;IAEF,yDAAyD;IACzD,IAAI,KAAK;QAAE,OAAO,QAAQ,CAAC;IAE3B,OAAO,CACL,KAAC,aAAa,CAAC,QAAQ,IAAC,KAAK,EAAE,OAAO,YAAG,QAAQ,GAA0B,CAC5E,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,SAAY,EACT,EAAE;IACL,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACxC,OAAO,CACL,KAAC,qBAAqB,cACpB,KAAC,SAAS,OAAM,KAAa,EAAE,GAAG,EAAE,GAAG,GAAI,GACrB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;IACH,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,WAAW,CAAC;IAC5C,OAAO,OAAc,CAAC;AACxB,CAAC,CAAC;AAQF,SAAS,gBAAgB,CAAC,OAA4C;IACpE,MAAM,OAAO,GAAG,UAAU,CAAC,aAAa,CAAC,CAAC;IAC1C,IAAI,CAAC,OAAO,EAAE,QAAQ,IAAI,CAAC,OAAO;QAChC,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;IACJ,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,MAAM,EAAE,eAAe,EAAE,oBAAoB,EAAE,GAAG,sBAAsB,CAC7E,gBAAgB,EAChB,iBAAiB,CAClB,CAAC"}
+{"version":3,"file":"SmoothContext.js","sourceRoot":"","sources":["../../../src/utils/smooth/SmoothContext.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAEL,aAAa,EAEb,UAAU,EAEV,UAAU,EACV,QAAQ,GACT,MAAM,OAAO,CAAC;AAEf,OAAO,EAAE,MAAM,EAAiB,MAAM,SAAS,CAAC;AAKhD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,4DAAyD;AAQ1F,MAAM,aAAa,GAAG,aAAa,CAA4B,IAAI,CAAC,CAAC;AAErE,MAAM,iBAAiB,GAAG,CACxB,YAA2D,EAC3D,EAAE;IACF,MAAM,eAAe,GAAG,MAAM,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC;IACnD,OAAO,EAAE,eAAe,EAAE,CAAC;AAC7B,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAA0B,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IAErB,MAAM,CAAC,OAAO,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAC9B,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAChD,CAAC;IAEF,yDAAyD;IACzD,IAAI,KAAK;QAAE,OAAO,QAAQ,CAAC;IAE3B,OAAO,CACL,KAAC,aAAa,CAAC,QAAQ,IAAC,KAAK,EAAE,OAAO,YAAG,QAAQ,GAA0B,CAC5E,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,SAAY,EACT,EAAE;IACL,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACxC,OAAO,CACL,KAAC,qBAAqB,cACpB,KAAC,SAAS,OAAM,KAAa,EAAE,GAAG,EAAE,GAAG,GAAI,GACrB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;IACH,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,WAAW,CAAC;IAC5C,OAAO,OAAc,CAAC;AACxB,CAAC,CAAC;AAQF,SAAS,gBAAgB,CAAC,OAA4C;IACpE,MAAM,OAAO,GAAG,UAAU,CAAC,aAAa,CAAC,CAAC;IAC1C,IAAI,CAAC,OAAO,EAAE,QAAQ,IAAI,CAAC,OAAO;QAChC,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;IACJ,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,MAAM,EAAE,eAAe,EAAE,oBAAoB,EAAE,GAAG,sBAAsB,CAC7E,gBAAgB,EAChB,iBAAiB,CAClB,CAAC"}

package/dist/utils/smooth/useSmooth.js

@@ -1,6 +1,6 @@
 "use client";
 import { useEffect, useMemo, useRef, useState } from "react";
-import { useAssistantState } from "../../context/index.js";
+import { useAuiState } from "@assistant-ui/store";
 import { useCallbackRef } from "@radix-ui/react-use-callback-ref";
 import { useSmoothStatusStore } from "./SmoothContext.js";
 import { writableStore } from "../../context/ReadonlyStore.js";
@@ -55,7 +55,7 @@ const SMOOTH_STATUS = Object.freeze({
 });
 export const useSmooth = (state, smooth = false) => {
     const { text } = state;
-    const id = useAssistantState(({ message }) => message.id);
+    const id = useAuiState(({ message }) => message.id);
     const idRef = useRef(id);
     const [displayedText, setDisplayedText] = useState(text);
     const smoothStatusStore = useSmoothStatusStore({ optional: true });

package/dist/utils/smooth/useSmooth.js.map

@@ -1 +1 @@
-{"version":3,"file":"useSmooth.js","sourceRoot":"","sources":["../../../src/utils/smooth/useSmooth.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,+BAAsB;AAMlD,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,oBAAoB,EAAE,2BAAwB;AACvD,OAAO,EAAE,aAAa,EAAE,uCAAoC;AAG5D,MAAM,kBAAkB;IAOb;IACC;IAPF,gBAAgB,GAAkB,IAAI,CAAC;IACvC,cAAc,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;IAErC,UAAU,GAAW,EAAE,CAAC;IAE/B,YACS,WAAmB,EAClB,OAAkC;QADnC,gBAAW,GAAX,WAAW,CAAQ;QAClB,YAAO,GAAP,OAAO,CAA2B;IACzC,CAAC;IAEJ,KAAK;QACH,IAAI,IAAI,CAAC,gBAAgB,KAAK,IAAI;YAAE,OAAO;QAC3C,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,gBAAgB,KAAK,IAAI,EAAE,CAAC;YACnC,oBAAoB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC5C,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;IACH,CAAC;IAEO,OAAO,GAAG,GAAG,EAAE;QACrB,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC;QACpD,IAAI,aAAa,GAAG,SAAS,CAAC;QAE9B,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QACxE,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,GAAG,cAAc,CAAC,CAAC;QAE1D,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,OAAO,aAAa,IAAI,eAAe,IAAI,UAAU,GAAG,cAAc,EAAE,CAAC;YACvE,UAAU,EAAE,CAAC;YACb,aAAa,IAAI,eAAe,CAAC;QACnC,CAAC;QAED,IAAI,UAAU,KAAK,cAAc,EAAE,CAAC;YAClC,IAAI,CAAC,gBAAgB,GAAG,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;QACD,IAAI,UAAU,KAAK,CAAC;YAAE,OAAO;QAE7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CACtC,CAAC,EACD,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,UAAU,CACrC,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,WAAW,GAAG,aAAa,CAAC;QAClD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACjC,CAAC,CAAC;CACH;AAED,MAAM,aAAa,GAAsB,MAAM,CAAC,MAAM,CAAC;IACrD,IAAI,EAAE,SAAS;CAChB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,SAAS,GAAG,CACvB,KAAkE,EAClE,SAAkB,KAAK,EACsC,EAAE;IAC/D,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;IACvB,MAAM,EAAE,GAAG,iBAAiB,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAE1D,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC;IACzB,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAEzD,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,MAAM,OAAO,GAAG,cAAc,CAAC,CAAC,IAAY,EAAE,EAAE;QAC9C,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACvB,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,MAAM,GACV,aAAa,KAAK,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS;gBACvD,CAAC,CAAC,aAAa;gBACf,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;YACnB,aAAa,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,qBAAqB;IACrB,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,MAAM,GACV,MAAM,IAAI,CAAC,aAAa,KAAK,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC;gBACnE,CAAC,CAAC,aAAa;gBACf,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;YACnB,aAAa,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnE,MAAM,CAAC,WAAW,CAAC,GAAG,QAAQ,CAC5B,IAAI,kBAAkB,CAAC,IAAI,EAAE,OAAO,CAAC,CACtC,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,WAAW,CAAC,IAAI,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC;YACrE,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,CAAC;YAEd,WAAW,CAAC,WAAW,GAAG,IAAI,CAAC;YAC/B,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC;YAC9B,WAAW,CAAC,IAAI,EAAE,CAAC;YAEnB,OAAO;QACT,CAAC;QAED,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC;QAC9B,WAAW,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;IAE7C,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,GAAG,EAAE;YACV,WAAW,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;IAElB,OAAO,OAAO,CACZ,GAAG,EAAE,CACH,MAAM;QACJ,CAAC,CAAC;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,aAAa;YACnB,MAAM,EAAE,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa;SAC9D;QACH,CAAC,CAAC,KAAK,EACX,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,IAAI,CAAC,CACrC,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"useSmooth.js","sourceRoot":"","sources":["../../../src/utils/smooth/useSmooth.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC7D,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAMlD,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,oBAAoB,EAAE,2BAAwB;AACvD,OAAO,EAAE,aAAa,EAAE,uCAAoC;AAG5D,MAAM,kBAAkB;IAOb;IACC;IAPF,gBAAgB,GAAkB,IAAI,CAAC;IACvC,cAAc,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;IAErC,UAAU,GAAW,EAAE,CAAC;IAE/B,YACS,WAAmB,EAClB,OAAkC;QADnC,gBAAW,GAAX,WAAW,CAAQ;QAClB,YAAO,GAAP,OAAO,CAA2B;IACzC,CAAC;IAEJ,KAAK;QACH,IAAI,IAAI,CAAC,gBAAgB,KAAK,IAAI;YAAE,OAAO;QAC3C,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,gBAAgB,KAAK,IAAI,EAAE,CAAC;YACnC,oBAAoB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC5C,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;IACH,CAAC;IAEO,OAAO,GAAG,GAAG,EAAE;QACrB,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC;QACpD,IAAI,aAAa,GAAG,SAAS,CAAC;QAE9B,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QACxE,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,GAAG,cAAc,CAAC,CAAC;QAE1D,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,OAAO,aAAa,IAAI,eAAe,IAAI,UAAU,GAAG,cAAc,EAAE,CAAC;YACvE,UAAU,EAAE,CAAC;YACb,aAAa,IAAI,eAAe,CAAC;QACnC,CAAC;QAED,IAAI,UAAU,KAAK,cAAc,EAAE,CAAC;YAClC,IAAI,CAAC,gBAAgB,GAAG,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;QACD,IAAI,UAAU,KAAK,CAAC;YAAE,OAAO;QAE7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CACtC,CAAC,EACD,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,UAAU,CACrC,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,WAAW,GAAG,aAAa,CAAC;QAClD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACjC,CAAC,CAAC;CACH;AAED,MAAM,aAAa,GAAsB,MAAM,CAAC,MAAM,CAAC;IACrD,IAAI,EAAE,SAAS;CAChB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,SAAS,GAAG,CACvB,KAAkE,EAClE,SAAkB,KAAK,EACsC,EAAE;IAC/D,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;IACvB,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAEpD,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC;IACzB,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAEzD,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,MAAM,OAAO,GAAG,cAAc,CAAC,CAAC,IAAY,EAAE,EAAE;QAC9C,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACvB,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,MAAM,GACV,aAAa,KAAK,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS;gBACvD,CAAC,CAAC,aAAa;gBACf,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;YACnB,aAAa,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,qBAAqB;IACrB,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,MAAM,GACV,MAAM,IAAI,CAAC,aAAa,KAAK,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC;gBACnE,CAAC,CAAC,aAAa;gBACf,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;YACnB,aAAa,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnE,MAAM,CAAC,WAAW,CAAC,GAAG,QAAQ,CAC5B,IAAI,kBAAkB,CAAC,IAAI,EAAE,OAAO,CAAC,CACtC,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,WAAW,CAAC,IAAI,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC;YACrE,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,CAAC;YAEd,WAAW,CAAC,WAAW,GAAG,IAAI,CAAC;YAC/B,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC;YAC9B,WAAW,CAAC,IAAI,EAAE,CAAC;YAEnB,OAAO;QACT,CAAC;QAED,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC;QAC9B,WAAW,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;IAE7C,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,GAAG,EAAE;YACV,WAAW,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;IAElB,OAAO,OAAO,CACZ,GAAG,EAAE,CACH,MAAM;QACJ,CAAC,CAAC;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,aAAa;YACnB,MAAM,EAAE,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa;SAC9D;QACH,CAAC,CAAC,KAAK,EACX,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,IAAI,CAAC,CACrC,CAAC;AACJ,CAAC,CAAC"}

package/dist/utils/useToolArgsFieldStatus.js

@@ -1,8 +1,8 @@
 import { getPartialJsonObjectFieldState } from "assistant-stream/utils";
-import { useAssistantState } from "../context/index.js";
+import { useAuiState } from "@assistant-ui/store";
 const COMPLETE_STATUS = { type: "complete" };
 export const useToolArgsFieldStatus = (fieldPath) => {
-    return useAssistantState(({ part }) => {
+    return useAuiState(({ part }) => {
         if (part.type !== "tool-call")
             throw new Error("useToolArgsFieldStatus can only be used inside tool-call message parts");
         const state = getPartialJsonObjectFieldState(part.args, fieldPath);

package/dist/utils/useToolArgsFieldStatus.js.map

@@ -1 +1 @@
-{"version":3,"file":"useToolArgsFieldStatus.js","sourceRoot":"","sources":["../../src/utils/useToolArgsFieldStatus.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,8BAA8B,EAAE,MAAM,wBAAwB,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,4BAAmB;AAE/C,MAAM,eAAe,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;AAE7C,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,SAA8B,EAAE,EAAE;IACvE,OAAO,iBAAiB,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE;QACpC,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW;YAC3B,MAAM,IAAI,KAAK,CACb,wEAAwE,CACzE,CAAC;QAEJ,MAAM,KAAK,GAAG,8BAA8B,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACnE,IAAI,KAAK,KAAK,UAAU,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,iBAAiB;YACjE,OAAO,eAAe,CAAC;QACzB,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC"}
+{"version":3,"file":"useToolArgsFieldStatus.js","sourceRoot":"","sources":["../../src/utils/useToolArgsFieldStatus.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,8BAA8B,EAAE,MAAM,wBAAwB,CAAC;AACxE,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,MAAM,eAAe,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;AAE7C,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,SAA8B,EAAE,EAAE;IACvE,OAAO,WAAW,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE;QAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW;YAC3B,MAAM,IAAI,KAAK,CACb,wEAAwE,CACzE,CAAC;QAEJ,MAAM,KAAK,GAAG,8BAA8B,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACnE,IAAI,KAAK,KAAK,UAAU,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,iBAAiB;YACjE,OAAO,eAAe,CAAC;QACzB,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react",
-  "version": "0.11.58",
+  "version": "0.12.1",
   "description": "TypeScript/React library for AI Chat",
   "keywords": [
     "radix-ui",
@@ -50,7 +50,8 @@
   "sideEffects": false,
   "dependencies": {
     "assistant-cloud": "^0.1.13",
-    "@assistant-ui/tap": "^0.3.6",
+    "@assistant-ui/tap": "^0.4.0",
+    "@assistant-ui/store": "^0.1.0",
     "@radix-ui/primitive": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.2",
     "@radix-ui/react-context": "^1.1.3",

package/src/client/ModelContextClient.ts

@@ -1,16 +1,23 @@
-import { resource, tapState } from "@assistant-ui/tap";
-import { tapApi } from "../utils/tap-store";
+import { resource, tapMemo, tapState } from "@assistant-ui/tap";
+import { type ClientOutput } from "@assistant-ui/store";
 import { CompositeContextProvider } from "../utils/CompositeContextProvider";
-import type { ModelContextState, ModelContextApi } from "./types/ModelContext";
+import type { ModelContextState } from "../types/scopes";
 
-export const ModelContext = resource(() => {
-  const [state] = tapState<ModelContextState>(() => ({}));
-  const composite = new CompositeContextProvider();
+const version = 1;
 
-  return tapApi<ModelContextApi>({
-    getState: () => state,
-    getModelContext: () => composite.getModelContext(),
-    subscribe: (callback) => composite.subscribe(callback),
-    register: (provider) => composite.registerModelContextProvider(provider),
-  });
+export const ModelContext = resource((): ClientOutput<"modelContext"> => {
+  const [state] = tapState<ModelContextState>(
+    () => ({ version: version + 1 }) as unknown as ModelContextState,
+  );
+  const composite = tapMemo(() => new CompositeContextProvider(), []);
+
+  return {
+    state,
+    methods: {
+      getState: () => state,
+      getModelContext: () => composite.getModelContext(),
+      subscribe: (callback) => composite.subscribe(callback),
+      register: (provider) => composite.registerModelContextProvider(provider),
+    },
+  };
 });

package/src/client/NoOpComposerClient.tsx

@@ -1,11 +1,11 @@
 "use client";
 import { resource, tapMemo } from "@assistant-ui/tap";
-import type { ComposerClientState, ComposerClientApi } from "./types/Composer";
-import { tapApi } from "../utils/tap-store";
+import type { ClientOutput } from "@assistant-ui/store";
+import type { ComposerState } from "../types/scopes";
 
 export const NoOpComposerClient = resource(
-  ({ type }: { type: "edit" | "thread" }) => {
-    const state = tapMemo<ComposerClientState>(() => {
+  ({ type }: { type: "edit" | "thread" }): ClientOutput<"composer"> => {
+    const state = tapMemo<ComposerState>(() => {
       return {
         isEditing: false,
         isEmpty: true,
@@ -20,44 +20,47 @@ export const NoOpComposerClient = resource(
       };
     }, [type]);
 
-    return tapApi<ComposerClientApi>({
-      getState: () => state,
-      setText: () => {
-        throw new Error("Not supported");
+    return {
+      state,
+      methods: {
+        getState: () => state,
+        setText: () => {
+          throw new Error("Not supported");
+        },
+        setRole: () => {
+          throw new Error("Not supported");
+        },
+        setRunConfig: () => {
+          throw new Error("Not supported");
+        },
+        addAttachment: () => {
+          throw new Error("Not supported");
+        },
+        clearAttachments: () => {
+          throw new Error("Not supported");
+        },
+        attachment: () => {
+          throw new Error("Not supported");
+        },
+        reset: () => {
+          throw new Error("Not supported");
+        },
+        send: () => {
+          throw new Error("Not supported");
+        },
+        cancel: () => {
+          throw new Error("Not supported");
+        },
+        startDictation: () => {
+          throw new Error("Not supported");
+        },
+        stopDictation: () => {
+          throw new Error("Not supported");
+        },
+        beginEdit: () => {
+          throw new Error("Not supported");
+        },
       },
-      setRole: () => {
-        throw new Error("Not supported");
-      },
-      setRunConfig: () => {
-        throw new Error("Not supported");
-      },
-      addAttachment: () => {
-        throw new Error("Not supported");
-      },
-      clearAttachments: () => {
-        throw new Error("Not supported");
-      },
-      attachment: () => {
-        throw new Error("Not supported");
-      },
-      reset: () => {
-        throw new Error("Not supported");
-      },
-      send: () => {
-        throw new Error("Not supported");
-      },
-      cancel: () => {
-        throw new Error("Not supported");
-      },
-      beginEdit: () => {
-        throw new Error("Not supported");
-      },
-      startDictation: () => {
-        throw new Error("Not supported");
-      },
-      stopDictation: () => {
-        throw new Error("Not supported");
-      },
-    });
+    };
   },
 );