@dudousxd/nestjs-agent-store-drizzle 0.29.12 → 0.30.0

package/dist/index.js

@@ -193,6 +193,37 @@ var agentTokenUsage = sqliteTable("agent_token_usage", {
 }, (table) => [
   index("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
 ]);
+var agentGenuiCall = sqliteTable("agent_genui_call", {
+  id: text("id").primaryKey(),
+  runId: text("run_id").notNull(),
+  threadId: text("thread_id").notNull().references(() => agentThread.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: text("actor_ref").notNull(),
+  agentName: text("agent_name"),
+  step: integer("step").notNull(),
+  toolName: text("tool_name").notNull(),
+  channel: text("channel").notNull(),
+  mode: text("mode").$type().notNull(),
+  modelId: text("model_id").notNull(),
+  attempt: integer("attempt").notNull(),
+  outcome: text("outcome").$type().notNull(),
+  validity: text("validity").$type().notNull(),
+  errorCategory: text("error_category").$type(),
+  errorMessage: text("error_message"),
+  inputTokens: integer("input_tokens").notNull(),
+  outputTokens: integer("output_tokens").notNull(),
+  costUsd: real("cost_usd"),
+  costSource: text("cost_source").$type(),
+  firstFrameMs: integer("first_frame_ms"),
+  completeFrameMs: integer("complete_frame_ms"),
+  payloadBytes: integer("payload_bytes").notNull(),
+  payload: text("payload"),
+  createdAt: integer("created_at").notNull()
+}, (table) => [
+  index("agent_genui_call_created_idx").on(table.createdAt),
+  index("agent_genui_call_run_idx").on(table.runId)
+]);
 var agentModelPricing = sqliteTable("agent_model_pricing", {
   id: text("id").primaryKey(),
   modelId: text("model_id").notNull(),
@@ -357,6 +388,7 @@ var agentSchema = {
   agentQueuedMessage,
   agentToolCall,
   agentTokenUsage,
+  agentGenuiCall,
   agentModelPricing,
   agentRun,
   agentMemory,
@@ -498,6 +530,45 @@ var agentTokenUsage2 = mysqlTable("agent_token_usage", {
 }, (table) => [
   index2("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
 ]);
+var agentGenuiCall2 = mysqlTable("agent_genui_call", {
+  id: key("id").primaryKey(),
+  runId: key("run_id").notNull(),
+  threadId: key("thread_id").notNull().references(() => agentThread2.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: key("actor_ref").notNull(),
+  agentName: key("agent_name"),
+  step: int("step").notNull(),
+  toolName: key("tool_name").notNull(),
+  channel: key("channel").notNull(),
+  mode: key("mode").$type().notNull(),
+  modelId: key("model_id").notNull(),
+  attempt: int("attempt").notNull(),
+  outcome: key("outcome").$type().notNull(),
+  validity: key("validity").$type().notNull(),
+  errorCategory: key("error_category").$type(),
+  errorMessage: longtext("error_message"),
+  inputTokens: int("input_tokens").notNull(),
+  outputTokens: int("output_tokens").notNull(),
+  costUsd: double("cost_usd"),
+  costSource: varchar("cost_source", {
+    length: 16
+  }).$type(),
+  firstFrameMs: bigint("first_frame_ms", {
+    mode: "number"
+  }),
+  completeFrameMs: bigint("complete_frame_ms", {
+    mode: "number"
+  }),
+  payloadBytes: int("payload_bytes").notNull(),
+  payload: longtext("payload"),
+  createdAt: bigint("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  index2("agent_genui_call_created_idx").on(table.createdAt),
+  index2("agent_genui_call_run_idx").on(table.runId)
+]);
 var agentModelPricing2 = mysqlTable("agent_model_pricing", {
   id: key("id").primaryKey(),
   modelId: key("model_id").notNull(),
@@ -653,6 +724,7 @@ var mysqlAgentSchema = {
   agentQueuedMessage: agentQueuedMessage2,
   agentToolCall: agentToolCall2,
   agentTokenUsage: agentTokenUsage2,
+  agentGenuiCall: agentGenuiCall2,
   agentModelPricing: agentModelPricing2,
   agentRun: agentRun2,
   agentMemory: agentMemory2,
@@ -664,7 +736,7 @@ var mysqlAgentSchema = {
 
 // src/schema-pg.ts
 import { canonicalActionProposalJson as canonicalActionProposalJson2 } from "@dudousxd/nestjs-agent-core";
-import { bigint as bigint2, boolean as boolean2, customType as customType2, doublePrecision, index as index3, integer as integer2, jsonb, pgTable, primaryKey as primaryKey3, text as text2, timestamp, uniqueIndex as uniqueIndex3 } from "drizzle-orm/pg-core";
+import { bigint as bigint2, boolean as boolean2, customType as customType2, doublePrecision, index as index3, integer as integer2, jsonb, pgTable, primaryKey as primaryKey3, text as text2, timestamp, uniqueIndex as uniqueIndex3, varchar as varchar2 } from "drizzle-orm/pg-core";
 var ts2 = /* @__PURE__ */ __name((name) => timestamp(name, {
   withTimezone: true,
   precision: 3,
@@ -781,6 +853,45 @@ var agentTokenUsage3 = pgTable("agent_token_usage", {
 }, (table) => [
   index3("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
 ]);
+var agentGenuiCall3 = pgTable("agent_genui_call", {
+  id: text2("id").primaryKey(),
+  runId: text2("run_id").notNull(),
+  threadId: text2("thread_id").notNull().references(() => agentThread3.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: text2("actor_ref").notNull(),
+  agentName: text2("agent_name"),
+  step: integer2("step").notNull(),
+  toolName: text2("tool_name").notNull(),
+  channel: text2("channel").notNull(),
+  mode: text2("mode").$type().notNull(),
+  modelId: text2("model_id").notNull(),
+  attempt: integer2("attempt").notNull(),
+  outcome: text2("outcome").$type().notNull(),
+  validity: text2("validity").$type().notNull(),
+  errorCategory: text2("error_category").$type(),
+  errorMessage: text2("error_message"),
+  inputTokens: integer2("input_tokens").notNull(),
+  outputTokens: integer2("output_tokens").notNull(),
+  costUsd: doublePrecision("cost_usd"),
+  costSource: varchar2("cost_source", {
+    length: 16
+  }).$type(),
+  firstFrameMs: bigint2("first_frame_ms", {
+    mode: "number"
+  }),
+  completeFrameMs: bigint2("complete_frame_ms", {
+    mode: "number"
+  }),
+  payloadBytes: integer2("payload_bytes").notNull(),
+  payload: text2("payload"),
+  createdAt: bigint2("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  index3("agent_genui_call_created_idx").on(table.createdAt),
+  index3("agent_genui_call_run_idx").on(table.runId)
+]);
 var agentModelPricing3 = pgTable("agent_model_pricing", {
   id: text2("id").primaryKey(),
   modelId: text2("model_id").notNull(),
@@ -934,6 +1045,7 @@ var pgAgentSchema = {
   agentQueuedMessage: agentQueuedMessage3,
   agentToolCall: agentToolCall3,
   agentTokenUsage: agentTokenUsage3,
+  agentGenuiCall: agentGenuiCall3,
   agentModelPricing: agentModelPricing3,
   agentRun: agentRun3,
   agentMemory: agentMemory3,
@@ -2538,6 +2650,44 @@ var DrizzleAgentStore = class {
       createdAt: /* @__PURE__ */ new Date()
     });
   }
+  /**
+  * Generative-UI telemetry: one `agent_genui_call` row per UI tool call, upserted by its tool call
+  * id — a replayed record replaces the row (keeping the first `created_at`) rather than adding one.
+  */
+  async recordGenuiCalls(rows) {
+    const table = this.t.agentGenuiCall;
+    for (const row of rows) {
+      const columns = {
+        runId: row.runId,
+        threadId: row.threadId,
+        actorRef: row.actorRef,
+        agentName: row.agentName ?? null,
+        step: row.step,
+        toolName: row.toolName,
+        channel: row.channel,
+        mode: row.mode,
+        modelId: row.modelId,
+        attempt: row.attempt,
+        outcome: row.outcome,
+        validity: row.validity,
+        errorCategory: row.errorCategory ?? null,
+        errorMessage: row.errorMessage ?? null,
+        inputTokens: row.inputTokens,
+        outputTokens: row.outputTokens,
+        costUsd: row.costUsd ?? null,
+        costSource: row.costSource ?? null,
+        firstFrameMs: row.firstFrameMs ?? null,
+        completeFrameMs: row.completeFrameMs ?? null,
+        payloadBytes: row.payloadBytes,
+        payload: row.payload ?? null
+      };
+      await upsert(this.dialect, this.db.insert(table).values({
+        id: row.toolCallId,
+        ...columns,
+        createdAt: Date.now()
+      }), table.id, columns);
+    }
+  }
   async quotaToday(actorRef, day) {
     return this.usageBetween(actorRef, day, day);
   }
@@ -2651,7 +2801,7 @@ function toQueuedMessage(row) {
 __name(toQueuedMessage, "toQueuedMessage");
 
 // src/drizzle-governance-queries.ts
-import { bucketByActor, bucketByModel, bucketByThread, bucketUsageTrend, dayBoundsUtc, rollupThreadUsage, truncateDetailContent } from "@dudousxd/nestjs-agent-core";
+import { bucketByActor, bucketByModel, bucketByThread, bucketUsageTrend, dayBoundsUtc, rollupThreadUsage, summarizeGenuiCalls, truncateDetailContent } from "@dudousxd/nestjs-agent-core";
 import { and as and3, asc as asc3, count, desc as desc2, eq as eq3, gte as gte2, inArray as inArray2, isNull as isNull3, like, lte as lte3, sql as sql4, sum } from "drizzle-orm";
 var UNCLASSIFIED_ERROR_CODE = "unknown";
 var DEFAULT_AGENT_BUCKET = "(default)";
@@ -2710,6 +2860,51 @@ function emptyPage(query) {
   };
 }
 __name(emptyPage, "emptyPage");
+function toGenuiCallRow(row) {
+  return {
+    toolCallId: row.id,
+    runId: row.runId,
+    threadId: row.threadId,
+    actorRef: row.actorRef,
+    ...row.agentName !== null ? {
+      agentName: row.agentName
+    } : {},
+    step: Number(row.step),
+    toolName: row.toolName,
+    channel: row.channel,
+    mode: row.mode,
+    modelId: row.modelId,
+    attempt: Number(row.attempt),
+    outcome: row.outcome,
+    validity: row.validity,
+    ...row.errorCategory !== null ? {
+      errorCategory: row.errorCategory
+    } : {},
+    ...row.errorMessage !== null ? {
+      errorMessage: row.errorMessage
+    } : {},
+    inputTokens: Number(row.inputTokens),
+    outputTokens: Number(row.outputTokens),
+    ...row.costUsd !== null ? {
+      costUsd: Number(row.costUsd)
+    } : {},
+    ...row.costSource !== null ? {
+      costSource: row.costSource
+    } : {},
+    ...row.firstFrameMs !== null ? {
+      firstFrameMs: Number(row.firstFrameMs)
+    } : {},
+    ...row.completeFrameMs !== null ? {
+      completeFrameMs: Number(row.completeFrameMs)
+    } : {},
+    payloadBytes: Number(row.payloadBytes),
+    ...row.payload !== null ? {
+      payload: row.payload
+    } : {},
+    createdAt: new Date(Number(row.createdAt))
+  };
+}
+__name(toGenuiCallRow, "toGenuiCallRow");
 function toRecentRunRow(run) {
   return {
     runId: run.id,
@@ -3295,6 +3490,13 @@ var DrizzleGovernanceQueries = class {
       })
     };
   }
+  /** Generative-UI stats over the `agent_genui_call` rows created in the (UTC, inclusive) day range. */
+  async genuiStats(range) {
+    const { start, end } = dayBoundsUtc(range);
+    const table = this.t.agentGenuiCall;
+    const rows = await this.db.select().from(table).where(and3(gte2(table.createdAt, start.getTime()), lte3(table.createdAt, end.getTime())));
+    return summarizeGenuiCalls(rows.map(toGenuiCallRow));
+  }
   /** Tool-call counts for a set of messages in ONE grouped query — the detail view's no-N+1 guarantee. */
   async toolCallCountsByMessage(messageIds) {
     const counts = /* @__PURE__ */ new Map();
@@ -4140,6 +4342,7 @@ export {
   agentChannelState,
   agentConfirmToken,
   agentDialectOf,
+  agentGenuiCall,
   agentMemory,
   agentMessage,
   agentModelPricing,