Class ThreadStateLibSQL

stores/libsql/src/storage/domains/thread-state/index.ts:22–139

 * payload (e.g. the task list for `type = 'task'`).
 */
export class ThreadStateLibSQL extends ThreadStateStorage {
  /**
   * `thread_state` grows as a side effect of thread activity (one row per
   * thread per state type). It anchors on `updatedAt` (last activity), so state
   * for a thread that is still being appended to is not pruned by creation age.
   */
  static readonly retentionTables: RetentionTablesDescriptor = {
    threadState: { table: TABLE_THREAD_STATE, column: 'updatedAt', indexed: true },
  };

  #db: LibSQLDB;
  #client: Client;

  constructor(config: LibSQLDomainConfig) {
    super();
    const client = resolveClient(config);
    this.#client = client;
    this.#db = new LibSQLDB({ client, maxRetries: config.maxRetries, initialBackoffMs: config.initialBackoffMs });
  }

  async init(): Promise<void> {
    await this.#db.createTable({
      tableName: TABLE_THREAD_STATE,
      schema: THREAD_STATE_SCHEMA,
      compositePrimaryKey: ['threadId', 'type'],
    });
  }

  /** Delete thread state older than the `threadState` policy's `maxAge`, batched. */
  async prune(policies: Record<string, TableRetentionPolicy>, options?: PruneOptions): Promise<PruneResult[]> {
    const targets = resolveTargets({
      policies,
      descriptor: ThreadStateLibSQL.retentionTables,
      order: ['threadState'],
    });
    return runPrune({ db: this.#db, domain: 'threadState', targets, options, logger: this.logger });
  }

  async dangerouslyClearAll(): Promise<void> {
    try {
      await this.#client.execute(`DELETE FROM "${TABLE_THREAD_STATE}"`);
    } catch (error) {
      throw new MastraError(
        {
          id: createStorageErrorId('LIBSQL', 'THREAD_STATE_CLEAR_ALL', 'FAILED'),
          domain: ErrorDomain.STORAGE,
          category: ErrorCategory.THIRD_PARTY,
        },
        error,
      );
    }
  }

  async getState<T = unknown>({ threadId, type }: { threadId: string; type: string }): Promise<T | undefined> {
    try {
      const result = await this.#client.execute({
        sql: `SELECT "value" FROM "${TABLE_THREAD_STATE}" WHERE "threadId" = ? AND "type" = ? LIMIT 1`,
        args: [threadId, type],