@dudousxd/nestjs-agent-store-drizzle 0.31.0 → 0.32.0

package/dist/index.cjs

@@ -311,6 +311,12 @@ var agentRun = (0, import_sqlite_core.sqliteTable)("agent_run", {
   agentName: (0, import_sqlite_core.text)("agent_name"),
   status: (0, import_sqlite_core.text)("status").$type().notNull(),
   durationMs: (0, import_sqlite_core.integer)("duration_ms"),
+  /** Turn start to the first `text` frame; null for a run that streamed none or did not complete. */
+  ttftMs: (0, import_sqlite_core.integer)("ttft_ms"),
+  /** The run's known spend (the sum of its model calls with a known cost); null when none had one. */
+  costUsd: (0, import_sqlite_core.real)("cost_usd"),
+  /** The model the run's last model call resolved to. */
+  modelId: (0, import_sqlite_core.text)("model_id"),
   errorCode: (0, import_sqlite_core.text)("error_code"),
   errorMessage: (0, import_sqlite_core.text)("error_message"),
   retries: (0, import_sqlite_core.integer)("retries").notNull().default(0),
@@ -743,6 +749,12 @@ var agentRun2 = (0, import_mysql_core.mysqlTable)("agent_run", {
   agentName: key("agent_name"),
   status: key("status").$type().notNull(),
   durationMs: (0, import_mysql_core.int)("duration_ms"),
+  /** Turn start to the first `text` frame; null for a run that streamed none or did not complete. */
+  ttftMs: (0, import_mysql_core.int)("ttft_ms"),
+  /** The run's known spend (the sum of its model calls with a known cost); null when none had one. */
+  costUsd: (0, import_mysql_core.double)("cost_usd"),
+  /** The model the run's last model call resolved to. */
+  modelId: key("model_id"),
   errorCode: key("error_code"),
   errorMessage: (0, import_mysql_core.longtext)("error_message"),
   retries: (0, import_mysql_core.int)("retries").notNull().default(0),
@@ -1189,6 +1201,12 @@ var agentRun3 = (0, import_pg_core.pgTable)("agent_run", {
   agentName: (0, import_pg_core.text)("agent_name"),
   status: (0, import_pg_core.text)("status").$type().notNull(),
   durationMs: (0, import_pg_core.integer)("duration_ms"),
+  /** Turn start to the first `text` frame; null for a run that streamed none or did not complete. */
+  ttftMs: (0, import_pg_core.integer)("ttft_ms"),
+  /** The run's known spend (the sum of its model calls with a known cost); null when none had one. */
+  costUsd: (0, import_pg_core.doublePrecision)("cost_usd"),
+  /** The model the run's last model call resolved to. */
+  modelId: (0, import_pg_core.text)("model_id"),
   errorCode: (0, import_pg_core.text)("error_code"),
   errorMessage: (0, import_pg_core.text)("error_message"),
   retries: (0, import_pg_core.integer)("retries").notNull().default(0),
@@ -2804,6 +2822,9 @@ var DrizzleAgentStore = class {
       agentName: run.agentName ?? null,
       status: "running",
       durationMs: null,
+      ttftMs: null,
+      costUsd: null,
+      modelId: null,
       errorCode: null,
       errorMessage: null,
       retries: 0,
@@ -2823,6 +2844,15 @@ var DrizzleAgentStore = class {
     if (end.durationMs !== void 0) {
       updates.durationMs = end.durationMs;
     }
+    if (end.ttftMs !== void 0) {
+      updates.ttftMs = end.ttftMs;
+    }
+    if (end.costUsd !== void 0) {
+      updates.costUsd = end.costUsd;
+    }
+    if (end.modelId !== void 0) {
+      updates.modelId = end.modelId;
+    }
     if (end.errorCode !== void 0) {
       updates.errorCode = end.errorCode;
     }
@@ -3308,6 +3338,11 @@ function toRecentRunRow(run) {
     agentName: run.agentName ?? null,
     status: run.status,
     durationMs: run.durationMs ?? null,
+    ttftMs: run.ttftMs ?? null,
+    // MySQL's DOUBLE and Postgres' double precision come back as numbers; a driver that hands a
+    // decimal back as a string is normalized here so the SPI row is always a number.
+    costUsd: run.costUsd == null ? null : Number(run.costUsd),
+    modelId: run.modelId ?? null,
     errorCode: run.errorCode ?? null,
     errorMessage: run.errorMessage ?? null,
     retries: run.retries,
@@ -3511,6 +3546,7 @@ var DrizzleGovernanceQueries = class {
     let failed = 0;
     let retries = 0;
     const settledDurationsMs = [];
+    const ttftsMs = [];
     for (const run of runs) {
       if (run.status === "completed") {
         completed += 1;
@@ -3521,8 +3557,12 @@ var DrizzleGovernanceQueries = class {
       if ((run.status === "completed" || run.status === "failed") && run.durationMs != null) {
         settledDurationsMs.push(run.durationMs);
       }
+      if (run.status === "completed" && run.parentRunId == null && run.ttftMs != null) {
+        ttftsMs.push(run.ttftMs);
+      }
     }
     settledDurationsMs.sort((left, right) => left - right);
+    ttftsMs.sort((left, right) => left - right);
     return {
       runs: runs.length,
       completed,
@@ -3530,7 +3570,9 @@ var DrizzleGovernanceQueries = class {
       successRate: runs.length === 0 ? 0 : completed / runs.length,
       retries,
       durationP50Ms: percentileMs(settledDurationsMs, 0.5),
-      durationP95Ms: percentileMs(settledDurationsMs, 0.95)
+      durationP95Ms: percentileMs(settledDurationsMs, 0.95),
+      ttftP50Ms: percentileMs(ttftsMs, 0.5),
+      ttftP95Ms: percentileMs(ttftsMs, 0.95)
     };
   }
   async runsByAgent(range) {