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

package/dist/index.cjs

@@ -37,6 +37,7 @@ __export(index_exports, {
   agentChannelState: () => agentChannelState,
   agentConfirmToken: () => agentConfirmToken,
   agentDialectOf: () => agentDialectOf,
+  agentGenuiCall: () => agentGenuiCall,
   agentMemory: () => agentMemory,
   agentMessage: () => agentMessage,
   agentModelPricing: () => agentModelPricing,
@@ -247,6 +248,37 @@ var agentTokenUsage = (0, import_sqlite_core.sqliteTable)("agent_token_usage", {
 }, (table) => [
   (0, import_sqlite_core.index)("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
 ]);
+var agentGenuiCall = (0, import_sqlite_core.sqliteTable)("agent_genui_call", {
+  id: (0, import_sqlite_core.text)("id").primaryKey(),
+  runId: (0, import_sqlite_core.text)("run_id").notNull(),
+  threadId: (0, import_sqlite_core.text)("thread_id").notNull().references(() => agentThread.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: (0, import_sqlite_core.text)("actor_ref").notNull(),
+  agentName: (0, import_sqlite_core.text)("agent_name"),
+  step: (0, import_sqlite_core.integer)("step").notNull(),
+  toolName: (0, import_sqlite_core.text)("tool_name").notNull(),
+  channel: (0, import_sqlite_core.text)("channel").notNull(),
+  mode: (0, import_sqlite_core.text)("mode").$type().notNull(),
+  modelId: (0, import_sqlite_core.text)("model_id").notNull(),
+  attempt: (0, import_sqlite_core.integer)("attempt").notNull(),
+  outcome: (0, import_sqlite_core.text)("outcome").$type().notNull(),
+  validity: (0, import_sqlite_core.text)("validity").$type().notNull(),
+  errorCategory: (0, import_sqlite_core.text)("error_category").$type(),
+  errorMessage: (0, import_sqlite_core.text)("error_message"),
+  inputTokens: (0, import_sqlite_core.integer)("input_tokens").notNull(),
+  outputTokens: (0, import_sqlite_core.integer)("output_tokens").notNull(),
+  costUsd: (0, import_sqlite_core.real)("cost_usd"),
+  costSource: (0, import_sqlite_core.text)("cost_source").$type(),
+  firstFrameMs: (0, import_sqlite_core.integer)("first_frame_ms"),
+  completeFrameMs: (0, import_sqlite_core.integer)("complete_frame_ms"),
+  payloadBytes: (0, import_sqlite_core.integer)("payload_bytes").notNull(),
+  payload: (0, import_sqlite_core.text)("payload"),
+  createdAt: (0, import_sqlite_core.integer)("created_at").notNull()
+}, (table) => [
+  (0, import_sqlite_core.index)("agent_genui_call_created_idx").on(table.createdAt),
+  (0, import_sqlite_core.index)("agent_genui_call_run_idx").on(table.runId)
+]);
 var agentModelPricing = (0, import_sqlite_core.sqliteTable)("agent_model_pricing", {
   id: (0, import_sqlite_core.text)("id").primaryKey(),
   modelId: (0, import_sqlite_core.text)("model_id").notNull(),
@@ -411,6 +443,7 @@ var agentSchema = {
   agentQueuedMessage,
   agentToolCall,
   agentTokenUsage,
+  agentGenuiCall,
   agentModelPricing,
   agentRun,
   agentMemory,
@@ -552,6 +585,45 @@ var agentTokenUsage2 = (0, import_mysql_core.mysqlTable)("agent_token_usage", {
 }, (table) => [
   (0, import_mysql_core.index)("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
 ]);
+var agentGenuiCall2 = (0, import_mysql_core.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: (0, import_mysql_core.int)("step").notNull(),
+  toolName: key("tool_name").notNull(),
+  channel: key("channel").notNull(),
+  mode: key("mode").$type().notNull(),
+  modelId: key("model_id").notNull(),
+  attempt: (0, import_mysql_core.int)("attempt").notNull(),
+  outcome: key("outcome").$type().notNull(),
+  validity: key("validity").$type().notNull(),
+  errorCategory: key("error_category").$type(),
+  errorMessage: (0, import_mysql_core.longtext)("error_message"),
+  inputTokens: (0, import_mysql_core.int)("input_tokens").notNull(),
+  outputTokens: (0, import_mysql_core.int)("output_tokens").notNull(),
+  costUsd: (0, import_mysql_core.double)("cost_usd"),
+  costSource: (0, import_mysql_core.varchar)("cost_source", {
+    length: 16
+  }).$type(),
+  firstFrameMs: (0, import_mysql_core.bigint)("first_frame_ms", {
+    mode: "number"
+  }),
+  completeFrameMs: (0, import_mysql_core.bigint)("complete_frame_ms", {
+    mode: "number"
+  }),
+  payloadBytes: (0, import_mysql_core.int)("payload_bytes").notNull(),
+  payload: (0, import_mysql_core.longtext)("payload"),
+  createdAt: (0, import_mysql_core.bigint)("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  (0, import_mysql_core.index)("agent_genui_call_created_idx").on(table.createdAt),
+  (0, import_mysql_core.index)("agent_genui_call_run_idx").on(table.runId)
+]);
 var agentModelPricing2 = (0, import_mysql_core.mysqlTable)("agent_model_pricing", {
   id: key("id").primaryKey(),
   modelId: key("model_id").notNull(),
@@ -707,6 +779,7 @@ var mysqlAgentSchema = {
   agentQueuedMessage: agentQueuedMessage2,
   agentToolCall: agentToolCall2,
   agentTokenUsage: agentTokenUsage2,
+  agentGenuiCall: agentGenuiCall2,
   agentModelPricing: agentModelPricing2,
   agentRun: agentRun2,
   agentMemory: agentMemory2,
@@ -835,6 +908,45 @@ var agentTokenUsage3 = (0, import_pg_core.pgTable)("agent_token_usage", {
 }, (table) => [
   (0, import_pg_core.index)("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
 ]);
+var agentGenuiCall3 = (0, import_pg_core.pgTable)("agent_genui_call", {
+  id: (0, import_pg_core.text)("id").primaryKey(),
+  runId: (0, import_pg_core.text)("run_id").notNull(),
+  threadId: (0, import_pg_core.text)("thread_id").notNull().references(() => agentThread3.id, {
+    onDelete: "cascade"
+  }),
+  actorRef: (0, import_pg_core.text)("actor_ref").notNull(),
+  agentName: (0, import_pg_core.text)("agent_name"),
+  step: (0, import_pg_core.integer)("step").notNull(),
+  toolName: (0, import_pg_core.text)("tool_name").notNull(),
+  channel: (0, import_pg_core.text)("channel").notNull(),
+  mode: (0, import_pg_core.text)("mode").$type().notNull(),
+  modelId: (0, import_pg_core.text)("model_id").notNull(),
+  attempt: (0, import_pg_core.integer)("attempt").notNull(),
+  outcome: (0, import_pg_core.text)("outcome").$type().notNull(),
+  validity: (0, import_pg_core.text)("validity").$type().notNull(),
+  errorCategory: (0, import_pg_core.text)("error_category").$type(),
+  errorMessage: (0, import_pg_core.text)("error_message"),
+  inputTokens: (0, import_pg_core.integer)("input_tokens").notNull(),
+  outputTokens: (0, import_pg_core.integer)("output_tokens").notNull(),
+  costUsd: (0, import_pg_core.doublePrecision)("cost_usd"),
+  costSource: (0, import_pg_core.varchar)("cost_source", {
+    length: 16
+  }).$type(),
+  firstFrameMs: (0, import_pg_core.bigint)("first_frame_ms", {
+    mode: "number"
+  }),
+  completeFrameMs: (0, import_pg_core.bigint)("complete_frame_ms", {
+    mode: "number"
+  }),
+  payloadBytes: (0, import_pg_core.integer)("payload_bytes").notNull(),
+  payload: (0, import_pg_core.text)("payload"),
+  createdAt: (0, import_pg_core.bigint)("created_at", {
+    mode: "number"
+  }).notNull()
+}, (table) => [
+  (0, import_pg_core.index)("agent_genui_call_created_idx").on(table.createdAt),
+  (0, import_pg_core.index)("agent_genui_call_run_idx").on(table.runId)
+]);
 var agentModelPricing3 = (0, import_pg_core.pgTable)("agent_model_pricing", {
   id: (0, import_pg_core.text)("id").primaryKey(),
   modelId: (0, import_pg_core.text)("model_id").notNull(),
@@ -988,6 +1100,7 @@ var pgAgentSchema = {
   agentQueuedMessage: agentQueuedMessage3,
   agentToolCall: agentToolCall3,
   agentTokenUsage: agentTokenUsage3,
+  agentGenuiCall: agentGenuiCall3,
   agentModelPricing: agentModelPricing3,
   agentRun: agentRun3,
   agentMemory: agentMemory3,
@@ -2592,6 +2705,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);
   }
@@ -2764,6 +2915,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,
@@ -3349,6 +3545,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 } = (0, import_nestjs_agent_core5.dayBoundsUtc)(range);
+    const table = this.t.agentGenuiCall;
+    const rows = await this.db.select().from(table).where((0, import_drizzle_orm5.and)((0, import_drizzle_orm5.gte)(table.createdAt, start.getTime()), (0, import_drizzle_orm5.lte)(table.createdAt, end.getTime())));
+    return (0, import_nestjs_agent_core5.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();
@@ -4195,6 +4398,7 @@ DrizzleAgentStoreModule = _ts_decorate2([
   agentChannelState,
   agentConfirmToken,
   agentDialectOf,
+  agentGenuiCall,
   agentMemory,
   agentMessage,
   agentModelPricing,