@elizaos/agent 2.0.0-alpha.98 → 2.0.0-beta.2

package/packages/agent/src/runtime/trajectory-query.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Trajectory query — read operations.
+ *
+ * Handles listing, loading, searching, and filtering trajectories.
+ */
+import type { IAgentRuntime } from "@elizaos/core";
+export declare function loadPersistedTrajectoryRows(runtime: IAgentRuntime, maxRows?: number): Promise<Record<string, unknown>[] | null>;
+//# sourceMappingURL=trajectory-query.d.ts.map

package/packages/agent/src/runtime/trajectory-query.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trajectory-query.d.ts","sourceRoot":"","sources":["../../../../../src/runtime/trajectory-query.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAcnD,wBAAsB,2BAA2B,CAC/C,OAAO,EAAE,aAAa,EACtB,OAAO,SAAO,GACb,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,IAAI,CAAC,CAkB3C"}

package/packages/agent/src/runtime/trajectory-query.js

@@ -0,0 +1,27 @@
+/**
+ * Trajectory query — read operations.
+ *
+ * Handles listing, loading, searching, and filtering trajectories.
+ */
+import { asRecord, ensureTrajectoriesTable, executeRawSql, extractRows, hasRuntimeDb, } from "./trajectory-internals.js";
+// ---------------------------------------------------------------------------
+// Public read API
+// ---------------------------------------------------------------------------
+export async function loadPersistedTrajectoryRows(runtime, maxRows = 5000) {
+    if (!hasRuntimeDb(runtime))
+        return null;
+    const tableReady = await ensureTrajectoriesTable(runtime);
+    if (!tableReady)
+        return [];
+    const safeLimit = Math.max(1, Math.min(10000, Math.trunc(maxRows)));
+    try {
+        const result = await executeRawSql(runtime, `SELECT * FROM trajectories ORDER BY created_at DESC LIMIT ${safeLimit}`);
+        const rows = extractRows(result);
+        return rows
+            .map((row) => asRecord(row))
+            .filter((row) => Boolean(row));
+    }
+    catch {
+        return null;
+    }
+}

package/packages/agent/src/runtime/trajectory-storage.d.ts

@@ -0,0 +1,91 @@
+/**
+ * Trajectory storage — write operations.
+ *
+ * Handles saving, updating, deleting trajectories, installing the database
+ * logger, and the DatabaseTrajectoryLogger service class.
+ */
+import { type IAgentRuntime, Service } from "@elizaos/core";
+import type { Trajectory, TrajectoryExportResult, TrajectoryListOptions, TrajectoryListResult, TrajectoryStatus, TrajectoryStepKind } from "../types/trajectory.ts";
+import { type RuntimeTrajectoryExportOptions } from "./trajectory-export.ts";
+import { type CompleteStepOptions, type StartStepOptions } from "./trajectory-internals.ts";
+export type { CompleteStepOptions, StartStepOptions, } from "./trajectory-internals.ts";
+export declare function installDatabaseTrajectoryLogger(runtime: IAgentRuntime): Promise<void>;
+export declare function startTrajectoryStepInDatabase({ runtime, stepId, source, metadata, }: StartStepOptions): Promise<boolean>;
+/**
+ * Annotate an existing trajectory step with structural metadata (kind
+ * discriminator, script, child step IDs, used skills). Safe to call for any
+ * of the new trajectory step fields; passing `undefined` for a field leaves
+ * the existing value alone, while passing an explicit value overwrites.
+ */
+export declare function annotateTrajectoryStep({ runtime, stepId, kind, script, childSteps, appendChildSteps, usedSkills, }: {
+    runtime: IAgentRuntime;
+    stepId: string;
+    kind?: TrajectoryStepKind;
+    script?: string;
+    /** Replace child steps wholesale. */
+    childSteps?: string[];
+    /** Append the given child step IDs (deduped, order preserved). */
+    appendChildSteps?: string[];
+    usedSkills?: string[];
+}): Promise<boolean>;
+export declare function completeTrajectoryStepInDatabase({ runtime, stepId, status, source, metadata, }: CompleteStepOptions): Promise<boolean>;
+export declare function deletePersistedTrajectoryRows(runtime: IAgentRuntime, trajectoryIds: string[]): Promise<number | null>;
+export declare function clearPersistedTrajectoryRows(runtime: IAgentRuntime): Promise<number | null>;
+/**
+ * Wait for all pending trajectory writes to complete.
+ * Useful for tests to ensure writes are flushed before assertions.
+ */
+export declare function flushTrajectoryWrites(runtime: IAgentRuntime): Promise<void>;
+/**
+ * Database-backed trajectory logger service that implements the full API
+ * expected by trajectory-routes.ts.
+ */
+export declare class DatabaseTrajectoryLogger extends Service {
+    static serviceType: string;
+    capabilityDescription: string;
+    private enabled;
+    /**
+     * Static start method required by @elizaos/core runtime.
+     */
+    static start(runtime: IAgentRuntime): Promise<Service>;
+    initialize(): Promise<void>;
+    stop(): Promise<void>;
+    isEnabled(): boolean;
+    setEnabled(enabled: boolean): void;
+    startTrajectory(stepIdOrAgentId: string, options?: {
+        agentId?: string;
+        roomId?: string;
+        entityId?: string;
+        source?: string;
+        metadata?: Record<string, unknown>;
+    }): Promise<string>;
+    startStep(trajectoryId: string): string;
+    annotateStep(params: {
+        stepId: string;
+        kind?: TrajectoryStepKind;
+        script?: string;
+        childSteps?: string[];
+        appendChildSteps?: string[];
+        usedSkills?: string[];
+    }): Promise<void>;
+    endTrajectory(stepIdOrTrajectoryId: string, status?: TrajectoryStatus): Promise<void>;
+    logLlmCall(params: Record<string, unknown>): void;
+    logProviderAccess(params: Record<string, unknown>): void;
+    getLlmCallLogs(): readonly unknown[];
+    getProviderAccessLogs(): readonly unknown[];
+    listTrajectories(options: TrajectoryListOptions): Promise<TrajectoryListResult>;
+    getTrajectoryDetail(trajectoryId: string): Promise<Trajectory | null>;
+    getStats(): Promise<unknown>;
+    deleteTrajectories(trajectoryIds: string[]): Promise<number>;
+    clearAllTrajectories(): Promise<number>;
+    exportTrajectories(options: RuntimeTrajectoryExportOptions): Promise<TrajectoryExportResult>;
+}
+/**
+ * Create and register a database-backed trajectory logger service on the runtime.
+ */
+export declare function createDatabaseTrajectoryLogger(runtime: IAgentRuntime): DatabaseTrajectoryLogger;
+/**
+ * Archive and then delete trajectories older than `maxAgeDays`.
+ */
+export declare function pruneOldTrajectories(runtime: IAgentRuntime, maxAgeDays?: number): Promise<number | null>;
+//# sourceMappingURL=trajectory-storage.d.ts.map

package/packages/agent/src/runtime/trajectory-storage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trajectory-storage.d.ts","sourceRoot":"","sources":["../../../../../src/runtime/trajectory-storage.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAEL,KAAK,aAAa,EAClB,OAAO,EACR,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EACV,UAAU,EACV,sBAAsB,EAEtB,qBAAqB,EACrB,oBAAoB,EACpB,gBAAgB,EAChB,kBAAkB,EACnB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAGL,KAAK,8BAA8B,EAEpC,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAEL,KAAK,mBAAmB,EA6BxB,KAAK,gBAAgB,EActB,MAAM,2BAA2B,CAAC;AAGnC,YAAY,EACV,mBAAmB,EACnB,gBAAgB,GACjB,MAAM,2BAA2B,CAAC;AAwZnC,wBAAsB,+BAA+B,CACnD,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,IAAI,CAAC,CA+Yf;AAED,wBAAsB,6BAA6B,CAAC,EAClD,OAAO,EACP,MAAM,EACN,MAAM,EACN,QAAQ,GACT,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC,CAkBrC;AAED;;;;;GAKG;AACH,wBAAsB,sBAAsB,CAAC,EAC3C,OAAO,EACP,MAAM,EACN,IAAI,EACJ,MAAM,EACN,UAAU,EACV,gBAAgB,EAChB,UAAU,GACX,EAAE;IACD,OAAO,EAAE,aAAa,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,kBAAkB,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,qCAAqC;IACrC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,kEAAkE;IAClE,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;CACvB,GAAG,OAAO,CAAC,OAAO,CAAC,CAmDnB;AAED,wBAAsB,gCAAgC,CAAC,EACrD,OAAO,EACP,MAAM,EACN,MAAoB,EACpB,MAAM,EACN,QAAQ,GACT,EAAE,mBAAmB,GAAG,OAAO,CAAC,OAAO,CAAC,CAmBxC;AAED,wBAAsB,6BAA6B,CACjD,OAAO,EAAE,aAAa,EACtB,aAAa,EAAE,MAAM,EAAE,GACtB,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CA4BxB;AAED,wBAAsB,4BAA4B,CAChD,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAiBxB;AAED;;;GAGG;AACH,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,IAAI,CAAC,CAaf;AAMD;;;GAGG;AACH,qBAAa,wBAAyB,SAAQ,OAAO;IACnD,MAAM,CAAC,WAAW,SAAkB;IACpC,qBAAqB,SACmD;IAExE,OAAO,CAAC,OAAO,CAA4C;IAE3D;;OAEG;WACU,KAAK,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;IAMtD,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAkB3B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B,SAAS,IAAI,OAAO;IAIpB,UAAU,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAI5B,eAAe,CACnB,eAAe,EAAE,MAAM,EACvB,OAAO,CAAC,EAAE;QACR,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GACA,OAAO,CAAC,MAAM,CAAC;IA0BlB,SAAS,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM;IAIjC,YAAY,CAAC,MAAM,EAAE;QACzB,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,CAAC,EAAE,kBAAkB,CAAC;QAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;QACtB,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;QAC5B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;KACvB,GAAG,OAAO,CAAC,IAAI,CAAC;IAQX,aAAa,CACjB,oBAAoB,EAAE,MAAM,EAC5B,MAAM,GAAE,gBAA8B,GACrC,OAAO,CAAC,IAAI,CAAC;IAsBhB,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI;IAkBjD,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI;IAsBxD,cAAc,IAAI,SAAS,OAAO,EAAE;IAIpC,qBAAqB,IAAI,SAAS,OAAO,EAAE;IAIrC,gBAAgB,CACpB,OAAO,EAAE,qBAAqB,GAC7B,OAAO,CAAC,oBAAoB,CAAC;IAwC1B,mBAAmB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAYrE,QAAQ,IAAI,OAAO,CAAC,OAAO,CAAC;IA+B5B,kBAAkB,CAAC,aAAa,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAQ5D,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;IAKvC,kBAAkB,CACtB,OAAO,EAAE,8BAA8B,GACtC,OAAO,CAAC,sBAAsB,CAAC;CAWnC;AAED;;GAEG;AACH,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,aAAa,GACrB,wBAAwB,CAG1B;AAuCD;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,aAAa,EACtB,UAAU,SAAK,GACd,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAiHxB"}