@dudousxd/nestjs-agent-store-drizzle 0.15.0 → 0.17.0

package/README.md

@@ -120,6 +120,7 @@ ALTER TABLE agent_message ADD COLUMN reasoning TEXT;
 ALTER TABLE agent_message ADD COLUMN reasoning_ms INTEGER;
 ALTER TABLE agent_message ADD COLUMN ui TEXT;
 ALTER TABLE agent_message ADD COLUMN feedback TEXT;
+ALTER TABLE agent_thread ADD COLUMN model TEXT;
 ALTER TABLE agent_tool_call ADD COLUMN approver TEXT;
 ALTER TABLE agent_tool_call ADD COLUMN expires_at INTEGER;
 ALTER TABLE agent_tool_call ADD COLUMN remember INTEGER;

package/dist/index.cjs

@@ -54,6 +54,8 @@ var agentThread = (0, import_sqlite_core.sqliteTable)("agent_thread", {
   activeStreamId: (0, import_sqlite_core.text)("active_stream_id"),
   /** The agent a new turn on this thread defaults to when the caller names none. */
   defaultAgent: (0, import_sqlite_core.text)("default_agent"),
+  /** The model every turn on this thread runs on unless the send names one; `null` → default. */
+  model: (0, import_sqlite_core.text)("model"),
   createdAt: (0, import_sqlite_core.integer)("created_at", {
     mode: "timestamp_ms"
   }).notNull(),
@@ -291,6 +293,7 @@ var statements = [
     transient INTEGER NOT NULL DEFAULT 0,
     active_stream_id TEXT,
     default_agent TEXT,
+    model TEXT,
     created_at INTEGER NOT NULL,
     updated_at INTEGER NOT NULL,
     deleted_at INTEGER
@@ -457,6 +460,11 @@ var additiveColumns = [
     column: "feedback",
     ddl: "ALTER TABLE agent_message ADD COLUMN feedback TEXT"
   },
+  {
+    table: "agent_thread",
+    column: "model",
+    ddl: "ALTER TABLE agent_thread ADD COLUMN model TEXT"
+  },
   {
     table: "agent_tool_call",
     column: "approver",
@@ -522,6 +530,7 @@ var DrizzleAgentStore = class {
       transient: input.transient ?? false,
       activeStreamId: null,
       defaultAgent: null,
+      model: null,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
@@ -627,6 +636,7 @@ var DrizzleAgentStore = class {
       transient: false,
       activeStreamId: null,
       defaultAgent: source.defaultAgent,
+      model: source.model,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
@@ -775,6 +785,9 @@ var DrizzleAgentStore = class {
     if (patch.defaultAgent !== void 0) {
       updates.defaultAgent = patch.defaultAgent;
     }
+    if (patch.model !== void 0) {
+      updates.model = patch.model;
+    }
     if (Object.keys(updates).length === 0) {
       return;
     }
@@ -786,6 +799,13 @@ var DrizzleAgentStore = class {
   * agent answers the next turn, and {@link getThread} would materialize the whole transcript to
   * hand it over. `null` when the thread is unknown, soft-deleted, or has no default set.
   */
+  /** The thread's pinned model, projected like {@link defaultAgentForThread}. */
+  async modelForThread(threadId) {
+    const [thread] = await this.db.select({
+      model: agentThread.model
+    }).from(agentThread).where((0, import_drizzle_orm2.and)((0, import_drizzle_orm2.eq)(agentThread.id, threadId), (0, import_drizzle_orm2.isNull)(agentThread.deletedAt)));
+    return thread?.model ?? null;
+  }
   async defaultAgentForThread(threadId) {
     const [thread] = await this.db.select({
       defaultAgent: agentThread.defaultAgent
@@ -998,8 +1018,11 @@ var DrizzleAgentStore = class {
     });
   }
   async quotaToday(actorRef, day) {
-    const start = /* @__PURE__ */ new Date(`${day}T00:00:00.000Z`);
-    const end = /* @__PURE__ */ new Date(`${day}T23:59:59.999Z`);
+    return this.usageBetween(actorRef, day, day);
+  }
+  async usageBetween(actorRef, fromDay, toDay) {
+    const start = /* @__PURE__ */ new Date(`${fromDay}T00:00:00.000Z`);
+    const end = /* @__PURE__ */ new Date(`${toDay}T23:59:59.999Z`);
     const rows = await this.db.select().from(agentTokenUsage).where((0, import_drizzle_orm2.and)((0, import_drizzle_orm2.eq)(agentTokenUsage.actorRef, actorRef), (0, import_drizzle_orm2.gte)(agentTokenUsage.createdAt, start), (0, import_drizzle_orm2.lte)(agentTokenUsage.createdAt, end)));
     const usedTokens = rows.reduce((sum2, row) => sum2 + row.inputTokens + row.outputTokens, 0);
     const costUsd = rows.reduce((sum2, row) => sum2 + (row.costUsd ?? 0), 0);
@@ -1016,6 +1039,9 @@ var DrizzleAgentStore = class {
       createdAt: thread.createdAt.toISOString(),
       updatedAt: thread.updatedAt.toISOString(),
       defaultAgent: thread.defaultAgent,
+      ...thread.model != null ? {
+        model: thread.model
+      } : {},
       ...lastContent !== void 0 ? {
         lastMessagePreview: lastContent.slice(0, 120)
       } : {}