@elizaos/plugin-trajectory-logger 2.0.0-alpha.8 → 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;
   }
@@ -2832,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) {
@@ -2886,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),
@@ -3561,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 {
@@ -3608,4 +3803,4 @@ export {
   RewardService
 };
 
-//# debugId=16B8470362B4AC2364756E2164756E21
+//# debugId=CDC29521B99CF06F64756E2164756E21