agentrein 1.0.8 → 1.0.9

package/README.md

@@ -0,0 +1,71 @@
+# AgentRein SDK - Node.js Client
+
+**AgentRein** is an AI Agent Reliability Platform that provides automatic rollback, approval gates, and intent verification for AI workflows. This SDK allows you to easily integrate AgentRein into your Node.js applications.
+
+---
+
+## 📦 Install
+
+```bash
+npm install agentrein
+```
+
+## 🚀 Quick Start
+
+```typescript
+import { AgentRein } from 'agentrein'
+
+const agentrein = new AgentRein({
+  apiKey: process.env.AGENTREIN_API_KEY, // Your Organization API Key
+  intentVerification: true,              // Enable LLM intent check
+})
+
+async function runAgent() {
+  // 1. Create a session with a clear intent
+  const session = await agentrein.newSession({
+    agentId: 'support-agent-01',
+    intent: 'Refund customer cus_123 for order ord_456'
+  })
+
+  try {
+    // 2. Wrap your external API calls
+    const result = await agentrein.call(
+      stripe.refunds.create, // The function to execute
+      session,               // The current session
+      { 
+        type: 'http-delete', 
+        url: 'https://api.stripe.com/v1/refunds/re_123' 
+      },                     // Undo configuration (if rollback is needed)
+      { charge: 'ch_789' }   // Original function arguments
+    )
+
+    console.log('Action executed safely:', result)
+  } catch (error) {
+    // 3. AgentRein automatically triggers rollback on failure
+    console.error('Action failed, rolling back previous steps...', error)
+  }
+}
+```
+
+## 🧠 Key Concepts
+
+### Sessions
+A `Session` groups multiple actions together. If any action in the session fails, AgentRein will attempt to roll back all previous successful actions in that session in reverse order.
+
+### Intent Verification
+By providing an `intent` when creating a session, AgentRein uses an LLM to verify that each subsequent action aligns with the user's original goal. If "drift" is detected, the action can be flagged or blocked.
+
+### Approval Gates
+You can configure rules in the [AgentRein Dashboard](https://agentrein.com/sessions) to require human approval for high-risk actions before they are executed.
+
+## 🛠️ Configuration
+
+| Option | Type | Description |
+| :--- | :--- | :--- |
+| `apiKey` | `string` | **Required**. Your organization's API key. |
+| `baseUrl` | `string` | Optional. Defaults to `https://api.agentrein.com`. |
+| `intentVerification` | `boolean` | Optional. Enable/disable LLM drift detection. |
+
+## 📄 License
+
+MIT

package/dist/cjs/agentreinClient.d.ts

@@ -0,0 +1,64 @@
+import { AgentReinUnavailableError } from './errors';
+export interface AgentReinOptions {
+    apiKey: string;
+    serverUrl?: string;
+    failureMode?: 'open' | 'closed';
+}
+export interface SessionOptions {
+    agentId?: string;
+    intent?: string;
+}
+export interface UndoConfig {
+    type: 'slack-correction' | 'http-delete' | 'none';
+    url?: string;
+    headers?: Record<string, string>;
+    body?: Record<string, unknown>;
+}
+export interface Session {
+    id: string;
+    organizationId: string;
+    agentId: string;
+    intent: string | null;
+    status: string;
+    startedAt: string;
+    endedAt: string | null;
+}
+export { AgentReinUnavailableError };
+export declare class AgentRein {
+    private readonly serverUrl;
+    private readonly apiKey;
+    private readonly failureMode;
+    private token;
+    constructor(options: AgentReinOptions);
+    /**
+     * Obtain a JWT token from the AgentRein server using the API key.
+     * Caches the token for subsequent requests.
+     */
+    private getToken;
+    /**
+     * Build authorization headers for server requests.
+     */
+    private authHeaders;
+    /**
+     * Create a new agent session on the AgentRein server.
+     *
+     * @param options - Session options (agentId + optional intent).
+     *                  Can also pass a plain string for backward compat (agentId).
+     *                  If omitted, a random agentId is generated.
+     */
+    newSession(options?: SessionOptions | string): Promise<Session>;
+    /**
+     * Execute an API call under AgentRein's protection.
+     *
+     * 1. Calls fn(...args)
+     * 2. Logs the action to the AgentRein server (async, non-blocking)
+     * 3. On failure, triggers server-side rollback
+     *
+     * @param fn      - The function to execute
+     * @param session - The active AgentRein session
+     * @param args    - Arguments forwarded to fn
+     */
+    call<T>(fn: Function, session: Session, ...args: any[]): Promise<T>;
+    call<T>(fn: Function, session: Session, undoConfig: UndoConfig, ...args: any[]): Promise<T>;
+}
+export default AgentRein;

package/dist/cjs/agentreinClient.js

@@ -0,0 +1,93 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentRein = exports.AgentReinUnavailableError = void 0;
+const axios_1 = require("axios");
+const errors_1 = require("./errors");
+Object.defineProperty(exports, "AgentReinUnavailableError", { enumerable: true, get: function () { return errors_1.AgentReinUnavailableError; } });
+// ─── AgentRein Client ─────────────────────────────────────
+class AgentRein {
+    constructor(options) {
+        this.token = null;
+        this.serverUrl = options.serverUrl || 'https://api.agentrein.com';
+        this.apiKey = options.apiKey;
+        this.failureMode = options.failureMode ?? 'open';
+    }
+    // ── Authentication ──────────────────────────────────
+    /**
+     * Obtain a JWT token from the AgentRein server using the API key.
+     * Caches the token for subsequent requests.
+     */
+    async getToken() {
+        if (this.token)
+            return this.token;
+        try {
+            const res = await axios_1.default.post(`${this.serverUrl}/auth/token`, {
+                apiKey: this.apiKey,
+            });
+            this.token = res.data.data.token;
+            return this.token;
+        }
+        catch (err) {
+            throw new errors_1.AgentReinUnavailableError(`Failed to authenticate with AgentRein server: ${err instanceof Error ? err.message : String(err)}`);
+        }
+    }
+    /**
+     * Build authorization headers for server requests.
+     */
+    async authHeaders() {
+        const token = await this.getToken();
+        return { Authorization: `Bearer ${token}` };
+    }
+    // ── newSession ───────────────────────────────────────
+    /**
+     * Create a new agent session on the AgentRein server.
+     *
+     * @param options - Session options (agentId + optional intent).
+     *                  Can also pass a plain string for backward compat (agentId).
+     *                  If omitted, a random agentId is generated.
+     */
+    async newSession(options) {
+        const resolved = typeof options === 'string'
+            ? { agentId: options, intent: undefined }
+            : options ?? {};
+        const agentId = resolved.agentId ?? crypto.randomUUID();
+        const intent = resolved.intent;
+        const headers = await this.authHeaders();
+        const res = await axios_1.default.post(`${this.serverUrl}/sessions`, { agentId, intent }, { headers });
+        return res.data.data;
+    }
+    async call(fn, session, ...args) {
+        // Detect if first extra arg is an UndoConfig object
+        let undoConfig;
+        let callArgs = args;
+        if (args.length > 0 &&
+            args[0] &&
+            typeof args[0] === 'object' &&
+            'type' in args[0] &&
+            ['slack-correction', 'http-delete', 'none'].includes(args[0].type)) {
+            undoConfig = args[0];
+            callArgs = args.slice(1);
+        }
+        const headers = await this.authHeaders();
+        try {
+            const result = await fn(...callArgs);
+            // Log action to server (fire-and-forget)
+            axios_1.default.post(`${this.serverUrl}/sessions/${session.id}/actions`, {
+                apiName: fn.name || 'anonymous',
+                operationType: 'CREATE',
+                payload: callArgs[0] ?? {},
+                response: result,
+                status: 'SUCCESS',
+                undoConfig,
+            }, { headers }).catch(() => { });
+            return result;
+        }
+        catch (error) {
+            // Trigger server-side rollback
+            await axios_1.default.post(`${this.serverUrl}/sessions/${session.id}/rollback`, {}, { headers }).catch(() => { });
+            throw error;
+        }
+    }
+}
+exports.AgentRein = AgentRein;
+exports.default = AgentRein;

package/dist/cjs/errors.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Custom errors for the AgentRein SDK.
+ */
+export declare class AgentReinUnavailableError extends Error {
+    constructor(message?: string);
+}
+export declare class ApprovalRejectedError extends Error {
+    constructor(apiName: string, reason?: string);
+}
+export declare class ApprovalTimeoutError extends Error {
+    constructor(apiName: string, timeoutMinutes: number);
+}

package/dist/cjs/errors.js

@@ -0,0 +1,27 @@
+"use strict";
+/**
+ * Custom errors for the AgentRein SDK.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ApprovalTimeoutError = exports.ApprovalRejectedError = exports.AgentReinUnavailableError = void 0;
+class AgentReinUnavailableError extends Error {
+    constructor(message) {
+        super(message ?? 'AgentRein safety layer is unavailable');
+        this.name = 'AgentReinUnavailableError';
+    }
+}
+exports.AgentReinUnavailableError = AgentReinUnavailableError;
+class ApprovalRejectedError extends Error {
+    constructor(apiName, reason) {
+        super(`Action "${apiName}" was rejected by a human reviewer${reason ? `: ${reason}` : ''}`);
+        this.name = 'ApprovalRejectedError';
+    }
+}
+exports.ApprovalRejectedError = ApprovalRejectedError;
+class ApprovalTimeoutError extends Error {
+    constructor(apiName, timeoutMinutes) {
+        super(`Action "${apiName}" was not approved within ${timeoutMinutes} minutes — auto-rejected`);
+        this.name = 'ApprovalTimeoutError';
+    }
+}
+exports.ApprovalTimeoutError = ApprovalTimeoutError;

package/dist/esm/agentreinClient.js

@@ -0,0 +1,90 @@
+import axios from 'axios';
+import { AgentReinUnavailableError } from './errors';
+// Re-export errors for consumer convenience
+export { AgentReinUnavailableError };
+// ─── AgentRein Client ─────────────────────────────────────
+export class AgentRein {
+    constructor(options) {
+        this.token = null;
+        this.serverUrl = options.serverUrl || 'https://api.agentrein.com';
+        this.apiKey = options.apiKey;
+        this.failureMode = options.failureMode ?? 'open';
+    }
+    // ── Authentication ──────────────────────────────────
+    /**
+     * Obtain a JWT token from the AgentRein server using the API key.
+     * Caches the token for subsequent requests.
+     */
+    async getToken() {
+        if (this.token)
+            return this.token;
+        try {
+            const res = await axios.post(`${this.serverUrl}/auth/token`, {
+                apiKey: this.apiKey,
+            });
+            this.token = res.data.data.token;
+            return this.token;
+        }
+        catch (err) {
+            throw new AgentReinUnavailableError(`Failed to authenticate with AgentRein server: ${err instanceof Error ? err.message : String(err)}`);
+        }
+    }
+    /**
+     * Build authorization headers for server requests.
+     */
+    async authHeaders() {
+        const token = await this.getToken();
+        return { Authorization: `Bearer ${token}` };
+    }
+    // ── newSession ───────────────────────────────────────
+    /**
+     * Create a new agent session on the AgentRein server.
+     *
+     * @param options - Session options (agentId + optional intent).
+     *                  Can also pass a plain string for backward compat (agentId).
+     *                  If omitted, a random agentId is generated.
+     */
+    async newSession(options) {
+        const resolved = typeof options === 'string'
+            ? { agentId: options, intent: undefined }
+            : options ?? {};
+        const agentId = resolved.agentId ?? crypto.randomUUID();
+        const intent = resolved.intent;
+        const headers = await this.authHeaders();
+        const res = await axios.post(`${this.serverUrl}/sessions`, { agentId, intent }, { headers });
+        return res.data.data;
+    }
+    async call(fn, session, ...args) {
+        // Detect if first extra arg is an UndoConfig object
+        let undoConfig;
+        let callArgs = args;
+        if (args.length > 0 &&
+            args[0] &&
+            typeof args[0] === 'object' &&
+            'type' in args[0] &&
+            ['slack-correction', 'http-delete', 'none'].includes(args[0].type)) {
+            undoConfig = args[0];
+            callArgs = args.slice(1);
+        }
+        const headers = await this.authHeaders();
+        try {
+            const result = await fn(...callArgs);
+            // Log action to server (fire-and-forget)
+            axios.post(`${this.serverUrl}/sessions/${session.id}/actions`, {
+                apiName: fn.name || 'anonymous',
+                operationType: 'CREATE',
+                payload: callArgs[0] ?? {},
+                response: result,
+                status: 'SUCCESS',
+                undoConfig,
+            }, { headers }).catch(() => { });
+            return result;
+        }
+        catch (error) {
+            // Trigger server-side rollback
+            await axios.post(`${this.serverUrl}/sessions/${session.id}/rollback`, {}, { headers }).catch(() => { });
+            throw error;
+        }
+    }
+}
+export default AgentRein;

package/dist/esm/errors.js

@@ -0,0 +1,21 @@
+/**
+ * Custom errors for the AgentRein SDK.
+ */
+export class AgentReinUnavailableError extends Error {
+    constructor(message) {
+        super(message ?? 'AgentRein safety layer is unavailable');
+        this.name = 'AgentReinUnavailableError';
+    }
+}
+export class ApprovalRejectedError extends Error {
+    constructor(apiName, reason) {
+        super(`Action "${apiName}" was rejected by a human reviewer${reason ? `: ${reason}` : ''}`);
+        this.name = 'ApprovalRejectedError';
+    }
+}
+export class ApprovalTimeoutError extends Error {
+    constructor(apiName, timeoutMinutes) {
+        super(`Action "${apiName}" was not approved within ${timeoutMinutes} minutes — auto-rejected`);
+        this.name = 'ApprovalTimeoutError';
+    }
+}

package/package.json

@@ -1,14 +1,24 @@
 {
   "name": "agentrein",
-  "version": "1.0.8",
-  "main": "dist/agentreinClient.js",
-  "types": "dist/agentreinClient.d.ts",
+  "version": "1.0.9",
+  "main": "./dist/cjs/agentreinClient.js",
+  "module": "./dist/esm/agentreinClient.js",
+  "types": "./dist/cjs/agentreinClient.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/esm/agentreinClient.js",
+      "require": "./dist/cjs/agentreinClient.js",
+      "types": "./dist/cjs/agentreinClient.d.ts"
+    }
+  },
   "files": [
     "dist/",
     "README.md"
   ],
   "scripts": {
-    "build": "tsc",
+    "build": "npm run build:cjs && npm run build:esm",
+    "build:cjs": "tsc --project tsconfig.cjs.json",
+    "build:esm": "tsc --project tsconfig.esm.json",
     "prepublishOnly": "npm run build"
   },
   "dependencies": {