@dudousxd/nestjs-agent-store-drizzle 0.27.0 → 0.28.0

package/dist/index.cjs

@@ -24,6 +24,7 @@ __export(index_exports, {
   DrizzleActionProposalStore: () => DrizzleActionProposalStore,
   DrizzleAgentStore: () => DrizzleAgentStore,
   DrizzleAgentStoreModule: () => DrizzleAgentStoreModule,
+  DrizzleChannelStore: () => DrizzleChannelStore,
   DrizzleConfirmTokenStore: () => DrizzleConfirmTokenStore,
   DrizzleGovernanceQueries: () => DrizzleGovernanceQueries,
   DrizzleMemoryProvider: () => DrizzleMemoryProvider,
@@ -33,6 +34,7 @@ __export(index_exports, {
   DrizzleTokenStreamSink: () => DrizzleTokenStreamSink,
   RAG_INGESTION_LOG_PAGE_ORDER: () => RAG_INGESTION_LOG_PAGE_ORDER,
   agentActionProposal: () => agentActionProposal,
+  agentChannelState: () => agentChannelState,
   agentConfirmToken: () => agentConfirmToken,
   agentDialectOf: () => agentDialectOf,
   agentMemory: () => agentMemory,
@@ -342,6 +344,14 @@ var ragIngestionLog = (0, import_sqlite_core.sqliteTable)("rag_ingestion_log", {
 }, (table) => [
   (0, import_sqlite_core.index)("rag_ingestion_log_collection_idx").on(table.collection, table.updatedAt)
 ]);
+var agentChannelState = (0, import_sqlite_core.sqliteTable)("agent_channel_state", {
+  key: (0, import_sqlite_core.text)("key").primaryKey(),
+  value: (0, import_sqlite_core.text)("value"),
+  expiresAt: (0, import_sqlite_core.integer)("expires_at").notNull(),
+  createdAt: (0, import_sqlite_core.integer)("created_at").notNull()
+}, (table) => [
+  (0, import_sqlite_core.index)("agent_channel_state_expires_idx").on(table.expiresAt)
+]);
 var agentConfirmToken = (0, import_sqlite_core.sqliteTable)("agent_confirm_token", {
   hash: (0, import_sqlite_core.text)("hash").primaryKey(),
   actorRef: (0, import_sqlite_core.text)("actor_ref").notNull(),
@@ -404,6 +414,7 @@ var agentSchema = {
   agentMemory,
   ragIngestionLog,
   agentConfirmToken,
+  agentChannelState,
   agentStreamFrame
 };
 
@@ -600,6 +611,18 @@ var ragIngestionLog2 = (0, import_mysql_core.mysqlTable)("rag_ingestion_log", {
 }, (table) => [
   (0, import_mysql_core.index)("rag_ingestion_log_collection_idx").on(table.collection, table.updatedAt)
 ]);
+var agentChannelState2 = (0, import_mysql_core.mysqlTable)("agent_channel_state", {
+  key: key("key").primaryKey(),
+  value: (0, import_mysql_core.longtext)("value"),
+  expiresAt: (0, import_mysql_core.bigint)("expires_at", {
+    mode: "number"
+  }).notNull(),
+  createdAt: (0, import_mysql_core.bigint)("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  (0, import_mysql_core.index)("agent_channel_state_expires_idx").on(table.expiresAt)
+]);
 var agentConfirmToken2 = (0, import_mysql_core.mysqlTable)("agent_confirm_token", {
   hash: key("hash").primaryKey(),
   actorRef: key("actor_ref").notNull(),
@@ -685,6 +708,7 @@ var mysqlAgentSchema = {
   agentMemory: agentMemory2,
   ragIngestionLog: ragIngestionLog2,
   agentConfirmToken: agentConfirmToken2,
+  agentChannelState: agentChannelState2,
   agentStreamFrame: agentStreamFrame2
 };
 
@@ -868,6 +892,18 @@ var ragIngestionLog3 = (0, import_pg_core.pgTable)("rag_ingestion_log", {
 }, (table) => [
   (0, import_pg_core.index)("rag_ingestion_log_collection_idx").on(table.collection, table.updatedAt)
 ]);
+var agentChannelState3 = (0, import_pg_core.pgTable)("agent_channel_state", {
+  key: (0, import_pg_core.text)("key").primaryKey(),
+  value: (0, import_pg_core.text)("value"),
+  expiresAt: (0, import_pg_core.bigint)("expires_at", {
+    mode: "number"
+  }).notNull(),
+  createdAt: (0, import_pg_core.bigint)("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  (0, import_pg_core.index)("agent_channel_state_expires_idx").on(table.expiresAt)
+]);
 var agentConfirmToken3 = (0, import_pg_core.pgTable)("agent_confirm_token", {
   hash: (0, import_pg_core.text)("hash").primaryKey(),
   actorRef: (0, import_pg_core.text)("actor_ref").notNull(),
@@ -951,6 +987,7 @@ var pgAgentSchema = {
   agentMemory: agentMemory3,
   ragIngestionLog: ragIngestionLog3,
   agentConfirmToken: agentConfirmToken3,
+  agentChannelState: agentChannelState3,
   agentStreamFrame: agentStreamFrame3
 };
 
@@ -3583,10 +3620,82 @@ var DrizzleConfirmTokenStore = class {
   }
 };
 
+// src/drizzle-channel-store.ts
+var import_node_crypto2 = require("crypto");
+var import_drizzle_orm10 = require("drizzle-orm");
+var channelStateKey = /* @__PURE__ */ __name((key2) => key2.length <= 255 ? key2 : `sha256:${(0, import_node_crypto2.createHash)("sha256").update(key2).digest("hex")}`, "channelStateKey");
+var DrizzleChannelStore = class {
+  static {
+    __name(this, "DrizzleChannelStore");
+  }
+  db;
+  dialect;
+  t;
+  purgeEveryMs;
+  lastPurge = 0;
+  constructor(db, options = {}) {
+    this.dialect = agentDialectOf(db);
+    this.t = agentTablesFor(this.dialect);
+    this.db = asBuilder(db);
+    this.purgeEveryMs = options.purgeEveryMs ?? 10 * 60 * 1e3;
+  }
+  async claim(key2, ttlMs) {
+    const table = this.t.agentChannelState;
+    const now = Date.now();
+    if (this.purgeEveryMs > 0 && now - this.lastPurge >= this.purgeEveryMs) {
+      this.lastPurge = now;
+      await this.purgeExpired(now);
+    }
+    const id = channelStateKey(key2);
+    await this.db.delete(table).where((0, import_drizzle_orm10.and)((0, import_drizzle_orm10.eq)(table.key, id), (0, import_drizzle_orm10.lte)(table.expiresAt, now)));
+    const row = {
+      key: id,
+      value: null,
+      expiresAt: now + ttlMs,
+      createdAt: now
+    };
+    if (this.dialect === "mysql") {
+      const insert = this.db.insert(table);
+      return mysqlAffectedRows(await insert.ignore().values(row)) === 1;
+    }
+    const inserted = await affectedRows(this.dialect, this.db.insert(table).values(row).onConflictDoNothing(), table.key);
+    return inserted === 1;
+  }
+  async get(key2) {
+    const table = this.t.agentChannelState;
+    const rows = await this.db.select({
+      value: table.value
+    }).from(table).where((0, import_drizzle_orm10.and)((0, import_drizzle_orm10.eq)(table.key, channelStateKey(key2)), (0, import_drizzle_orm10.gt)(table.expiresAt, Date.now())));
+    const value = rows[0]?.value;
+    return typeof value === "string" ? value : null;
+  }
+  async set(key2, value, ttlMs) {
+    const table = this.t.agentChannelState;
+    const now = Date.now();
+    await upsert(this.dialect, this.db.insert(table).values({
+      key: channelStateKey(key2),
+      value,
+      expiresAt: now + ttlMs,
+      createdAt: now
+    }), table.key, {
+      value,
+      expiresAt: now + ttlMs
+    });
+  }
+  async delete(key2) {
+    const table = this.t.agentChannelState;
+    await this.db.delete(table).where((0, import_drizzle_orm10.eq)(table.key, channelStateKey(key2)));
+  }
+  async purgeExpired(now = Date.now()) {
+    const table = this.t.agentChannelState;
+    return affectedRows(this.dialect, this.db.delete(table).where((0, import_drizzle_orm10.lte)(table.expiresAt, now)), table.key);
+  }
+};
+
 // src/drizzle-rag-ingestion-log.ts
 var import_node_diagnostics_channel = require("diagnostics_channel");
 var import_common = require("@nestjs/common");
-var import_drizzle_orm10 = require("drizzle-orm");
+var import_drizzle_orm11 = require("drizzle-orm");
 function _ts_decorate(decorators, target, key2, desc4) {
   var c = arguments.length, r = c < 3 ? target : desc4 === null ? desc4 = Object.getOwnPropertyDescriptor(target, key2) : desc4, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
@@ -3634,8 +3743,8 @@ function num(value) {
 }
 __name(num, "num");
 var RAG_INGESTION_LOG_PAGE_ORDER = [
-  (0, import_drizzle_orm10.desc)(ragIngestionLog.updatedAt),
-  (0, import_drizzle_orm10.asc)(ragIngestionLog.documentId)
+  (0, import_drizzle_orm11.desc)(ragIngestionLog.updatedAt),
+  (0, import_drizzle_orm11.asc)(ragIngestionLog.documentId)
 ];
 var DEFAULT_PAGE_SIZE = 200;
 var DEFAULT_BATCH_SIZE = 200;
@@ -3706,7 +3815,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
   */
   async listPage(query = {}) {
     const [totalRow] = await this.db.select({
-      value: (0, import_drizzle_orm10.count)()
+      value: (0, import_drizzle_orm11.count)()
     }).from(this.t.ragIngestionLog).where(this.where(query));
     return {
       rows: await this.list(query),
@@ -3785,17 +3894,17 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
   }
   /** The latest recorded outcome for one document, or null if it was never attempted. */
   async get(documentId) {
-    const [row] = await this.db.select().from(this.t.ragIngestionLog).where((0, import_drizzle_orm10.eq)(this.t.ragIngestionLog.documentId, documentId));
+    const [row] = await this.db.select().from(this.t.ragIngestionLog).where((0, import_drizzle_orm11.eq)(this.t.ragIngestionLog.documentId, documentId));
     return row ?? null;
   }
   /** Forget one document's record. Returns whether a row was actually removed. */
   async remove(documentId) {
-    const removed = await affectedRows(this.dialect, this.db.delete(this.t.ragIngestionLog).where((0, import_drizzle_orm10.eq)(this.t.ragIngestionLog.documentId, documentId)), this.t.ragIngestionLog.documentId);
+    const removed = await affectedRows(this.dialect, this.db.delete(this.t.ragIngestionLog).where((0, import_drizzle_orm11.eq)(this.t.ragIngestionLog.documentId, documentId)), this.t.ragIngestionLog.documentId);
     return removed > 0;
   }
   /** Forget every record for a collection — for when the collection itself is deleted. */
   async removeByCollection(collection) {
-    const removed = await affectedRows(this.dialect, this.db.delete(this.t.ragIngestionLog).where((0, import_drizzle_orm10.eq)(this.t.ragIngestionLog.collection, collection)), this.t.ragIngestionLog.documentId);
+    const removed = await affectedRows(this.dialect, this.db.delete(this.t.ragIngestionLog).where((0, import_drizzle_orm11.eq)(this.t.ragIngestionLog.collection, collection)), this.t.ragIngestionLog.documentId);
     return removed;
   }
   /**
@@ -3825,7 +3934,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
     }
   }
   where(query) {
-    return (0, import_drizzle_orm10.and)(query.collection !== void 0 ? (0, import_drizzle_orm10.eq)(this.t.ragIngestionLog.collection, query.collection) : void 0, query.status !== void 0 ? (0, import_drizzle_orm10.eq)(this.t.ragIngestionLog.status, query.status) : void 0);
+    return (0, import_drizzle_orm11.and)(query.collection !== void 0 ? (0, import_drizzle_orm11.eq)(this.t.ragIngestionLog.collection, query.collection) : void 0, query.status !== void 0 ? (0, import_drizzle_orm11.eq)(this.t.ragIngestionLog.status, query.status) : void 0);
   }
   /**
   * The coordinates filter AND'd with "strictly after `cursor` in
@@ -3840,7 +3949,7 @@ var DrizzleRagIngestionLog = class _DrizzleRagIngestionLog {
     if (cursor === void 0) {
       return base;
     }
-    return (0, import_drizzle_orm10.and)(base, (0, import_drizzle_orm10.or)((0, import_drizzle_orm10.lt)(this.t.ragIngestionLog.updatedAt, cursor.updatedAt), (0, import_drizzle_orm10.and)((0, import_drizzle_orm10.eq)(this.t.ragIngestionLog.updatedAt, cursor.updatedAt), (0, import_drizzle_orm10.gt)(this.t.ragIngestionLog.documentId, cursor.documentId))));
+    return (0, import_drizzle_orm11.and)(base, (0, import_drizzle_orm11.or)((0, import_drizzle_orm11.lt)(this.t.ragIngestionLog.updatedAt, cursor.updatedAt), (0, import_drizzle_orm11.and)((0, import_drizzle_orm11.eq)(this.t.ragIngestionLog.updatedAt, cursor.updatedAt), (0, import_drizzle_orm11.gt)(this.t.ragIngestionLog.documentId, cursor.documentId))));
   }
   /**
   * Upsert one outcome. Keyed by document id so the row always reflects the *current* state — a
@@ -3992,6 +4101,15 @@ var DrizzleAgentStoreModule = class _DrizzleAgentStoreModule {
         {
           provide: import_nestjs_agent_core8.AGENT_CONFIRM_TOKEN_STORE,
           useExisting: DrizzleConfirmTokenStore
+        },
+        // Text channels' state (`@dudousxd/nestjs-agent-channels`), shared by every replica.
+        {
+          provide: DrizzleChannelStore,
+          useFactory: /* @__PURE__ */ __name(() => new DrizzleChannelStore(options.db), "useFactory")
+        },
+        {
+          provide: import_nestjs_agent_core8.AGENT_CHANNEL_STORE,
+          useExisting: DrizzleChannelStore
         }
       ],
       exports: [
@@ -4006,7 +4124,9 @@ var DrizzleAgentStoreModule = class _DrizzleAgentStoreModule {
         DrizzlePricingStore,
         import_nestjs_agent_core8.AGENT_PRICING_STORE,
         DrizzleConfirmTokenStore,
-        import_nestjs_agent_core8.AGENT_CONFIRM_TOKEN_STORE
+        import_nestjs_agent_core8.AGENT_CONFIRM_TOKEN_STORE,
+        DrizzleChannelStore,
+        import_nestjs_agent_core8.AGENT_CHANNEL_STORE
       ]
     };
   }
@@ -4019,6 +4139,7 @@ DrizzleAgentStoreModule = _ts_decorate2([
   DrizzleActionProposalStore,
   DrizzleAgentStore,
   DrizzleAgentStoreModule,
+  DrizzleChannelStore,
   DrizzleConfirmTokenStore,
   DrizzleGovernanceQueries,
   DrizzleMemoryProvider,
@@ -4028,6 +4149,7 @@ DrizzleAgentStoreModule = _ts_decorate2([
   DrizzleTokenStreamSink,
   RAG_INGESTION_LOG_PAGE_ORDER,
   agentActionProposal,
+  agentChannelState,
   agentConfirmToken,
   agentDialectOf,
   agentMemory,