@elizaos/plugin-trajectory-logger 2.0.0-alpha.7 → 2.0.0-alpha.9

package/dist/node/index.node.js

@@ -10,7 +10,7 @@ var __export = (target, all) => {
 };
 var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/entity.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/entity.js
 function is(value, type) {
   if (!value || typeof value !== "object") {
     return false;
@@ -38,7 +38,7 @@ var init_entity = __esm(() => {
   hasOwnEntityKind = Symbol.for("drizzle:hasOwnEntityKind");
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/column.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/column.js
 var Column;
 var init_column = __esm(() => {
   init_entity();
@@ -92,7 +92,7 @@ var init_column = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/column-builder.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/column-builder.js
 var ColumnBuilder;
 var init_column_builder = __esm(() => {
   init_entity();
@@ -152,19 +152,19 @@ var init_column_builder = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/table.utils.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/table.utils.js
 var TableName;
 var init_table_utils = __esm(() => {
   TableName = Symbol.for("drizzle:Name");
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/tracing-utils.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/tracing-utils.js
 function iife(fn, ...args) {
   return fn(...args);
 }
 var init_tracing_utils = () => {};
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/pg-core/unique-constraint.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/pg-core/unique-constraint.js
 function uniqueKeyName(table, columns) {
   return `${table[TableName]}_${columns.join("_")}_unique`;
 }
@@ -172,7 +172,7 @@ var init_unique_constraint = __esm(() => {
   init_table_utils();
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/pg-core/columns/common.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/pg-core/columns/common.js
 var PgColumn, ExtraConfigColumn;
 var init_common = __esm(() => {
   init_column();
@@ -226,7 +226,7 @@ var init_common = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/pg-core/columns/enum.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/pg-core/columns/enum.js
 function isPgEnum(obj) {
   return !!obj && typeof obj === "function" && isPgEnumSym in obj && obj[isPgEnumSym] === true;
 }
@@ -261,7 +261,7 @@ var init_enum = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/subquery.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/subquery.js
 var Subquery, WithSubquery;
 var init_subquery = __esm(() => {
   init_entity();
@@ -283,11 +283,11 @@ var init_subquery = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/version.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/version.js
 var version = "0.45.1";
 var init_version = () => {};
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/tracing.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/tracing.js
 var otel, rawTracer, tracer;
 var init_tracing = __esm(() => {
   init_tracing_utils();
@@ -317,13 +317,13 @@ var init_tracing = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/view-common.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/view-common.js
 var ViewBaseConfig;
 var init_view_common = __esm(() => {
   ViewBaseConfig = Symbol.for("drizzle:ViewBaseConfig");
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/table.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/table.js
 function isTable(table) {
   return typeof table === "object" && table !== null && IsDrizzleTable in table;
 }
@@ -374,7 +374,7 @@ var init_table = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/sql/sql.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/sql/sql.js
 function isSQLWrapper(value) {
   return value !== null && value !== undefined && typeof value.getSQL === "function";
 }
@@ -776,7 +776,7 @@ var init_sql = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/alias.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/alias.js
 function aliasedTable(table, tableAlias) {
   return new Proxy(table, new TableAliasProxyHandler(tableAlias, false));
 }
@@ -877,7 +877,7 @@ var init_alias = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/errors.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/errors.js
 var DrizzleError, DrizzleQueryError, TransactionRollbackError;
 var init_errors = __esm(() => {
   init_entity();
@@ -909,7 +909,7 @@ params: ${params}`);
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/logger.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/logger.js
 var ConsoleLogWriter, DefaultLogger, NoopLogger;
 var init_logger = __esm(() => {
   init_entity();
@@ -943,7 +943,7 @@ var init_logger = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/query-promise.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/query-promise.js
 var QueryPromise;
 var init_query_promise = __esm(() => {
   init_entity();
@@ -968,7 +968,7 @@ var init_query_promise = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/utils.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/utils.js
 function mapResultRow(columns, row, joinsNotNullableMap) {
   const nullifyMap = {};
   const result = columns.reduce((result2, { path, field }, columnIndex) => {
@@ -1134,7 +1134,7 @@ var init_utils = __esm(() => {
   textDecoder = typeof TextDecoder === "undefined" ? null : new TextDecoder;
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/pg-core/table.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/pg-core/table.js
 var InlineForeignKeys, EnableRLS, PgTable;
 var init_table2 = __esm(() => {
   init_entity();
@@ -1154,7 +1154,7 @@ var init_table2 = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/pg-core/primary-keys.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/pg-core/primary-keys.js
 var PrimaryKeyBuilder, PrimaryKey;
 var init_primary_keys = __esm(() => {
   init_entity();
@@ -1186,7 +1186,7 @@ var init_primary_keys = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/sql/expressions/conditions.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/sql/expressions/conditions.js
 function bindIfParam(value, column) {
   if (isDriverValueEncoder(column) && !isSQLWrapper(value) && !is(value, Param) && !is(value, Placeholder) && !is(value, Column) && !is(value, Table) && !is(value, View)) {
     return new Param(value, column);
@@ -1322,7 +1322,7 @@ var init_conditions = __esm(() => {
   init_sql();
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/sql/expressions/select.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/sql/expressions/select.js
 function asc(column) {
   return sql`${column} asc`;
 }
@@ -1333,13 +1333,13 @@ var init_select = __esm(() => {
   init_sql();
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/sql/expressions/index.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/sql/expressions/index.js
 var init_expressions = __esm(() => {
   init_conditions();
   init_select();
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/relations.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/relations.js
 function getOperators() {
   return {
     and,
@@ -1571,7 +1571,7 @@ var init_relations = __esm(() => {
   };
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/sql/functions/aggregate.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/sql/functions/aggregate.js
 function count(expression) {
   return sql`count(${expression || sql.raw("*")})`.mapWith(Number);
 }
@@ -1602,7 +1602,7 @@ var init_aggregate = __esm(() => {
   init_sql();
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/sql/functions/vector.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/sql/functions/vector.js
 function toSql(value) {
   return JSON.stringify(value);
 }
@@ -1646,20 +1646,20 @@ var init_vector = __esm(() => {
   init_sql();
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/sql/functions/index.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/sql/functions/index.js
 var init_functions = __esm(() => {
   init_aggregate();
   init_vector();
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/sql/index.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/sql/index.js
 var init_sql2 = __esm(() => {
   init_expressions();
   init_functions();
   init_sql();
 });
 
-// ../../../node_modules/.bun/drizzle-orm@0.45.1/node_modules/drizzle-orm/index.js
+// ../../../node_modules/.pnpm/drizzle-orm@0.45.1_@electric-sql+pglite@0.3.15_@neondatabase+serverless@1.0.2_@opentelemetry+_7qzarrdo6qvvshsh3fvwkgjp5i/node_modules/drizzle-orm/index.js
 var exports_drizzle_orm = {};
 __export(exports_drizzle_orm, {
   textDecoder: () => textDecoder,
@@ -1797,6 +1797,12 @@ var init_drizzle_orm = __esm(() => {
   init_view_common();
 });
 
+// index.ts
+import {
+  createUniqueUuid
+} from "@elizaos/core";
+import crypto from "node:crypto";
+
 // TrajectoryLoggerService.ts
 import { logger as logger2, Service } from "@elizaos/core";
 import { v4 as uuidv4 } from "uuid";
@@ -1851,6 +1857,9 @@ function sqlLiteral(v) {
 
 class TrajectoryLoggerService extends Service {
   static serviceType = "trajectory_logger";
+  get serviceType() {
+    return TrajectoryLoggerService.serviceType;
+  }
   capabilityDescription = "Captures and persists LLM calls, provider accesses, and full trajectories for debugging, analysis, and RL training";
   enabled = true;
   initialized = false;
@@ -1858,7 +1867,7 @@ class TrajectoryLoggerService extends Service {
   stepToTrajectory = new Map;
   writeQueues = new Map;
   static async start(runtime) {
-    const service = new TrajectoryLoggerService(runtime);
+    const service = new this(runtime);
     await service.initialize();
     return service;
   }
@@ -1897,6 +1906,7 @@ class TrajectoryLoggerService extends Service {
       return;
     }
     await this.ensureTablesExist();
+    await this.backfillTrajectoriesMissingLlmCalls();
     this.initialized = true;
     logger2.info("[trajectory-logger] Trajectory logger service initialized");
   }
@@ -2041,6 +2051,56 @@ class TrajectoryLoggerService extends Service {
     await this.executeRawSql(`CREATE INDEX IF NOT EXISTS idx_trajectory_step_index_trajectory_id ON trajectory_step_index(trajectory_id)`);
     await this.executeRawSql(`CREATE INDEX IF NOT EXISTS idx_trajectory_step_index_is_active ON trajectory_step_index(is_active)`);
   }
+  normalizeTrajectoryStatus(value) {
+    switch (value) {
+      case "active":
+      case "completed":
+      case "error":
+      case "timeout":
+      case "terminated":
+        return value;
+      default:
+        return "completed";
+    }
+  }
+  async backfillTrajectoriesMissingLlmCalls() {
+    const maxRows = 2000;
+    let fixed = 0;
+    try {
+      const rows = await this.executeRawSql(`
+        SELECT id, status
+        FROM trajectories
+        WHERE COALESCE(llm_call_count, 0) = 0
+          AND status <> 'active'
+        ORDER BY created_at ASC
+        LIMIT ${maxRows}
+      `);
+      for (const row of rows.rows) {
+        const trajectoryId = asString(pickCell(row, "id"));
+        if (!trajectoryId)
+          continue;
+        const status = this.normalizeTrajectoryStatus(asString(pickCell(row, "status")));
+        await this.withTrajectoryWriteLock(trajectoryId, async () => {
+          const trajectory = await this.getTrajectoryById(trajectoryId);
+          if (!trajectory)
+            return;
+          const inserted = this.ensureAtLeastOneLlmCall(trajectory, "backfill");
+          if (!inserted)
+            return;
+          await this.persistTrajectory(trajectoryId, trajectory, status);
+          fixed += 1;
+        });
+      }
+      if (fixed > 0) {
+        logger2.info(`[trajectory-logger] Backfilled ${fixed} completed trajectories with synthetic LLM calls`);
+      }
+      if (rows.rows.length >= maxRows) {
+        logger2.warn(`[trajectory-logger] Backfill hit safety cap (${maxRows}); remaining older trajectories will be fixed lazily when touched`);
+      }
+    } catch (err) {
+      logger2.warn(`[trajectory-logger] Failed to backfill trajectories missing LLM calls (non-fatal): ${err instanceof Error ? err.message : String(err)}`);
+    }
+  }
   normalizePurpose(value) {
     switch (value) {
       case "action":
@@ -2087,15 +2147,57 @@ class TrajectoryLoggerService extends Service {
       done: false
     };
   }
+  ensureAtLeastOneLlmCall(trajectory, source) {
+    if (this.computeTotals(trajectory.steps).llmCallCount > 0) {
+      return false;
+    }
+    const timestamp = Date.now();
+    let step = trajectory.steps[trajectory.steps.length - 1];
+    if (!step) {
+      step = this.createStep(uuidv4(), 0, this.defaultEnvironmentState(timestamp));
+      step.done = true;
+      trajectory.steps.push(step);
+    }
+    if (!Array.isArray(step.llmCalls)) {
+      step.llmCalls = [];
+    }
+    const syntheticCall = {
+      callId: uuidv4(),
+      timestamp,
+      model: "milady/synthetic-trajectory-fallback",
+      systemPrompt: "[synthetic] inserted by trajectory logger because no LLM calls were captured",
+      userPrompt: "[synthetic] this trajectory completed without recorded model activity",
+      response: "[synthetic] placeholder call inserted to enforce minimum llm_call_count=1",
+      temperature: 0,
+      maxTokens: 0,
+      promptTokens: 0,
+      completionTokens: 0,
+      latencyMs: 0,
+      purpose: "other",
+      actionType: "TRAJECTORY_FALLBACK"
+    };
+    step.llmCalls.push(syntheticCall);
+    if (!step.metadata)
+      step.metadata = {};
+    step.metadata.syntheticLlmCall = true;
+    step.metadata.syntheticLlmCallSource = source;
+    trajectory.metadata.syntheticLlmCall = true;
+    trajectory.metadata.syntheticLlmCallSource = source;
+    const existingFallbackCount = trajectory.metrics.syntheticLlmFallbackCount;
+    trajectory.metrics.syntheticLlmFallbackCount = typeof existingFallbackCount === "number" && Number.isFinite(existingFallbackCount) ? existingFallbackCount + 1 : 1;
+    return true;
+  }
   computeTotals(steps) {
     let llmCallCount = 0;
     let providerAccessCount = 0;
     let totalPromptTokens = 0;
     let totalCompletionTokens = 0;
     for (const step of steps) {
-      llmCallCount += step.llmCalls.length;
-      providerAccessCount += step.providerAccesses.length;
-      for (const call of step.llmCalls) {
+      const llmCalls = Array.isArray(step.llmCalls) ? step.llmCalls : [];
+      const providerAccesses = Array.isArray(step.providerAccesses) ? step.providerAccesses : [];
+      llmCallCount += llmCalls.length;
+      providerAccessCount += providerAccesses.length;
+      for (const call of llmCalls) {
         totalPromptTokens += call.promptTokens ?? 0;
         totalCompletionTokens += call.completionTokens ?? 0;
       }
@@ -2190,6 +2292,9 @@ class TrajectoryLoggerService extends Service {
     return row ? asString(pickCell(row, "step_id")) : null;
   }
   async persistTrajectory(trajectoryId, trajectory, status = "active") {
+    if (status !== "active") {
+      this.ensureAtLeastOneLlmCall(trajectory, "finalize");
+    }
     const totals = this.computeTotals(trajectory.steps);
     const isFinalStatus = status !== "active";
     const persistedEndTime = isFinalStatus ? trajectory.endTime : null;
@@ -2238,8 +2343,13 @@ class TrajectoryLoggerService extends Service {
   }
   async ensureStepExists(trajectory, stepId) {
     let step = trajectory.steps.find((entry) => entry.stepId === stepId);
-    if (step)
+    if (step) {
+      if (!Array.isArray(step.llmCalls))
+        step.llmCalls = [];
+      if (!Array.isArray(step.providerAccesses))
+        step.providerAccesses = [];
       return step;
+    }
     const index = await this.getStepIndex(stepId);
     const stepNumber = index?.stepNumber ?? trajectory.steps.length;
     step = this.createStep(stepId, stepNumber, this.defaultEnvironmentState());
@@ -2589,23 +2699,28 @@ class TrajectoryLoggerService extends Service {
       ORDER BY created_at DESC
       LIMIT ${limit} OFFSET ${offset}
     `);
-    const trajectories = rowsResult.rows.map((row) => ({
-      id: asString(pickCell(row, "id")) ?? "",
-      agentId: asString(pickCell(row, "agent_id")) ?? "",
-      source: asString(pickCell(row, "source")) ?? "chat",
-      status: asString(pickCell(row, "status")) ?? "completed",
-      startTime: asNumber(pickCell(row, "start_time")) ?? 0,
-      endTime: asNumber(pickCell(row, "end_time")),
-      durationMs: asNumber(pickCell(row, "duration_ms")),
-      stepCount: asNumber(pickCell(row, "step_count")) ?? 0,
-      llmCallCount: asNumber(pickCell(row, "llm_call_count")) ?? 0,
-      totalPromptTokens: asNumber(pickCell(row, "total_prompt_tokens")) ?? 0,
-      totalCompletionTokens: asNumber(pickCell(row, "total_completion_tokens")) ?? 0,
-      totalReward: asNumber(pickCell(row, "total_reward")) ?? 0,
-      scenarioId: asString(pickCell(row, "scenario_id")),
-      batchId: asString(pickCell(row, "batch_id")),
-      createdAt: asIsoString(pickCell(row, "created_at"))
-    }));
+    const trajectories = rowsResult.rows.map((row) => {
+      const status = asString(pickCell(row, "status")) ?? "completed";
+      const rawLlmCallCount = asNumber(pickCell(row, "llm_call_count")) ?? 0;
+      const llmCallCount = status === "active" ? rawLlmCallCount : Math.max(1, rawLlmCallCount);
+      return {
+        id: asString(pickCell(row, "id")) ?? "",
+        agentId: asString(pickCell(row, "agent_id")) ?? "",
+        source: asString(pickCell(row, "source")) ?? "chat",
+        status,
+        startTime: asNumber(pickCell(row, "start_time")) ?? 0,
+        endTime: asNumber(pickCell(row, "end_time")),
+        durationMs: asNumber(pickCell(row, "duration_ms")),
+        stepCount: asNumber(pickCell(row, "step_count")) ?? 0,
+        llmCallCount,
+        totalPromptTokens: asNumber(pickCell(row, "total_prompt_tokens")) ?? 0,
+        totalCompletionTokens: asNumber(pickCell(row, "total_completion_tokens")) ?? 0,
+        totalReward: asNumber(pickCell(row, "total_reward")) ?? 0,
+        scenarioId: asString(pickCell(row, "scenario_id")),
+        batchId: asString(pickCell(row, "batch_id")),
+        createdAt: asIsoString(pickCell(row, "created_at"))
+      };
+    });
     return { trajectories, total, offset, limit };
   }
   async getTrajectoryDetail(trajectoryId) {
@@ -2617,7 +2732,17 @@ class TrajectoryLoggerService extends Service {
     if (result.rows.length === 0)
       return null;
     const row = result.rows[0];
-    return this.rowToTrajectory(row);
+    const trajectory = this.rowToTrajectory(row);
+    const status = this.normalizeTrajectoryStatus(asString(pickCell(row, "status")));
+    if (status !== "active") {
+      const inserted = this.ensureAtLeastOneLlmCall(trajectory, "backfill");
+      if (inserted) {
+        await this.withTrajectoryWriteLock(trajectoryId, async () => {
+          await this.persistTrajectory(trajectoryId, trajectory, status);
+        });
+      }
+    }
+    return trajectory;
   }
   async getStats() {
     const runtime = this.runtime;
@@ -2716,6 +2841,113 @@ class TrajectoryLoggerService extends Service {
     await this.executeRawSql(`DELETE FROM trajectories`);
     return count2;
   }
+  sanitizeZipFolderName(value) {
+    const sanitized = value.trim().replace(/[^a-zA-Z0-9._-]+/g, "_").replace(/^_+|_+$/g, "");
+    return sanitized || "trajectory";
+  }
+  redactTrajectoryPrompts(trajectory) {
+    return {
+      ...trajectory,
+      steps: trajectory.steps.map((step) => ({
+        ...step,
+        llmCalls: step.llmCalls.map((call) => ({
+          ...call,
+          systemPrompt: "[redacted]",
+          userPrompt: "[redacted]",
+          response: "[redacted]"
+        }))
+      }))
+    };
+  }
+  buildZipSummary(trajectory) {
+    const finalStatus = trajectory.metrics?.finalStatus ?? "completed";
+    const normalizedEndTime = typeof trajectory.endTime === "number" && trajectory.endTime > 0 ? trajectory.endTime : null;
+    const status = finalStatus === "timeout" || finalStatus === "terminated" || finalStatus === "error" ? "error" : finalStatus === "completed" ? "completed" : normalizedEndTime ? "completed" : "active";
+    let llmCallCount = 0;
+    let providerAccessCount = 0;
+    let totalPromptTokens = 0;
+    let totalCompletionTokens = 0;
+    for (const step of trajectory.steps) {
+      providerAccessCount += step.providerAccesses.length;
+      llmCallCount += step.llmCalls.length;
+      for (const call of step.llmCalls) {
+        totalPromptTokens += call.promptTokens ?? 0;
+        totalCompletionTokens += call.completionTokens ?? 0;
+      }
+    }
+    const metadata = trajectory.metadata ?? {};
+    const asNullableString = (value) => typeof value === "string" ? value : null;
+    const source = typeof metadata.source === "string" ? metadata.source : "chat";
+    const normalizedDurationMs = status === "active" ? null : typeof trajectory.durationMs === "number" ? trajectory.durationMs : null;
+    const updatedAtMs = normalizedEndTime ?? (trajectory.startTime || Date.now());
+    return {
+      id: trajectory.trajectoryId,
+      agentId: trajectory.agentId,
+      roomId: asNullableString(metadata.roomId),
+      entityId: asNullableString(metadata.entityId),
+      conversationId: asNullableString(metadata.conversationId),
+      source,
+      status,
+      startTime: trajectory.startTime,
+      endTime: normalizedEndTime,
+      durationMs: normalizedDurationMs,
+      llmCallCount,
+      providerAccessCount,
+      totalPromptTokens,
+      totalCompletionTokens,
+      metadata,
+      createdAt: new Date(trajectory.startTime).toISOString(),
+      updatedAt: new Date(updatedAtMs).toISOString()
+    };
+  }
+  async exportTrajectoriesZip(options = {}) {
+    let targetIds = Array.isArray(options.trajectoryIds) ? options.trajectoryIds.filter((id) => typeof id === "string" && id.trim().length > 0) : [];
+    if (targetIds.length === 0) {
+      const list = await this.listTrajectories({
+        limit: 500,
+        startDate: options.startDate,
+        endDate: options.endDate,
+        scenarioId: options.scenarioId,
+        batchId: options.batchId
+      });
+      targetIds = list.trajectories.map((trajectory) => trajectory.id);
+    }
+    const entries = [];
+    const manifestRows = [];
+    for (const trajectoryId of targetIds) {
+      const detail = await this.getTrajectoryDetail(trajectoryId);
+      if (!detail)
+        continue;
+      const exportTrajectory = options.includePrompts === false ? this.redactTrajectoryPrompts(detail) : detail;
+      const summary = this.buildZipSummary(exportTrajectory);
+      const folderName = this.sanitizeZipFolderName(trajectoryId);
+      entries.push({
+        name: `${folderName}/trajectory.json`,
+        data: JSON.stringify(exportTrajectory, null, 2)
+      });
+      entries.push({
+        name: `${folderName}/summary.json`,
+        data: JSON.stringify(summary, null, 2)
+      });
+      manifestRows.push({
+        trajectoryId,
+        folder: folderName,
+        createdAt: summary.createdAt
+      });
+    }
+    entries.unshift({
+      name: "manifest.json",
+      data: JSON.stringify({
+        exportedAt: new Date().toISOString(),
+        trajectories: manifestRows
+      }, null, 2)
+    });
+    const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
+    return {
+      filename: `trajectories-${timestamp}.zip`,
+      entries
+    };
+  }
   async exportTrajectories(options) {
     const runtime = this.runtime;
     if (!runtime?.adapter) {
@@ -2770,18 +3002,7 @@ class TrajectoryLoggerService extends Service {
     }
     let exportData = trajectories;
     if (!options.includePrompts) {
-      exportData = trajectories.map((t) => ({
-        ...t,
-        steps: t.steps.map((s) => ({
-          ...s,
-          llmCalls: s.llmCalls.map((c) => ({
-            ...c,
-            systemPrompt: "[redacted]",
-            userPrompt: "[redacted]",
-            response: "[redacted]"
-          }))
-        }))
-      }));
+      exportData = trajectories.map((trajectory) => this.redactTrajectoryPrompts(trajectory));
     }
     return {
       data: JSON.stringify(exportData, null, 2),
@@ -3445,11 +3666,101 @@ async function scoreTrajectoryGroup(trajectories) {
 }
 
 // index.ts
+var pendingTrajectoryStepByReplyId = new Map;
+var pendingTrajectoryEndTargetByStepId = new Map;
 var trajectoryLoggerPlugin = {
   name: "@elizaos/plugin-trajectory-logger",
   description: "Captures and persists complete agent interaction trajectories for debugging, analysis, and RL training. " + "Records LLM calls, provider accesses, actions, environment state, and computes rewards.",
   dependencies: ["@elizaos/plugin-sql"],
-  services: [TrajectoryLoggerService]
+  services: [TrajectoryLoggerService],
+  events: {
+    MESSAGE_RECEIVED: [
+      async (payload) => {
+        const { runtime, message, source } = payload;
+        if (!message || !runtime)
+          return;
+        if (!message.metadata) {
+          message.metadata = {
+            type: "message"
+          };
+        }
+        const meta = message.metadata;
+        const logger5 = runtime.getService("trajectory_logger");
+        if (!logger5)
+          return;
+        let trajectoryStepId = crypto.randomUUID();
+        meta.trajectoryStepId = trajectoryStepId;
+        try {
+          const trajectoryId = await logger5.startTrajectory(runtime.agentId, {
+            source: source ?? meta.source ?? "chat",
+            metadata: {
+              roomId: message.roomId,
+              entityId: message.entityId,
+              messageId: message.id,
+              channelType: meta.channelType ?? message.content?.channelType,
+              conversationId: meta.sessionKey
+            }
+          });
+          const normalizedTrajectoryId = typeof trajectoryId === "string" && trajectoryId.trim().length > 0 ? trajectoryId : null;
+          if (normalizedTrajectoryId) {
+            const runtimeStepId = logger5.startStep(normalizedTrajectoryId, {
+              timestamp: Date.now(),
+              agentBalance: 0,
+              agentPoints: 0,
+              agentPnL: 0,
+              openPositions: 0
+            });
+            const normalizedStepId = typeof runtimeStepId === "string" && runtimeStepId.trim().length > 0 ? runtimeStepId : trajectoryStepId;
+            trajectoryStepId = normalizedStepId;
+            meta.trajectoryStepId = trajectoryStepId;
+            pendingTrajectoryEndTargetByStepId.set(trajectoryStepId, normalizedTrajectoryId);
+          } else {}
+          if (message.id) {
+            const replyId = createUniqueUuid(runtime, message.id);
+            pendingTrajectoryStepByReplyId.set(replyId, trajectoryStepId);
+          }
+        } catch (err) {
+          runtime.logger?.warn({
+            err,
+            src: "plugin-trajectory-logger",
+            roomId: message.roomId
+          }, "Failed to start trajectory logging");
+        }
+      }
+    ],
+    MESSAGE_SENT: [
+      async (payload) => {
+        const { runtime, message } = payload;
+        if (!message || !runtime)
+          return;
+        const meta = message.metadata;
+        const inReplyTo = typeof message.content === "object" && message.content !== null && "inReplyTo" in message.content && typeof message.content.inReplyTo === "string" ? message.content.inReplyTo : undefined;
+        let trajectoryStepId = meta?.trajectoryStepId;
+        if (!trajectoryStepId && inReplyTo) {
+          trajectoryStepId = pendingTrajectoryStepByReplyId.get(inReplyTo);
+        }
+        if (!trajectoryStepId)
+          return;
+        const logger5 = runtime.getService("trajectory_logger");
+        if (!logger5)
+          return;
+        try {
+          const endTarget = pendingTrajectoryEndTargetByStepId.get(trajectoryStepId) ?? trajectoryStepId;
+          await logger5.endTrajectory(endTarget, "completed");
+        } catch (err) {
+          runtime.logger?.warn({
+            err,
+            src: "plugin-trajectory-logger",
+            trajectoryStepId
+          }, "Failed to end trajectory logging");
+        }
+        if (inReplyTo) {
+          pendingTrajectoryStepByReplyId.delete(inReplyTo);
+        }
+        pendingTrajectoryEndTargetByStepId.delete(trajectoryStepId);
+      }
+    ]
+  }
 };
 var typescript_default = trajectoryLoggerPlugin;
 export {
@@ -3492,4 +3803,4 @@ export {
   RewardService
 };
 
-//# debugId=29EE15F427B068BA64756E2164756E21
+//# debugId=CDC29521B99CF06F64756E2164756E21