agentrein 1.0.11 → 1.0.12

package/dist/cjs/agentreinClient.d.ts

@@ -33,6 +33,7 @@ export declare class AgentRein {
     private readonly apiKey;
     private readonly failureMode;
     private token;
+    private currentSession;
     constructor(options: AgentReinOptions);
     /**
      * Obtain a JWT token from the AgentRein server using the API key.
@@ -51,11 +52,22 @@ export declare class AgentRein {
      *                  If omitted, a random agentId is generated.
      */
     newSession(options?: SessionOptions | string): Promise<Session>;
+    /**
+     * Start a new workflow session.
+     * All subsequent agentrein.call() will be logged under this session automatically.
+     */
+    startWorkflow(options?: SessionOptions | string): Promise<Session>;
+    /**
+     * End the current workflow and clear the internal session state.
+     */
+    endWorkflow(): Promise<void>;
+    /**
+     * Get the current active session, or create one automatically if none exists.
+     */
+    getCurrentSession(): Promise<Session>;
     /**
      * Resume an existing session by ID.
-     * Use this when multiple tools or steps need to share the same session.
-     *
-     * @param sessionId - The ID of the existing session to resume
+     * Use this when you need to explicitly attach to a known session.
      */
     resumeSession(sessionId: string): Promise<Session>;
     /**
@@ -66,10 +78,9 @@ export declare class AgentRein {
      * 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>;
+    call<T>(fn: Function, ...args: any[]): Promise<T>;
+    call<T>(fn: Function, undoConfig: UndoConfig, ...args: any[]): Promise<T>;
 }
 export default AgentRein;

package/dist/cjs/agentreinClient.js

@@ -12,6 +12,7 @@ function createUndoConfig(config) {
 class AgentRein {
     constructor(options) {
         this.token = null;
+        this.currentSession = null;
         this.serverUrl = options.serverUrl || 'https://api.agentrein.com';
         this.apiKey = options.apiKey;
         this.failureMode = options.failureMode ?? 'open';
@@ -60,18 +61,41 @@ class AgentRein {
         const res = await axios_1.default.post(`${this.serverUrl}/sessions`, { agentId, intent }, { headers });
         return res.data.data;
     }
+    /**
+     * Start a new workflow session.
+     * All subsequent agentrein.call() will be logged under this session automatically.
+     */
+    async startWorkflow(options) {
+        this.currentSession = await this.newSession(options);
+        return this.currentSession;
+    }
+    /**
+     * End the current workflow and clear the internal session state.
+     */
+    async endWorkflow() {
+        this.currentSession = null;
+    }
+    /**
+     * Get the current active session, or create one automatically if none exists.
+     */
+    async getCurrentSession() {
+        if (!this.currentSession) {
+            this.currentSession = await this.newSession();
+        }
+        return this.currentSession;
+    }
     /**
      * Resume an existing session by ID.
-     * Use this when multiple tools or steps need to share the same session.
-     *
-     * @param sessionId - The ID of the existing session to resume
+     * Use this when you need to explicitly attach to a known session.
      */
     async resumeSession(sessionId) {
         const headers = await this.authHeaders();
         const res = await axios_1.default.get(`${this.serverUrl}/sessions/${sessionId}`, { headers });
-        return res.data.data;
+        this.currentSession = res.data.data;
+        return this.currentSession;
     }
-    async call(fn, session, ...args) {
+    async call(fn, ...args) {
+        const session = await this.getCurrentSession();
         // Detect if first extra arg is an UndoConfig object
         let undoConfig;
         let callArgs = args;

package/dist/esm/agentreinClient.js

@@ -9,6 +9,7 @@ export { AgentReinUnavailableError };
 export class AgentRein {
     constructor(options) {
         this.token = null;
+        this.currentSession = null;
         this.serverUrl = options.serverUrl || 'https://api.agentrein.com';
         this.apiKey = options.apiKey;
         this.failureMode = options.failureMode ?? 'open';
@@ -57,18 +58,41 @@ export class AgentRein {
         const res = await axios.post(`${this.serverUrl}/sessions`, { agentId, intent }, { headers });
         return res.data.data;
     }
+    /**
+     * Start a new workflow session.
+     * All subsequent agentrein.call() will be logged under this session automatically.
+     */
+    async startWorkflow(options) {
+        this.currentSession = await this.newSession(options);
+        return this.currentSession;
+    }
+    /**
+     * End the current workflow and clear the internal session state.
+     */
+    async endWorkflow() {
+        this.currentSession = null;
+    }
+    /**
+     * Get the current active session, or create one automatically if none exists.
+     */
+    async getCurrentSession() {
+        if (!this.currentSession) {
+            this.currentSession = await this.newSession();
+        }
+        return this.currentSession;
+    }
     /**
      * Resume an existing session by ID.
-     * Use this when multiple tools or steps need to share the same session.
-     *
-     * @param sessionId - The ID of the existing session to resume
+     * Use this when you need to explicitly attach to a known session.
      */
     async resumeSession(sessionId) {
         const headers = await this.authHeaders();
         const res = await axios.get(`${this.serverUrl}/sessions/${sessionId}`, { headers });
-        return res.data.data;
+        this.currentSession = res.data.data;
+        return this.currentSession;
     }
-    async call(fn, session, ...args) {
+    async call(fn, ...args) {
+        const session = await this.getCurrentSession();
         // Detect if first extra arg is an UndoConfig object
         let undoConfig;
         let callArgs = args;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentrein",
-  "version": "1.0.11",
+  "version": "1.0.12",
   "main": "./dist/cjs/agentreinClient.js",
   "module": "./dist/esm/agentreinClient.js",
   "types": "./dist/cjs/agentreinClient.d.ts",