agents 0.14.0 → 0.14.2

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

@@ -1 +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"}
+{"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  async onStart(): Promise<void> {\n    this.migrate();\n    await 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,MAAM,UAAyB;EAC7B,KAAK,QAAQ;EACb,MAAM,KAAK,oBAAoB;CACjC;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,CAAC,CACA,QACQ,CAAC,CAAC,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,CAAC,CACA,QACO,CAAC,CAAC,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,CAAC,CACA,IACM,CAAC,CAAC;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,CAAC,CACA,QACY,CAAC,CAAC;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,EAAE,EAAE,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,EAAE,EAAE,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,CAAC,CAAC,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,CAAC,CAAC,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,EAAE,EAAE,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,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,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,CAAC,CAAC;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,EAAA,CAAG,UAAU,QAAQ;CAC5D;CAEA,MAAM,YAAY,UAAiC;EACjD,OAAO,MAAM,KAAK,WAAW,QAAQ,EAAA,CAAG,YAAY,QAAQ;CAC9D;CAEA,MAAM,aAAa,UAAoC;EACrD,QAAQ,MAAM,KAAK,WAAW,QAAQ,EAAA,CAAG,aAAa,QAAQ;CAChE;CAEA,MAAM,YACJ,UACA,OAC6B;EAC7B,QAAQ,MAAM,KAAK,WAAW,QAAQ,EAAA,CAAG,YAAY,UAAU,KAAK;CACtE;CAEA,MAAM,YAAY,MAAkC;EAClD,OACE,MAAM,KAAK,WAAW,KAAK,QAAQ,EAAA,CACnC,YAAY,KAAK,UAAU,KAAK,KAAK;CACzC;CAEA,MAAM,WAAW,MAAmB,OAAiC;EACnE,QAAQ,MAAM,KAAK,WAAW,KAAK,QAAQ,EAAA,CAAG,WAC5C,KAAK,UACL,KAAK,OACL,KACF;CACF;CAEA,MAAM,iBAAiB,UAAiC;EACtD,OAAO,MAAM,KAAK,WAAW,QAAQ,EAAA,CAAG,iBAAiB,QAAQ;CACnE;CAEA,MAAM,QACJ,UACA,OACA,SACiB;EACjB,QAAQ,MAAM,KAAK,WAAW,QAAQ,EAAA,CAAG,QACvC,UACA,KAAK,UAAU,KAAK,GACpB,OACF;CACF;CAEA,MAAM,QAAQ,UAAqD;EACjE,MAAM,MAAM,OAAO,MAAM,KAAK,WAAW,QAAQ,EAAA,CAAG,SAAS,QAAQ;EACrE,OAAO,QAAQ,OACX,OACA,gBAAmC,KAAK,mBAAmB,UAAU;CAC3E;CAEA,MAAM,WAAW,UAAmC;EAClD,QAAQ,MAAM,KAAK,WAAW,QAAQ,EAAA,CAAG,WAAW,QAAQ;CAC9D;CAEA,MAAM,aACJ,KACA,OACA,SACe;EACf,OACE,MAAM,KAAK,iBAAiB,GAAG,EAAA,CAC/B,WACA,KACA,KAAK,UAAU,KAAK,GACpB,SAAS,WACT,SAAS,KACX;CACF;CAEA,MAAM,QAAqB,KAA2B;EAEpD,QAAO,OADY,MAAM,KAAK,iBAAiB,GAAG,EAAA,CAAG,QAAQ,GAAG,EAAA,CACrD,KAAK,UACd,gBAAmB,OAAO,kBAAkB,KAAK,CACnD;CACF;CAEA,MAAM,IAAiB,KAAgC;EACrD,MAAM,MAAM,OAAO,MAAM,KAAK,iBAAiB,GAAG,EAAA,CAAG,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,EAAA,CAC/B,SAAS,KAAK,KAAK,UAAU,KAAK,GAAG,KAAK;CAC9C;CAEA,MAAM,eACJ,KACA,OACA,OACkB;EAClB,QAAQ,MAAM,KAAK,iBAAiB,GAAG,EAAA,CAAG,oBACxC,KACA,KAAK,UAAU,KAAK,GACpB,KACF;CACF;CAEA,MAAM,OAAO,KAA4B;EACvC,OAAO,MAAM,KAAK,iBAAiB,GAAG,EAAA,CAAG,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/{classPrivateFieldGet2-Evpt0SEr.js → classPrivateFieldGet2-Beqsfu2Z.js}

@@ -1,25 +1,25 @@
-//#region \0@oxc-project+runtime@0.130.0/helpers/checkPrivateRedeclaration.js
+//#region \0@oxc-project+runtime@0.134.0/helpers/esm/checkPrivateRedeclaration.js
 function _checkPrivateRedeclaration(e, t) {
 	if (t.has(e)) throw new TypeError("Cannot initialize the same private elements twice on an object");
 }
 //#endregion
-//#region \0@oxc-project+runtime@0.130.0/helpers/classPrivateFieldInitSpec.js
+//#region \0@oxc-project+runtime@0.134.0/helpers/esm/classPrivateFieldInitSpec.js
 function _classPrivateFieldInitSpec(e, t, a) {
 	_checkPrivateRedeclaration(e, t), t.set(e, a);
 }
 //#endregion
-//#region \0@oxc-project+runtime@0.130.0/helpers/assertClassBrand.js
+//#region \0@oxc-project+runtime@0.134.0/helpers/esm/assertClassBrand.js
 function _assertClassBrand(e, t, n) {
 	if ("function" == typeof e ? e === t : e.has(t)) return arguments.length < 3 ? t : n;
 	throw new TypeError("Private element is not present on this object");
 }
 //#endregion
-//#region \0@oxc-project+runtime@0.130.0/helpers/classPrivateFieldSet2.js
+//#region \0@oxc-project+runtime@0.134.0/helpers/esm/classPrivateFieldSet2.js
 function _classPrivateFieldSet2(s, a, r) {
 	return s.set(_assertClassBrand(s, a), r), r;
 }
 //#endregion
-//#region \0@oxc-project+runtime@0.130.0/helpers/classPrivateFieldGet2.js
+//#region \0@oxc-project+runtime@0.134.0/helpers/esm/classPrivateFieldGet2.js
 function _classPrivateFieldGet2(s, a) {
 	return s.get(_assertClassBrand(s, a));
 }

package/dist/{classPrivateMethodInitSpec-bG0tD96O.js → classPrivateMethodInitSpec-B5ko1s2R.js}

@@ -1,5 +1,5 @@
-import { a as _checkPrivateRedeclaration } from "./classPrivateFieldGet2-Evpt0SEr.js";
-//#region \0@oxc-project+runtime@0.130.0/helpers/classPrivateMethodInitSpec.js
+import { a as _checkPrivateRedeclaration } from "./classPrivateFieldGet2-Beqsfu2Z.js";
+//#region \0@oxc-project+runtime@0.134.0/helpers/esm/classPrivateMethodInitSpec.js
 function _classPrivateMethodInitSpec(e, a) {
 	_checkPrivateRedeclaration(e, a), a.add(e);
 }

package/dist/cli/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../src/cli/create.ts","../../src/cli/index.ts"],"sourcesContent":["import yargs from \"yargs\";\nimport { hideBin } from \"yargs/helpers\";\n\nexport function createCli(argv = process.argv) {\n  return yargs(hideBin(argv))\n    .scriptName(\"agents\")\n    .usage(\"$0 <command> [options]\")\n    .command(\n      [\"init\", \"create\"],\n      \"Initialize an agents project\",\n      (cmd) => cmd,\n      async () => {\n        console.log(\"agents init: not implemented yet\");\n        process.exit(0);\n      }\n    )\n    .command(\n      \"dev\",\n      \"Start development server\",\n      (cmd) => cmd,\n      async () => {\n        console.log(\"agents dev: not implemented yet\");\n        process.exit(0);\n      }\n    )\n    .command(\n      \"deploy\",\n      \"Deploy agents to Cloudflare\",\n      (cmd) => cmd,\n      async () => {\n        console.log(\"agents deploy: not implemented yet\");\n        process.exit(0);\n      }\n    )\n    .command(\n      \"mcp\",\n      \"The agents mcp server\",\n      (cmd) => cmd,\n      async () => {\n        console.log(\"agents mcp: not implemented yet\");\n        process.exit(0);\n      }\n    )\n    .demandCommand(1, \"Please provide a command\")\n    .strict()\n    .help();\n}\n","#!/usr/bin/env node\n\nimport { createCli } from \"./create\";\n\nvoid createCli().parse();\n"],"mappings":";;;;AAGA,SAAgB,UAAU,OAAO,QAAQ,MAAM;CAC7C,OAAO,MAAM,QAAQ,IAAI,CAAC,EACvB,WAAW,QAAQ,EACnB,MAAM,wBAAwB,EAC9B,QACC,CAAC,QAAQ,QAAQ,GACjB,iCACC,QAAQ,KACT,YAAY;EACV,QAAQ,IAAI,kCAAkC;EAC9C,QAAQ,KAAK,CAAC;CAChB,CACF,EACC,QACC,OACA,6BACC,QAAQ,KACT,YAAY;EACV,QAAQ,IAAI,iCAAiC;EAC7C,QAAQ,KAAK,CAAC;CAChB,CACF,EACC,QACC,UACA,gCACC,QAAQ,KACT,YAAY;EACV,QAAQ,IAAI,oCAAoC;EAChD,QAAQ,KAAK,CAAC;CAChB,CACF,EACC,QACC,OACA,0BACC,QAAQ,KACT,YAAY;EACV,QAAQ,IAAI,iCAAiC;EAC7C,QAAQ,KAAK,CAAC;CAChB,CACF,EACC,cAAc,GAAG,0BAA0B,EAC3C,OAAO,EACP,KAAK;AACV;;;AC1CK,UAAU,EAAE,MAAM"}
+{"version":3,"file":"index.js","names":[],"sources":["../../src/cli/create.ts","../../src/cli/index.ts"],"sourcesContent":["import yargs from \"yargs\";\nimport { hideBin } from \"yargs/helpers\";\n\nexport function createCli(argv = process.argv) {\n  return yargs(hideBin(argv))\n    .scriptName(\"agents\")\n    .usage(\"$0 <command> [options]\")\n    .command(\n      [\"init\", \"create\"],\n      \"Initialize an agents project\",\n      (cmd) => cmd,\n      async () => {\n        console.log(\"agents init: not implemented yet\");\n        process.exit(0);\n      }\n    )\n    .command(\n      \"dev\",\n      \"Start development server\",\n      (cmd) => cmd,\n      async () => {\n        console.log(\"agents dev: not implemented yet\");\n        process.exit(0);\n      }\n    )\n    .command(\n      \"deploy\",\n      \"Deploy agents to Cloudflare\",\n      (cmd) => cmd,\n      async () => {\n        console.log(\"agents deploy: not implemented yet\");\n        process.exit(0);\n      }\n    )\n    .command(\n      \"mcp\",\n      \"The agents mcp server\",\n      (cmd) => cmd,\n      async () => {\n        console.log(\"agents mcp: not implemented yet\");\n        process.exit(0);\n      }\n    )\n    .demandCommand(1, \"Please provide a command\")\n    .strict()\n    .help();\n}\n","#!/usr/bin/env node\n\nimport { createCli } from \"./create\";\n\nvoid createCli().parse();\n"],"mappings":";;;;AAGA,SAAgB,UAAU,OAAO,QAAQ,MAAM;CAC7C,OAAO,MAAM,QAAQ,IAAI,CAAC,CAAC,CACxB,WAAW,QAAQ,CAAC,CACpB,MAAM,wBAAwB,CAAC,CAC/B,QACC,CAAC,QAAQ,QAAQ,GACjB,iCACC,QAAQ,KACT,YAAY;EACV,QAAQ,IAAI,kCAAkC;EAC9C,QAAQ,KAAK,CAAC;CAChB,CACF,CAAC,CACA,QACC,OACA,6BACC,QAAQ,KACT,YAAY;EACV,QAAQ,IAAI,iCAAiC;EAC7C,QAAQ,KAAK,CAAC;CAChB,CACF,CAAC,CACA,QACC,UACA,gCACC,QAAQ,KACT,YAAY;EACV,QAAQ,IAAI,oCAAoC;EAChD,QAAQ,KAAK,CAAC;CAChB,CACF,CAAC,CACA,QACC,OACA,0BACC,QAAQ,KACT,YAAY;EACV,QAAQ,IAAI,iCAAiC;EAC7C,QAAQ,KAAK,CAAC;CAChB,CACF,CAAC,CACA,cAAc,GAAG,0BAA0B,CAAC,CAC5C,OAAO,CAAC,CACR,KAAK;AACV;;;AC1CK,UAAU,CAAC,CAAC,MAAM"}

package/dist/{client-NradHZZz.js → client-FUizKzj2.js}

@@ -401,11 +401,12 @@ var MCPClientConnection = class {
 	/**
 	* Complete OAuth authorization
 	*/
-	async completeAuthorization(code) {
-		if (this.connectionState !== MCPConnectionState.AUTHENTICATING) throw new Error("Connection must be in authenticating state to complete authorization");
+	async completeAuthorization(code, options = {}) {
+		const expectedState = options.alreadyAccepted ? MCPConnectionState.CONNECTING : MCPConnectionState.AUTHENTICATING;
+		if (this.connectionState !== expectedState) throw new Error(`Connection must be in ${expectedState} state to complete authorization`);
+		if (!options.alreadyAccepted) this.connectionState = MCPConnectionState.CONNECTING;
 		try {
 			await this.finishAuthProbe(code);
-			this.connectionState = MCPConnectionState.CONNECTING;
 		} catch (error) {
 			this.connectionState = MCPConnectionState.FAILED;
 			throw error;
@@ -1031,6 +1032,54 @@ var MCPClientManager = class {
 			authError: error
 		};
 	}
+	isAuthAcceptedConnection(conn) {
+		return conn.connectionState === MCPConnectionState.READY || conn.connectionState === MCPConnectionState.CONNECTED || conn.connectionState === MCPConnectionState.CONNECTING || conn.connectionState === MCPConnectionState.DISCOVERING;
+	}
+	oauthCallbackSuccess(serverId, conn) {
+		this.clearServerAuthUrl(serverId);
+		conn.connectionError = null;
+		return {
+			serverId,
+			authSuccess: true
+		};
+	}
+	async runWithCodeVerifierState(authProvider, state, callback) {
+		if (authProvider.runWithCodeVerifierState) return authProvider.runWithCodeVerifierState(state, callback);
+		return callback();
+	}
+	async consumeStaleOAuthState(serverId, authProvider, state) {
+		try {
+			const stateValidation = await authProvider.checkState(state);
+			if (!stateValidation.valid) {
+				console.warn(`[MCPClientManager] Ignoring stale OAuth callback with invalid state for server "${serverId}": ${stateValidation.error ?? "Invalid state"}`);
+				return;
+			}
+			await authProvider.consumeState(state);
+		} catch (cleanupError) {
+			console.warn(`[MCPClientManager] Failed to clean up stale OAuth callback state for server "${serverId}":`, cleanupError);
+		}
+	}
+	async completeAuthorizationAndCleanupVerifier(serverId, conn, authProvider, state, code) {
+		await this.runWithCodeVerifierState(authProvider, state, async () => {
+			let completeError;
+			let cleanupError;
+			try {
+				await conn.completeAuthorization(code, { alreadyAccepted: true });
+			} catch (error) {
+				completeError = error;
+			}
+			try {
+				await authProvider.deleteCodeVerifier();
+			} catch (deleteError) {
+				cleanupError = deleteError;
+			}
+			if (completeError) {
+				if (cleanupError) console.warn(`[MCPClientManager] Failed to clean up OAuth code verifier for server "${serverId}":`, cleanupError);
+				throw completeError;
+			}
+			if (cleanupError) throw cleanupError;
+		});
+	}
 	/**
 	* Create an auth provider for a server
 	* @internal
@@ -1213,7 +1262,19 @@ var MCPClientManager = class {
 		}
 		await this.mcpConnections[id].init();
 		if (options.reconnect?.oauthCode) try {
-			await this.mcpConnections[id].completeAuthorization(options.reconnect.oauthCode);
+			const authProvider = this.mcpConnections[id].options.transport.authProvider;
+			let completeError;
+			try {
+				await this.mcpConnections[id].completeAuthorization(options.reconnect.oauthCode);
+			} catch (error) {
+				completeError = error;
+			}
+			try {
+				await authProvider?.deleteCodeVerifier();
+			} catch (cleanupError) {
+				console.warn(`[MCPClientManager] Failed to clean up OAuth code verifier for server "${id}":`, cleanupError);
+			}
+			if (completeError) throw completeError;
 			await this.mcpConnections[id].init();
 		} catch (error) {
 			this._onObservabilityEvent.fire({
@@ -1404,11 +1465,13 @@ var MCPClientManager = class {
 		};
 		if (error) return {
 			serverId,
+			state,
 			valid: false,
 			error: errorDescription || error
 		};
 		if (!code) return {
 			serverId,
+			state,
 			valid: false,
 			error: "Unauthorized: no code provided"
 		};
@@ -1432,7 +1495,18 @@ var MCPClientManager = class {
 	async handleCallbackRequest(req) {
 		const validation = this.validateCallbackRequest(req);
 		if (!validation.valid) {
-			if (validation.serverId && this.mcpConnections[validation.serverId]) return this.failConnection(validation.serverId, validation.error);
+			const conn = validation.serverId ? this.mcpConnections[validation.serverId] : void 0;
+			if (validation.serverId && conn) {
+				if (this.isAuthAcceptedConnection(conn)) {
+					const authProvider = conn.options.transport.authProvider;
+					if (validation.state && authProvider) {
+						authProvider.serverId = validation.serverId;
+						await this.consumeStaleOAuthState(validation.serverId, authProvider, validation.state);
+					}
+					return this.oauthCallbackSuccess(validation.serverId, conn);
+				}
+				return this.failConnection(validation.serverId, validation.error);
+			}
 			return {
 				serverId: validation.serverId,
 				authSuccess: false,
@@ -1446,26 +1520,25 @@ var MCPClientManager = class {
 			const authProvider = conn.options.transport.authProvider;
 			authProvider.serverId = serverId;
 			const stateValidation = await authProvider.checkState(state);
-			if (!stateValidation.valid) throw new Error(stateValidation.error || "Invalid state");
-			if (conn.connectionState === MCPConnectionState.READY || conn.connectionState === MCPConnectionState.CONNECTED) {
-				this.clearServerAuthUrl(serverId);
-				return {
-					serverId,
-					authSuccess: true
-				};
+			if (!stateValidation.valid) {
+				if (this.isAuthAcceptedConnection(conn)) {
+					await this.consumeStaleOAuthState(serverId, authProvider, state);
+					return this.oauthCallbackSuccess(serverId, conn);
+				}
+				throw new Error(stateValidation.error || "Invalid state");
+			}
+			if (this.isAuthAcceptedConnection(conn)) {
+				await this.consumeStaleOAuthState(serverId, authProvider, state);
+				return this.oauthCallbackSuccess(serverId, conn);
 			}
 			if (conn.connectionState !== MCPConnectionState.AUTHENTICATING) throw new Error(`Failed to authenticate: the client is in "${conn.connectionState}" state, expected "authenticating"`);
+			conn.connectionState = MCPConnectionState.CONNECTING;
 			await authProvider.consumeState(state);
-			await conn.completeAuthorization(code);
+			await this.completeAuthorizationAndCleanupVerifier(serverId, conn, authProvider, state, code);
 			this.updateStoredSessionId(serverId, conn.sessionId);
-			await authProvider.deleteCodeVerifier();
-			this.clearServerAuthUrl(serverId);
-			conn.connectionError = null;
+			const result = this.oauthCallbackSuccess(serverId, conn);
 			this._onServerStateChanged.fire();
-			return {
-				serverId,
-				authSuccess: true
-			};
+			return result;
 		} catch (err) {
 			const message = err instanceof Error ? err.message : String(err);
 			return this.failConnection(serverId, message);
@@ -1756,4 +1829,4 @@ function getNamespacedData(mcpClients, type) {
 //#endregion
 export { MCPConnectionState as a, RPC_DO_PREFIX as c, normalizeServerId as i, DisposableStore as l, MCP_SERVER_ID_MAX_LENGTH as n, RPCClientTransport as o, getNamespacedData as r, RPCServerTransport as s, MCPClientManager as t };
 
-//# sourceMappingURL=client-NradHZZz.js.map
+//# sourceMappingURL=client-FUizKzj2.js.map