agentrein 1.0.11 → 1.0.13

package/dist/cjs/agentreinClient.d.ts

@@ -33,6 +33,7 @@ export declare class AgentRein {
     private readonly apiKey;
     private readonly failureMode;
     private token;
+    private static activeSessions;
     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

@@ -60,18 +60,45 @@ 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) {
+        const session = await this.newSession(options);
+        AgentRein.activeSessions.set(this.apiKey, session);
+        return session;
+    }
+    /**
+     * End the current workflow and clear the internal session state.
+     */
+    async endWorkflow() {
+        AgentRein.activeSessions.delete(this.apiKey);
+    }
+    /**
+     * Get the current active session, or create one automatically if none exists.
+     */
+    async getCurrentSession() {
+        const cached = AgentRein.activeSessions.get(this.apiKey);
+        if (cached)
+            return cached;
+        const session = await this.newSession();
+        AgentRein.activeSessions.set(this.apiKey, session);
+        return 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.
      */
     async resumeSession(sessionId) {
         const headers = await this.authHeaders();
         const res = await axios_1.default.get(`${this.serverUrl}/sessions/${sessionId}`, { headers });
-        return res.data.data;
+        const session = res.data.data;
+        AgentRein.activeSessions.set(this.apiKey, session);
+        return session;
     }
-    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;
@@ -104,4 +131,5 @@ class AgentRein {
     }
 }
 exports.AgentRein = AgentRein;
+AgentRein.activeSessions = new Map();
 exports.default = AgentRein;

package/dist/esm/agentreinClient.js

@@ -57,18 +57,45 @@ 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) {
+        const session = await this.newSession(options);
+        AgentRein.activeSessions.set(this.apiKey, session);
+        return session;
+    }
+    /**
+     * End the current workflow and clear the internal session state.
+     */
+    async endWorkflow() {
+        AgentRein.activeSessions.delete(this.apiKey);
+    }
+    /**
+     * Get the current active session, or create one automatically if none exists.
+     */
+    async getCurrentSession() {
+        const cached = AgentRein.activeSessions.get(this.apiKey);
+        if (cached)
+            return cached;
+        const session = await this.newSession();
+        AgentRein.activeSessions.set(this.apiKey, session);
+        return 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.
      */
     async resumeSession(sessionId) {
         const headers = await this.authHeaders();
         const res = await axios.get(`${this.serverUrl}/sessions/${sessionId}`, { headers });
-        return res.data.data;
+        const session = res.data.data;
+        AgentRein.activeSessions.set(this.apiKey, session);
+        return session;
     }
-    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;
@@ -100,4 +127,5 @@ export class AgentRein {
         }
     }
 }
+AgentRein.activeSessions = new Map();
 export default AgentRein;

package/package.json

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