@contractspec/lib.ai-agent 1.61.0 → 2.0.0

package/dist/node/session/index.js

@@ -15,10 +15,19 @@ var __require = /* @__PURE__ */ createRequire(import.meta.url);
 // src/session/store.ts
 class InMemorySessionStore {
   sessions = new Map;
+  maxSessions;
+  maxMessagesPerSession;
+  maxStepsPerSession;
+  constructor(options = {}) {
+    this.maxSessions = options.maxSessions ?? 500;
+    this.maxMessagesPerSession = options.maxMessagesPerSession ?? 200;
+    this.maxStepsPerSession = options.maxStepsPerSession ?? 200;
+  }
   async get(sessionId) {
     return this.sessions.get(sessionId) ?? null;
   }
   async create(session) {
+    this.evictIfNeeded();
     const now = new Date;
     const fullSession = {
       ...session,
@@ -32,6 +41,7 @@ class InMemorySessionStore {
     const session = this.sessions.get(sessionId);
     if (session) {
       session.steps.push(step);
+      this.trimArray(session.steps, this.maxStepsPerSession);
       session.updatedAt = new Date;
     }
   }
@@ -39,6 +49,7 @@ class InMemorySessionStore {
     const session = this.sessions.get(sessionId);
     if (session) {
       session.messages.push(message);
+      this.trimArray(session.messages, this.maxMessagesPerSession);
       session.updatedAt = new Date;
     }
   }
@@ -76,9 +87,33 @@ class InMemorySessionStore {
   clear() {
     this.sessions.clear();
   }
+  evictIfNeeded() {
+    if (this.sessions.size < this.maxSessions) {
+      return;
+    }
+    let oldestSessionId = null;
+    let oldestUpdatedAt = Number.POSITIVE_INFINITY;
+    for (const [sessionId, session] of this.sessions.entries()) {
+      const updatedAt = session.updatedAt.getTime();
+      if (updatedAt < oldestUpdatedAt) {
+        oldestUpdatedAt = updatedAt;
+        oldestSessionId = sessionId;
+      }
+    }
+    if (oldestSessionId) {
+      this.sessions.delete(oldestSessionId);
+    }
+  }
+  trimArray(items, maxItems) {
+    if (items.length <= maxItems) {
+      return;
+    }
+    const overflow = items.length - maxItems;
+    items.splice(0, overflow);
+  }
 }
-function createInMemorySessionStore() {
-  return new InMemorySessionStore;
+function createInMemorySessionStore(options) {
+  return new InMemorySessionStore(options);
 }
 function generateSessionId() {
   return `sess_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`;

package/dist/node/session/store.js

@@ -15,10 +15,19 @@ var __require = /* @__PURE__ */ createRequire(import.meta.url);
 // src/session/store.ts
 class InMemorySessionStore {
   sessions = new Map;
+  maxSessions;
+  maxMessagesPerSession;
+  maxStepsPerSession;
+  constructor(options = {}) {
+    this.maxSessions = options.maxSessions ?? 500;
+    this.maxMessagesPerSession = options.maxMessagesPerSession ?? 200;
+    this.maxStepsPerSession = options.maxStepsPerSession ?? 200;
+  }
   async get(sessionId) {
     return this.sessions.get(sessionId) ?? null;
   }
   async create(session) {
+    this.evictIfNeeded();
     const now = new Date;
     const fullSession = {
       ...session,
@@ -32,6 +41,7 @@ class InMemorySessionStore {
     const session = this.sessions.get(sessionId);
     if (session) {
       session.steps.push(step);
+      this.trimArray(session.steps, this.maxStepsPerSession);
       session.updatedAt = new Date;
     }
   }
@@ -39,6 +49,7 @@ class InMemorySessionStore {
     const session = this.sessions.get(sessionId);
     if (session) {
       session.messages.push(message);
+      this.trimArray(session.messages, this.maxMessagesPerSession);
       session.updatedAt = new Date;
     }
   }
@@ -76,9 +87,33 @@ class InMemorySessionStore {
   clear() {
     this.sessions.clear();
   }
+  evictIfNeeded() {
+    if (this.sessions.size < this.maxSessions) {
+      return;
+    }
+    let oldestSessionId = null;
+    let oldestUpdatedAt = Number.POSITIVE_INFINITY;
+    for (const [sessionId, session] of this.sessions.entries()) {
+      const updatedAt = session.updatedAt.getTime();
+      if (updatedAt < oldestUpdatedAt) {
+        oldestUpdatedAt = updatedAt;
+        oldestSessionId = sessionId;
+      }
+    }
+    if (oldestSessionId) {
+      this.sessions.delete(oldestSessionId);
+    }
+  }
+  trimArray(items, maxItems) {
+    if (items.length <= maxItems) {
+      return;
+    }
+    const overflow = items.length - maxItems;
+    items.splice(0, overflow);
+  }
 }
-function createInMemorySessionStore() {
-  return new InMemorySessionStore;
+function createInMemorySessionStore(options) {
+  return new InMemorySessionStore(options);
 }
 function generateSessionId() {
   return `sess_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`;