assistant-cloud 0.1.27 → 0.1.29

package/dist/CloudMessagePersistence.js

@@ -1,103 +1,92 @@
+//#region src/CloudMessagePersistence.ts
 /**
- * Shared persistence logic for cloud message storage.
- *
- * Handles ID mapping (local → remote) and parent_id chaining for both:
- * - AssistantCloudThreadHistoryAdapter (assistant-ui runtime)
- * - useCloudChat (standalone AI SDK hook)
- *
- * The promise-based ID resolution handles concurrent appends — if message B's
- * parent is message A, and A is still being created, we await A's promise
- * to get its remote ID before creating B.
- */
-export class CloudMessagePersistence {
-    cloud;
-    idMapping = {};
-    constructor(cloud) {
-        this.cloud = cloud;
-    }
-    /**
-     * Persist a message to the cloud.
-     *
-     * @param threadId - Remote thread ID
-     * @param messageId - Local message ID (used for tracking)
-     * @param parentId - Local parent message ID (or null for first message)
-     * @param format - Message format (e.g., "aui/v0", "ai-sdk/v6")
-     * @param content - Message content (format-specific)
-     */
-    async append(threadId, messageId, parentId, format, content) {
-        // Resolve parent's remote ID if it exists (may be a promise if concurrent)
-        const resolvedParentId = parentId
-            ? ((await this.idMapping[parentId]) ?? parentId)
-            : null;
-        const task = this.cloud.threads.messages
-            .create(threadId, {
-            parent_id: resolvedParentId,
-            format,
-            content,
-        })
-            .then(({ message_id }) => {
-            this.idMapping[messageId] = message_id;
-            return message_id;
-        })
-            .catch((err) => {
-            // Only delete if we're still the active task (avoids clobbering a retry)
-            if (this.idMapping[messageId] === task) {
-                delete this.idMapping[messageId];
-            }
-            throw err;
-        });
-        // Store the promise immediately so concurrent appends can await it
-        this.idMapping[messageId] = task;
-        return task.then(() => { });
-    }
-    /**
-     * Update an already-persisted message in the cloud.
-     */
-    async update(threadId, messageId, _format, content) {
-        const remoteId = await this.getRemoteId(messageId);
-        if (!remoteId)
-            return; // not persisted yet, skip
-        await this.cloud.threads.messages.update(threadId, remoteId, { content });
-    }
-    /**
-     * Check if a message has been persisted (or is currently being persisted).
-     */
-    isPersisted(messageId) {
-        return messageId in this.idMapping;
-    }
-    /**
-     * Get the remote ID for a local message ID (resolved).
-     * Returns undefined if not persisted.
-     */
-    async getRemoteId(messageId) {
-        const entry = this.idMapping[messageId];
-        if (!entry)
-            return undefined;
-        return entry;
-    }
-    /**
-     * Load messages from the cloud and populate the ID mapping.
-     *
-     * The ID mapping is populated so that `isPersisted()` returns true for
-     * loaded messages, preventing re-persistence of already-stored messages.
-     *
-     * @param threadId - Remote thread ID
-     * @param format - Optional format filter
-     * @returns Array of cloud messages
-     */
-    async load(threadId, format) {
-        const { messages } = await this.cloud.threads.messages.list(threadId, format ? { format } : undefined);
-        // Populate ID mapping so isPersisted() recognizes loaded messages
-        for (const m of messages) {
-            this.idMapping[m.id] = m.id;
-        }
-        return messages;
-    }
-    /**
-     * Reset the ID mapping (call when switching threads).
-     */
-    reset() {
-        this.idMapping = {};
-    }
-}
+* Shared persistence logic for cloud message storage.
+*
+* Handles ID mapping (local → remote) and parent_id chaining for both:
+* - AssistantCloudThreadHistoryAdapter (assistant-ui runtime)
+* - useCloudChat (standalone AI SDK hook)
+*
+* The promise-based ID resolution handles concurrent appends — if message B's
+* parent is message A, and A is still being created, we await A's promise
+* to get its remote ID before creating B.
+*/
+var CloudMessagePersistence = class {
+	cloud;
+	idMapping = {};
+	constructor(cloud) {
+		this.cloud = cloud;
+	}
+	/**
+	* Persist a message to the cloud.
+	*
+	* @param threadId - Remote thread ID
+	* @param messageId - Local message ID (used for tracking)
+	* @param parentId - Local parent message ID (or null for first message)
+	* @param format - Message format (e.g., "aui/v0", "ai-sdk/v6")
+	* @param content - Message content (format-specific)
+	*/
+	async append(threadId, messageId, parentId, format, content) {
+		const resolvedParentId = parentId ? await this.idMapping[parentId] ?? parentId : null;
+		const task = this.cloud.threads.messages.create(threadId, {
+			parent_id: resolvedParentId,
+			format,
+			content
+		}).then(({ message_id }) => {
+			this.idMapping[messageId] = message_id;
+			return message_id;
+		}).catch((err) => {
+			if (this.idMapping[messageId] === task) delete this.idMapping[messageId];
+			throw err;
+		});
+		this.idMapping[messageId] = task;
+		return task.then(() => {});
+	}
+	/**
+	* Update an already-persisted message in the cloud.
+	*/
+	async update(threadId, messageId, _format, content) {
+		const remoteId = await this.getRemoteId(messageId);
+		if (!remoteId) return;
+		await this.cloud.threads.messages.update(threadId, remoteId, { content });
+	}
+	/**
+	* Check if a message has been persisted (or is currently being persisted).
+	*/
+	isPersisted(messageId) {
+		return messageId in this.idMapping;
+	}
+	/**
+	* Get the remote ID for a local message ID (resolved).
+	* Returns undefined if not persisted.
+	*/
+	async getRemoteId(messageId) {
+		const entry = this.idMapping[messageId];
+		if (!entry) return void 0;
+		return entry;
+	}
+	/**
+	* Load messages from the cloud and populate the ID mapping.
+	*
+	* The ID mapping is populated so that `isPersisted()` returns true for
+	* loaded messages, preventing re-persistence of already-stored messages.
+	*
+	* @param threadId - Remote thread ID
+	* @param format - Optional format filter
+	* @returns Array of cloud messages
+	*/
+	async load(threadId, format) {
+		const { messages } = await this.cloud.threads.messages.list(threadId, format ? { format } : void 0);
+		for (const m of messages) this.idMapping[m.id] = m.id;
+		return messages;
+	}
+	/**
+	* Reset the ID mapping (call when switching threads).
+	*/
+	reset() {
+		this.idMapping = {};
+	}
+};
+//#endregion
+export { CloudMessagePersistence };
+
 //# sourceMappingURL=CloudMessagePersistence.js.map

package/dist/CloudMessagePersistence.js.map

@@ -1 +1 @@
-{"version":3,"file":"CloudMessagePersistence.js","sourceRoot":"","sources":["../src/CloudMessagePersistence.ts"],"names":[],"mappings":"AAGA;;;;;;;;;;GAUG;AACH,MAAM,OAAO,uBAAuB;IAGd;IAFZ,SAAS,GAA6C,EAAE,CAAC;IAEjE,YAAoB,KAAqB;QAArB,UAAK,GAAL,KAAK,CAAgB;IAAG,CAAC;IAE7C;;;;;;;;OAQG;IACH,KAAK,CAAC,MAAM,CACV,QAAgB,EAChB,SAAiB,EACjB,QAAuB,EACvB,MAAc,EACd,OAA2B;QAE3B,2EAA2E;QAC3E,MAAM,gBAAgB,GAAG,QAAQ;YAC/B,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,IAAI,QAAQ,CAAC;YAChD,CAAC,CAAC,IAAI,CAAC;QAET,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ;aACrC,MAAM,CAAC,QAAQ,EAAE;YAChB,SAAS,EAAE,gBAAgB;YAC3B,MAAM;YACN,OAAO;SACR,CAAC;aACD,IAAI,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE;YACvB,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,UAAU,CAAC;YACvC,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACb,yEAAyE;YACzE,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE,CAAC;gBACvC,OAAO,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACnC,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC,CAAC,CAAC;QAEL,mEAAmE;QACnE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;QACjC,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CACV,QAAgB,EAChB,SAAiB,EACjB,OAAe,EACf,OAA2B;QAE3B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,QAAQ;YAAE,OAAO,CAAC,0BAA0B;QACjD,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,SAAiB;QAC3B,OAAO,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;IACrC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,WAAW,CAAC,SAAiB;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAC7B,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE,MAAe;QAC1C,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CACzD,QAAQ,EACR,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAChC,CAAC;QACF,kEAAkE;QAClE,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC9B,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;IACtB,CAAC;CACF"}
+{"version":3,"file":"CloudMessagePersistence.js","names":[],"sources":["../src/CloudMessagePersistence.ts"],"sourcesContent":["import type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport type { AssistantCloud } from \"./AssistantCloud\";\n\n/**\n * Shared persistence logic for cloud message storage.\n *\n * Handles ID mapping (local → remote) and parent_id chaining for both:\n * - AssistantCloudThreadHistoryAdapter (assistant-ui runtime)\n * - useCloudChat (standalone AI SDK hook)\n *\n * The promise-based ID resolution handles concurrent appends — if message B's\n * parent is message A, and A is still being created, we await A's promise\n * to get its remote ID before creating B.\n */\nexport class CloudMessagePersistence {\n  private idMapping: Record<string, string | Promise<string>> = {};\n\n  constructor(private cloud: AssistantCloud) {}\n\n  /**\n   * Persist a message to the cloud.\n   *\n   * @param threadId - Remote thread ID\n   * @param messageId - Local message ID (used for tracking)\n   * @param parentId - Local parent message ID (or null for first message)\n   * @param format - Message format (e.g., \"aui/v0\", \"ai-sdk/v6\")\n   * @param content - Message content (format-specific)\n   */\n  async append(\n    threadId: string,\n    messageId: string,\n    parentId: string | null,\n    format: string,\n    content: ReadonlyJSONObject,\n  ): Promise<void> {\n    // Resolve parent's remote ID if it exists (may be a promise if concurrent)\n    const resolvedParentId = parentId\n      ? ((await this.idMapping[parentId]) ?? parentId)\n      : null;\n\n    const task = this.cloud.threads.messages\n      .create(threadId, {\n        parent_id: resolvedParentId,\n        format,\n        content,\n      })\n      .then(({ message_id }) => {\n        this.idMapping[messageId] = message_id;\n        return message_id;\n      })\n      .catch((err) => {\n        // Only delete if we're still the active task (avoids clobbering a retry)\n        if (this.idMapping[messageId] === task) {\n          delete this.idMapping[messageId];\n        }\n        throw err;\n      });\n\n    // Store the promise immediately so concurrent appends can await it\n    this.idMapping[messageId] = task;\n    return task.then(() => {});\n  }\n\n  /**\n   * Update an already-persisted message in the cloud.\n   */\n  async update(\n    threadId: string,\n    messageId: string,\n    _format: string,\n    content: ReadonlyJSONObject,\n  ): Promise<void> {\n    const remoteId = await this.getRemoteId(messageId);\n    if (!remoteId) return; // not persisted yet, skip\n    await this.cloud.threads.messages.update(threadId, remoteId, { content });\n  }\n\n  /**\n   * Check if a message has been persisted (or is currently being persisted).\n   */\n  isPersisted(messageId: string): boolean {\n    return messageId in this.idMapping;\n  }\n\n  /**\n   * Get the remote ID for a local message ID (resolved).\n   * Returns undefined if not persisted.\n   */\n  async getRemoteId(messageId: string): Promise<string | undefined> {\n    const entry = this.idMapping[messageId];\n    if (!entry) return undefined;\n    return entry;\n  }\n\n  /**\n   * Load messages from the cloud and populate the ID mapping.\n   *\n   * The ID mapping is populated so that `isPersisted()` returns true for\n   * loaded messages, preventing re-persistence of already-stored messages.\n   *\n   * @param threadId - Remote thread ID\n   * @param format - Optional format filter\n   * @returns Array of cloud messages\n   */\n  async load(threadId: string, format?: string) {\n    const { messages } = await this.cloud.threads.messages.list(\n      threadId,\n      format ? { format } : undefined,\n    );\n    // Populate ID mapping so isPersisted() recognizes loaded messages\n    for (const m of messages) {\n      this.idMapping[m.id] = m.id;\n    }\n    return messages;\n  }\n\n  /**\n   * Reset the ID mapping (call when switching threads).\n   */\n  reset() {\n    this.idMapping = {};\n  }\n}\n"],"mappings":";;;;;;;;;;;;AAcA,IAAa,0BAAb,MAAqC;CAGf;CAFpB,YAA8D,CAAC;CAE/D,YAAY,OAA+B;EAAvB,KAAA,QAAA;CAAwB;;;;;;;;;;CAW5C,MAAM,OACJ,UACA,WACA,UACA,QACA,SACe;EAEf,MAAM,mBAAmB,WACnB,MAAM,KAAK,UAAU,aAAc,WACrC;EAEJ,MAAM,OAAO,KAAK,MAAM,QAAQ,SAC7B,OAAO,UAAU;GAChB,WAAW;GACX;GACA;EACF,CAAC,EACA,MAAM,EAAE,iBAAiB;GACxB,KAAK,UAAU,aAAa;GAC5B,OAAO;EACT,CAAC,EACA,OAAO,QAAQ;GAEd,IAAI,KAAK,UAAU,eAAe,MAChC,OAAO,KAAK,UAAU;GAExB,MAAM;EACR,CAAC;EAGH,KAAK,UAAU,aAAa;EAC5B,OAAO,KAAK,WAAW,CAAC,CAAC;CAC3B;;;;CAKA,MAAM,OACJ,UACA,WACA,SACA,SACe;EACf,MAAM,WAAW,MAAM,KAAK,YAAY,SAAS;EACjD,IAAI,CAAC,UAAU;EACf,MAAM,KAAK,MAAM,QAAQ,SAAS,OAAO,UAAU,UAAU,EAAE,QAAQ,CAAC;CAC1E;;;;CAKA,YAAY,WAA4B;EACtC,OAAO,aAAa,KAAK;CAC3B;;;;;CAMA,MAAM,YAAY,WAAgD;EAChE,MAAM,QAAQ,KAAK,UAAU;EAC7B,IAAI,CAAC,OAAO,OAAO,KAAA;EACnB,OAAO;CACT;;;;;;;;;;;CAYA,MAAM,KAAK,UAAkB,QAAiB;EAC5C,MAAM,EAAE,aAAa,MAAM,KAAK,MAAM,QAAQ,SAAS,KACrD,UACA,SAAS,EAAE,OAAO,IAAI,KAAA,CACxB;EAEA,KAAK,MAAM,KAAK,UACd,KAAK,UAAU,EAAE,MAAM,EAAE;EAE3B,OAAO;CACT;;;;CAKA,QAAQ;EACN,KAAK,YAAY,CAAC;CACpB;AACF"}

package/dist/FormattedCloudPersistence.d.ts

@@ -1,25 +1,27 @@
-import type { ReadonlyJSONObject } from "assistant-stream/utils";
+import { ReadonlyJSONObject } from "assistant-stream/utils";
+
+//#region src/FormattedCloudPersistence.d.ts
 /**
  * Format adapter shape — structurally identical to the MessageFormatAdapter
  * in @assistant-ui/react, but defined here to avoid cross-package type moves.
  * TypeScript's structural typing ensures these are interchangeable.
  */
-export type MessageFormatAdapter<TMessage, TStorageFormat> = {
+type MessageFormatAdapter<TMessage, TStorageFormat> = {
+  format: string;
+  encode(item: {
+    parentId: string | null;
+    message: TMessage;
+  }): TStorageFormat;
+  decode(stored: {
+    id: string;
+    parent_id: string | null;
     format: string;
-    encode(item: {
-        parentId: string | null;
-        message: TMessage;
-    }): TStorageFormat;
-    decode(stored: {
-        id: string;
-        parent_id: string | null;
-        format: string;
-        content: TStorageFormat;
-    }): {
-        parentId: string | null;
-        message: TMessage;
-    };
-    getId(message: TMessage): string;
+    content: TStorageFormat;
+  }): {
+    parentId: string | null;
+    message: TMessage;
+  };
+  getId(message: TMessage): string;
 };
 /**
  * Wraps a CloudMessagePersistence instance with format-aware encode/decode.
@@ -31,26 +33,28 @@ export type MessageFormatAdapter<TMessage, TStorageFormat> = {
  * The persistence parameter is typed structurally (not by class) so callers
  * don't need to import CloudMessagePersistence directly.
  */
-export declare const createFormattedPersistence: <TMessage, TStorageFormat>(persistence: {
-    append: (threadId: string, messageId: string, parentId: string | null, format: string, content: ReadonlyJSONObject) => Promise<void>;
-    load: (threadId: string, format?: string) => Promise<any[]>;
-    isPersisted: (messageId: string) => boolean;
-    update?: (threadId: string, messageId: string, format: string, content: ReadonlyJSONObject) => Promise<void>;
+declare const createFormattedPersistence: <TMessage, TStorageFormat>(persistence: {
+  append: (threadId: string, messageId: string, parentId: string | null, format: string, content: ReadonlyJSONObject) => Promise<void>;
+  load: (threadId: string, format?: string) => Promise<any[]>;
+  isPersisted: (messageId: string) => boolean;
+  update?: (threadId: string, messageId: string, format: string, content: ReadonlyJSONObject) => Promise<void>;
 }, adapter: MessageFormatAdapter<TMessage, TStorageFormat>) => {
-    append: (threadId: string, item: {
-        parentId: string | null;
-        message: TMessage;
-    }) => Promise<void>;
-    update: ((threadId: string, item: {
-        parentId: string | null;
-        message: TMessage;
-    }, messageId: string) => Promise<void>) | undefined;
-    load: (threadId: string) => Promise<{
-        messages: {
-            parentId: string | null;
-            message: TMessage;
-        }[];
-    }>;
-    isPersisted: (messageId: string) => boolean;
+  append: (threadId: string, item: {
+    parentId: string | null;
+    message: TMessage;
+  }) => Promise<void>;
+  update: ((threadId: string, item: {
+    parentId: string | null;
+    message: TMessage;
+  }, messageId: string) => Promise<void>) | undefined;
+  load: (threadId: string) => Promise<{
+    messages: {
+      parentId: string | null;
+      message: TMessage;
+    }[];
+  }>;
+  isPersisted: (messageId: string) => boolean;
 };
+//#endregion
+export { MessageFormatAdapter, createFormattedPersistence };
 //# sourceMappingURL=FormattedCloudPersistence.d.ts.map

package/dist/FormattedCloudPersistence.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"FormattedCloudPersistence.d.ts","sourceRoot":"","sources":["../src/FormattedCloudPersistence.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAEjE;;;;GAIG;AACH,MAAM,MAAM,oBAAoB,CAAC,QAAQ,EAAE,cAAc,IAAI;IAC3D,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,OAAO,EAAE,QAAQ,CAAA;KAAE,GAAG,cAAc,CAAC;IAC7E,MAAM,CAAC,MAAM,EAAE;QACb,EAAE,EAAE,MAAM,CAAC;QACX,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;QACzB,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,cAAc,CAAC;KACzB,GAAG;QAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,OAAO,EAAE,QAAQ,CAAA;KAAE,CAAC;IACnD,KAAK,CAAC,OAAO,EAAE,QAAQ,GAAG,MAAM,CAAC;CAClC,CAAC;AAEF;;;;;;;;;GASG;AACH,eAAO,MAAM,0BAA0B,GAAI,QAAQ,EAAE,cAAc,EACjE,aAAa;IACX,MAAM,EAAE,CACN,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,MAAM,GAAG,IAAI,EACvB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,kBAAkB,KACxB,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,IAAI,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IAC5D,WAAW,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC;IAC5C,MAAM,CAAC,EAAE,CACP,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,kBAAkB,KACxB,OAAO,CAAC,IAAI,CAAC,CAAC;CACpB,EACD,SAAS,oBAAoB,CAAC,QAAQ,EAAE,cAAc,CAAC;uBAG3C,MAAM,QACV;QAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,OAAO,EAAE,QAAQ,CAAA;KAAE,KACnD,OAAO,CAAC,IAAI,CAAC;wBAaA,MAAM,QACV;QAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,OAAO,EAAE,QAAQ,CAAA;KAAE,aACzC,MAAM,KAChB,OAAO,CAAC,IAAI,CAAC;qBAUG,MAAM;;sBA/Db,MAAM,GAAG,IAAI;;;;6BA+EJ,MAAM;CAC/B,CAAC"}
+{"version":3,"file":"FormattedCloudPersistence.d.ts","names":[],"sources":["../src/FormattedCloudPersistence.ts"],"mappings":";;;;;AAOA;;;KAAY,oBAAA;EACV,MAAA;EACA,MAAA,CAAO,IAAA;IAAQ,QAAA;IAAyB,OAAA,EAAS,QAAA;EAAA,IAAa,cAAA;EAC9D,MAAA,CAAO,MAAA;IACL,EAAA;IACA,SAAA;IACA,MAAA;IACA,OAAA,EAAS,cAAA;EAAA;IACL,QAAA;IAAyB,OAAA,EAAS,QAAA;EAAA;EACxC,KAAA,CAAM,OAAA,EAAS,QAAA;AAAA;;;;;;;;;;;cAaJ,0BAAA,6BACX,WAAA;EACE,MAAA,GACE,QAAA,UACA,SAAA,UACA,QAAA,iBACA,MAAA,UACA,OAAA,EAAS,kBAAA,KACN,OAAA;EACL,IAAA,GAAO,QAAA,UAAkB,MAAA,cAAoB,OAAA;EAC7C,WAAA,GAAc,SAAA;EACd,MAAA,IACE,QAAA,UACA,SAAA,UACA,MAAA,UACA,OAAA,EAAS,kBAAA,KACN,OAAA;AAAA,GAEP,OAAA,EAAS,oBAAA,CAAqB,QAAA,EAAU,cAAA;6BAGtB,IAAA;IACR,QAAA;IAAyB,OAAA,EAAS,QAAA;EAAA,MACzC,OAAA;8BAamB,IAAA;IACR,QAAA;IAAyB,OAAA,EAAS,QAAA;EAAA,GAAU,SAAA,aAEnD,OAAA;8BAUsB,OAAA"}

package/dist/FormattedCloudPersistence.js

@@ -1,39 +1,35 @@
+//#region src/FormattedCloudPersistence.ts
 /**
- * Wraps a CloudMessagePersistence instance with format-aware encode/decode.
- *
- * This centralizes the pattern used by both:
- * - useCloudChat (standalone AI SDK hook)
- * - AssistantCloudThreadHistoryAdapter.withFormat() (assistant-ui runtime)
- *
- * The persistence parameter is typed structurally (not by class) so callers
- * don't need to import CloudMessagePersistence directly.
- */
-export const createFormattedPersistence = (persistence, adapter) => ({
-    append: async (threadId, item) => {
-        const messageId = adapter.getId(item.message);
-        const encoded = adapter.encode(item);
-        return persistence.append(threadId, messageId, item.parentId, adapter.format, encoded);
-    },
-    update: persistence.update
-        ? async (threadId, item, messageId) => {
-            const encoded = adapter.encode(item);
-            return persistence.update(threadId, messageId, adapter.format, encoded);
-        }
-        : undefined,
-    load: async (threadId) => {
-        const messages = await persistence.load(threadId, adapter.format);
-        return {
-            messages: messages
-                .filter((m) => m.format === adapter.format)
-                .map((m) => adapter.decode({
-                id: m.id,
-                parent_id: m.parent_id,
-                format: m.format,
-                content: m.content,
-            }))
-                .reverse(),
-        };
-    },
-    isPersisted: (messageId) => persistence.isPersisted(messageId),
+* Wraps a CloudMessagePersistence instance with format-aware encode/decode.
+*
+* This centralizes the pattern used by both:
+* - useCloudChat (standalone AI SDK hook)
+* - AssistantCloudThreadHistoryAdapter.withFormat() (assistant-ui runtime)
+*
+* The persistence parameter is typed structurally (not by class) so callers
+* don't need to import CloudMessagePersistence directly.
+*/
+const createFormattedPersistence = (persistence, adapter) => ({
+	append: async (threadId, item) => {
+		const messageId = adapter.getId(item.message);
+		const encoded = adapter.encode(item);
+		return persistence.append(threadId, messageId, item.parentId, adapter.format, encoded);
+	},
+	update: persistence.update ? async (threadId, item, messageId) => {
+		const encoded = adapter.encode(item);
+		return persistence.update(threadId, messageId, adapter.format, encoded);
+	} : void 0,
+	load: async (threadId) => {
+		return { messages: (await persistence.load(threadId, adapter.format)).filter((m) => m.format === adapter.format).map((m) => adapter.decode({
+			id: m.id,
+			parent_id: m.parent_id,
+			format: m.format,
+			content: m.content
+		})).reverse() };
+	},
+	isPersisted: (messageId) => persistence.isPersisted(messageId)
 });
+//#endregion
+export { createFormattedPersistence };
+
 //# sourceMappingURL=FormattedCloudPersistence.js.map

package/dist/FormattedCloudPersistence.js.map

@@ -1 +1 @@
-{"version":3,"file":"FormattedCloudPersistence.js","sourceRoot":"","sources":["../src/FormattedCloudPersistence.ts"],"names":[],"mappings":"AAmBA;;;;;;;;;GASG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CACxC,WAgBC,EACD,OAAuD,EACvD,EAAE,CAAC,CAAC;IACJ,MAAM,EAAE,KAAK,EACX,QAAgB,EAChB,IAAoD,EACrC,EAAE;QACjB,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACrC,OAAO,WAAW,CAAC,MAAM,CACvB,QAAQ,EACR,SAAS,EACT,IAAI,CAAC,QAAQ,EACb,OAAO,CAAC,MAAM,EACd,OAA6B,CAC9B,CAAC;IACJ,CAAC;IACD,MAAM,EAAE,WAAW,CAAC,MAAM;QACxB,CAAC,CAAC,KAAK,EACH,QAAgB,EAChB,IAAoD,EACpD,SAAiB,EACF,EAAE;YACjB,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACrC,OAAO,WAAW,CAAC,MAAO,CACxB,QAAQ,EACR,SAAS,EACT,OAAO,CAAC,MAAM,EACd,OAA6B,CAC9B,CAAC;QACJ,CAAC;QACH,CAAC,CAAC,SAAS;IACb,IAAI,EAAE,KAAK,EAAE,QAAgB,EAAE,EAAE;QAC/B,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QAClE,OAAO;YACL,QAAQ,EAAE,QAAQ;iBACf,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC;iBAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACT,OAAO,CAAC,MAAM,CAAC;gBACb,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,MAAM,EAAE,CAAC,CAAC,MAAM;gBAChB,OAAO,EAAE,CAAC,CAAC,OAAyB;aACrC,CAAC,CACH;iBACA,OAAO,EAAE;SACb,CAAC;IACJ,CAAC;IACD,WAAW,EAAE,CAAC,SAAiB,EAAE,EAAE,CAAC,WAAW,CAAC,WAAW,CAAC,SAAS,CAAC;CACvE,CAAC,CAAC"}
+{"version":3,"file":"FormattedCloudPersistence.js","names":[],"sources":["../src/FormattedCloudPersistence.ts"],"sourcesContent":["import type { ReadonlyJSONObject } from \"assistant-stream/utils\";\n\n/**\n * Format adapter shape — structurally identical to the MessageFormatAdapter\n * in @assistant-ui/react, but defined here to avoid cross-package type moves.\n * TypeScript's structural typing ensures these are interchangeable.\n */\nexport type MessageFormatAdapter<TMessage, TStorageFormat> = {\n  format: string;\n  encode(item: { parentId: string | null; message: TMessage }): TStorageFormat;\n  decode(stored: {\n    id: string;\n    parent_id: string | null;\n    format: string;\n    content: TStorageFormat;\n  }): { parentId: string | null; message: TMessage };\n  getId(message: TMessage): string;\n};\n\n/**\n * Wraps a CloudMessagePersistence instance with format-aware encode/decode.\n *\n * This centralizes the pattern used by both:\n * - useCloudChat (standalone AI SDK hook)\n * - AssistantCloudThreadHistoryAdapter.withFormat() (assistant-ui runtime)\n *\n * The persistence parameter is typed structurally (not by class) so callers\n * don't need to import CloudMessagePersistence directly.\n */\nexport const createFormattedPersistence = <TMessage, TStorageFormat>(\n  persistence: {\n    append: (\n      threadId: string,\n      messageId: string,\n      parentId: string | null,\n      format: string,\n      content: ReadonlyJSONObject,\n    ) => Promise<void>;\n    load: (threadId: string, format?: string) => Promise<any[]>;\n    isPersisted: (messageId: string) => boolean;\n    update?: (\n      threadId: string,\n      messageId: string,\n      format: string,\n      content: ReadonlyJSONObject,\n    ) => Promise<void>;\n  },\n  adapter: MessageFormatAdapter<TMessage, TStorageFormat>,\n) => ({\n  append: async (\n    threadId: string,\n    item: { parentId: string | null; message: TMessage },\n  ): Promise<void> => {\n    const messageId = adapter.getId(item.message);\n    const encoded = adapter.encode(item);\n    return persistence.append(\n      threadId,\n      messageId,\n      item.parentId,\n      adapter.format,\n      encoded as ReadonlyJSONObject,\n    );\n  },\n  update: persistence.update\n    ? async (\n        threadId: string,\n        item: { parentId: string | null; message: TMessage },\n        messageId: string,\n      ): Promise<void> => {\n        const encoded = adapter.encode(item);\n        return persistence.update!(\n          threadId,\n          messageId,\n          adapter.format,\n          encoded as ReadonlyJSONObject,\n        );\n      }\n    : undefined,\n  load: async (threadId: string) => {\n    const messages = await persistence.load(threadId, adapter.format);\n    return {\n      messages: messages\n        .filter((m) => m.format === adapter.format)\n        .map((m) =>\n          adapter.decode({\n            id: m.id,\n            parent_id: m.parent_id,\n            format: m.format,\n            content: m.content as TStorageFormat,\n          }),\n        )\n        .reverse(),\n    };\n  },\n  isPersisted: (messageId: string) => persistence.isPersisted(messageId),\n});\n"],"mappings":";;;;;;;;;;;AA6BA,MAAa,8BACX,aAiBA,aACI;CACJ,QAAQ,OACN,UACA,SACkB;EAClB,MAAM,YAAY,QAAQ,MAAM,KAAK,OAAO;EAC5C,MAAM,UAAU,QAAQ,OAAO,IAAI;EACnC,OAAO,YAAY,OACjB,UACA,WACA,KAAK,UACL,QAAQ,QACR,OACF;CACF;CACA,QAAQ,YAAY,SAChB,OACE,UACA,MACA,cACkB;EAClB,MAAM,UAAU,QAAQ,OAAO,IAAI;EACnC,OAAO,YAAY,OACjB,UACA,WACA,QAAQ,QACR,OACF;CACF,IACA,KAAA;CACJ,MAAM,OAAO,aAAqB;EAEhC,OAAO,EACL,WAAU,MAFW,YAAY,KAAK,UAAU,QAAQ,MAAM,GAG3D,QAAQ,MAAM,EAAE,WAAW,QAAQ,MAAM,EACzC,KAAK,MACJ,QAAQ,OAAO;GACb,IAAI,EAAE;GACN,WAAW,EAAE;GACb,QAAQ,EAAE;GACV,SAAS,EAAE;EACb,CAAC,CACH,EACC,QAAQ,EACb;CACF;CACA,cAAc,cAAsB,YAAY,YAAY,SAAS;AACvE"}

package/dist/index.d.ts

@@ -1,8 +1,8 @@
-export type { CloudMessage } from "./AssistantCloudThreadMessages.js";
-export type { AssistantCloudTelemetryConfig } from "./AssistantCloudAPI.js";
-export type { AssistantCloudRunReport } from "./AssistantCloudRuns.js";
-export { AssistantCloud } from "./AssistantCloud.js";
-export { CloudMessagePersistence } from "./CloudMessagePersistence.js";
-export { createFormattedPersistence, type MessageFormatAdapter, } from "./FormattedCloudPersistence.js";
-export { wrapSamplingHandler, createSamplingCollector, type SamplingCallData, type McpSamplingHandler, } from "./instrumentMcpSampling.js";
-//# sourceMappingURL=index.d.ts.map
+import { McpSamplingHandler, SamplingCallData, createSamplingCollector, wrapSamplingHandler } from "./instrumentMcpSampling.js";
+import { AssistantCloudRunReport } from "./AssistantCloudRuns.js";
+import { AssistantCloudTelemetryConfig } from "./AssistantCloudAPI.js";
+import { CloudMessage } from "./AssistantCloudThreadMessages.js";
+import { AssistantCloud } from "./AssistantCloud.js";
+import { CloudMessagePersistence } from "./CloudMessagePersistence.js";
+import { MessageFormatAdapter, createFormattedPersistence } from "./FormattedCloudPersistence.js";
+export { AssistantCloud, type AssistantCloudRunReport, type AssistantCloudTelemetryConfig, type CloudMessage, CloudMessagePersistence, type McpSamplingHandler, type MessageFormatAdapter, type SamplingCallData, createFormattedPersistence, createSamplingCollector, wrapSamplingHandler };

package/dist/index.js

@@ -1,5 +1,5 @@
-export { AssistantCloud } from "./AssistantCloud.js";
-export { CloudMessagePersistence } from "./CloudMessagePersistence.js";
-export { createFormattedPersistence, } from "./FormattedCloudPersistence.js";
-export { wrapSamplingHandler, createSamplingCollector, } from "./instrumentMcpSampling.js";
-//# sourceMappingURL=index.js.map
+import { AssistantCloud } from "./AssistantCloud.js";
+import { CloudMessagePersistence } from "./CloudMessagePersistence.js";
+import { createFormattedPersistence } from "./FormattedCloudPersistence.js";
+import { createSamplingCollector, wrapSamplingHandler } from "./instrumentMcpSampling.js";
+export { AssistantCloud, CloudMessagePersistence, createFormattedPersistence, createSamplingCollector, wrapSamplingHandler };

package/dist/instrumentMcpSampling.d.ts

@@ -1,3 +1,4 @@
+//#region src/instrumentMcpSampling.d.ts
 /**
  * MCP sampling instrumentation utility.
  *
@@ -5,40 +6,40 @@
  * (sampling/createMessage requests) made during tool execution.
  * The captured data can be reported as child generation spans.
  */
-export type SamplingCallData = {
-    model_id?: string;
-    input_tokens?: number;
-    output_tokens?: number;
-    reasoning_tokens?: number;
-    cached_input_tokens?: number;
-    duration_ms?: number;
+type SamplingCallData = {
+  model_id?: string;
+  input_tokens?: number;
+  output_tokens?: number;
+  reasoning_tokens?: number;
+  cached_input_tokens?: number;
+  duration_ms?: number;
 };
-export type McpSamplingHandler = (request: McpSamplingRequest) => Promise<McpSamplingResponse>;
-export type McpSamplingRequest = {
-    method: "sampling/createMessage";
-    params: {
-        messages: unknown[];
-        modelPreferences?: {
-            hints?: {
-                name?: string;
-            }[];
-        };
-        maxTokens?: number;
-        [key: string]: unknown;
-    };
-};
-export type McpSamplingResponse = {
-    model?: string;
-    content: unknown;
-    usage?: {
-        inputTokens?: number;
-        outputTokens?: number;
-        promptTokens?: number;
-        completionTokens?: number;
-        reasoningTokens?: number;
-        cachedInputTokens?: number;
+type McpSamplingHandler = (request: McpSamplingRequest) => Promise<McpSamplingResponse>;
+type McpSamplingRequest = {
+  method: "sampling/createMessage";
+  params: {
+    messages: unknown[];
+    modelPreferences?: {
+      hints?: {
+        name?: string;
+      }[];
     };
+    maxTokens?: number;
     [key: string]: unknown;
+  };
+};
+type McpSamplingResponse = {
+  model?: string;
+  content: unknown;
+  usage?: {
+    inputTokens?: number;
+    outputTokens?: number;
+    promptTokens?: number;
+    completionTokens?: number;
+    reasoningTokens?: number;
+    cachedInputTokens?: number;
+  };
+  [key: string]: unknown;
 };
 /**
  * Wraps an MCP sampling handler to intercept and measure sampling calls.
@@ -58,7 +59,7 @@ export type McpSamplingResponse = {
  * // After tool execution, `samplingCalls` contains metrics for all nested LLM calls
  * ```
  */
-export declare function wrapSamplingHandler(handler: McpSamplingHandler, onSamplingCall: (data: SamplingCallData) => void): McpSamplingHandler;
+declare function wrapSamplingHandler(handler: McpSamplingHandler, onSamplingCall: (data: SamplingCallData) => void): McpSamplingHandler;
 /**
  * Creates a collector that accumulates sampling call data during tool execution.
  * Use with `wrapSamplingHandler` to capture all sampling calls for a tool invocation.
@@ -71,9 +72,11 @@ export declare function wrapSamplingHandler(handler: McpSamplingHandler, onSampl
  * const calls = collector.getCalls(); // SamplingCallData[]
  * ```
  */
-export declare function createSamplingCollector(): {
-    collect: (data: SamplingCallData) => number;
-    getCalls: () => SamplingCallData[];
-    reset: () => void;
+declare function createSamplingCollector(): {
+  collect: (data: SamplingCallData) => number;
+  getCalls: () => SamplingCallData[];
+  reset: () => void;
 };
+//#endregion
+export { McpSamplingHandler, McpSamplingRequest, McpSamplingResponse, SamplingCallData, createSamplingCollector, wrapSamplingHandler };
 //# sourceMappingURL=instrumentMcpSampling.d.ts.map

package/dist/instrumentMcpSampling.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"instrumentMcpSampling.d.ts","sourceRoot":"","sources":["../src/instrumentMcpSampling.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,MAAM,MAAM,gBAAgB,GAAG;IAC7B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG,CAC/B,OAAO,EAAE,kBAAkB,KACxB,OAAO,CAAC,mBAAmB,CAAC,CAAC;AAElC,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,wBAAwB,CAAC;IACjC,MAAM,EAAE;QACN,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,gBAAgB,CAAC,EAAE;YAAE,KAAK,CAAC,EAAE;gBAAE,IAAI,CAAC,EAAE,MAAM,CAAA;aAAE,EAAE,CAAA;SAAE,CAAC;QACnD,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACxB,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE;QACN,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B,CAAC;IACF,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB,CAAC;AAEF;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,kBAAkB,EAC3B,cAAc,EAAE,CAAC,IAAI,EAAE,gBAAgB,KAAK,IAAI,GAC/C,kBAAkB,CA+BpB;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,uBAAuB;oBAGnB,gBAAgB;;;EAMnC"}
+{"version":3,"file":"instrumentMcpSampling.d.ts","names":[],"sources":["../src/instrumentMcpSampling.ts"],"mappings":";;AAQA;;;;;;KAAY,gBAAA;EACV,QAAA;EACA,YAAA;EACA,aAAA;EACA,gBAAA;EACA,mBAAA;EACA,WAAA;AAAA;AAAA,KAGU,kBAAA,IACV,OAAA,EAAS,kBAAA,KACN,OAAA,CAAQ,mBAAA;AAAA,KAED,kBAAA;EACV,MAAA;EACA,MAAA;IACE,QAAA;IACA,gBAAA;MAAqB,KAAA;QAAU,IAAA;MAAA;IAAA;IAC/B,SAAA;IAAA,CACC,GAAA;EAAA;AAAA;AAAA,KAIO,mBAAA;EACV,KAAA;EACA,OAAA;EACA,KAAA;IACE,WAAA;IACA,YAAA;IACA,YAAA;IACA,gBAAA;IACA,eAAA;IACA,iBAAA;EAAA;EAAA,CAED,GAAA;AAAA;;;;;;;;;;;;;;;AAAW;AAqBd;;;iBAAgB,mBAAA,CACd,OAAA,EAAS,kBAAA,EACT,cAAA,GAAiB,IAAA,EAAM,gBAAA,YACtB,kBAAA;;;;;;;;;;;;AAAkB;iBA6CL,uBAAA,CAAA;kBAGI,gBAAA;kBAAA,gBAAA"}