agents 0.13.1 → 0.13.3

package/dist/chat-sdk/index.js

@@ -0,0 +1,453 @@
+import { Agent, getCurrentAgent } from "../index.js";
+//#region src/chat-sdk/agent.ts
+const NEXT_CLEANUP_AT_KEY = "next_cleanup_at";
+const CLEANUP_SCHEDULE_ID_KEY = "cleanup_schedule_id";
+var ChatSdkStateAgent = class extends Agent {
+	onStart() {
+		this.migrate();
+		this.scheduleNextCleanup();
+	}
+	subscribe(threadId) {
+		this.sql`
+      INSERT OR IGNORE INTO chat_sdk_state_subscriptions (thread_id)
+      VALUES (${threadId})
+    `;
+	}
+	unsubscribe(threadId) {
+		this.sql`
+      DELETE FROM chat_sdk_state_subscriptions
+      WHERE thread_id = ${threadId}
+    `;
+	}
+	isSubscribed(threadId) {
+		return this.sql`
+      SELECT 1 as found
+      FROM chat_sdk_state_subscriptions
+      WHERE thread_id = ${threadId}
+      LIMIT 1
+    `.length > 0;
+	}
+	async acquireLock(threadId, ttlMs) {
+		const result = this.ctx.storage.transactionSync(() => {
+			const now = Date.now();
+			this.ctx.storage.sql.exec("DELETE FROM chat_sdk_state_locks WHERE thread_id = ? AND expires_at <= ?", threadId, now);
+			if (this.ctx.storage.sql.exec("SELECT 1 FROM chat_sdk_state_locks WHERE thread_id = ? LIMIT 1", threadId).toArray().length > 0) return null;
+			const token = crypto.randomUUID();
+			const expiresAt = now + ttlMs;
+			this.ctx.storage.sql.exec("INSERT INTO chat_sdk_state_locks (thread_id, token, expires_at) VALUES (?, ?, ?)", threadId, token, expiresAt);
+			return {
+				threadId,
+				token,
+				expiresAt
+			};
+		});
+		await this.scheduleCleanupForExpiry(result?.expiresAt ?? null);
+		return result;
+	}
+	releaseLock(threadId, token) {
+		this.sql`
+      DELETE FROM chat_sdk_state_locks
+      WHERE thread_id = ${threadId} AND token = ${token}
+    `;
+	}
+	async extendLock(threadId, token, ttlMs) {
+		const result = this.ctx.storage.transactionSync(() => {
+			const now = Date.now();
+			return this.ctx.storage.sql.exec(`UPDATE chat_sdk_state_locks SET expires_at = ?
+           WHERE thread_id = ? AND token = ? AND expires_at > ?
+           RETURNING thread_id`, now + ttlMs, threadId, token, now).toArray().length > 0;
+		});
+		if (result) await this.scheduleCleanupForExpiry(Date.now() + ttlMs);
+		return result;
+	}
+	forceReleaseLock(threadId) {
+		this.sql`
+      DELETE FROM chat_sdk_state_locks
+      WHERE thread_id = ${threadId}
+    `;
+	}
+	async enqueue(threadId, value, maxSize) {
+		const parsed = parseQueueEntry(value);
+		const count = this.ctx.storage.transactionSync(() => {
+			this.ctx.storage.sql.exec("INSERT INTO chat_sdk_state_queue (thread_id, value, enqueued_at, expires_at) VALUES (?, ?, ?, ?)", threadId, value, parsed.enqueuedAt, parsed.expiresAt);
+			this.ctx.storage.sql.exec(`DELETE FROM chat_sdk_state_queue WHERE thread_id = ? AND id NOT IN (
+          SELECT id FROM chat_sdk_state_queue
+          WHERE thread_id = ?
+          ORDER BY id DESC
+          LIMIT ?
+        )`, threadId, threadId, maxSize);
+			return this.ctx.storage.sql.exec("SELECT COUNT(*) as count FROM chat_sdk_state_queue WHERE thread_id = ?", threadId).one().count;
+		});
+		await this.scheduleCleanupForExpiry(parsed.expiresAt);
+		return count;
+	}
+	popQueue(threadId) {
+		return this.ctx.storage.transactionSync(() => {
+			const now = Date.now();
+			this.ctx.storage.sql.exec("DELETE FROM chat_sdk_state_queue WHERE thread_id = ? AND expires_at <= ?", threadId, now);
+			const row = this.ctx.storage.sql.exec("SELECT id, value FROM chat_sdk_state_queue WHERE thread_id = ? ORDER BY id ASC LIMIT 1", threadId).toArray()[0];
+			if (!row) return null;
+			this.ctx.storage.sql.exec("DELETE FROM chat_sdk_state_queue WHERE id = ?", row.id);
+			return row.value;
+		});
+	}
+	queueDepth(threadId) {
+		return this.sql`
+      SELECT COUNT(*) as count
+      FROM chat_sdk_state_queue
+      WHERE thread_id = ${threadId} AND expires_at > ${Date.now()}
+    `[0]?.count ?? 0;
+	}
+	async listAppend(key, value, maxLength, ttlMs) {
+		const expiresAt = ttlMs && ttlMs > 0 ? Date.now() + ttlMs : null;
+		this.ctx.storage.transactionSync(() => {
+			this.ctx.storage.sql.exec("INSERT INTO chat_sdk_state_lists (key, value, expires_at) VALUES (?, ?, ?)", key, value, expiresAt);
+			if (expiresAt !== null) this.ctx.storage.sql.exec("UPDATE chat_sdk_state_lists SET expires_at = ? WHERE key = ?", expiresAt, key);
+			if (maxLength != null && maxLength > 0) this.ctx.storage.sql.exec(`DELETE FROM chat_sdk_state_lists WHERE key = ? AND id NOT IN (
+            SELECT id FROM chat_sdk_state_lists
+            WHERE key = ?
+            ORDER BY id DESC
+            LIMIT ?
+          )`, key, key, maxLength);
+		});
+		await this.scheduleCleanupForExpiry(expiresAt);
+	}
+	listGet(key) {
+		const now = Date.now();
+		this.sql`
+      DELETE FROM chat_sdk_state_lists
+      WHERE key = ${key}
+        AND expires_at IS NOT NULL
+        AND expires_at <= ${now}
+    `;
+		return this.sql`
+      SELECT value
+      FROM chat_sdk_state_lists
+      WHERE key = ${key}
+      ORDER BY id ASC
+    `.map((row) => row.value);
+	}
+	cacheGet(key) {
+		return this.readCacheValue(key, Date.now());
+	}
+	async cacheSet(key, value, ttlMs) {
+		const expiresAt = ttlMs && ttlMs > 0 ? Date.now() + ttlMs : null;
+		this.upsertCacheValue(key, value, expiresAt);
+		await this.scheduleCleanupForExpiry(expiresAt);
+	}
+	async cacheSetIfNotExists(key, value, ttlMs) {
+		const now = Date.now();
+		const inserted = this.ctx.storage.transactionSync(() => {
+			this.ctx.storage.sql.exec("DELETE FROM chat_sdk_state_cache WHERE key = ? AND expires_at IS NOT NULL AND expires_at <= ?", key, now);
+			if (this.readCacheValue(key, now) !== null) return false;
+			const expiresAt = ttlMs && ttlMs > 0 ? now + ttlMs : null;
+			this.upsertCacheValue(key, value, expiresAt);
+			return true;
+		});
+		if (inserted) {
+			const expiresAt = ttlMs && ttlMs > 0 ? Date.now() + ttlMs : null;
+			await this.scheduleCleanupForExpiry(expiresAt);
+		}
+		return inserted;
+	}
+	cacheDelete(key) {
+		this.sql`
+      DELETE FROM chat_sdk_state_cache
+      WHERE key = ${key}
+    `;
+	}
+	async cleanupExpired(payload) {
+		const current = this.readCleanupMetadata();
+		if (payload?.expiresAt !== void 0 && current.nextCleanupAt !== null && payload.expiresAt !== current.nextCleanupAt) return;
+		const now = Date.now();
+		this.clearCleanupMetadata();
+		this.sql`
+      DELETE FROM chat_sdk_state_locks
+      WHERE expires_at <= ${now}
+    `;
+		this.sql`
+      DELETE FROM chat_sdk_state_cache
+      WHERE expires_at IS NOT NULL AND expires_at <= ${now}
+    `;
+		this.sql`
+      DELETE FROM chat_sdk_state_queue
+      WHERE expires_at <= ${now}
+    `;
+		this.sql`
+      DELETE FROM chat_sdk_state_lists
+      WHERE expires_at IS NOT NULL AND expires_at <= ${now}
+    `;
+		await this.scheduleNextCleanup();
+	}
+	migrate() {
+		this.sql`
+      CREATE TABLE IF NOT EXISTS chat_sdk_state_subscriptions (
+        thread_id TEXT PRIMARY KEY
+      )
+    `;
+		this.sql`
+      CREATE TABLE IF NOT EXISTS chat_sdk_state_locks (
+        thread_id TEXT PRIMARY KEY,
+        token TEXT NOT NULL,
+        expires_at INTEGER NOT NULL
+      )
+    `;
+		this.sql`
+      CREATE TABLE IF NOT EXISTS chat_sdk_state_cache (
+        key TEXT PRIMARY KEY,
+        value TEXT NOT NULL,
+        expires_at INTEGER
+      )
+    `;
+		this.sql`
+      CREATE TABLE IF NOT EXISTS chat_sdk_state_queue (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        thread_id TEXT NOT NULL,
+        value TEXT NOT NULL,
+        enqueued_at INTEGER NOT NULL,
+        expires_at INTEGER NOT NULL
+      )
+    `;
+		this.sql`
+      CREATE TABLE IF NOT EXISTS chat_sdk_state_lists (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        key TEXT NOT NULL,
+        value TEXT NOT NULL,
+        expires_at INTEGER
+      )
+    `;
+		this.sql`
+      CREATE TABLE IF NOT EXISTS chat_sdk_state_metadata (
+        key TEXT PRIMARY KEY,
+        value TEXT NOT NULL
+      )
+    `;
+		this.sql`
+      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_locks_expires
+      ON chat_sdk_state_locks(expires_at)
+    `;
+		this.sql`
+      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_cache_expires
+      ON chat_sdk_state_cache(expires_at)
+      WHERE expires_at IS NOT NULL
+    `;
+		this.sql`
+      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_queue_thread
+      ON chat_sdk_state_queue(thread_id, id)
+    `;
+		this.sql`
+      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_queue_expires
+      ON chat_sdk_state_queue(expires_at)
+    `;
+		this.sql`
+      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_lists_key
+      ON chat_sdk_state_lists(key, id)
+    `;
+		this.sql`
+      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_lists_expires
+      ON chat_sdk_state_lists(expires_at)
+      WHERE expires_at IS NOT NULL
+    `;
+	}
+	readCacheValue(key, now) {
+		return this.sql`
+      SELECT value
+      FROM chat_sdk_state_cache
+      WHERE key = ${key}
+        AND (expires_at IS NULL OR expires_at > ${now})
+    `[0]?.value ?? null;
+	}
+	upsertCacheValue(key, value, expiresAt) {
+		this.sql`
+      INSERT OR REPLACE INTO chat_sdk_state_cache (key, value, expires_at)
+      VALUES (${key}, ${value}, ${expiresAt})
+    `;
+	}
+	async scheduleCleanupForExpiry(expiresAt) {
+		if (expiresAt === null) return;
+		await this.ensureCleanupScheduled(expiresAt);
+	}
+	async scheduleNextCleanup() {
+		const next = this.nextExpiry();
+		if (next === null) {
+			const current = this.readCleanupMetadata();
+			if (current.scheduleId) await this.cancelSchedule(current.scheduleId).catch(() => false);
+			this.clearCleanupMetadata();
+			return;
+		}
+		await this.ensureCleanupScheduled(next);
+	}
+	async ensureCleanupScheduled(expiresAt) {
+		const current = this.readCleanupMetadata();
+		if (current.nextCleanupAt !== null && current.nextCleanupAt <= expiresAt) return;
+		if (current.scheduleId) await this.cancelSchedule(current.scheduleId).catch(() => false);
+		const delaySeconds = Math.max(0, Math.ceil((expiresAt - Date.now()) / 1e3));
+		const schedule = await this.schedule(delaySeconds, "cleanupExpired", { expiresAt });
+		this.writeCleanupMetadata(expiresAt, schedule.id);
+	}
+	nextExpiry() {
+		return this.sql`
+      SELECT MIN(expires_at) as expires_at
+      FROM (
+        SELECT expires_at FROM chat_sdk_state_locks
+        UNION ALL
+        SELECT expires_at FROM chat_sdk_state_queue
+        UNION ALL
+        SELECT expires_at FROM chat_sdk_state_cache WHERE expires_at IS NOT NULL
+        UNION ALL
+        SELECT expires_at FROM chat_sdk_state_lists WHERE expires_at IS NOT NULL
+      )
+    `[0]?.expires_at ?? null;
+	}
+	readCleanupMetadata() {
+		const rows = this.sql`
+      SELECT key, value
+      FROM chat_sdk_state_metadata
+      WHERE key IN (${NEXT_CLEANUP_AT_KEY}, ${CLEANUP_SCHEDULE_ID_KEY})
+    `;
+		const values = new Map(rows.map((row) => [row.key, row.value]));
+		const nextCleanupAt = Number(values.get(NEXT_CLEANUP_AT_KEY));
+		return {
+			nextCleanupAt: Number.isFinite(nextCleanupAt) ? nextCleanupAt : null,
+			scheduleId: values.get(CLEANUP_SCHEDULE_ID_KEY) ?? null
+		};
+	}
+	writeCleanupMetadata(expiresAt, scheduleId) {
+		this.sql`
+      INSERT OR REPLACE INTO chat_sdk_state_metadata (key, value)
+      VALUES (${NEXT_CLEANUP_AT_KEY}, ${String(expiresAt)})
+    `;
+		this.sql`
+      INSERT OR REPLACE INTO chat_sdk_state_metadata (key, value)
+      VALUES (${CLEANUP_SCHEDULE_ID_KEY}, ${scheduleId})
+    `;
+	}
+	clearCleanupMetadata() {
+		this.sql`
+      DELETE FROM chat_sdk_state_metadata
+      WHERE key IN (${NEXT_CLEANUP_AT_KEY}, ${CLEANUP_SCHEDULE_ID_KEY})
+    `;
+	}
+};
+function parseQueueEntry(value) {
+	const raw = JSON.parse(value);
+	if (typeof raw.enqueuedAt !== "number" || typeof raw.expiresAt !== "number") throw new Error("ChatSdkStateAgent expected QueueEntry JSON with numeric TTLs");
+	return {
+		enqueuedAt: raw.enqueuedAt,
+		expiresAt: raw.expiresAt
+	};
+}
+//#endregion
+//#region src/chat-sdk/adapter.ts
+const THREAD_STATE_PREFIX = "thread-state:";
+const CHANNEL_STATE_PREFIX = "channel-state:";
+const MESSAGE_HISTORY_PREFIX = "msg-history:";
+const TRANSCRIPTS_USER_PREFIX = "transcripts:user:";
+function parseStoredJson(raw, label) {
+	try {
+		return JSON.parse(raw);
+	} catch (error) {
+		throw new Error(`ChatSdkStateAdapter expected JSON-encoded ${label}`, { cause: error });
+	}
+}
+function defaultThreadShard(threadId) {
+	return threadId.split(":").slice(0, 2).join(":") || "default";
+}
+function defaultKeyShard(key, shardThread = defaultThreadShard) {
+	for (const prefix of [
+		THREAD_STATE_PREFIX,
+		CHANNEL_STATE_PREFIX,
+		MESSAGE_HISTORY_PREFIX,
+		TRANSCRIPTS_USER_PREFIX
+	]) if (key.startsWith(prefix)) return shardThread(key.slice(prefix.length));
+}
+var ChatSdkStateAdapter = class {
+	constructor(options = {}) {
+		this.connected = false;
+		const parent = options.parent ?? getCurrentAgent().agent;
+		if (!parent) throw new Error("ChatSdkStateAdapter requires a parent Agent. Pass `parent` or create it inside an Agent context.");
+		this.parent = parent;
+		this.agentClass = options.agent ?? ChatSdkStateAgent;
+		this.defaultName = options.name ?? "default";
+		this.keyShard = options.keyShard;
+		this.shardKey = options.shardKey ?? defaultThreadShard;
+	}
+	async connect() {
+		this.connected = true;
+	}
+	async disconnect() {
+		this.connected = false;
+	}
+	async subscribe(threadId) {
+		await (await this.stateAgent(threadId)).subscribe(threadId);
+	}
+	async unsubscribe(threadId) {
+		await (await this.stateAgent(threadId)).unsubscribe(threadId);
+	}
+	async isSubscribed(threadId) {
+		return (await this.stateAgent(threadId)).isSubscribed(threadId);
+	}
+	async acquireLock(threadId, ttlMs) {
+		return (await this.stateAgent(threadId)).acquireLock(threadId, ttlMs);
+	}
+	async releaseLock(lock) {
+		await (await this.stateAgent(lock.threadId)).releaseLock(lock.threadId, lock.token);
+	}
+	async extendLock(lock, ttlMs) {
+		return (await this.stateAgent(lock.threadId)).extendLock(lock.threadId, lock.token, ttlMs);
+	}
+	async forceReleaseLock(threadId) {
+		await (await this.stateAgent(threadId)).forceReleaseLock(threadId);
+	}
+	async enqueue(threadId, entry, maxSize) {
+		return (await this.stateAgent(threadId)).enqueue(threadId, JSON.stringify(entry), maxSize);
+	}
+	async dequeue(threadId) {
+		const raw = await (await this.stateAgent(threadId)).popQueue(threadId);
+		return raw === null ? null : parseStoredJson(raw, `queue entry for ${threadId}`);
+	}
+	async queueDepth(threadId) {
+		return (await this.stateAgent(threadId)).queueDepth(threadId);
+	}
+	async appendToList(key, value, options) {
+		await (await this.stateAgentForKey(key)).listAppend(key, JSON.stringify(value), options?.maxLength, options?.ttlMs);
+	}
+	async getList(key) {
+		return (await (await this.stateAgentForKey(key)).listGet(key)).map((value) => parseStoredJson(value, `list entry for ${key}`));
+	}
+	async get(key) {
+		const raw = await (await this.stateAgentForKey(key)).cacheGet(key);
+		return raw === null ? null : parseStoredJson(raw, `cache key ${key}`);
+	}
+	async set(key, value, ttlMs) {
+		await (await this.stateAgentForKey(key)).cacheSet(key, JSON.stringify(value), ttlMs);
+	}
+	async setIfNotExists(key, value, ttlMs) {
+		return (await this.stateAgentForKey(key)).cacheSetIfNotExists(key, JSON.stringify(value), ttlMs);
+	}
+	async delete(key) {
+		await (await this.stateAgentForKey(key)).cacheDelete(key);
+	}
+	async stateAgent(threadId) {
+		this.ensureConnected();
+		const name = threadId ? this.shardKey(threadId) : this.defaultName;
+		return this.parent.subAgent(this.agentClass, name);
+	}
+	async stateAgentForKey(key) {
+		this.ensureConnected();
+		const name = this.keyShard?.(key) ?? defaultKeyShard(key, this.shardKey) ?? this.defaultName;
+		return this.parent.subAgent(this.agentClass, name);
+	}
+	ensureConnected() {
+		if (!this.connected) throw new Error("ChatSdkStateAdapter is not connected");
+	}
+};
+//#endregion
+//#region src/chat-sdk/index.ts
+function createChatSdkState(options = {}) {
+	return new ChatSdkStateAdapter(options);
+}
+//#endregion
+export { ChatSdkStateAdapter, ChatSdkStateAgent, createChatSdkState, defaultKeyShard, defaultThreadShard };
+
+//# sourceMappingURL=index.js.map

package/dist/chat-sdk/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":[],"sources":["../../src/chat-sdk/agent.ts","../../src/chat-sdk/adapter.ts","../../src/chat-sdk/index.ts"],"sourcesContent":["import { Agent } from \"../index\";\n\ninterface StoredLock {\n  threadId: string;\n  token: string;\n  expiresAt: number;\n}\n\ninterface StoredQueueEntry {\n  enqueuedAt: number;\n  expiresAt: number;\n}\n\nconst NEXT_CLEANUP_AT_KEY = \"next_cleanup_at\";\nconst CLEANUP_SCHEDULE_ID_KEY = \"cleanup_schedule_id\";\n\nexport class ChatSdkStateAgent extends Agent {\n  onStart(): void {\n    this.migrate();\n    void this.scheduleNextCleanup();\n  }\n\n  subscribe(threadId: string): void {\n    this.sql`\n      INSERT OR IGNORE INTO chat_sdk_state_subscriptions (thread_id)\n      VALUES (${threadId})\n    `;\n  }\n\n  unsubscribe(threadId: string): void {\n    this.sql`\n      DELETE FROM chat_sdk_state_subscriptions\n      WHERE thread_id = ${threadId}\n    `;\n  }\n\n  isSubscribed(threadId: string): boolean {\n    const rows = this.sql<{ found: number }>`\n      SELECT 1 as found\n      FROM chat_sdk_state_subscriptions\n      WHERE thread_id = ${threadId}\n      LIMIT 1\n    `;\n    return rows.length > 0;\n  }\n\n  async acquireLock(\n    threadId: string,\n    ttlMs: number\n  ): Promise<StoredLock | null> {\n    const result = this.ctx.storage.transactionSync(() => {\n      const now = Date.now();\n\n      this.ctx.storage.sql.exec(\n        \"DELETE FROM chat_sdk_state_locks WHERE thread_id = ? AND expires_at <= ?\",\n        threadId,\n        now\n      );\n\n      const existing = this.ctx.storage.sql\n        .exec(\n          \"SELECT 1 FROM chat_sdk_state_locks WHERE thread_id = ? LIMIT 1\",\n          threadId\n        )\n        .toArray();\n      if (existing.length > 0) {\n        return null;\n      }\n\n      const token = crypto.randomUUID();\n      const expiresAt = now + ttlMs;\n\n      this.ctx.storage.sql.exec(\n        \"INSERT INTO chat_sdk_state_locks (thread_id, token, expires_at) VALUES (?, ?, ?)\",\n        threadId,\n        token,\n        expiresAt\n      );\n\n      return { threadId, token, expiresAt };\n    });\n\n    await this.scheduleCleanupForExpiry(result?.expiresAt ?? null);\n    return result;\n  }\n\n  releaseLock(threadId: string, token: string): void {\n    this.sql`\n      DELETE FROM chat_sdk_state_locks\n      WHERE thread_id = ${threadId} AND token = ${token}\n    `;\n  }\n\n  async extendLock(\n    threadId: string,\n    token: string,\n    ttlMs: number\n  ): Promise<boolean> {\n    const result = this.ctx.storage.transactionSync(() => {\n      const now = Date.now();\n      const rows = this.ctx.storage.sql\n        .exec<{ thread_id: string }>(\n          `UPDATE chat_sdk_state_locks SET expires_at = ?\n           WHERE thread_id = ? AND token = ? AND expires_at > ?\n           RETURNING thread_id`,\n          now + ttlMs,\n          threadId,\n          token,\n          now\n        )\n        .toArray();\n      return rows.length > 0;\n    });\n    if (result) {\n      await this.scheduleCleanupForExpiry(Date.now() + ttlMs);\n    }\n    return result;\n  }\n\n  forceReleaseLock(threadId: string): void {\n    this.sql`\n      DELETE FROM chat_sdk_state_locks\n      WHERE thread_id = ${threadId}\n    `;\n  }\n\n  async enqueue(\n    threadId: string,\n    value: string,\n    maxSize: number\n  ): Promise<number> {\n    const parsed = parseQueueEntry(value);\n\n    const count = this.ctx.storage.transactionSync(() => {\n      this.ctx.storage.sql.exec(\n        \"INSERT INTO chat_sdk_state_queue (thread_id, value, enqueued_at, expires_at) VALUES (?, ?, ?, ?)\",\n        threadId,\n        value,\n        parsed.enqueuedAt,\n        parsed.expiresAt\n      );\n\n      this.ctx.storage.sql.exec(\n        `DELETE FROM chat_sdk_state_queue WHERE thread_id = ? AND id NOT IN (\n          SELECT id FROM chat_sdk_state_queue\n          WHERE thread_id = ?\n          ORDER BY id DESC\n          LIMIT ?\n        )`,\n        threadId,\n        threadId,\n        maxSize\n      );\n\n      const row = this.ctx.storage.sql\n        .exec<{ count: number }>(\n          \"SELECT COUNT(*) as count FROM chat_sdk_state_queue WHERE thread_id = ?\",\n          threadId\n        )\n        .one();\n      return row.count;\n    });\n    await this.scheduleCleanupForExpiry(parsed.expiresAt);\n    return count;\n  }\n\n  popQueue(threadId: string): string | null {\n    return this.ctx.storage.transactionSync(() => {\n      const now = Date.now();\n\n      this.ctx.storage.sql.exec(\n        \"DELETE FROM chat_sdk_state_queue WHERE thread_id = ? AND expires_at <= ?\",\n        threadId,\n        now\n      );\n\n      const rows = this.ctx.storage.sql\n        .exec<{ id: number; value: string }>(\n          \"SELECT id, value FROM chat_sdk_state_queue WHERE thread_id = ? ORDER BY id ASC LIMIT 1\",\n          threadId\n        )\n        .toArray();\n      const row = rows[0];\n      if (!row) {\n        return null;\n      }\n\n      this.ctx.storage.sql.exec(\n        \"DELETE FROM chat_sdk_state_queue WHERE id = ?\",\n        row.id\n      );\n      return row.value;\n    });\n  }\n\n  queueDepth(threadId: string): number {\n    const rows = this.sql<{ count: number }>`\n      SELECT COUNT(*) as count\n      FROM chat_sdk_state_queue\n      WHERE thread_id = ${threadId} AND expires_at > ${Date.now()}\n    `;\n    return rows[0]?.count ?? 0;\n  }\n\n  async listAppend(\n    key: string,\n    value: string,\n    maxLength?: number,\n    ttlMs?: number\n  ): Promise<void> {\n    const expiresAt = ttlMs && ttlMs > 0 ? Date.now() + ttlMs : null;\n\n    this.ctx.storage.transactionSync(() => {\n      this.ctx.storage.sql.exec(\n        \"INSERT INTO chat_sdk_state_lists (key, value, expires_at) VALUES (?, ?, ?)\",\n        key,\n        value,\n        expiresAt\n      );\n\n      if (expiresAt !== null) {\n        // Chat SDK history lists use a list-level TTL: any append refreshes the\n        // expiry for the whole logical list, not only the new row.\n        this.ctx.storage.sql.exec(\n          \"UPDATE chat_sdk_state_lists SET expires_at = ? WHERE key = ?\",\n          expiresAt,\n          key\n        );\n      }\n\n      if (maxLength != null && maxLength > 0) {\n        this.ctx.storage.sql.exec(\n          `DELETE FROM chat_sdk_state_lists WHERE key = ? AND id NOT IN (\n            SELECT id FROM chat_sdk_state_lists\n            WHERE key = ?\n            ORDER BY id DESC\n            LIMIT ?\n          )`,\n          key,\n          key,\n          maxLength\n        );\n      }\n    });\n    await this.scheduleCleanupForExpiry(expiresAt);\n  }\n\n  listGet(key: string): string[] {\n    const now = Date.now();\n\n    this.sql`\n      DELETE FROM chat_sdk_state_lists\n      WHERE key = ${key}\n        AND expires_at IS NOT NULL\n        AND expires_at <= ${now}\n    `;\n\n    return this.sql<{ value: string }>`\n      SELECT value\n      FROM chat_sdk_state_lists\n      WHERE key = ${key}\n      ORDER BY id ASC\n    `.map((row) => row.value);\n  }\n\n  cacheGet(key: string): string | null {\n    return this.readCacheValue(key, Date.now());\n  }\n\n  async cacheSet(key: string, value: string, ttlMs?: number): Promise<void> {\n    const expiresAt = ttlMs && ttlMs > 0 ? Date.now() + ttlMs : null;\n    this.upsertCacheValue(key, value, expiresAt);\n    await this.scheduleCleanupForExpiry(expiresAt);\n  }\n\n  async cacheSetIfNotExists(\n    key: string,\n    value: string,\n    ttlMs?: number\n  ): Promise<boolean> {\n    const now = Date.now();\n\n    const inserted = this.ctx.storage.transactionSync(() => {\n      this.ctx.storage.sql.exec(\n        \"DELETE FROM chat_sdk_state_cache WHERE key = ? AND expires_at IS NOT NULL AND expires_at <= ?\",\n        key,\n        now\n      );\n\n      if (this.readCacheValue(key, now) !== null) {\n        return false;\n      }\n\n      const expiresAt = ttlMs && ttlMs > 0 ? now + ttlMs : null;\n      this.upsertCacheValue(key, value, expiresAt);\n      return true;\n    });\n    if (inserted) {\n      const expiresAt = ttlMs && ttlMs > 0 ? Date.now() + ttlMs : null;\n      await this.scheduleCleanupForExpiry(expiresAt);\n    }\n    return inserted;\n  }\n\n  cacheDelete(key: string): void {\n    this.sql`\n      DELETE FROM chat_sdk_state_cache\n      WHERE key = ${key}\n    `;\n  }\n\n  async cleanupExpired(payload?: { expiresAt?: number }): Promise<void> {\n    const current = this.readCleanupMetadata();\n    if (\n      payload?.expiresAt !== undefined &&\n      current.nextCleanupAt !== null &&\n      payload.expiresAt !== current.nextCleanupAt\n    ) {\n      return;\n    }\n\n    const now = Date.now();\n    this.clearCleanupMetadata();\n\n    this.sql`\n      DELETE FROM chat_sdk_state_locks\n      WHERE expires_at <= ${now}\n    `;\n    this.sql`\n      DELETE FROM chat_sdk_state_cache\n      WHERE expires_at IS NOT NULL AND expires_at <= ${now}\n    `;\n    this.sql`\n      DELETE FROM chat_sdk_state_queue\n      WHERE expires_at <= ${now}\n    `;\n    this.sql`\n      DELETE FROM chat_sdk_state_lists\n      WHERE expires_at IS NOT NULL AND expires_at <= ${now}\n    `;\n    await this.scheduleNextCleanup();\n  }\n\n  private migrate(): void {\n    this.sql`\n      CREATE TABLE IF NOT EXISTS chat_sdk_state_subscriptions (\n        thread_id TEXT PRIMARY KEY\n      )\n    `;\n\n    this.sql`\n      CREATE TABLE IF NOT EXISTS chat_sdk_state_locks (\n        thread_id TEXT PRIMARY KEY,\n        token TEXT NOT NULL,\n        expires_at INTEGER NOT NULL\n      )\n    `;\n\n    this.sql`\n      CREATE TABLE IF NOT EXISTS chat_sdk_state_cache (\n        key TEXT PRIMARY KEY,\n        value TEXT NOT NULL,\n        expires_at INTEGER\n      )\n    `;\n\n    this.sql`\n      CREATE TABLE IF NOT EXISTS chat_sdk_state_queue (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        thread_id TEXT NOT NULL,\n        value TEXT NOT NULL,\n        enqueued_at INTEGER NOT NULL,\n        expires_at INTEGER NOT NULL\n      )\n    `;\n\n    this.sql`\n      CREATE TABLE IF NOT EXISTS chat_sdk_state_lists (\n        id INTEGER PRIMARY KEY AUTOINCREMENT,\n        key TEXT NOT NULL,\n        value TEXT NOT NULL,\n        expires_at INTEGER\n      )\n    `;\n\n    this.sql`\n      CREATE TABLE IF NOT EXISTS chat_sdk_state_metadata (\n        key TEXT PRIMARY KEY,\n        value TEXT NOT NULL\n      )\n    `;\n\n    this.sql`\n      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_locks_expires\n      ON chat_sdk_state_locks(expires_at)\n    `;\n    this.sql`\n      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_cache_expires\n      ON chat_sdk_state_cache(expires_at)\n      WHERE expires_at IS NOT NULL\n    `;\n    this.sql`\n      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_queue_thread\n      ON chat_sdk_state_queue(thread_id, id)\n    `;\n    this.sql`\n      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_queue_expires\n      ON chat_sdk_state_queue(expires_at)\n    `;\n    this.sql`\n      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_lists_key\n      ON chat_sdk_state_lists(key, id)\n    `;\n    this.sql`\n      CREATE INDEX IF NOT EXISTS idx_chat_sdk_state_lists_expires\n      ON chat_sdk_state_lists(expires_at)\n      WHERE expires_at IS NOT NULL\n    `;\n  }\n\n  private readCacheValue(key: string, now: number): string | null {\n    const rows = this.sql<{ value: string }>`\n      SELECT value\n      FROM chat_sdk_state_cache\n      WHERE key = ${key}\n        AND (expires_at IS NULL OR expires_at > ${now})\n    `;\n    return rows[0]?.value ?? null;\n  }\n\n  private upsertCacheValue(\n    key: string,\n    value: string,\n    expiresAt: number | null\n  ): void {\n    this.sql`\n      INSERT OR REPLACE INTO chat_sdk_state_cache (key, value, expires_at)\n      VALUES (${key}, ${value}, ${expiresAt})\n    `;\n  }\n\n  private async scheduleCleanupForExpiry(\n    expiresAt: number | null\n  ): Promise<void> {\n    if (expiresAt === null) {\n      return;\n    }\n\n    await this.ensureCleanupScheduled(expiresAt);\n  }\n\n  private async scheduleNextCleanup(): Promise<void> {\n    const next = this.nextExpiry();\n    if (next === null) {\n      const current = this.readCleanupMetadata();\n      if (current.scheduleId) {\n        await this.cancelSchedule(current.scheduleId).catch(() => false);\n      }\n      this.clearCleanupMetadata();\n      return;\n    }\n\n    await this.ensureCleanupScheduled(next);\n  }\n\n  private async ensureCleanupScheduled(expiresAt: number): Promise<void> {\n    const current = this.readCleanupMetadata();\n    if (current.nextCleanupAt !== null && current.nextCleanupAt <= expiresAt) {\n      return;\n    }\n\n    if (current.scheduleId) {\n      await this.cancelSchedule(current.scheduleId).catch(() => false);\n    }\n\n    const delaySeconds = Math.max(\n      0,\n      Math.ceil((expiresAt - Date.now()) / 1000)\n    );\n    const schedule = await this.schedule(delaySeconds, \"cleanupExpired\", {\n      expiresAt\n    });\n    this.writeCleanupMetadata(expiresAt, schedule.id);\n  }\n\n  private nextExpiry(): number | null {\n    const rows = this.sql<{ expires_at: number | null }>`\n      SELECT MIN(expires_at) as expires_at\n      FROM (\n        SELECT expires_at FROM chat_sdk_state_locks\n        UNION ALL\n        SELECT expires_at FROM chat_sdk_state_queue\n        UNION ALL\n        SELECT expires_at FROM chat_sdk_state_cache WHERE expires_at IS NOT NULL\n        UNION ALL\n        SELECT expires_at FROM chat_sdk_state_lists WHERE expires_at IS NOT NULL\n      )\n    `;\n    return rows[0]?.expires_at ?? null;\n  }\n\n  private readCleanupMetadata(): {\n    nextCleanupAt: number | null;\n    scheduleId: string | null;\n  } {\n    const rows = this.sql<{ key: string; value: string }>`\n      SELECT key, value\n      FROM chat_sdk_state_metadata\n      WHERE key IN (${NEXT_CLEANUP_AT_KEY}, ${CLEANUP_SCHEDULE_ID_KEY})\n    `;\n    const values = new Map(rows.map((row) => [row.key, row.value]));\n    const nextCleanupAt = Number(values.get(NEXT_CLEANUP_AT_KEY));\n    return {\n      nextCleanupAt: Number.isFinite(nextCleanupAt) ? nextCleanupAt : null,\n      scheduleId: values.get(CLEANUP_SCHEDULE_ID_KEY) ?? null\n    };\n  }\n\n  private writeCleanupMetadata(expiresAt: number, scheduleId: string): void {\n    this.sql`\n      INSERT OR REPLACE INTO chat_sdk_state_metadata (key, value)\n      VALUES (${NEXT_CLEANUP_AT_KEY}, ${String(expiresAt)})\n    `;\n    this.sql`\n      INSERT OR REPLACE INTO chat_sdk_state_metadata (key, value)\n      VALUES (${CLEANUP_SCHEDULE_ID_KEY}, ${scheduleId})\n    `;\n  }\n\n  private clearCleanupMetadata(): void {\n    this.sql`\n      DELETE FROM chat_sdk_state_metadata\n      WHERE key IN (${NEXT_CLEANUP_AT_KEY}, ${CLEANUP_SCHEDULE_ID_KEY})\n    `;\n  }\n}\n\nfunction parseQueueEntry(value: string): StoredQueueEntry {\n  const raw = JSON.parse(value) as Record<string, unknown>;\n  if (typeof raw.enqueuedAt !== \"number\" || typeof raw.expiresAt !== \"number\") {\n    throw new Error(\n      \"ChatSdkStateAgent expected QueueEntry JSON with numeric TTLs\"\n    );\n  }\n\n  return {\n    enqueuedAt: raw.enqueuedAt,\n    expiresAt: raw.expiresAt\n  };\n}\n","import type {\n  Lock as ChatSdkLock,\n  QueueEntry as ChatSdkQueueEntry,\n  StateAdapter as ChatSdkStateAdapterInterface\n} from \"chat\";\nimport {\n  getCurrentAgent,\n  type SubAgentClass,\n  type SubAgentStub\n} from \"../index\";\nimport { ChatSdkStateAgent } from \"./agent\";\nimport type { ChatSdkStateAdapterOptions } from \"./types\";\n\nconst THREAD_STATE_PREFIX = \"thread-state:\";\nconst CHANNEL_STATE_PREFIX = \"channel-state:\";\nconst MESSAGE_HISTORY_PREFIX = \"msg-history:\";\nconst TRANSCRIPTS_USER_PREFIX = \"transcripts:user:\";\n\nfunction parseStoredJson<T>(raw: string, label: string): T {\n  try {\n    return JSON.parse(raw) as T;\n  } catch (error) {\n    throw new Error(`ChatSdkStateAdapter expected JSON-encoded ${label}`, {\n      cause: error\n    });\n  }\n}\n\nexport function defaultThreadShard(threadId: string): string {\n  return threadId.split(\":\").slice(0, 2).join(\":\") || \"default\";\n}\n\nexport function defaultKeyShard(\n  key: string,\n  shardThread: (threadId: string) => string = defaultThreadShard\n): string | undefined {\n  for (const prefix of [\n    THREAD_STATE_PREFIX,\n    CHANNEL_STATE_PREFIX,\n    MESSAGE_HISTORY_PREFIX,\n    TRANSCRIPTS_USER_PREFIX\n  ]) {\n    if (key.startsWith(prefix)) {\n      return shardThread(key.slice(prefix.length));\n    }\n  }\n\n  return undefined;\n}\n\nexport class ChatSdkStateAdapter implements ChatSdkStateAdapterInterface {\n  private readonly parent: NonNullable<ChatSdkStateAdapterOptions[\"parent\"]>;\n  private readonly agentClass: SubAgentClass<ChatSdkStateAgent>;\n  private readonly defaultName: string;\n  private readonly keyShard?: (key: string) => string | undefined;\n  private readonly shardKey: (threadId: string) => string;\n  private connected = false;\n\n  constructor(options: ChatSdkStateAdapterOptions = {}) {\n    const parent = options.parent ?? getCurrentAgent().agent;\n    if (!parent) {\n      throw new Error(\n        \"ChatSdkStateAdapter requires a parent Agent. Pass `parent` or create it inside an Agent context.\"\n      );\n    }\n\n    this.parent = parent;\n    this.agentClass = options.agent ?? ChatSdkStateAgent;\n    this.defaultName = options.name ?? \"default\";\n    this.keyShard = options.keyShard;\n    this.shardKey = options.shardKey ?? defaultThreadShard;\n  }\n\n  async connect(): Promise<void> {\n    this.connected = true;\n  }\n\n  async disconnect(): Promise<void> {\n    this.connected = false;\n  }\n\n  async subscribe(threadId: string): Promise<void> {\n    await (await this.stateAgent(threadId)).subscribe(threadId);\n  }\n\n  async unsubscribe(threadId: string): Promise<void> {\n    await (await this.stateAgent(threadId)).unsubscribe(threadId);\n  }\n\n  async isSubscribed(threadId: string): Promise<boolean> {\n    return (await this.stateAgent(threadId)).isSubscribed(threadId);\n  }\n\n  async acquireLock(\n    threadId: string,\n    ttlMs: number\n  ): Promise<ChatSdkLock | null> {\n    return (await this.stateAgent(threadId)).acquireLock(threadId, ttlMs);\n  }\n\n  async releaseLock(lock: ChatSdkLock): Promise<void> {\n    await (\n      await this.stateAgent(lock.threadId)\n    ).releaseLock(lock.threadId, lock.token);\n  }\n\n  async extendLock(lock: ChatSdkLock, ttlMs: number): Promise<boolean> {\n    return (await this.stateAgent(lock.threadId)).extendLock(\n      lock.threadId,\n      lock.token,\n      ttlMs\n    );\n  }\n\n  async forceReleaseLock(threadId: string): Promise<void> {\n    await (await this.stateAgent(threadId)).forceReleaseLock(threadId);\n  }\n\n  async enqueue(\n    threadId: string,\n    entry: ChatSdkQueueEntry,\n    maxSize: number\n  ): Promise<number> {\n    return (await this.stateAgent(threadId)).enqueue(\n      threadId,\n      JSON.stringify(entry),\n      maxSize\n    );\n  }\n\n  async dequeue(threadId: string): Promise<ChatSdkQueueEntry | null> {\n    const raw = await (await this.stateAgent(threadId)).popQueue(threadId);\n    return raw === null\n      ? null\n      : parseStoredJson<ChatSdkQueueEntry>(raw, `queue entry for ${threadId}`);\n  }\n\n  async queueDepth(threadId: string): Promise<number> {\n    return (await this.stateAgent(threadId)).queueDepth(threadId);\n  }\n\n  async appendToList(\n    key: string,\n    value: unknown,\n    options?: { maxLength?: number; ttlMs?: number }\n  ): Promise<void> {\n    await (\n      await this.stateAgentForKey(key)\n    ).listAppend(\n      key,\n      JSON.stringify(value),\n      options?.maxLength,\n      options?.ttlMs\n    );\n  }\n\n  async getList<T = unknown>(key: string): Promise<T[]> {\n    const raw = await (await this.stateAgentForKey(key)).listGet(key);\n    return raw.map((value) =>\n      parseStoredJson<T>(value, `list entry for ${key}`)\n    );\n  }\n\n  async get<T = unknown>(key: string): Promise<T | null> {\n    const raw = await (await this.stateAgentForKey(key)).cacheGet(key);\n    return raw === null ? null : parseStoredJson<T>(raw, `cache key ${key}`);\n  }\n\n  async set<T = unknown>(key: string, value: T, ttlMs?: number): Promise<void> {\n    await (\n      await this.stateAgentForKey(key)\n    ).cacheSet(key, JSON.stringify(value), ttlMs);\n  }\n\n  async setIfNotExists<T = unknown>(\n    key: string,\n    value: T,\n    ttlMs?: number\n  ): Promise<boolean> {\n    return (await this.stateAgentForKey(key)).cacheSetIfNotExists(\n      key,\n      JSON.stringify(value),\n      ttlMs\n    );\n  }\n\n  async delete(key: string): Promise<void> {\n    await (await this.stateAgentForKey(key)).cacheDelete(key);\n  }\n\n  private async stateAgent(\n    threadId?: string\n  ): Promise<SubAgentStub<ChatSdkStateAgent>> {\n    this.ensureConnected();\n    const name = threadId ? this.shardKey(threadId) : this.defaultName;\n    return this.parent.subAgent(this.agentClass, name);\n  }\n\n  private async stateAgentForKey(\n    key: string\n  ): Promise<SubAgentStub<ChatSdkStateAgent>> {\n    this.ensureConnected();\n    const name =\n      this.keyShard?.(key) ??\n      defaultKeyShard(key, this.shardKey) ??\n      this.defaultName;\n    return this.parent.subAgent(this.agentClass, name);\n  }\n\n  private ensureConnected(): void {\n    if (!this.connected) {\n      throw new Error(\"ChatSdkStateAdapter is not connected\");\n    }\n  }\n}\n","export {\n  ChatSdkStateAdapter,\n  defaultKeyShard,\n  defaultThreadShard\n} from \"./adapter\";\nexport { ChatSdkStateAgent } from \"./agent\";\nexport type { ChatSdkStateParent, ChatSdkStateAdapterOptions } from \"./types\";\n\nimport { ChatSdkStateAdapter } from \"./adapter\";\nimport type { ChatSdkStateAdapterOptions } from \"./types\";\n\nexport function createChatSdkState(\n  options: ChatSdkStateAdapterOptions = {}\n): ChatSdkStateAdapter {\n  return new ChatSdkStateAdapter(options);\n}\n"],"mappings":";;AAaA,MAAM,sBAAsB;AAC5B,MAAM,0BAA0B;AAEhC,IAAa,oBAAb,cAAuC,MAAM;CAC3C,UAAgB;EACd,KAAK,QAAQ;EACb,KAAU,oBAAoB;CAChC;CAEA,UAAU,UAAwB;EAChC,KAAK,GAAG;;gBAEI,SAAS;;CAEvB;CAEA,YAAY,UAAwB;EAClC,KAAK,GAAG;;0BAEc,SAAS;;CAEjC;CAEA,aAAa,UAA2B;EAOtC,OAAO,KANW,GAAsB;;;0BAGlB,SAAS;;MAGnB,SAAS;CACvB;CAEA,MAAM,YACJ,UACA,OAC4B;EAC5B,MAAM,SAAS,KAAK,IAAI,QAAQ,sBAAsB;GACpD,MAAM,MAAM,KAAK,IAAI;GAErB,KAAK,IAAI,QAAQ,IAAI,KACnB,4EACA,UACA,GACF;GAQA,IANiB,KAAK,IAAI,QAAQ,IAC/B,KACC,kEACA,QACF,EACC,QACQ,EAAE,SAAS,GACpB,OAAO;GAGT,MAAM,QAAQ,OAAO,WAAW;GAChC,MAAM,YAAY,MAAM;GAExB,KAAK,IAAI,QAAQ,IAAI,KACnB,oFACA,UACA,OACA,SACF;GAEA,OAAO;IAAE;IAAU;IAAO;GAAU;EACtC,CAAC;EAED,MAAM,KAAK,yBAAyB,QAAQ,aAAa,IAAI;EAC7D,OAAO;CACT;CAEA,YAAY,UAAkB,OAAqB;EACjD,KAAK,GAAG;;0BAEc,SAAS,eAAe,MAAM;;CAEtD;CAEA,MAAM,WACJ,UACA,OACA,OACkB;EAClB,MAAM,SAAS,KAAK,IAAI,QAAQ,sBAAsB;GACpD,MAAM,MAAM,KAAK,IAAI;GAYrB,OAXa,KAAK,IAAI,QAAQ,IAC3B,KACC;;iCAGA,MAAM,OACN,UACA,OACA,GACF,EACC,QACO,EAAE,SAAS;EACvB,CAAC;EACD,IAAI,QACF,MAAM,KAAK,yBAAyB,KAAK,IAAI,IAAI,KAAK;EAExD,OAAO;CACT;CAEA,iBAAiB,UAAwB;EACvC,KAAK,GAAG;;0BAEc,SAAS;;CAEjC;CAEA,MAAM,QACJ,UACA,OACA,SACiB;EACjB,MAAM,SAAS,gBAAgB,KAAK;EAEpC,MAAM,QAAQ,KAAK,IAAI,QAAQ,sBAAsB;GACnD,KAAK,IAAI,QAAQ,IAAI,KACnB,oGACA,UACA,OACA,OAAO,YACP,OAAO,SACT;GAEA,KAAK,IAAI,QAAQ,IAAI,KACnB;;;;;YAMA,UACA,UACA,OACF;GAQA,OANY,KAAK,IAAI,QAAQ,IAC1B,KACC,0EACA,QACF,EACC,IACM,EAAE;EACb,CAAC;EACD,MAAM,KAAK,yBAAyB,OAAO,SAAS;EACpD,OAAO;CACT;CAEA,SAAS,UAAiC;EACxC,OAAO,KAAK,IAAI,QAAQ,sBAAsB;GAC5C,MAAM,MAAM,KAAK,IAAI;GAErB,KAAK,IAAI,QAAQ,IAAI,KACnB,4EACA,UACA,GACF;GAQA,MAAM,MANO,KAAK,IAAI,QAAQ,IAC3B,KACC,0FACA,QACF,EACC,QACY,EAAE;GACjB,IAAI,CAAC,KACH,OAAO;GAGT,KAAK,IAAI,QAAQ,IAAI,KACnB,iDACA,IAAI,EACN;GACA,OAAO,IAAI;EACb,CAAC;CACH;CAEA,WAAW,UAA0B;EAMnC,OAAO,KALW,GAAsB;;;0BAGlB,SAAS,oBAAoB,KAAK,IAAI,EAAE;MAElD,IAAI,SAAS;CAC3B;CAEA,MAAM,WACJ,KACA,OACA,WACA,OACe;EACf,MAAM,YAAY,SAAS,QAAQ,IAAI,KAAK,IAAI,IAAI,QAAQ;EAE5D,KAAK,IAAI,QAAQ,sBAAsB;GACrC,KAAK,IAAI,QAAQ,IAAI,KACnB,8EACA,KACA,OACA,SACF;GAEA,IAAI,cAAc,MAGhB,KAAK,IAAI,QAAQ,IAAI,KACnB,gEACA,WACA,GACF;GAGF,IAAI,aAAa,QAAQ,YAAY,GACnC,KAAK,IAAI,QAAQ,IAAI,KACnB;;;;;cAMA,KACA,KACA,SACF;EAEJ,CAAC;EACD,MAAM,KAAK,yBAAyB,SAAS;CAC/C;CAEA,QAAQ,KAAuB;EAC7B,MAAM,MAAM,KAAK,IAAI;EAErB,KAAK,GAAG;;oBAEQ,IAAI;;4BAEI,IAAI;;EAG5B,OAAO,KAAK,GAAsB;;;oBAGlB,IAAI;;MAElB,KAAK,QAAQ,IAAI,KAAK;CAC1B;CAEA,SAAS,KAA4B;EACnC,OAAO,KAAK,eAAe,KAAK,KAAK,IAAI,CAAC;CAC5C;CAEA,MAAM,SAAS,KAAa,OAAe,OAA+B;EACxE,MAAM,YAAY,SAAS,QAAQ,IAAI,KAAK,IAAI,IAAI,QAAQ;EAC5D,KAAK,iBAAiB,KAAK,OAAO,SAAS;EAC3C,MAAM,KAAK,yBAAyB,SAAS;CAC/C;CAEA,MAAM,oBACJ,KACA,OACA,OACkB;EAClB,MAAM,MAAM,KAAK,IAAI;EAErB,MAAM,WAAW,KAAK,IAAI,QAAQ,sBAAsB;GACtD,KAAK,IAAI,QAAQ,IAAI,KACnB,iGACA,KACA,GACF;GAEA,IAAI,KAAK,eAAe,KAAK,GAAG,MAAM,MACpC,OAAO;GAGT,MAAM,YAAY,SAAS,QAAQ,IAAI,MAAM,QAAQ;GACrD,KAAK,iBAAiB,KAAK,OAAO,SAAS;GAC3C,OAAO;EACT,CAAC;EACD,IAAI,UAAU;GACZ,MAAM,YAAY,SAAS,QAAQ,IAAI,KAAK,IAAI,IAAI,QAAQ;GAC5D,MAAM,KAAK,yBAAyB,SAAS;EAC/C;EACA,OAAO;CACT;CAEA,YAAY,KAAmB;EAC7B,KAAK,GAAG;;oBAEQ,IAAI;;CAEtB;CAEA,MAAM,eAAe,SAAiD;EACpE,MAAM,UAAU,KAAK,oBAAoB;EACzC,IACE,SAAS,cAAc,KAAA,KACvB,QAAQ,kBAAkB,QAC1B,QAAQ,cAAc,QAAQ,eAE9B;EAGF,MAAM,MAAM,KAAK,IAAI;EACrB,KAAK,qBAAqB;EAE1B,KAAK,GAAG;;4BAEgB,IAAI;;EAE5B,KAAK,GAAG;;uDAE2C,IAAI;;EAEvD,KAAK,GAAG;;4BAEgB,IAAI;;EAE5B,KAAK,GAAG;;uDAE2C,IAAI;;EAEvD,MAAM,KAAK,oBAAoB;CACjC;CAEA,UAAwB;EACtB,KAAK,GAAG;;;;;EAMR,KAAK,GAAG;;;;;;;EAQR,KAAK,GAAG;;;;;;;EAQR,KAAK,GAAG;;;;;;;;;EAUR,KAAK,GAAG;;;;;;;;EASR,KAAK,GAAG;;;;;;EAOR,KAAK,GAAG;;;;EAIR,KAAK,GAAG;;;;;EAKR,KAAK,GAAG;;;;EAIR,KAAK,GAAG;;;;EAIR,KAAK,GAAG;;;;EAIR,KAAK,GAAG;;;;;CAKV;CAEA,eAAuB,KAAa,KAA4B;EAO9D,OAAO,KANW,GAAsB;;;oBAGxB,IAAI;kDAC0B,IAAI;MAEtC,IAAI,SAAS;CAC3B;CAEA,iBACE,KACA,OACA,WACM;EACN,KAAK,GAAG;;gBAEI,IAAI,IAAI,MAAM,IAAI,UAAU;;CAE1C;CAEA,MAAc,yBACZ,WACe;EACf,IAAI,cAAc,MAChB;EAGF,MAAM,KAAK,uBAAuB,SAAS;CAC7C;CAEA,MAAc,sBAAqC;EACjD,MAAM,OAAO,KAAK,WAAW;EAC7B,IAAI,SAAS,MAAM;GACjB,MAAM,UAAU,KAAK,oBAAoB;GACzC,IAAI,QAAQ,YACV,MAAM,KAAK,eAAe,QAAQ,UAAU,EAAE,YAAY,KAAK;GAEjE,KAAK,qBAAqB;GAC1B;EACF;EAEA,MAAM,KAAK,uBAAuB,IAAI;CACxC;CAEA,MAAc,uBAAuB,WAAkC;EACrE,MAAM,UAAU,KAAK,oBAAoB;EACzC,IAAI,QAAQ,kBAAkB,QAAQ,QAAQ,iBAAiB,WAC7D;EAGF,IAAI,QAAQ,YACV,MAAM,KAAK,eAAe,QAAQ,UAAU,EAAE,YAAY,KAAK;EAGjE,MAAM,eAAe,KAAK,IACxB,GACA,KAAK,MAAM,YAAY,KAAK,IAAI,KAAK,GAAI,CAC3C;EACA,MAAM,WAAW,MAAM,KAAK,SAAS,cAAc,kBAAkB,EACnE,UACF,CAAC;EACD,KAAK,qBAAqB,WAAW,SAAS,EAAE;CAClD;CAEA,aAAoC;EAalC,OAAO,KAZW,GAAkC;;;;;;;;;;;MAYxC,IAAI,cAAc;CAChC;CAEA,sBAGE;EACA,MAAM,OAAO,KAAK,GAAmC;;;sBAGnC,oBAAoB,IAAI,wBAAwB;;EAElE,MAAM,SAAS,IAAI,IAAI,KAAK,KAAK,QAAQ,CAAC,IAAI,KAAK,IAAI,KAAK,CAAC,CAAC;EAC9D,MAAM,gBAAgB,OAAO,OAAO,IAAI,mBAAmB,CAAC;EAC5D,OAAO;GACL,eAAe,OAAO,SAAS,aAAa,IAAI,gBAAgB;GAChE,YAAY,OAAO,IAAI,uBAAuB,KAAK;EACrD;CACF;CAEA,qBAA6B,WAAmB,YAA0B;EACxE,KAAK,GAAG;;gBAEI,oBAAoB,IAAI,OAAO,SAAS,EAAE;;EAEtD,KAAK,GAAG;;gBAEI,wBAAwB,IAAI,WAAW;;CAErD;CAEA,uBAAqC;EACnC,KAAK,GAAG;;sBAEU,oBAAoB,IAAI,wBAAwB;;CAEpE;AACF;AAEA,SAAS,gBAAgB,OAAiC;CACxD,MAAM,MAAM,KAAK,MAAM,KAAK;CAC5B,IAAI,OAAO,IAAI,eAAe,YAAY,OAAO,IAAI,cAAc,UACjE,MAAM,IAAI,MACR,8DACF;CAGF,OAAO;EACL,YAAY,IAAI;EAChB,WAAW,IAAI;CACjB;AACF;;;ACxhBA,MAAM,sBAAsB;AAC5B,MAAM,uBAAuB;AAC7B,MAAM,yBAAyB;AAC/B,MAAM,0BAA0B;AAEhC,SAAS,gBAAmB,KAAa,OAAkB;CACzD,IAAI;EACF,OAAO,KAAK,MAAM,GAAG;CACvB,SAAS,OAAO;EACd,MAAM,IAAI,MAAM,6CAA6C,SAAS,EACpE,OAAO,MACT,CAAC;CACH;AACF;AAEA,SAAgB,mBAAmB,UAA0B;CAC3D,OAAO,SAAS,MAAM,GAAG,EAAE,MAAM,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK;AACtD;AAEA,SAAgB,gBACd,KACA,cAA4C,oBACxB;CACpB,KAAK,MAAM,UAAU;EACnB;EACA;EACA;EACA;CACF,GACE,IAAI,IAAI,WAAW,MAAM,GACvB,OAAO,YAAY,IAAI,MAAM,OAAO,MAAM,CAAC;AAKjD;AAEA,IAAa,sBAAb,MAAyE;CAQvE,YAAY,UAAsC,CAAC,GAAG;EAFtD,KAAQ,YAAY;EAGlB,MAAM,SAAS,QAAQ,UAAU,gBAAgB,EAAE;EACnD,IAAI,CAAC,QACH,MAAM,IAAI,MACR,kGACF;EAGF,KAAK,SAAS;EACd,KAAK,aAAa,QAAQ,SAAS;EACnC,KAAK,cAAc,QAAQ,QAAQ;EACnC,KAAK,WAAW,QAAQ;EACxB,KAAK,WAAW,QAAQ,YAAY;CACtC;CAEA,MAAM,UAAyB;EAC7B,KAAK,YAAY;CACnB;CAEA,MAAM,aAA4B;EAChC,KAAK,YAAY;CACnB;CAEA,MAAM,UAAU,UAAiC;EAC/C,OAAO,MAAM,KAAK,WAAW,QAAQ,GAAG,UAAU,QAAQ;CAC5D;CAEA,MAAM,YAAY,UAAiC;EACjD,OAAO,MAAM,KAAK,WAAW,QAAQ,GAAG,YAAY,QAAQ;CAC9D;CAEA,MAAM,aAAa,UAAoC;EACrD,QAAQ,MAAM,KAAK,WAAW,QAAQ,GAAG,aAAa,QAAQ;CAChE;CAEA,MAAM,YACJ,UACA,OAC6B;EAC7B,QAAQ,MAAM,KAAK,WAAW,QAAQ,GAAG,YAAY,UAAU,KAAK;CACtE;CAEA,MAAM,YAAY,MAAkC;EAClD,OACE,MAAM,KAAK,WAAW,KAAK,QAAQ,GACnC,YAAY,KAAK,UAAU,KAAK,KAAK;CACzC;CAEA,MAAM,WAAW,MAAmB,OAAiC;EACnE,QAAQ,MAAM,KAAK,WAAW,KAAK,QAAQ,GAAG,WAC5C,KAAK,UACL,KAAK,OACL,KACF;CACF;CAEA,MAAM,iBAAiB,UAAiC;EACtD,OAAO,MAAM,KAAK,WAAW,QAAQ,GAAG,iBAAiB,QAAQ;CACnE;CAEA,MAAM,QACJ,UACA,OACA,SACiB;EACjB,QAAQ,MAAM,KAAK,WAAW,QAAQ,GAAG,QACvC,UACA,KAAK,UAAU,KAAK,GACpB,OACF;CACF;CAEA,MAAM,QAAQ,UAAqD;EACjE,MAAM,MAAM,OAAO,MAAM,KAAK,WAAW,QAAQ,GAAG,SAAS,QAAQ;EACrE,OAAO,QAAQ,OACX,OACA,gBAAmC,KAAK,mBAAmB,UAAU;CAC3E;CAEA,MAAM,WAAW,UAAmC;EAClD,QAAQ,MAAM,KAAK,WAAW,QAAQ,GAAG,WAAW,QAAQ;CAC9D;CAEA,MAAM,aACJ,KACA,OACA,SACe;EACf,OACE,MAAM,KAAK,iBAAiB,GAAG,GAC/B,WACA,KACA,KAAK,UAAU,KAAK,GACpB,SAAS,WACT,SAAS,KACX;CACF;CAEA,MAAM,QAAqB,KAA2B;EAEpD,QAAO,OADY,MAAM,KAAK,iBAAiB,GAAG,GAAG,QAAQ,GAAG,GACrD,KAAK,UACd,gBAAmB,OAAO,kBAAkB,KAAK,CACnD;CACF;CAEA,MAAM,IAAiB,KAAgC;EACrD,MAAM,MAAM,OAAO,MAAM,KAAK,iBAAiB,GAAG,GAAG,SAAS,GAAG;EACjE,OAAO,QAAQ,OAAO,OAAO,gBAAmB,KAAK,aAAa,KAAK;CACzE;CAEA,MAAM,IAAiB,KAAa,OAAU,OAA+B;EAC3E,OACE,MAAM,KAAK,iBAAiB,GAAG,GAC/B,SAAS,KAAK,KAAK,UAAU,KAAK,GAAG,KAAK;CAC9C;CAEA,MAAM,eACJ,KACA,OACA,OACkB;EAClB,QAAQ,MAAM,KAAK,iBAAiB,GAAG,GAAG,oBACxC,KACA,KAAK,UAAU,KAAK,GACpB,KACF;CACF;CAEA,MAAM,OAAO,KAA4B;EACvC,OAAO,MAAM,KAAK,iBAAiB,GAAG,GAAG,YAAY,GAAG;CAC1D;CAEA,MAAc,WACZ,UAC0C;EAC1C,KAAK,gBAAgB;EACrB,MAAM,OAAO,WAAW,KAAK,SAAS,QAAQ,IAAI,KAAK;EACvD,OAAO,KAAK,OAAO,SAAS,KAAK,YAAY,IAAI;CACnD;CAEA,MAAc,iBACZ,KAC0C;EAC1C,KAAK,gBAAgB;EACrB,MAAM,OACJ,KAAK,WAAW,GAAG,KACnB,gBAAgB,KAAK,KAAK,QAAQ,KAClC,KAAK;EACP,OAAO,KAAK,OAAO,SAAS,KAAK,YAAY,IAAI;CACnD;CAEA,kBAAgC;EAC9B,IAAI,CAAC,KAAK,WACR,MAAM,IAAI,MAAM,sCAAsC;CAE1D;AACF;;;AC3MA,SAAgB,mBACd,UAAsC,CAAC,GAClB;CACrB,OAAO,IAAI,oBAAoB,OAAO;AACxC"}

package/dist/client.d.ts

@@ -1,4 +1,4 @@
-import { v as Agent } from "./agent-tool-types-Dn9n-3SI.js";
+import { v as Agent } from "./agent-tool-types-l98LCbBl.js";
 import {
   ClientParameters,
   Method,

package/dist/{compaction-helpers-DAe-xiVY.d.ts → compaction-helpers-B-pG5J22.d.ts}

@@ -8,6 +8,7 @@ import { ToolSet } from "ai";
 interface SessionMessagePart {
   type: string;
   text?: string;
+  reasoning?: string;
   toolCallId?: string;
   toolName?: string;
   input?: unknown;
@@ -15,6 +16,28 @@ interface SessionMessagePart {
   state?: string;
   result?: unknown;
 }
+interface SessionTokenCounterInput {
+  /** Messages returned by `session.getHistory()` for the active branch. */
+  messages: SessionMessage[];
+  /** Frozen system prompt managed by the Session context system. */
+  systemPrompt: string;
+  /** Loaded context blocks that were used to build `systemPrompt`. */
+  contextBlocks: ContextBlock[];
+}
+type SessionTokenCounter = (
+  input: SessionTokenCounterInput
+) => number | Promise<number>;
+interface CompactAfterOptions {
+  /**
+   * Override the token estimate used by auto-compaction and status broadcasts.
+   *
+   * The default is a Workers-safe heuristic over message parts plus the
+   * Session-managed frozen system prompt. Callers that have model-reported
+   * usage or a tokenizer can provide a more precise counter here.
+   */
+  tokenCounter?: SessionTokenCounter;
+}
+type CompactionErrorHandler = (error: unknown) => void | Promise<void>;
 /**
  * Minimal message shape used by Session internals.
  * Vercel AI SDK's `UIMessage` is structurally compatible — you can pass
@@ -34,6 +57,10 @@ interface SessionOptions {
   context?: ContextConfig[];
   /** Provider for persisting the frozen system prompt. */
   promptStore?: WritableContextProvider;
+  /** Custom token counter for auto-compaction/status estimates. */
+  tokenCounter?: SessionTokenCounter;
+  /** Called when automatic compaction fails after a threshold trigger. */
+  onCompactionError?: CompactionErrorHandler;
 }
 //#endregion
 //#region src/experimental/memory/session/provider.d.ts
@@ -432,11 +459,14 @@ declare function createCompactFunction(
 ): (messages: SessionMessage[]) => Promise<CompactResult | null>;
 //#endregion
 export {
-  SessionOptions as A,
+  SessionMessage as A,
   AgentSessionProvider as C,
   StoredCompaction as D,
   SessionProvider as E,
-  SessionMessage as O,
+  SessionOptions as M,
+  SessionTokenCounter as N,
+  CompactAfterOptions as O,
+  SessionTokenCounterInput as P,
   isSearchProvider as S,
   SearchResult as T,
   R2SkillProvider as _,
@@ -448,7 +478,8 @@ export {
   isWritableProvider as g,
   WritableContextProvider as h,
   alignBoundaryBackward as i,
-  SessionMessagePart as k,
+  SessionMessagePart as j,
+  CompactionErrorHandler as k,
   findTailCutByTokens as l,
   ContextProvider as m,
   CompactOptions as n,
@@ -463,4 +494,4 @@ export {
   SearchProvider as x,
   isSkillProvider as y
 };
-//# sourceMappingURL=compaction-helpers-DAe-xiVY.d.ts.map
+//# sourceMappingURL=compaction-helpers-B-pG5J22.d.ts.map

package/dist/{compaction-helpers-DvcZnvQ1.js → compaction-helpers-fJyf8j4m.js}

@@ -20,10 +20,19 @@ function estimateStringTokens(text) {
 	const wordEstimate = text.split(/\s+/).filter(Boolean).length * WORDS_TOKEN_MULTIPLIER;
 	return Math.ceil(Math.max(charEstimate, wordEstimate));
 }
+function estimateUnknownTokens(value) {
+	if (value === null || value === void 0) return 0;
+	if (typeof value === "string") return estimateStringTokens(value);
+	try {
+		return estimateStringTokens(JSON.stringify(value));
+	} catch {
+		return estimateStringTokens(String(value));
+	}
+}
 /**
 * Estimate total token count for an array of UIMessages.
 *
-* Walks each message's parts (text, tool invocations, tool results)
+* Walks each message's parts (text, reasoning, tool invocations, tool results)
 * and applies per-message overhead.
 *
 * This is a heuristic. Do not use where exact counts are required.
@@ -32,12 +41,12 @@ function estimateMessageTokens(messages) {
 	let tokens = 0;
 	for (const msg of messages) {
 		tokens += 4;
-		for (const part of msg.parts) if (part.type === "text") tokens += estimateStringTokens(part.text);
+		for (const part of msg.parts) if (part.type === "text" || part.type === "reasoning") tokens += estimateUnknownTokens(part.text ?? part.reasoning);
 		else if (part.type.startsWith("tool-") || part.type === "dynamic-tool") {
-			const toolPart = part;
-			if (toolPart.input) tokens += estimateStringTokens(JSON.stringify(toolPart.input));
-			if (toolPart.output) tokens += estimateStringTokens(JSON.stringify(toolPart.output));
-		}
+			tokens += estimateUnknownTokens(part.input);
+			tokens += estimateUnknownTokens(part.output ?? part.result);
+		} else if (part.text !== void 0) tokens += estimateUnknownTokens(part.text);
+		else if (part.result !== void 0) tokens += estimateUnknownTokens(part.result);
 	}
 	return tokens;
 }
@@ -309,4 +318,4 @@ function createCompactFunction(opts) {
 //#endregion
 export { computeSummaryBudget as a, isCompactionMessage as c, TOKENS_PER_MESSAGE as d, WORDS_TOKEN_MULTIPLIER as f, buildSummaryPrompt as i, sanitizeToolPairs as l, estimateStringTokens as m, alignBoundaryBackward as n, createCompactFunction as o, estimateMessageTokens as p, alignBoundaryForward as r, findTailCutByTokens as s, COMPACTION_PREFIX as t, CHARS_PER_TOKEN as u };
 
-//# sourceMappingURL=compaction-helpers-DvcZnvQ1.js.map
+//# sourceMappingURL=compaction-helpers-fJyf8j4m.js.map