@elizaos/plugin-trajectory-logger 2.0.0-alpha.3 → 2.0.0-alpha.5

package/README.md

@@ -0,0 +1,26 @@
+# @elizaos/plugin-trajectory-logger
+
+Trajectory logger plugin for ElizaOS.
+
+This plugin captures rich agent interaction trajectories and provides utilities to:
+
+- Wrap actions/providers with logging hooks
+- Record LLM calls, provider accesses, and action attempts
+- Convert trajectories to ART-compatible message format (OpenPipe ART / GRPO)
+
+## Install
+
+```bash
+npm install @elizaos/plugin-trajectory-logger
+```
+
+## Usage
+
+```ts
+import { trajectoryLoggerPlugin } from "@elizaos/plugin-trajectory-logger";
+
+export const character = {
+  plugins: [trajectoryLoggerPlugin],
+};
+```
+

package/package.json

@@ -1,11 +1,11 @@
 {
   "name": "@elizaos/plugin-trajectory-logger",
+  "version": "2.0.0-alpha.5",
   "description": "Trajectory logging plugin for ElizaOS - capture agent interaction trajectories for RL training",
-  "version": "2.0.0-alpha.3",
   "type": "module",
-  "main": "dist/node/index.node.js",
-  "module": "dist/node/index.node.js",
-  "types": "dist/index.d.ts",
+  "main": "typescript/dist/node/index.node.js",
+  "module": "typescript/dist/node/index.node.js",
+  "types": "typescript/dist/index.d.ts",
   "packageType": "plugin",
   "platform": "node",
   "license": "MIT",
@@ -23,29 +23,34 @@
     "url": "https://github.com/elizaos/elizaos"
   },
   "homepage": "https://elizaos.ai",
+  "bugs": {
+    "url": "https://github.com/elizaos/elizaos/issues"
+  },
   "exports": {
     "./package.json": "./package.json",
     ".": {
-      "types": "./dist/index.d.ts",
+      "types": "./typescript/dist/index.d.ts",
       "node": {
-        "types": "./dist/index.d.ts",
-        "import": "./dist/node/index.node.js",
-        "default": "./dist/node/index.node.js"
+        "types": "./typescript/dist/node/index.d.ts",
+        "import": "./typescript/dist/node/index.node.js",
+        "default": "./typescript/dist/node/index.node.js"
       },
       "bun": {
-        "types": "./dist/index.d.ts",
-        "default": "./dist/node/index.node.js"
+        "types": "./typescript/dist/node/index.d.ts",
+        "default": "./typescript/dist/node/index.node.js"
       },
-      "default": "./dist/node/index.node.js"
+      "default": "./typescript/dist/node/index.node.js"
     }
   },
   "files": [
-    "dist",
+    "typescript/dist",
+    "python",
+    "rust",
     "README.md"
   ],
   "sideEffects": false,
   "dependencies": {
-    "@elizaos/core": "2.0.0-alpha.3",
+    "@elizaos/core": "next",
     "uuid": "^13.0.0"
   },
   "devDependencies": {
@@ -56,34 +61,30 @@
     "typescript": "^5.9.3"
   },
   "scripts": {
-    "dev": "bun --hot build.ts",
-    "clean": "rm -rf dist .turbo node_modules tsconfig.tsbuildinfo",
-    "format": "bunx @biomejs/biome format --write .",
-    "format:check": "bunx @biomejs/biome format .",
-    "typecheck": "tsc --noEmit -p tsconfig.json",
-    "test": "vitest run tests/",
-    "test:unit": "vitest run tests/",
-    "test:watch": "vitest",
-    "lint": "bunx @biomejs/biome check --write --unsafe .",
-    "lint:check": "bunx @biomejs/biome check .",
-    "build": "bun run build.ts",
-    "build:ts": "bun run build.ts"
-  },
-  "publishConfig": {
-    "access": "public"
+    "build": "bun run build:ts && bun run build:rust && bun run build:python",
+    "build:ts": "cd typescript && bun run build.ts",
+    "build:rust": "test -d rust && cd rust && cargo build --release || echo 'Rust build skipped - no rust directory'",
+    "build:python": "test -n \"$SKIP_PYTHON_BUILD\" && echo 'Python build skipped (SKIP_PYTHON_BUILD set)' || (test -d python && cd python && (timeout 120 python3 -m build 2>/dev/null || timeout 120 pyproject-build 2>/dev/null) || echo 'Python build skipped or timed out')",
+    "dev": "cd typescript && bun --hot build.ts",
+    "test": "bun run test:ts && bun run test:rust && bun run test:python",
+    "test:ts": "cd typescript && vitest run || echo 'TypeScript tests skipped - no tests found'",
+    "test:rust": "test -d rust && cd rust && cargo test || echo 'Rust tests skipped'",
+    "test:python": "test -d python && cd python && pytest -p no:anchorpy --asyncio-mode=auto || echo 'Python tests skipped'",
+    "typecheck": "tsc --noEmit -p typescript/tsconfig.json",
+    "lint": "bunx @biomejs/biome check --write ./typescript",
+    "lint:check": "bunx @biomejs/biome check ./typescript",
+    "lint:rust": "test -d rust && cd rust && cargo clippy --all-targets --fix --allow-dirty --allow-staged -- -D warnings && cargo fmt || echo 'Rust lint skipped'",
+    "lint:python": "test -d python && cd python && ruff check --fix . && ruff format . || echo 'Python lint skipped'",
+    "clean": "rm -rf typescript/dist python/dist rust/target .turbo node_modules",
+    "format": "bunx @biomejs/biome format --write ./typescript",
+    "format:check": "bunx @biomejs/biome format ./typescript",
+    "typecheck:python": "test -d python && cd python && mypy . --ignore-missing-imports || echo 'Python typecheck skipped'",
+    "typecheck:rust": "test -d rust && cd rust && cargo check || echo 'Rust typecheck skipped'"
   },
   "agentConfig": {
     "pluginType": "elizaos:plugin:1.0.0",
     "pluginParameters": {},
     "description": "Collects complete agent interaction trajectories for RL training."
   },
-  "milaidy": {
-    "platforms": [
-      "node"
-    ],
-    "runtime": "node",
-    "platformDetails": {
-      "node": "Node.js only (platform: node)"
-    }
-  }
+  "gitHead": "401e0d2502fea9d6f459aa80faad1369798e9600"
 }

package/python/README.md

@@ -0,0 +1,10 @@
+# elizaOS Trajectory Logger Plugin (Python)
+
+Python implementation of the elizaOS Trajectory Logger plugin.
+
+This package provides:
+
+- In-memory trajectory logging (`TrajectoryLoggerService`)
+- ART / GRPO formatting helpers (OpenPipe ART message format)
+- Simple heuristic reward scoring
+

package/python/elizaos_plugin_discord/generated/specs/__init__.py

@@ -0,0 +1 @@
+"""Auto-generated module package."""

package/python/elizaos_plugin_discord/generated/specs/specs.py

@@ -0,0 +1,77 @@
+"""
+Auto-generated canonical action/provider/evaluator docs for plugin-trajectory-logger.
+DO NOT EDIT - Generated from prompts/specs/**.
+"""
+
+from __future__ import annotations
+
+import json
+from typing import TypedDict
+
+
+class ActionDoc(TypedDict, total=False):
+    name: str
+    description: str
+    similes: list[str]
+    parameters: list[object]
+    examples: list[list[object]]
+
+
+class ProviderDoc(TypedDict, total=False):
+    name: str
+    description: str
+    position: int
+    dynamic: bool
+
+
+class EvaluatorDoc(TypedDict, total=False):
+    name: str
+    description: str
+    similes: list[str]
+    alwaysRun: bool
+    examples: list[object]
+
+
+_CORE_ACTION_DOCS_JSON = """{
+  "version": "1.0.0",
+  "actions": []
+}"""
+_ALL_ACTION_DOCS_JSON = """{
+  "version": "1.0.0",
+  "actions": []
+}"""
+_CORE_PROVIDER_DOCS_JSON = """{
+  "version": "1.0.0",
+  "providers": []
+}"""
+_ALL_PROVIDER_DOCS_JSON = """{
+  "version": "1.0.0",
+  "providers": []
+}"""
+_CORE_EVALUATOR_DOCS_JSON = """{
+  "version": "1.0.0",
+  "evaluators": []
+}"""
+_ALL_EVALUATOR_DOCS_JSON = """{
+  "version": "1.0.0",
+  "evaluators": []
+}"""
+
+core_action_docs: dict[str, object] = json.loads(_CORE_ACTION_DOCS_JSON)
+all_action_docs: dict[str, object] = json.loads(_ALL_ACTION_DOCS_JSON)
+core_provider_docs: dict[str, object] = json.loads(_CORE_PROVIDER_DOCS_JSON)
+all_provider_docs: dict[str, object] = json.loads(_ALL_PROVIDER_DOCS_JSON)
+core_evaluator_docs: dict[str, object] = json.loads(_CORE_EVALUATOR_DOCS_JSON)
+all_evaluator_docs: dict[str, object] = json.loads(_ALL_EVALUATOR_DOCS_JSON)
+
+__all__ = [
+    "ActionDoc",
+    "ProviderDoc",
+    "EvaluatorDoc",
+    "core_action_docs",
+    "all_action_docs",
+    "core_provider_docs",
+    "all_provider_docs",
+    "core_evaluator_docs",
+    "all_evaluator_docs",
+]

package/python/elizaos_plugin_trajectory_logger/__init__.py

@@ -0,0 +1,98 @@
+"""
+elizaOS Trajectory Logger Plugin - capture and export agent trajectories.
+
+This package provides:
+- In-memory trajectory logging service
+- ART / GRPO formatting helpers
+- Simple heuristic reward scoring
+"""
+
+from elizaos_plugin_trajectory_logger.art_format import (
+    extract_shared_prefix,
+    group_trajectories,
+    prepare_for_ruler,
+    remove_shared_prefix,
+    to_art_messages,
+    to_art_trajectory,
+    validate_art_compatibility,
+)
+from elizaos_plugin_trajectory_logger.eliza_service import (
+    TRAJECTORY_LOGGER_SERVICE_TYPE,
+    TrajectoryExportConfig,
+    TrajectoryLoggerElizaService,
+)
+from elizaos_plugin_trajectory_logger.export import (
+    ExportOptions,
+    ExportResult,
+    export_for_openpipe_art,
+    export_grouped_for_grpo,
+)
+from elizaos_plugin_trajectory_logger.plugin import get_trajectory_logger_plugin
+from elizaos_plugin_trajectory_logger.reward_service import RewardService, create_reward_service
+from elizaos_plugin_trajectory_logger.runtime_service import (
+    TrajectoryExportConfig as TrajectoryRuntimeExportConfig,
+)
+from elizaos_plugin_trajectory_logger.runtime_service import (
+    TrajectoryLoggerRuntimeService,
+)
+from elizaos_plugin_trajectory_logger.service import TrajectoryLoggerService
+from elizaos_plugin_trajectory_logger.types import (
+    ActionAttempt,
+    ARTTrajectory,
+    ChatMessage,
+    EnvironmentState,
+    JsonValue,
+    LLMCall,
+    ProviderAccess,
+    RewardComponents,
+    TrainingBatch,
+    Trajectory,
+    TrajectoryGroup,
+    TrajectoryMetrics,
+    TrajectoryStep,
+)
+
+__all__ = [
+    # Service
+    "TrajectoryLoggerService",
+    "get_trajectory_logger_plugin",
+    "TrajectoryLoggerElizaService",
+    "TRAJECTORY_LOGGER_SERVICE_TYPE",
+    "TrajectoryExportConfig",
+    "TrajectoryLoggerRuntimeService",
+    "TrajectoryRuntimeExportConfig",
+    # Types
+    "JsonValue",
+    "LLMCall",
+    "ProviderAccess",
+    "ActionAttempt",
+    "EnvironmentState",
+    "TrajectoryStep",
+    "RewardComponents",
+    "TrajectoryMetrics",
+    "Trajectory",
+    "ChatMessage",
+    "ARTTrajectory",
+    "TrajectoryGroup",
+    "TrainingBatch",
+    # ART helpers
+    "to_art_messages",
+    "to_art_trajectory",
+    "group_trajectories",
+    "extract_shared_prefix",
+    "remove_shared_prefix",
+    "prepare_for_ruler",
+    "validate_art_compatibility",
+    # Rewards
+    "RewardService",
+    "create_reward_service",
+    # Export
+    "ExportOptions",
+    "ExportResult",
+    "export_for_openpipe_art",
+    "export_grouped_for_grpo",
+]
+
+__version__ = "1.2.0"
+PLUGIN_NAME = "trajectory-logger"
+PLUGIN_DESCRIPTION = "Trajectory logging utilities for elizaOS agents"