@elizaos/plugin-trajectory-logger 2.0.0-alpha.20 → 2.0.0-alpha.21

package/dist/node/index.node.js

@@ -14,7 +14,7 @@ var __export = (target, all) => {
 };
 var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
 
-// ../../../../node_modules/drizzle-orm/entity.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/entity.js
 function is(value, type) {
   if (!value || typeof value !== "object") {
     return false;
@@ -42,7 +42,7 @@ var init_entity = __esm(() => {
   hasOwnEntityKind = Symbol.for("drizzle:hasOwnEntityKind");
 });
 
-// ../../../../node_modules/drizzle-orm/column.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/column.js
 var Column;
 var init_column = __esm(() => {
   init_entity();
@@ -96,7 +96,7 @@ var init_column = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/column-builder.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/column-builder.js
 var ColumnBuilder;
 var init_column_builder = __esm(() => {
   init_entity();
@@ -156,19 +156,19 @@ var init_column_builder = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/table.utils.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/table.utils.js
 var TableName;
 var init_table_utils = __esm(() => {
   TableName = Symbol.for("drizzle:Name");
 });
 
-// ../../../../node_modules/drizzle-orm/tracing-utils.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/tracing-utils.js
 function iife(fn, ...args) {
   return fn(...args);
 }
 var init_tracing_utils = () => {};
 
-// ../../../../node_modules/drizzle-orm/pg-core/unique-constraint.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/pg-core/unique-constraint.js
 function uniqueKeyName(table, columns) {
   return `${table[TableName]}_${columns.join("_")}_unique`;
 }
@@ -176,7 +176,7 @@ var init_unique_constraint = __esm(() => {
   init_table_utils();
 });
 
-// ../../../../node_modules/drizzle-orm/pg-core/columns/common.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/pg-core/columns/common.js
 var PgColumn, ExtraConfigColumn;
 var init_common = __esm(() => {
   init_column();
@@ -230,7 +230,7 @@ var init_common = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/pg-core/columns/enum.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/pg-core/columns/enum.js
 function isPgEnum(obj) {
   return !!obj && typeof obj === "function" && isPgEnumSym in obj && obj[isPgEnumSym] === true;
 }
@@ -265,7 +265,7 @@ var init_enum = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/subquery.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/subquery.js
 var Subquery, WithSubquery;
 var init_subquery = __esm(() => {
   init_entity();
@@ -287,11 +287,11 @@ var init_subquery = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/version.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/version.js
 var version = "0.45.1";
 var init_version = () => {};
 
-// ../../../../node_modules/drizzle-orm/tracing.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/tracing.js
 var otel, rawTracer, tracer;
 var init_tracing = __esm(() => {
   init_tracing_utils();
@@ -321,13 +321,13 @@ var init_tracing = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/view-common.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/view-common.js
 var ViewBaseConfig;
 var init_view_common = __esm(() => {
   ViewBaseConfig = Symbol.for("drizzle:ViewBaseConfig");
 });
 
-// ../../../../node_modules/drizzle-orm/table.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/table.js
 function isTable(table) {
   return typeof table === "object" && table !== null && IsDrizzleTable in table;
 }
@@ -378,7 +378,7 @@ var init_table = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/sql/sql.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/sql/sql.js
 function isSQLWrapper(value) {
   return value !== null && value !== undefined && typeof value.getSQL === "function";
 }
@@ -780,7 +780,7 @@ var init_sql = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/alias.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/alias.js
 function aliasedTable(table, tableAlias) {
   return new Proxy(table, new TableAliasProxyHandler(tableAlias, false));
 }
@@ -881,7 +881,7 @@ var init_alias = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/errors.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/errors.js
 var DrizzleError, DrizzleQueryError, TransactionRollbackError;
 var init_errors = __esm(() => {
   init_entity();
@@ -913,7 +913,7 @@ params: ${params}`);
   };
 });
 
-// ../../../../node_modules/drizzle-orm/logger.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/logger.js
 var ConsoleLogWriter, DefaultLogger, NoopLogger;
 var init_logger = __esm(() => {
   init_entity();
@@ -947,7 +947,7 @@ var init_logger = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/query-promise.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/query-promise.js
 var QueryPromise;
 var init_query_promise = __esm(() => {
   init_entity();
@@ -972,7 +972,7 @@ var init_query_promise = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/utils.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/utils.js
 function mapResultRow(columns, row, joinsNotNullableMap) {
   const nullifyMap = {};
   const result = columns.reduce((result2, { path, field }, columnIndex) => {
@@ -1138,7 +1138,7 @@ var init_utils = __esm(() => {
   textDecoder = typeof TextDecoder === "undefined" ? null : new TextDecoder;
 });
 
-// ../../../../node_modules/drizzle-orm/pg-core/table.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/pg-core/table.js
 var InlineForeignKeys, EnableRLS, PgTable;
 var init_table2 = __esm(() => {
   init_entity();
@@ -1158,7 +1158,7 @@ var init_table2 = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/pg-core/primary-keys.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/pg-core/primary-keys.js
 var PrimaryKeyBuilder, PrimaryKey;
 var init_primary_keys = __esm(() => {
   init_entity();
@@ -1190,7 +1190,7 @@ var init_primary_keys = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/sql/expressions/conditions.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/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);
@@ -1326,7 +1326,7 @@ var init_conditions = __esm(() => {
   init_sql();
 });
 
-// ../../../../node_modules/drizzle-orm/sql/expressions/select.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/sql/expressions/select.js
 function asc(column) {
   return sql`${column} asc`;
 }
@@ -1337,13 +1337,13 @@ var init_select = __esm(() => {
   init_sql();
 });
 
-// ../../../../node_modules/drizzle-orm/sql/expressions/index.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/sql/expressions/index.js
 var init_expressions = __esm(() => {
   init_conditions();
   init_select();
 });
 
-// ../../../../node_modules/drizzle-orm/relations.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/relations.js
 function getOperators() {
   return {
     and,
@@ -1575,7 +1575,7 @@ var init_relations = __esm(() => {
   };
 });
 
-// ../../../../node_modules/drizzle-orm/sql/functions/aggregate.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/sql/functions/aggregate.js
 function count(expression) {
   return sql`count(${expression || sql.raw("*")})`.mapWith(Number);
 }
@@ -1606,7 +1606,7 @@ var init_aggregate = __esm(() => {
   init_sql();
 });
 
-// ../../../../node_modules/drizzle-orm/sql/functions/vector.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/sql/functions/vector.js
 function toSql(value) {
   return JSON.stringify(value);
 }
@@ -1650,20 +1650,20 @@ var init_vector = __esm(() => {
   init_sql();
 });
 
-// ../../../../node_modules/drizzle-orm/sql/functions/index.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/sql/functions/index.js
 var init_functions = __esm(() => {
   init_aggregate();
   init_vector();
 });
 
-// ../../../../node_modules/drizzle-orm/sql/index.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/sql/index.js
 var init_sql2 = __esm(() => {
   init_expressions();
   init_functions();
   init_sql();
 });
 
-// ../../../../node_modules/drizzle-orm/index.js
+// ../../../node_modules/.bun/drizzle-orm@0.45.1+6e0ea2f0f122d214/node_modules/drizzle-orm/index.js
 var exports_drizzle_orm = {};
 __export(exports_drizzle_orm, {
   textDecoder: () => textDecoder,
@@ -1808,8 +1808,24 @@ import {
 import crypto from "node:crypto";
 
 // TrajectoryLoggerService.ts
-import { logger as logger2, Service } from "@elizaos/core";
+import {
+  logger as logger2,
+  Service,
+  getTrajectoryContext,
+  runWithTrajectoryContext,
+  setTrajectoryContextManager
+} from "@elizaos/core";
 import { v4 as uuidv4 } from "uuid";
+import { AsyncLocalStorage } from "node:async_hooks";
+var trajectoryStorage = new AsyncLocalStorage;
+setTrajectoryContextManager({
+  run(context, fn) {
+    return trajectoryStorage.run(context, fn);
+  },
+  active() {
+    return trajectoryStorage.getStore();
+  }
+});
 function asNumber(value) {
   if (typeof value === "number" && Number.isFinite(value))
     return value;
@@ -1865,11 +1881,49 @@ class TrajectoryLoggerService extends Service {
     return TrajectoryLoggerService.serviceType;
   }
   capabilityDescription = "Captures and persists LLM calls, provider accesses, and full trajectories for debugging, analysis, and RL training";
+  static resolveFromRuntime(runtime) {
+    const first = runtime.getService(TrajectoryLoggerService.serviceType);
+    if (first && typeof first.startTrajectory === "function") {
+      return first;
+    }
+    const all = typeof runtime.getServicesByType === "function" ? runtime.getServicesByType(TrajectoryLoggerService.serviceType) : [];
+    for (const svc of all) {
+      if (typeof svc.startTrajectory === "function") {
+        return svc;
+      }
+    }
+    return null;
+  }
+  static async waitForService(runtime, timeoutMs = 1e4) {
+    const deadline = Date.now() + timeoutMs;
+    while (Date.now() < deadline) {
+      const svc = TrajectoryLoggerService.resolveFromRuntime(runtime);
+      if (svc)
+        return svc;
+      await new Promise((r) => setTimeout(r, 50));
+    }
+    return null;
+  }
   enabled = true;
   initialized = false;
   activeStepIds = new Map;
   stepToTrajectory = new Map;
   writeQueues = new Map;
+  exposeBoundMethods() {
+    const service = this;
+    service.startTrajectory = this.startTrajectory.bind(this);
+    service.endTrajectory = this.endTrajectory.bind(this);
+    service.startStep = this.startStep.bind(this);
+    service.getCurrentStepId = this.getCurrentStepId.bind(this);
+    service.completeStep = this.completeStep.bind(this);
+    service.logLLMCall = this.logLLMCall.bind(this);
+    service.logProviderAccess = this.logProviderAccess.bind(this);
+    service.logProviderAccessByTrajectoryId = this.logProviderAccessByTrajectoryId.bind(this);
+    service.isEnabled = this.isEnabled.bind(this);
+    service.listTrajectories = this.listTrajectories.bind(this);
+    service.getTrajectoryDetail = this.getTrajectoryDetail.bind(this);
+    service.flushWriteQueue = this.flushWriteQueue.bind(this);
+  }
   static async start(runtime) {
     const service = new this(runtime);
     await service.initialize();
@@ -1904,6 +1958,7 @@ class TrajectoryLoggerService extends Service {
   async initialize() {
     if (this.initialized)
       return;
+    this.exposeBoundMethods();
     const runtime = this.runtime;
     if (!runtime?.adapter) {
       logger2.warn("[trajectory-logger] No runtime adapter available, skipping initialization");
@@ -1911,6 +1966,29 @@ class TrajectoryLoggerService extends Service {
     }
     await this.ensureTablesExist();
     await this.backfillTrajectoriesMissingLlmCalls();
+    const originalUseModel = runtime.useModel;
+    if (typeof originalUseModel === "function") {
+      const patchedUseModel = async function(...args) {
+        const context = getTrajectoryContext();
+        if (context?.trajectoryStepId) {
+          return originalUseModel.call(this, ...args);
+        }
+        const trajLogger = TrajectoryLoggerService.resolveFromRuntime(runtime);
+        if (!trajLogger || !trajLogger.isEnabled()) {
+          return originalUseModel.call(this, ...args);
+        }
+        const stepId = await trajLogger.startTrajectory(runtime.agentId, {
+          source: "chat-orphan",
+          metadata: { autoPatch: true }
+        });
+        try {
+          return await runWithTrajectoryContext({ trajectoryStepId: stepId }, () => originalUseModel.call(this, ...args));
+        } finally {
+          await trajLogger.endTrajectory(stepId, "completed");
+        }
+      };
+      runtime.useModel = patchedUseModel;
+    }
     this.initialized = true;
     logger2.info("[trajectory-logger] Trajectory logger service initialized");
   }
@@ -2178,7 +2256,15 @@ class TrajectoryLoggerService extends Service {
       completionTokens: 0,
       latencyMs: 0,
       purpose: "other",
-      actionType: "TRAJECTORY_FALLBACK"
+      actionType: "TRAJECTORY_FALLBACK",
+      stepType: "synthetic",
+      tags: [
+        "llm",
+        "step:synthetic",
+        "purpose:other",
+        "action:trajectory_fallback",
+        "synthetic"
+      ]
     };
     step.llmCalls.push(syntheticCall);
     if (!step.metadata)
@@ -2214,6 +2300,12 @@ class TrajectoryLoggerService extends Service {
       totalCompletionTokens
     };
   }
+  async flushWriteQueue(trajectoryId) {
+    const pending = this.writeQueues.get(trajectoryId);
+    if (pending) {
+      await pending.catch(() => {});
+    }
+  }
   async withTrajectoryWriteLock(trajectoryId, task) {
     const previous = this.writeQueues.get(trajectoryId) ?? Promise.resolve();
     const next = previous.catch(() => {}).then(task);
@@ -2364,41 +2456,49 @@ class TrajectoryLoggerService extends Service {
   logLlmCall(params) {
     if (!this.enabled)
       return;
-    (async () => {
-      const trajectoryId = await this.resolveTrajectoryId(params.stepId);
-      if (!trajectoryId) {
-        logger2.debug({ stepId: params.stepId }, "[trajectory-logger] No trajectory mapping for LLM call");
-        return;
-      }
-      await this.withTrajectoryWriteLock(trajectoryId, async () => {
-        const trajectory = await this.getTrajectoryById(trajectoryId);
-        if (!trajectory)
+    const trajectoryId = this.stepToTrajectory.get(params.stepId);
+    if (!trajectoryId) {
+      (async () => {
+        const resolved = await this.resolveTrajectoryId(params.stepId);
+        if (!resolved)
           return;
-        const step = await this.ensureStepExists(trajectory, params.stepId);
-        const llmCall = {
-          callId: uuidv4(),
-          timestamp: Date.now(),
-          model: params.model,
-          modelVersion: params.modelVersion,
-          systemPrompt: params.systemPrompt,
-          userPrompt: params.userPrompt,
-          response: params.response,
-          reasoning: params.reasoning,
-          temperature: params.temperature,
-          maxTokens: params.maxTokens,
-          purpose: this.normalizePurpose(params.purpose),
-          actionType: params.actionType,
-          promptTokens: params.promptTokens,
-          completionTokens: params.completionTokens,
-          latencyMs: params.latencyMs
-        };
-        step.llmCalls.push(llmCall);
-        await this.persistTrajectory(trajectoryId, trajectory, "active");
+        await this._persistLlmCall(resolved, params);
+      })().catch((err) => {
+        logger2.warn({ err, stepId: params.stepId }, "[trajectory-logger] Failed to persist LLM call (fallback)");
       });
-    })().catch((err) => {
+      return;
+    }
+    this._persistLlmCall(trajectoryId, params).catch((err) => {
       logger2.warn({ err, stepId: params.stepId }, "[trajectory-logger] Failed to persist LLM call");
     });
   }
+  async _persistLlmCall(trajectoryId, params) {
+    await this.withTrajectoryWriteLock(trajectoryId, async () => {
+      const trajectory = await this.getTrajectoryById(trajectoryId);
+      if (!trajectory)
+        return;
+      const step = await this.ensureStepExists(trajectory, params.stepId);
+      const llmCall = {
+        callId: uuidv4(),
+        timestamp: Date.now(),
+        model: params.model,
+        modelVersion: params.modelVersion,
+        systemPrompt: params.systemPrompt,
+        userPrompt: params.userPrompt,
+        response: params.response,
+        reasoning: params.reasoning,
+        temperature: params.temperature,
+        maxTokens: params.maxTokens,
+        purpose: this.normalizePurpose(params.purpose),
+        actionType: params.actionType,
+        promptTokens: params.promptTokens,
+        completionTokens: params.completionTokens,
+        latencyMs: params.latencyMs
+      };
+      step.llmCalls.push(llmCall);
+      await this.persistTrajectory(trajectoryId, trajectory, "active");
+    });
+  }
   logLLMCall(stepId, details) {
     this.logLlmCall({
       stepId,
@@ -2632,21 +2732,28 @@ class TrajectoryLoggerService extends Service {
         return;
       }
       const now = Date.now();
-      trajectory.endTime = now;
-      trajectory.durationMs = now - trajectory.startTime;
-      trajectory.metrics.finalStatus = status;
-      trajectory.metrics.episodeLength = trajectory.steps.length;
+      const durationMs = now - trajectory.startTime;
+      const metrics = { ...trajectory.metrics, finalStatus: status, episodeLength: trajectory.steps.length };
       if (finalMetrics) {
-        trajectory.metrics = {
-          ...trajectory.metrics,
-          ...finalMetrics
-        };
+        Object.assign(metrics, finalMetrics);
       }
       await this.markAllStepsInactive(trajectoryId);
       this.activeStepIds.delete(trajectoryId);
-      await this.persistTrajectory(trajectoryId, trajectory, status);
+      const updatedAtIso = new Date().toISOString();
+      await this.executeRawSql(`
+				UPDATE trajectories SET
+					status = ${sqlLiteral(status)},
+					end_time = ${now},
+					duration_ms = ${durationMs},
+					metrics_json = ${sqlLiteral(metrics)},
+					updated_at = ${sqlLiteral(updatedAtIso)}
+				WHERE id = ${sqlLiteral(trajectoryId)}
+			`);
     });
-    for (const [stepId, mappedTrajectoryId] of this.stepToTrajectory.entries()) {
+    for (const [
+      stepId,
+      mappedTrajectoryId
+    ] of this.stepToTrajectory.entries()) {
       if (mappedTrajectoryId === trajectoryId) {
         this.stepToTrajectory.delete(stepId);
       }
@@ -3067,7 +3174,7 @@ var trajectoryContexts = new WeakMap;
 function setTrajectoryContext(runtime, trajectoryId, trajectoryLogger) {
   trajectoryContexts.set(runtime, { trajectoryId, logger: trajectoryLogger });
 }
-function getTrajectoryContext(runtime) {
+function getTrajectoryContext2(runtime) {
   return trajectoryContexts.get(runtime) || null;
 }
 function clearTrajectoryContext(runtime) {
@@ -3078,7 +3185,7 @@ function wrapActionWithLogging(action, _trajectoryLogger) {
   return {
     ...action,
     handler: async (runtime, message, state, options, callback) => {
-      const context = getTrajectoryContext(runtime);
+      const context = getTrajectoryContext2(runtime);
       if (!context) {
         const result = await originalHandler(runtime, message, state, options, callback);
         return result ?? undefined;
@@ -3185,7 +3292,7 @@ function wrapProviderWithLogging(provider, _trajectoryLogger) {
   return {
     ...provider,
     get: async (runtime, message, state) => {
-      const context = getTrajectoryContext(runtime);
+      const context = getTrajectoryContext2(runtime);
       if (!context) {
         return originalGet?.(runtime, message, state) || { text: "" };
       }
@@ -3520,6 +3627,7 @@ async function recomputeTrajectoryRewards(_trajectoryIds) {}
 import { logger as logger4 } from "@elizaos/core";
 async function startAutonomousTick(trajectoryLogger, context) {
   const trajectoryId = await trajectoryLogger.startTrajectory(context.agentId, {
+    source: context.source,
     scenarioId: context.scenarioId,
     episodeId: context.episodeId,
     batchId: context.batchId,
@@ -3537,6 +3645,10 @@ async function startAutonomousTick(trajectoryLogger, context) {
   return trajectoryId;
 }
 async function endAutonomousTick(trajectoryLogger, trajectoryId, status = "completed", finalMetrics) {
+  const flush = trajectoryLogger.flushWriteQueue;
+  if (typeof flush === "function") {
+    await flush(trajectoryId);
+  }
   await trajectoryLogger.endTrajectory(trajectoryId, status, finalMetrics);
   logger4.info({ trajectoryId, status }, "Ended autonomous tick trajectory");
 }
@@ -3689,7 +3801,7 @@ var trajectoryLoggerPlugin = {
           };
         }
         const meta = message.metadata;
-        const logger5 = runtime.getService("trajectory_logger");
+        const logger5 = TrajectoryLoggerService.resolveFromRuntime(runtime);
         if (!logger5)
           return;
         let trajectoryStepId = crypto.randomUUID();
@@ -3745,7 +3857,7 @@ var trajectoryLoggerPlugin = {
         }
         if (!trajectoryStepId)
           return;
-        const logger5 = runtime.getService("trajectory_logger");
+        const logger5 = TrajectoryLoggerService.resolveFromRuntime(runtime);
         if (!logger5)
           return;
         try {
@@ -3790,7 +3902,7 @@ export {
   logProviderAccess,
   logLLMCallFromAction,
   groupTrajectories,
-  getTrajectoryContext,
+  getTrajectoryContext2 as getTrajectoryContext,
   extractSharedPrefix,
   exportToHuggingFace,
   exportGroupedForGRPO,
@@ -3807,4 +3919,4 @@ export {
   RewardService
 };
 
-//# debugId=621E07B14B9A695B64756E2164756E21
+//# debugId=82D6338D5E8EB6DA64756E2164756E21