@dudousxd/nestjs-agent-store-drizzle 0.5.0 → 0.6.0

package/dist/index.d.cts

@@ -1,6 +1,6 @@
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
 import { BaseSQLiteDatabase } from 'drizzle-orm/sqlite-core';
-import { MessageRole, ToolCallRequest, ToolResult, MessageUsage, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, CreateThreadInput, ThreadSummary, ThreadDetail, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
+import { MessageRole, ToolCallRequest, ToolResult, MessageUsage, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, CreateThreadInput, ThreadSummary, ThreadDetail, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
 import { Type, DynamicModule } from '@nestjs/common';
 
 /**
@@ -2531,6 +2531,25 @@ declare class DrizzleGovernanceQueries implements AgentGovernanceQueries {
      * their final status); `null` when none did.
      */
     toolStats(range: GovernanceRange): Promise<ToolStatRow[]>;
+    /**
+     * 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.
+     */
+    toolCallsPage(query: GovernancePageQuery<ToolCallWhere>): Promise<GovernancePage<ToolCallActivityRow>>;
+    /**
+     * 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`.
+     */
+    threadsPage(query: GovernancePageQuery<ThreadWhere>): Promise<GovernancePage<ThreadActivityRow>>;
+    /**
+     * 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.
+     */
+    runsPage(query: GovernancePageQuery<RunWhere>): Promise<GovernancePage<RecentRunRow>>;
 }
 
 /**

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
 import { BaseSQLiteDatabase } from 'drizzle-orm/sqlite-core';
-import { MessageRole, ToolCallRequest, ToolResult, MessageUsage, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, CreateThreadInput, ThreadSummary, ThreadDetail, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
+import { MessageRole, ToolCallRequest, ToolResult, MessageUsage, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, CreateThreadInput, ThreadSummary, ThreadDetail, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
 import { Type, DynamicModule } from '@nestjs/common';
 
 /**
@@ -2531,6 +2531,25 @@ declare class DrizzleGovernanceQueries implements AgentGovernanceQueries {
      * their final status); `null` when none did.
      */
     toolStats(range: GovernanceRange): Promise<ToolStatRow[]>;
+    /**
+     * 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.
+     */
+    toolCallsPage(query: GovernancePageQuery<ToolCallWhere>): Promise<GovernancePage<ToolCallActivityRow>>;
+    /**
+     * 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`.
+     */
+    threadsPage(query: GovernancePageQuery<ThreadWhere>): Promise<GovernancePage<ThreadActivityRow>>;
+    /**
+     * 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.
+     */
+    runsPage(query: GovernancePageQuery<RunWhere>): Promise<GovernancePage<RecentRunRow>>;
 }
 
 /**

package/dist/index.js

@@ -550,9 +550,61 @@ var DrizzleAgentStore = class {
 
 // src/drizzle-governance-queries.ts
 import { bucketByActor, bucketByModel, bucketByThread, bucketUsageTrend, dayBoundsUtc } from "@dudousxd/nestjs-agent-core";
-import { and as and2, count, desc as desc2, eq as eq2, gte as gte2, inArray as inArray2, isNull as isNull2, lte as lte2 } from "drizzle-orm";
+import { and as and2, count, desc as desc2, eq as eq2, gte as gte2, inArray as inArray2, isNull as isNull2, like, lte as lte2, sql as sql3 } from "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 dayBoundsUtc({
+    fromDay: day,
+    toDay: day
+  }).start;
+}
+__name(dayStartUtc, "dayStartUtc");
+function dayEndUtc(day) {
+  return 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,
@@ -863,6 +915,115 @@ 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 = and2(filters?.toolName !== void 0 ? eq2(agentToolCall.toolName, filters.toolName) : void 0, filters?.toolType !== void 0 && isToolKind(filters.toolType) ? eq2(agentToolCall.toolType, filters.toolType) : void 0, filters?.status !== void 0 && isToolCallStatus(filters.status) ? eq2(agentToolCall.status, filters.status) : void 0, filters?.threadId !== void 0 ? eq2(agentMessage.threadId, filters.threadId) : void 0, filters?.fromDay !== void 0 ? gte2(agentToolCall.createdAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(agentToolCall.createdAt, dayEndUtc(filters.toDay)) : void 0);
+    const [totalRow] = await this.db.select({
+      value: count()
+    }).from(agentToolCall).innerJoin(agentMessage, eq2(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
+    }).from(agentToolCall).innerJoin(agentMessage, eq2(agentToolCall.messageId, agentMessage.id)).where(whereClause).orderBy(desc2(agentToolCall.createdAt), desc2(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()
+      })),
+      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 = and2(isNull2(agentThread.deletedAt), filters?.actorRef !== void 0 ? eq2(agentThread.actorRef, filters.actorRef) : void 0, filters?.title !== void 0 ? like(sql3`lower(${agentThread.title})`, `%${filters.title.toLowerCase()}%`) : void 0, filters?.fromDay !== void 0 ? gte2(agentThread.updatedAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(agentThread.updatedAt, dayEndUtc(filters.toDay)) : void 0);
+    const [totalRow] = await this.db.select({
+      value: count()
+    }).from(agentThread).where(whereClause);
+    const threads = await this.db.select().from(agentThread).where(whereClause).orderBy(desc2(agentThread.updatedAt), desc2(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: count()
+      }).from(agentMessage).where(eq2(agentMessage.threadId, thread.id));
+      const usageRows = await this.db.select().from(agentTokenUsage).where(eq2(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 = and2(filters?.agentName !== void 0 ? eq2(agentRun.agentName, filters.agentName) : void 0, filters?.status !== void 0 && isRunStatus(filters.status) ? eq2(agentRun.status, filters.status) : void 0, filters?.errorCode !== void 0 ? eq2(agentRun.errorCode, filters.errorCode) : void 0, filters?.fromDay !== void 0 ? gte2(agentRun.startedAt, dayStartUtc(filters.fromDay)) : void 0, filters?.toDay !== void 0 ? lte2(agentRun.startedAt, dayEndUtc(filters.toDay)) : void 0);
+    const [totalRow] = await this.db.select({
+      value: count()
+    }).from(agentRun).where(whereClause);
+    const runs = await this.db.select().from(agentRun).where(whereClause).orderBy(desc2(agentRun.startedAt), desc2(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