@dudousxd/nestjs-agent-store-drizzle 0.5.1 → 0.7.0

package/dist/index.cjs

@@ -107,7 +107,9 @@ var agentToolCall = (0, import_sqlite_core.sqliteTable)("agent_tool_call", {
   }).notNull(),
   executedAt: (0, import_sqlite_core.integer)("executed_at", {
     mode: "timestamp_ms"
-  })
+  }),
+  /** The run (turn) this call belongs to, for a trace deep-link; null for a pre-rollout row. */
+  runId: (0, import_sqlite_core.text)("run_id")
 });
 var agentTokenUsage = (0, import_sqlite_core.sqliteTable)("agent_token_usage", {
   id: (0, import_sqlite_core.text)("id").primaryKey(),
@@ -222,7 +224,8 @@ var statements = [
     execution_ms INTEGER,
     error TEXT,
     created_at INTEGER NOT NULL,
-    executed_at INTEGER
+    executed_at INTEGER,
+    run_id TEXT
   )`,
   `CREATE TABLE IF NOT EXISTS agent_token_usage (
     id TEXT PRIMARY KEY NOT NULL,
@@ -495,7 +498,8 @@ var DrizzleAgentStore = class {
       executionMs: null,
       error: null,
       createdAt: /* @__PURE__ */ new Date(),
-      executedAt: null
+      executedAt: null,
+      runId: input.runId ?? null
     });
   }
   async updateToolCall(input) {
@@ -588,6 +592,58 @@ var import_nestjs_agent_core = require("@dudousxd/nestjs-agent-core");
 var import_drizzle_orm3 = require("drizzle-orm");
 var UNCLASSIFIED_ERROR_CODE = "unknown";
 var DEFAULT_AGENT_BUCKET = "(default)";
+var TOOL_CALL_STATUSES = [
+  "auto_executed",
+  "pending_approval",
+  "executed",
+  "rejected",
+  "failed"
+];
+function isToolCallStatus(value) {
+  return TOOL_CALL_STATUSES.includes(value);
+}
+__name(isToolCallStatus, "isToolCallStatus");
+var TOOL_KINDS = [
+  "read",
+  "action",
+  "agent"
+];
+function isToolKind(value) {
+  return TOOL_KINDS.includes(value);
+}
+__name(isToolKind, "isToolKind");
+var RUN_STATUSES = [
+  "running",
+  "completed",
+  "failed"
+];
+function isRunStatus(value) {
+  return RUN_STATUSES.includes(value);
+}
+__name(isRunStatus, "isRunStatus");
+function dayStartUtc(day) {
+  return (0, import_nestjs_agent_core.dayBoundsUtc)({
+    fromDay: day,
+    toDay: day
+  }).start;
+}
+__name(dayStartUtc, "dayStartUtc");
+function dayEndUtc(day) {
+  return (0, import_nestjs_agent_core.dayBoundsUtc)({
+    fromDay: day,
+    toDay: day
+  }).end;
+}
+__name(dayEndUtc, "dayEndUtc");
+function emptyPage(query) {
+  return {
+    rows: [],
+    total: 0,
+    page: query.page,
+    pageSize: query.pageSize
+  };
+}
+__name(emptyPage, "emptyPage");
 function toUsageInput(row) {
   return {
     modelId: row.modelId,
@@ -684,7 +740,8 @@ var DrizzleGovernanceQueries = class {
       toolType: agentToolCall.toolType,
       status: agentToolCall.status,
       threadId: agentMessage.threadId,
-      createdAt: agentToolCall.createdAt
+      createdAt: agentToolCall.createdAt,
+      runId: agentToolCall.runId
     }).from(agentToolCall).innerJoin(agentMessage, (0, import_drizzle_orm3.eq)(agentToolCall.messageId, agentMessage.id)).orderBy((0, import_drizzle_orm3.desc)(agentToolCall.createdAt), (0, import_drizzle_orm3.desc)(agentToolCall.id)).limit(limit);
     return rows.map((row) => ({
       toolCallId: row.toolCallId,
@@ -692,7 +749,8 @@ var DrizzleGovernanceQueries = class {
       toolType: row.toolType,
       status: row.status,
       threadId: row.threadId,
-      createdAt: row.createdAt.toISOString()
+      createdAt: row.createdAt.toISOString(),
+      runId: row.runId
     }));
   }
   async recentThreads(limit) {
@@ -843,7 +901,8 @@ var DrizzleGovernanceQueries = class {
       threadTitle: agentThread.title,
       actorRef: agentThread.actorRef,
       agentName: agentMessage.agentName,
-      requestedAt: agentToolCall.createdAt
+      requestedAt: agentToolCall.createdAt,
+      runId: agentToolCall.runId
     }).from(agentToolCall).innerJoin(agentMessage, (0, import_drizzle_orm3.eq)(agentToolCall.messageId, agentMessage.id)).innerJoin(agentThread, (0, import_drizzle_orm3.eq)(agentMessage.threadId, agentThread.id)).where((0, import_drizzle_orm3.eq)(agentToolCall.status, "pending_approval")).orderBy(agentToolCall.createdAt, agentToolCall.id).limit(limit);
     return rows.map((row) => ({
       toolCallId: row.toolCallId,
@@ -853,7 +912,8 @@ var DrizzleGovernanceQueries = class {
       threadTitle: row.threadTitle,
       actorRef: row.actorRef,
       agentName: row.agentName ?? null,
-      requestedAt: row.requestedAt.toISOString()
+      requestedAt: row.requestedAt.toISOString(),
+      runId: row.runId
     }));
   }
   /**
@@ -898,6 +958,117 @@ var DrizzleGovernanceQueries = class {
     result.sort((left, right) => right.calls - left.calls || left.toolName.localeCompare(right.toolName));
     return result;
   }
+  /**
+  * Paged, filterable tool-call activity, newest-first (same ordering as {@link recentToolCalls}).
+  * `toolType`/`status` values outside the known literals can never match a row — short-circuits to
+  * an empty page rather than issuing a query. Joins `agent_message` unconditionally (like
+  * `recentToolCalls`) to resolve `threadId` and to support the `threadId` filter.
+  */
+  async toolCallsPage(query) {
+    const filters = query.where;
+    if (filters?.status !== void 0 && !isToolCallStatus(filters.status)) {
+      return emptyPage(query);
+    }
+    if (filters?.toolType !== void 0 && !isToolKind(filters.toolType)) {
+      return emptyPage(query);
+    }
+    const whereClause = (0, import_drizzle_orm3.and)(filters?.toolName !== void 0 ? (0, import_drizzle_orm3.eq)(agentToolCall.toolName, filters.toolName) : void 0, filters?.toolType !== void 0 && isToolKind(filters.toolType) ? (0, import_drizzle_orm3.eq)(agentToolCall.toolType, filters.toolType) : void 0, filters?.status !== void 0 && isToolCallStatus(filters.status) ? (0, import_drizzle_orm3.eq)(agentToolCall.status, filters.status) : void 0, filters?.threadId !== void 0 ? (0, import_drizzle_orm3.eq)(agentMessage.threadId, filters.threadId) : void 0, filters?.fromDay !== void 0 ? (0, import_drizzle_orm3.gte)(agentToolCall.createdAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? (0, import_drizzle_orm3.lte)(agentToolCall.createdAt, dayEndUtc(filters.toDay)) : void 0);
+    const [totalRow] = await this.db.select({
+      value: (0, import_drizzle_orm3.count)()
+    }).from(agentToolCall).innerJoin(agentMessage, (0, import_drizzle_orm3.eq)(agentToolCall.messageId, agentMessage.id)).where(whereClause);
+    const rows = await this.db.select({
+      toolCallId: agentToolCall.id,
+      toolName: agentToolCall.toolName,
+      toolType: agentToolCall.toolType,
+      status: agentToolCall.status,
+      threadId: agentMessage.threadId,
+      createdAt: agentToolCall.createdAt,
+      runId: agentToolCall.runId
+    }).from(agentToolCall).innerJoin(agentMessage, (0, import_drizzle_orm3.eq)(agentToolCall.messageId, agentMessage.id)).where(whereClause).orderBy((0, import_drizzle_orm3.desc)(agentToolCall.createdAt), (0, import_drizzle_orm3.desc)(agentToolCall.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
+    return {
+      rows: rows.map((row) => ({
+        toolCallId: row.toolCallId,
+        toolName: row.toolName,
+        toolType: row.toolType,
+        status: row.status,
+        threadId: row.threadId,
+        createdAt: row.createdAt.toISOString(),
+        runId: row.runId
+      })),
+      total: totalRow?.value ?? 0,
+      page: query.page,
+      pageSize: query.pageSize
+    };
+  }
+  /**
+  * Paged, filterable thread activity, newest-first (same ordering as {@link recentThreads}).
+  * `title` is a case-insensitive substring match via `lower(...)` on both sides (portable across
+  * dialects, regardless of the underlying column collation). Soft-deleted threads are excluded,
+  * mirroring `recentThreads`.
+  */
+  async threadsPage(query) {
+    const filters = query.where;
+    const whereClause = (0, import_drizzle_orm3.and)((0, import_drizzle_orm3.isNull)(agentThread.deletedAt), filters?.actorRef !== void 0 ? (0, import_drizzle_orm3.eq)(agentThread.actorRef, filters.actorRef) : void 0, filters?.title !== void 0 ? (0, import_drizzle_orm3.like)(import_drizzle_orm3.sql`lower(${agentThread.title})`, `%${filters.title.toLowerCase()}%`) : void 0, filters?.fromDay !== void 0 ? (0, import_drizzle_orm3.gte)(agentThread.updatedAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? (0, import_drizzle_orm3.lte)(agentThread.updatedAt, dayEndUtc(filters.toDay)) : void 0);
+    const [totalRow] = await this.db.select({
+      value: (0, import_drizzle_orm3.count)()
+    }).from(agentThread).where(whereClause);
+    const threads = await this.db.select().from(agentThread).where(whereClause).orderBy((0, import_drizzle_orm3.desc)(agentThread.updatedAt), (0, import_drizzle_orm3.desc)(agentThread.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
+    const rows = [];
+    for (const thread of threads) {
+      const [messageCountRow] = await this.db.select({
+        value: (0, import_drizzle_orm3.count)()
+      }).from(agentMessage).where((0, import_drizzle_orm3.eq)(agentMessage.threadId, thread.id));
+      const usageRows = await this.db.select().from(agentTokenUsage).where((0, import_drizzle_orm3.eq)(agentTokenUsage.threadId, thread.id));
+      const totalTokens = usageRows.reduce((sum, row) => sum + row.inputTokens + row.outputTokens, 0);
+      rows.push({
+        threadId: thread.id,
+        title: thread.title,
+        actorRef: thread.actorRef,
+        messageCount: messageCountRow?.value ?? 0,
+        totalTokens,
+        lastActivityAt: thread.updatedAt.toISOString()
+      });
+    }
+    return {
+      rows,
+      total: totalRow?.value ?? 0,
+      page: query.page,
+      pageSize: query.pageSize
+    };
+  }
+  /**
+  * Paged, filterable run activity, newest-first (same ordering as {@link recentRuns}). `status`
+  * values outside the known literals can never match a row — short-circuits to an empty page.
+  */
+  async runsPage(query) {
+    const filters = query.where;
+    if (filters?.status !== void 0 && !isRunStatus(filters.status)) {
+      return emptyPage(query);
+    }
+    const whereClause = (0, import_drizzle_orm3.and)(filters?.agentName !== void 0 ? (0, import_drizzle_orm3.eq)(agentRun.agentName, filters.agentName) : void 0, filters?.status !== void 0 && isRunStatus(filters.status) ? (0, import_drizzle_orm3.eq)(agentRun.status, filters.status) : void 0, filters?.errorCode !== void 0 ? (0, import_drizzle_orm3.eq)(agentRun.errorCode, filters.errorCode) : void 0, filters?.threadId !== void 0 ? (0, import_drizzle_orm3.eq)(agentRun.threadId, filters.threadId) : void 0, filters?.fromDay !== void 0 ? (0, import_drizzle_orm3.gte)(agentRun.startedAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? (0, import_drizzle_orm3.lte)(agentRun.startedAt, dayEndUtc(filters.toDay)) : void 0);
+    const [totalRow] = await this.db.select({
+      value: (0, import_drizzle_orm3.count)()
+    }).from(agentRun).where(whereClause);
+    const runs = await this.db.select().from(agentRun).where(whereClause).orderBy((0, import_drizzle_orm3.desc)(agentRun.startedAt), (0, import_drizzle_orm3.desc)(agentRun.id)).limit(query.pageSize).offset((query.page - 1) * query.pageSize);
+    return {
+      rows: runs.map((run) => ({
+        runId: run.id,
+        threadId: run.threadId,
+        actorRef: run.actorRef,
+        agentName: run.agentName ?? null,
+        status: run.status,
+        durationMs: run.durationMs ?? null,
+        errorCode: run.errorCode ?? null,
+        errorMessage: run.errorMessage ?? null,
+        retries: run.retries,
+        startedAt: run.startedAt.toISOString(),
+        promptHash: run.promptHash ?? null
+      })),
+      total: totalRow?.value ?? 0,
+      page: query.page,
+      pageSize: query.pageSize
+    };
+  }
 };
 
 // src/drizzle-pricing-store.ts