@farukada/aws-langgraph-dynamodb-ts 0.1.1 → 0.2.2

package/dist/history/actions/add-messages.js

@@ -1,57 +1,52 @@
 "use strict";
 /**
- * Add multiple messages to a session
- * Uses an atomic counter to claim a unique message index range, preventing race conditions
- * Uses transactWrite for message items (≤100, enforced by validation)
+ * Add multiple messages to a session using optimistic concurrency control.
+ *
+ * Reads the current messageCount, writes N message items and the updated counter
+ * in a single TransactWrite (N ≤ 99, leaving room for the metadata item in the
+ * 100-item transaction limit). The metadata update has a conditional check on the
+ * observed count; on concurrent modification we re-read and retry.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.addMessagesAction = void 0;
 const utils_1 = require("../utils");
-/**
- * Add multiple messages to a session
- * Atomically claims message indices via DynamoDB's ADD operation, then writes message items
- *
- * @param params - Parameters for the add messages operation
- * @throws Error if the operation fails or validation fails
- */
 const addMessagesAction = async (params) => {
-    const { client, tableName, userId, sessionId, messages, title, ttlDays } = params;
-    // Validate inputs (enforces max 100 messages per batch)
+    const { client, tableName, userId, sessionId, messages, title, ttlDays, signal } = params;
     (0, utils_1.validateUserId)(userId);
     (0, utils_1.validateSessionId)(sessionId);
     (0, utils_1.validateMessages)(messages);
     (0, utils_1.validateMessagesSize)(messages);
     (0, utils_1.validateTitle)(title);
     (0, utils_1.validateTTLDays)(ttlDays);
-    // Generate a title if this is the first message batch and no title provided
     const sessionTitle = title ?? (0, utils_1.generateTitle)(messages[0]);
-    // Atomically claim a range of message indices via ADD messageCount
-    const { updateExpression, expressionAttributeValues } = (0, utils_1.buildAtomicMetadataUpdate)(sessionTitle, messages.length, // Claiming N indices
-    ttlDays);
-    const updateResult = await (0, utils_1.withDynamoDBRetry)(async () => {
-        return await client.update({
-            TableName: tableName,
-            Key: { userId, sessionId },
-            UpdateExpression: updateExpression,
-            ExpressionAttributeValues: expressionAttributeValues,
-            ReturnValues: 'ALL_NEW',
-        });
-    });
-    // The new messageCount is the count AFTER incrementing by messages.length
-    // So our claimed starting index is newCount - messages.length
-    const newCount = updateResult.Attributes?.messageCount ?? messages.length;
-    const startIndex = newCount - messages.length;
-    // Build all message items
-    const messageItems = (0, utils_1.buildMessageItems)(userId, sessionId, messages, startIndex, ttlDays);
-    // Write all message items in a single transaction (max 100 items, enforced by validation)
-    await (0, utils_1.withDynamoDBRetry)(async () => {
+    await (0, utils_1.withOptimisticRetry)(`addMessages:${sessionId}`, async () => {
+        signal?.throwIfAborted();
+        const existing = await (0, utils_1.withDynamoDBRetry)(async () => {
+            return await client.get({
+                TableName: tableName,
+                Key: { userId, sessionId },
+                ConsistentRead: true,
+                ProjectionExpression: 'messageCount',
+            });
+        }, { signal });
+        const expectedCount = existing.Item?.messageCount;
+        const currentCount = expectedCount ?? 0;
+        const newCount = currentCount + messages.length;
+        const messageItems = (0, utils_1.buildMessageItems)(userId, sessionId, messages, currentCount, ttlDays);
+        const { updateExpression, conditionExpression, expressionAttributeValues } = (0, utils_1.buildOptimisticMetadataUpdate)(sessionTitle, newCount, expectedCount, ttlDays);
         await client.transactWrite({
-            TransactItems: messageItems.map((item) => ({
-                Put: {
-                    TableName: tableName,
-                    Item: item,
+            TransactItems: [
+                ...messageItems.map((item) => ({ Put: { TableName: tableName, Item: item } })),
+                {
+                    Update: {
+                        TableName: tableName,
+                        Key: { userId, sessionId },
+                        UpdateExpression: updateExpression,
+                        ConditionExpression: conditionExpression,
+                        ExpressionAttributeValues: expressionAttributeValues,
+                    },
                 },
-            })),
+            ],
         });
     });
 };

package/dist/history/actions/add-messages.js.map

@@ -1 +1 @@
-{"version":3,"file":"add-messages.js","sourceRoot":"","sources":["../../../src/history/actions/add-messages.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAGH,oCAWkB;AAElB;;;;;;GAMG;AACI,MAAM,iBAAiB,GAAG,KAAK,EAAE,MAA+B,EAAiB,EAAE;IACxF,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC;IAElF,wDAAwD;IACxD,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,yBAAiB,EAAC,SAAS,CAAC,CAAC;IAC7B,IAAA,wBAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,IAAA,4BAAoB,EAAC,QAAQ,CAAC,CAAC;IAC/B,IAAA,qBAAa,EAAC,KAAK,CAAC,CAAC;IACrB,IAAA,uBAAe,EAAC,OAAO,CAAC,CAAC;IAEzB,4EAA4E;IAC5E,MAAM,YAAY,GAAG,KAAK,IAAI,IAAA,qBAAa,EAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAEzD,mEAAmE;IACnE,MAAM,EAAE,gBAAgB,EAAE,yBAAyB,EAAE,GAAG,IAAA,iCAAyB,EAC/E,YAAY,EACZ,QAAQ,CAAC,MAAM,EAAE,qBAAqB;IACtC,OAAO,CACR,CAAC;IAEF,MAAM,YAAY,GAAG,MAAM,IAAA,yBAAiB,EAAC,KAAK,IAAI,EAAE;QACtD,OAAO,MAAM,MAAM,CAAC,MAAM,CAAC;YACzB,SAAS,EAAE,SAAS;YACpB,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;YAC1B,gBAAgB,EAAE,gBAAgB;YAClC,yBAAyB,EAAE,yBAAyB;YACpD,YAAY,EAAE,SAAS;SACxB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,0EAA0E;IAC1E,8DAA8D;IAC9D,MAAM,QAAQ,GAAI,YAAY,CAAC,UAAU,EAAE,YAAuB,IAAI,QAAQ,CAAC,MAAM,CAAC;IACtF,MAAM,UAAU,GAAG,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC;IAE9C,0BAA0B;IAC1B,MAAM,YAAY,GAAG,IAAA,yBAAiB,EAAC,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IAEzF,0FAA0F;IAC1F,MAAM,IAAA,yBAAiB,EAAC,KAAK,IAAI,EAAE;QACjC,MAAM,MAAM,CAAC,aAAa,CAAC;YACzB,aAAa,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACzC,GAAG,EAAE;oBACH,SAAS,EAAE,SAAS;oBACpB,IAAI,EAAE,IAAI;iBACX;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAlDW,QAAA,iBAAiB,qBAkD5B"}
+{"version":3,"file":"add-messages.js","sourceRoot":"","sources":["../../../src/history/actions/add-messages.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;AAGH,oCAYkB;AAEX,MAAM,iBAAiB,GAAG,KAAK,EAAE,MAA+B,EAAiB,EAAE;IACxF,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE1F,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,yBAAiB,EAAC,SAAS,CAAC,CAAC;IAC7B,IAAA,wBAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,IAAA,4BAAoB,EAAC,QAAQ,CAAC,CAAC;IAC/B,IAAA,qBAAa,EAAC,KAAK,CAAC,CAAC;IACrB,IAAA,uBAAe,EAAC,OAAO,CAAC,CAAC;IAEzB,MAAM,YAAY,GAAG,KAAK,IAAI,IAAA,qBAAa,EAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAEzD,MAAM,IAAA,2BAAmB,EAAC,eAAe,SAAS,EAAE,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,EAAE,cAAc,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,IAAA,yBAAiB,EACtC,KAAK,IAAI,EAAE;YACT,OAAO,MAAM,MAAM,CAAC,GAAG,CAAC;gBACtB,SAAS,EAAE,SAAS;gBACpB,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;gBAC1B,cAAc,EAAE,IAAI;gBACpB,oBAAoB,EAAE,cAAc;aACrC,CAAC,CAAC;QACL,CAAC,EACD,EAAE,MAAM,EAAE,CACX,CAAC;QAEF,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,EAAE,YAAkC,CAAC;QACxE,MAAM,YAAY,GAAG,aAAa,IAAI,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC;QAEhD,MAAM,YAAY,GAAG,IAAA,yBAAiB,EAAC,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;QAC3F,MAAM,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,yBAAyB,EAAE,GACxE,IAAA,qCAA6B,EAAC,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QAEhF,MAAM,MAAM,CAAC,aAAa,CAAC;YACzB,aAAa,EAAE;gBACb,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC9E;oBACE,MAAM,EAAE;wBACN,SAAS,EAAE,SAAS;wBACpB,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;wBAC1B,gBAAgB,EAAE,gBAAgB;wBAClC,mBAAmB,EAAE,mBAAmB;wBACxC,yBAAyB,EAAE,yBAAyB;qBACrD;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAjDW,QAAA,iBAAiB,qBAiD5B"}

package/dist/history/actions/clear.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"clear.d.ts","sourceRoot":"","sources":["../../../src/history/actions/clear.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAOH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAGlD;;;;;;GAMG;AACH,eAAO,MAAM,WAAW,GAAU,QAAQ,iBAAiB,KAAG,OAAO,CAAC,IAAI,CA4DzE,CAAC"}
+{"version":3,"file":"clear.d.ts","sourceRoot":"","sources":["../../../src/history/actions/clear.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAOH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAGlD;;;;;;GAMG;AACH,eAAO,MAAM,WAAW,GAAU,QAAQ,iBAAiB,KAAG,OAAO,CAAC,IAAI,CA+DzE,CAAC"}

package/dist/history/actions/clear.js

@@ -15,7 +15,7 @@ const utils_1 = require("../utils");
  * @throws Error if the operation fails or validation fails
  */
 const clearAction = async (params) => {
-    const { client, tableName, userId, sessionId } = params;
+    const { client, tableName, userId, sessionId, signal } = params;
     // Validate inputs
     (0, utils_1.validateUserId)(userId);
     (0, utils_1.validateSessionId)(sessionId);
@@ -44,7 +44,7 @@ const clearAction = async (params) => {
                 ProjectionExpression: 'userId, sessionId',
                 ExclusiveStartKey: lastEvaluatedKey,
             });
-        });
+        }, { signal });
         if (result.Items && result.Items.length > 0) {
             // Prevent memory exhaustion
             if (keysToDelete.length + result.Items.length > shared_1.MAX_TOTAL_ITEMS_IN_MEMORY) {
@@ -61,7 +61,7 @@ const clearAction = async (params) => {
     const deleteRequests = keysToDelete.map((key) => ({
         DeleteRequest: { Key: key },
     }));
-    await (0, shared_1.batchWriteAllWithRetry)(client, tableName, deleteRequests);
+    await (0, shared_1.batchWriteAllWithRetry)(client, tableName, deleteRequests, { signal });
 };
 exports.clearAction = clearAction;
 //# sourceMappingURL=clear.js.map

package/dist/history/actions/clear.js.map

@@ -1 +1 @@
-{"version":3,"file":"clear.js","sourceRoot":"","sources":["../../../src/history/actions/clear.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,yCAIsB;AAEtB,oCAAgF;AAEhF;;;;;;GAMG;AACI,MAAM,WAAW,GAAG,KAAK,EAAE,MAAyB,EAAiB,EAAE;IAC5E,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC;IAExD,kBAAkB;IAClB,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,yBAAiB,EAAC,SAAS,CAAC,CAAC;IAE7B,qEAAqE;IACrE,MAAM,YAAY,GAAiD,EAAE,CAAC;IAEtE,4BAA4B;IAC5B,YAAY,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;IAEzC,uCAAuC;IACvC,MAAM,aAAa,GAAG,GAAG,SAAS,OAAO,CAAC;IAC1C,IAAI,gBAAiD,CAAC;IACtD,IAAI,cAAc,GAAG,CAAC,CAAC;IAEvB,GAAG,CAAC;QACF,yBAAyB;QACzB,cAAc,EAAE,CAAC;QACjB,IAAI,cAAc,GAAG,4BAAmB,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,yBAAiB,EAAC,KAAK,IAAI,EAAE;YAChD,OAAO,MAAM,MAAM,CAAC,KAAK,CAAC;gBACxB,SAAS,EAAE,SAAS;gBACpB,sBAAsB,EAAE,sDAAsD;gBAC9E,yBAAyB,EAAE;oBACzB,SAAS,EAAE,MAAM;oBACjB,SAAS,EAAE,aAAa;iBACzB;gBACD,oBAAoB,EAAE,mBAAmB;gBACzC,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5C,4BAA4B;YAC5B,IAAI,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,kCAAyB,EAAE,CAAC;gBAC1E,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;YAC5E,CAAC;YACD,YAAY,CAAC,IAAI,CACf,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC7B,MAAM,EAAE,IAAI,CAAC,MAAgB;gBAC7B,SAAS,EAAE,IAAI,CAAC,SAAmB;aACpC,CAAC,CAAC,CACJ,CAAC;QACJ,CAAC;QAED,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAC7C,CAAC,QAAQ,gBAAgB,EAAE;IAE3B,oEAAoE;IACpE,MAAM,cAAc,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAChD,aAAa,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE;KAC5B,CAAC,CAAC,CAAC;IAEJ,MAAM,IAAA,+BAAsB,EAAC,MAAM,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;AAClE,CAAC,CAAC;AA5DW,QAAA,WAAW,eA4DtB"}
+{"version":3,"file":"clear.js","sourceRoot":"","sources":["../../../src/history/actions/clear.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,yCAIsB;AAEtB,oCAAgF;AAEhF;;;;;;GAMG;AACI,MAAM,WAAW,GAAG,KAAK,EAAE,MAAyB,EAAiB,EAAE;IAC5E,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAEhE,kBAAkB;IAClB,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,yBAAiB,EAAC,SAAS,CAAC,CAAC;IAE7B,qEAAqE;IACrE,MAAM,YAAY,GAAiD,EAAE,CAAC;IAEtE,4BAA4B;IAC5B,YAAY,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;IAEzC,uCAAuC;IACvC,MAAM,aAAa,GAAG,GAAG,SAAS,OAAO,CAAC;IAC1C,IAAI,gBAAiD,CAAC;IACtD,IAAI,cAAc,GAAG,CAAC,CAAC;IAEvB,GAAG,CAAC;QACF,yBAAyB;QACzB,cAAc,EAAE,CAAC;QACjB,IAAI,cAAc,GAAG,4BAAmB,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,yBAAiB,EACpC,KAAK,IAAI,EAAE;YACT,OAAO,MAAM,MAAM,CAAC,KAAK,CAAC;gBACxB,SAAS,EAAE,SAAS;gBACpB,sBAAsB,EAAE,sDAAsD;gBAC9E,yBAAyB,EAAE;oBACzB,SAAS,EAAE,MAAM;oBACjB,SAAS,EAAE,aAAa;iBACzB;gBACD,oBAAoB,EAAE,mBAAmB;gBACzC,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;QACL,CAAC,EACD,EAAE,MAAM,EAAE,CACX,CAAC;QAEF,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5C,4BAA4B;YAC5B,IAAI,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,kCAAyB,EAAE,CAAC;gBAC1E,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;YAC5E,CAAC;YACD,YAAY,CAAC,IAAI,CACf,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC7B,MAAM,EAAE,IAAI,CAAC,MAAgB;gBAC7B,SAAS,EAAE,IAAI,CAAC,SAAmB;aACpC,CAAC,CAAC,CACJ,CAAC;QACJ,CAAC;QAED,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAC7C,CAAC,QAAQ,gBAAgB,EAAE;IAE3B,oEAAoE;IACpE,MAAM,cAAc,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAChD,aAAa,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE;KAC5B,CAAC,CAAC,CAAC;IAEJ,MAAM,IAAA,+BAAsB,EAAC,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;AAC9E,CAAC,CAAC;AA/DW,QAAA,WAAW,eA+DtB"}

package/dist/history/actions/get-messages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"get-messages.d.ts","sourceRoot":"","sources":["../../../src/history/actions/get-messages.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,WAAW,EAAmC,MAAM,0BAA0B,CAAC;AAGxF,OAAO,KAAK,EAAE,uBAAuB,EAAuB,MAAM,UAAU,CAAC;AAG7E;;;;;;;GAOG;AACH,eAAO,MAAM,iBAAiB,GAC5B,QAAQ,uBAAuB,KAC9B,OAAO,CAAC,WAAW,EAAE,CAsDvB,CAAC"}
+{"version":3,"file":"get-messages.d.ts","sourceRoot":"","sources":["../../../src/history/actions/get-messages.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,WAAW,EAAmC,MAAM,0BAA0B,CAAC;AAGxF,OAAO,KAAK,EAAE,uBAAuB,EAAuB,MAAM,UAAU,CAAC;AAG7E;;;;;;;GAOG;AACH,eAAO,MAAM,iBAAiB,GAC5B,QAAQ,uBAAuB,KAC9B,OAAO,CAAC,WAAW,EAAE,CAyDvB,CAAC"}

package/dist/history/actions/get-messages.js

@@ -17,7 +17,7 @@ const utils_1 = require("../utils");
  * @throws Error if the operation fails or validation fails
  */
 const getMessagesAction = async (params) => {
-    const { client, tableName, userId, sessionId } = params;
+    const { client, tableName, userId, sessionId, signal } = params;
     // Validate inputs
     (0, utils_1.validateUserId)(userId);
     (0, utils_1.validateSessionId)(sessionId);
@@ -42,7 +42,7 @@ const getMessagesAction = async (params) => {
                 },
                 ExclusiveStartKey: lastEvaluatedKey,
             });
-        });
+        }, { signal });
         if (result.Items && result.Items.length > 0) {
             // Prevent memory exhaustion
             if (messages.length + result.Items.length > shared_1.MAX_TOTAL_ITEMS_IN_MEMORY) {

package/dist/history/actions/get-messages.js.map

@@ -1 +1 @@
-{"version":3,"file":"get-messages.js","sourceRoot":"","sources":["../../../src/history/actions/get-messages.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,uDAAwF;AAExF,yCAA8E;AAE9E,oCAAgF;AAEhF;;;;;;;GAOG;AACI,MAAM,iBAAiB,GAAG,KAAK,EACpC,MAA+B,EACP,EAAE;IAC1B,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC;IAExD,kBAAkB;IAClB,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,yBAAiB,EAAC,SAAS,CAAC,CAAC;IAE7B,0DAA0D;IAC1D,MAAM,aAAa,GAAG,GAAG,SAAS,OAAO,CAAC;IAC1C,MAAM,QAAQ,GAA0B,EAAE,CAAC;IAC3C,IAAI,gBAAiD,CAAC;IACtD,IAAI,cAAc,GAAG,CAAC,CAAC;IAEvB,GAAG,CAAC;QACF,yBAAyB;QACzB,cAAc,EAAE,CAAC;QACjB,IAAI,cAAc,GAAG,4BAAmB,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,yBAAiB,EAAC,KAAK,IAAI,EAAE;YAChD,OAAO,MAAM,MAAM,CAAC,KAAK,CAAC;gBACxB,SAAS,EAAE,SAAS;gBACpB,sBAAsB,EAAE,sDAAsD;gBAC9E,yBAAyB,EAAE;oBACzB,SAAS,EAAE,MAAM;oBACjB,SAAS,EAAE,aAAa;iBACzB;gBACD,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5C,4BAA4B;YAC5B,IAAI,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,kCAAyB,EAAE,CAAC;gBACtE,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;YAClF,CAAC;YACD,QAAQ,CAAC,IAAI,CAAC,GAAI,MAAM,CAAC,KAA+B,CAAC,CAAC;QAC5D,CAAC;QAED,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAC7C,CAAC,QAAQ,gBAAgB,EAAE;IAE3B,0CAA0C;IAC1C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,qDAAqD;IACrD,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC;IAEzD,oEAAoE;IACpE,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5D,OAAO,IAAA,0CAA+B,EAAC,cAAc,CAAC,CAAC;AACzD,CAAC,CAAC;AAxDW,QAAA,iBAAiB,qBAwD5B"}
+{"version":3,"file":"get-messages.js","sourceRoot":"","sources":["../../../src/history/actions/get-messages.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,uDAAwF;AAExF,yCAA8E;AAE9E,oCAAgF;AAEhF;;;;;;;GAOG;AACI,MAAM,iBAAiB,GAAG,KAAK,EACpC,MAA+B,EACP,EAAE;IAC1B,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAEhE,kBAAkB;IAClB,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,yBAAiB,EAAC,SAAS,CAAC,CAAC;IAE7B,0DAA0D;IAC1D,MAAM,aAAa,GAAG,GAAG,SAAS,OAAO,CAAC;IAC1C,MAAM,QAAQ,GAA0B,EAAE,CAAC;IAC3C,IAAI,gBAAiD,CAAC;IACtD,IAAI,cAAc,GAAG,CAAC,CAAC;IAEvB,GAAG,CAAC;QACF,yBAAyB;QACzB,cAAc,EAAE,CAAC;QACjB,IAAI,cAAc,GAAG,4BAAmB,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,yBAAiB,EACpC,KAAK,IAAI,EAAE;YACT,OAAO,MAAM,MAAM,CAAC,KAAK,CAAC;gBACxB,SAAS,EAAE,SAAS;gBACpB,sBAAsB,EAAE,sDAAsD;gBAC9E,yBAAyB,EAAE;oBACzB,SAAS,EAAE,MAAM;oBACjB,SAAS,EAAE,aAAa;iBACzB;gBACD,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;QACL,CAAC,EACD,EAAE,MAAM,EAAE,CACX,CAAC;QAEF,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5C,4BAA4B;YAC5B,IAAI,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,kCAAyB,EAAE,CAAC;gBACtE,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;YAClF,CAAC;YACD,QAAQ,CAAC,IAAI,CAAC,GAAI,MAAM,CAAC,KAA+B,CAAC,CAAC;QAC5D,CAAC;QAED,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAC7C,CAAC,QAAQ,gBAAgB,EAAE;IAE3B,0CAA0C;IAC1C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,qDAAqD;IACrD,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC;IAEzD,oEAAoE;IACpE,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5D,OAAO,IAAA,0CAA+B,EAAC,cAAc,CAAC,CAAC;AACzD,CAAC,CAAC;AA3DW,QAAA,iBAAiB,qBA2D5B"}

package/dist/history/actions/list-sessions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"list-sessions.d.ts","sourceRoot":"","sources":["../../../src/history/actions/list-sessions.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,KAAK,EAAE,wBAAwB,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAG1E;;;;;;;;;;;;;;;;;;GAkBG;AACH,eAAO,MAAM,kBAAkB,GAC7B,QAAQ,wBAAwB,KAC/B,OAAO,CAAC,eAAe,EAAE,CAwD3B,CAAC"}
+{"version":3,"file":"list-sessions.d.ts","sourceRoot":"","sources":["../../../src/history/actions/list-sessions.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,KAAK,EAAE,wBAAwB,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAG1E;;;;;;;;;;;;;;;;;;GAkBG;AACH,eAAO,MAAM,kBAAkB,GAC7B,QAAQ,wBAAwB,KAC/B,OAAO,CAAC,eAAe,EAAE,CAgE3B,CAAC"}

package/dist/history/actions/list-sessions.js

@@ -26,7 +26,7 @@ const utils_1 = require("../utils");
  * @throws Error if the operation fails or validation fails
  */
 const listSessionsAction = async (params) => {
-    const { client, tableName, userId, limit } = params;
+    const { client, tableName, userId, limit, signal } = params;
     // Validate inputs
     (0, utils_1.validateUserId)(userId);
     (0, utils_1.validateLimit)(limit);
@@ -48,7 +48,7 @@ const listSessionsAction = async (params) => {
                 ProjectionExpression: 'sessionId, title, createdAt, updatedAt, messageCount',
                 ExclusiveStartKey: lastEvaluatedKey,
             });
-        });
+        }, { signal });
         if (!result.Items || result.Items.length === 0) {
             break;
         }
@@ -61,8 +61,11 @@ const listSessionsAction = async (params) => {
             messageCount: item.messageCount,
         }));
         sessions.push(...mappedSessions);
-        // Safety limit: prevent unbounded memory growth
+        // Safety limit: prevent unbounded memory growth. Users near this bound should
+        // add a (userId, updatedAt) GSI — see method JSDoc.
         if (sessions.length >= shared_1.MAX_TOTAL_ITEMS_IN_MEMORY) {
+            (0, shared_1.getLogger)().warn(`listSessions truncated at ${shared_1.MAX_TOTAL_ITEMS_IN_MEMORY} items for userId="${userId}"; ` +
+                `older sessions are not returned. Add a GSI on (userId, updatedAt) for full coverage.`);
             break;
         }
         lastEvaluatedKey = result.LastEvaluatedKey;

package/dist/history/actions/list-sessions.js.map

@@ -1 +1 @@
-{"version":3,"file":"list-sessions.js","sourceRoot":"","sources":["../../../src/history/actions/list-sessions.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH,yCAAyD;AAEzD,oCAA4E;AAE5E;;;;;;;;;;;;;;;;;;GAkBG;AACI,MAAM,kBAAkB,GAAG,KAAK,EACrC,MAAgC,EACJ,EAAE;IAC9B,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;IAEpD,kBAAkB;IAClB,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,qBAAa,EAAC,KAAK,CAAC,CAAC;IAErB,MAAM,QAAQ,GAAsB,EAAE,CAAC;IACvC,IAAI,gBAAiD,CAAC;IAEtD,wEAAwE;IACxE,wEAAwE;IACxE,sDAAsD;IACtD,GAAG,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAA,yBAAiB,EAAC,KAAK,IAAI,EAAE;YAChD,OAAO,MAAM,MAAM,CAAC,KAAK,CAAC;gBACxB,SAAS,EAAE,SAAS;gBACpB,sBAAsB,EAAE,kBAAkB;gBAC1C,gBAAgB,EAAE,sBAAsB;gBACxC,yBAAyB,EAAE;oBACzB,SAAS,EAAE,MAAM;oBACjB,WAAW,EAAE,UAAU;iBACxB;gBACD,oBAAoB,EAAE,sDAAsD;gBAC5E,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/C,MAAM;QACR,CAAC;QAED,+BAA+B;QAC/B,MAAM,cAAc,GAAsB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACpE,SAAS,EAAE,IAAI,CAAC,SAAmB;YACnC,KAAK,EAAE,IAAI,CAAC,KAAe;YAC3B,SAAS,EAAE,IAAI,CAAC,SAAmB;YACnC,SAAS,EAAE,IAAI,CAAC,SAAmB;YACnC,YAAY,EAAE,IAAI,CAAC,YAAsB;SAC1C,CAAC,CAAC,CAAC;QAEJ,QAAQ,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;QAEjC,gDAAgD;QAChD,IAAI,QAAQ,CAAC,MAAM,IAAI,kCAAyB,EAAE,CAAC;YACjD,MAAM;QACR,CAAC;QAED,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAC7C,CAAC,QAAQ,gBAAgB,EAAE;IAE3B,6CAA6C;IAC7C,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IAEnD,gCAAgC;IAChC,OAAO,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;AACrD,CAAC,CAAC;AA1DW,QAAA,kBAAkB,sBA0D7B"}
+{"version":3,"file":"list-sessions.js","sourceRoot":"","sources":["../../../src/history/actions/list-sessions.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH,yCAAoE;AAEpE,oCAA4E;AAE5E;;;;;;;;;;;;;;;;;;GAkBG;AACI,MAAM,kBAAkB,GAAG,KAAK,EACrC,MAAgC,EACJ,EAAE;IAC9B,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE5D,kBAAkB;IAClB,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,qBAAa,EAAC,KAAK,CAAC,CAAC;IAErB,MAAM,QAAQ,GAAsB,EAAE,CAAC;IACvC,IAAI,gBAAiD,CAAC;IAEtD,wEAAwE;IACxE,wEAAwE;IACxE,sDAAsD;IACtD,GAAG,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAA,yBAAiB,EACpC,KAAK,IAAI,EAAE;YACT,OAAO,MAAM,MAAM,CAAC,KAAK,CAAC;gBACxB,SAAS,EAAE,SAAS;gBACpB,sBAAsB,EAAE,kBAAkB;gBAC1C,gBAAgB,EAAE,sBAAsB;gBACxC,yBAAyB,EAAE;oBACzB,SAAS,EAAE,MAAM;oBACjB,WAAW,EAAE,UAAU;iBACxB;gBACD,oBAAoB,EAAE,sDAAsD;gBAC5E,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;QACL,CAAC,EACD,EAAE,MAAM,EAAE,CACX,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/C,MAAM;QACR,CAAC;QAED,+BAA+B;QAC/B,MAAM,cAAc,GAAsB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACpE,SAAS,EAAE,IAAI,CAAC,SAAmB;YACnC,KAAK,EAAE,IAAI,CAAC,KAAe;YAC3B,SAAS,EAAE,IAAI,CAAC,SAAmB;YACnC,SAAS,EAAE,IAAI,CAAC,SAAmB;YACnC,YAAY,EAAE,IAAI,CAAC,YAAsB;SAC1C,CAAC,CAAC,CAAC;QAEJ,QAAQ,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;QAEjC,8EAA8E;QAC9E,oDAAoD;QACpD,IAAI,QAAQ,CAAC,MAAM,IAAI,kCAAyB,EAAE,CAAC;YACjD,IAAA,kBAAS,GAAE,CAAC,IAAI,CACd,6BAA6B,kCAAyB,sBAAsB,MAAM,KAAK;gBACrF,sFAAsF,CACzF,CAAC;YACF,MAAM;QACR,CAAC;QAED,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAC7C,CAAC,QAAQ,gBAAgB,EAAE;IAE3B,6CAA6C;IAC7C,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IAEnD,gCAAgC;IAChC,OAAO,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;AACrD,CAAC,CAAC;AAlEW,QAAA,kBAAkB,sBAkE7B"}

package/dist/history/index.d.ts

@@ -3,7 +3,16 @@
  * Provides persistent storage for chat messages with automatic title generation
  */
 import { BaseMessage } from '@langchain/core/messages';
+import { DynamoDBSessionChatMessageHistory } from './session-adapter';
 import type { DynamoDBChatMessageHistoryOptions, SessionMetadata } from './types';
+/**
+ * DynamoDB-backed chat message history with per-message-item storage.
+ *
+ * @remarks
+ * Each session is capped at ~999 999 messages by the 6-digit message-index
+ * sort-key padding. Sessions approaching this bound should be sharded by the
+ * caller; see `formatMessageIndex` for details.
+ */
 export declare class DynamoDBChatMessageHistory {
     private readonly ddbClient;
     private readonly client;
@@ -35,7 +44,9 @@ export declare class DynamoDBChatMessageHistory {
      * @returns Array of BaseMessage objects in chronological order
      * @throws Error if the operation fails or validation fails
      */
-    getMessages(userId: string, sessionId: string): Promise<BaseMessage[]>;
+    getMessages(userId: string, sessionId: string, options?: {
+        signal?: AbortSignal;
+    }): Promise<BaseMessage[]>;
     /**
      * Add a single message to a session
      * Generates title from the first message if this is a new session
@@ -46,7 +57,9 @@ export declare class DynamoDBChatMessageHistory {
      * @param title - Optional session title (auto-generated from the first message if not provided)
      * @throws Error if the operation fails or validation fails
      */
-    addMessage(userId: string, sessionId: string, message: BaseMessage, title?: string): Promise<void>;
+    addMessage(userId: string, sessionId: string, message: BaseMessage, title?: string, options?: {
+        signal?: AbortSignal;
+    }): Promise<void>;
     /**
      * Add multiple messages to a session
      * Generates title from the first message if this is a new session
@@ -58,7 +71,9 @@ export declare class DynamoDBChatMessageHistory {
      * @param title - Optional session title (auto-generated from the first message if not provided)
      * @throws Error if the operation fails or validation fails
      */
-    addMessages(userId: string, sessionId: string, messages: BaseMessage[], title?: string): Promise<void>;
+    addMessages(userId: string, sessionId: string, messages: BaseMessage[], title?: string, options?: {
+        signal?: AbortSignal;
+    }): Promise<void>;
     /**
      * Clear all messages in a session
      * Deletes the session item from DynamoDB
@@ -67,7 +82,9 @@ export declare class DynamoDBChatMessageHistory {
      * @param sessionId - Session identifier
      * @throws Error if the operation fails or validation fails
      */
-    clear(userId: string, sessionId: string): Promise<void>;
+    clear(userId: string, sessionId: string, options?: {
+        signal?: AbortSignal;
+    }): Promise<void>;
     /**
      * List all sessions for a user, sorted by most recent
      * Returns metadata only (excludes messages for performance)
@@ -77,6 +94,25 @@ export declare class DynamoDBChatMessageHistory {
      * @returns Array of session metadata, sorted by most recent first
      * @throws Error if the operation fails or validation fails
      */
-    listSessions(userId: string, limit?: number): Promise<SessionMetadata[]>;
+    listSessions(userId: string, limit?: number, options?: {
+        signal?: AbortSignal;
+    }): Promise<SessionMetadata[]>;
+    /**
+     * Bind this store to a single `(userId, sessionId)` pair and return a
+     * `BaseListChatMessageHistory` compatible instance — the shape LangChain's
+     * `RunnableWithMessageHistory` expects from `getMessageHistory(sessionId)`.
+     *
+     * @example
+     * ```ts
+     * const store = new DynamoDBChatMessageHistory({ tableName });
+     * new RunnableWithMessageHistory({
+     *   runnable,
+     *   getMessageHistory: (sessionId) => store.forSession(userId, sessionId),
+     *   // ...
+     * });
+     * ```
+     */
+    forSession(userId: string, sessionId: string): DynamoDBSessionChatMessageHistory;
 }
+export { DynamoDBSessionChatMessageHistory } from './session-adapter';
 //# sourceMappingURL=index.d.ts.map

package/dist/history/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/history/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AASvD,OAAO,KAAK,EAAE,iCAAiC,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAElF,qBAAa,0BAA0B;IACrC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA6B;IACvD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAmB;IAC1C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU;IAErC;;;;;;;;OAQG;gBACS,OAAO,EAAE,iCAAiC;IActD;;;;OAIG;IACH,OAAO,IAAI,IAAI;IAMf;;;;;;;;OAQG;IACG,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAS5E;;;;;;;;;OASG;IACG,UAAU,CACd,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,WAAW,EACpB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAYhB;;;;;;;;;;OAUG;IACG,WAAW,CACf,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,WAAW,EAAE,EACvB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAYhB;;;;;;;OAOG;IACG,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAS7D;;;;;;;;OAQG;IACG,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;CAQ/E"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/history/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAUvD,OAAO,EAAE,iCAAiC,EAAE,MAAM,mBAAmB,CAAC;AACtE,OAAO,KAAK,EAAE,iCAAiC,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAElF;;;;;;;GAOG;AACH,qBAAa,0BAA0B;IACrC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA6B;IACvD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAmB;IAC1C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU;IAErC;;;;;;;;OAQG;gBACS,OAAO,EAAE,iCAAiC;IAUtD;;;;OAIG;IACH,OAAO,IAAI,IAAI;IAMf;;;;;;;;OAQG;IACG,WAAW,CACf,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAO,GACrC,OAAO,CAAC,WAAW,EAAE,CAAC;IAUzB;;;;;;;;;OASG;IACG,UAAU,CACd,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,WAAW,EACpB,KAAK,CAAC,EAAE,MAAM,EACd,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAO,GACrC,OAAO,CAAC,IAAI,CAAC;IAahB;;;;;;;;;;OAUG;IACG,WAAW,CACf,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,WAAW,EAAE,EACvB,KAAK,CAAC,EAAE,MAAM,EACd,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAO,GACrC,OAAO,CAAC,IAAI,CAAC;IAahB;;;;;;;OAOG;IACG,KAAK,CACT,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAO,GACrC,OAAO,CAAC,IAAI,CAAC;IAUhB;;;;;;;;OAQG;IACG,YAAY,CAChB,MAAM,EAAE,MAAM,EACd,KAAK,CAAC,EAAE,MAAM,EACd,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAO,GACrC,OAAO,CAAC,eAAe,EAAE,CAAC;IAU7B;;;;;;;;;;;;;;OAcG;IACH,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,iCAAiC;CASjF;AAED,OAAO,EAAE,iCAAiC,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/history/index.js

@@ -4,10 +4,18 @@
  * Provides persistent storage for chat messages with automatic title generation
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DynamoDBChatMessageHistory = void 0;
-const client_dynamodb_1 = require("@aws-sdk/client-dynamodb");
-const lib_dynamodb_1 = require("@aws-sdk/lib-dynamodb");
+exports.DynamoDBSessionChatMessageHistory = exports.DynamoDBChatMessageHistory = void 0;
+const shared_1 = require("../shared");
 const actions_1 = require("./actions");
+const session_adapter_1 = require("./session-adapter");
+/**
+ * DynamoDB-backed chat message history with per-message-item storage.
+ *
+ * @remarks
+ * Each session is capped at ~999 999 messages by the 6-digit message-index
+ * sort-key padding. Sessions approaching this bound should be sharded by the
+ * caller; see `formatMessageIndex` for details.
+ */
 class DynamoDBChatMessageHistory {
     ddbClient;
     client;
@@ -24,16 +32,11 @@ class DynamoDBChatMessageHistory {
      * @param options.client - Optional pre-built DynamoDBDocument client (takes precedence over clientConfig)
      */
     constructor(options) {
-        if (options.client) {
-            this.client = options.client;
-            this.ddbClient = undefined;
-            this.ownsClient = false;
-        }
-        else {
-            this.ddbClient = new client_dynamodb_1.DynamoDBClient(options.clientConfig || {});
-            this.client = lib_dynamodb_1.DynamoDBDocument.from(this.ddbClient);
-            this.ownsClient = true;
-        }
+        ({
+            ddbClient: this.ddbClient,
+            client: this.client,
+            ownsClient: this.ownsClient,
+        } = (0, shared_1.resolveDynamoDBClient)(options));
         this.tableName = options.tableName;
         this.ttlDays = options.ttlDays;
     }
@@ -56,12 +59,13 @@ class DynamoDBChatMessageHistory {
      * @returns Array of BaseMessage objects in chronological order
      * @throws Error if the operation fails or validation fails
      */
-    async getMessages(userId, sessionId) {
+    async getMessages(userId, sessionId, options = {}) {
         return await (0, actions_1.getMessagesAction)({
             client: this.client,
             tableName: this.tableName,
             userId,
             sessionId,
+            signal: options.signal,
         });
     }
     /**
@@ -74,7 +78,7 @@ class DynamoDBChatMessageHistory {
      * @param title - Optional session title (auto-generated from the first message if not provided)
      * @throws Error if the operation fails or validation fails
      */
-    async addMessage(userId, sessionId, message, title) {
+    async addMessage(userId, sessionId, message, title, options = {}) {
         await (0, actions_1.addMessageAction)({
             client: this.client,
             tableName: this.tableName,
@@ -83,6 +87,7 @@ class DynamoDBChatMessageHistory {
             message,
             title,
             ttlDays: this.ttlDays,
+            signal: options.signal,
         });
     }
     /**
@@ -96,7 +101,7 @@ class DynamoDBChatMessageHistory {
      * @param title - Optional session title (auto-generated from the first message if not provided)
      * @throws Error if the operation fails or validation fails
      */
-    async addMessages(userId, sessionId, messages, title) {
+    async addMessages(userId, sessionId, messages, title, options = {}) {
         await (0, actions_1.addMessagesAction)({
             client: this.client,
             tableName: this.tableName,
@@ -105,6 +110,7 @@ class DynamoDBChatMessageHistory {
             messages,
             title,
             ttlDays: this.ttlDays,
+            signal: options.signal,
         });
     }
     /**
@@ -115,12 +121,13 @@ class DynamoDBChatMessageHistory {
      * @param sessionId - Session identifier
      * @throws Error if the operation fails or validation fails
      */
-    async clear(userId, sessionId) {
+    async clear(userId, sessionId, options = {}) {
         await (0, actions_1.clearAction)({
             client: this.client,
             tableName: this.tableName,
             userId,
             sessionId,
+            signal: options.signal,
         });
     }
     /**
@@ -132,14 +139,41 @@ class DynamoDBChatMessageHistory {
      * @returns Array of session metadata, sorted by most recent first
      * @throws Error if the operation fails or validation fails
      */
-    async listSessions(userId, limit) {
+    async listSessions(userId, limit, options = {}) {
         return await (0, actions_1.listSessionsAction)({
             client: this.client,
             tableName: this.tableName,
             userId,
             limit,
+            signal: options.signal,
+        });
+    }
+    /**
+     * Bind this store to a single `(userId, sessionId)` pair and return a
+     * `BaseListChatMessageHistory` compatible instance — the shape LangChain's
+     * `RunnableWithMessageHistory` expects from `getMessageHistory(sessionId)`.
+     *
+     * @example
+     * ```ts
+     * const store = new DynamoDBChatMessageHistory({ tableName });
+     * new RunnableWithMessageHistory({
+     *   runnable,
+     *   getMessageHistory: (sessionId) => store.forSession(userId, sessionId),
+     *   // ...
+     * });
+     * ```
+     */
+    forSession(userId, sessionId) {
+        return new session_adapter_1.DynamoDBSessionChatMessageHistory({
+            client: this.client,
+            tableName: this.tableName,
+            userId,
+            sessionId,
+            ttlDays: this.ttlDays,
         });
     }
 }
 exports.DynamoDBChatMessageHistory = DynamoDBChatMessageHistory;
+var session_adapter_2 = require("./session-adapter");
+Object.defineProperty(exports, "DynamoDBSessionChatMessageHistory", { enumerable: true, get: function () { return session_adapter_2.DynamoDBSessionChatMessageHistory; } });
 //# sourceMappingURL=index.js.map

package/dist/history/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/history/index.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,8DAA0D;AAC1D,wDAAyD;AAGzD,uCAMmB;AAGnB,MAAa,0BAA0B;IACpB,SAAS,CAA6B;IACtC,MAAM,CAAmB;IACzB,SAAS,CAAS;IAClB,OAAO,CAAU;IACjB,UAAU,CAAU;IAErC;;;;;;;;OAQG;IACH,YAAY,OAA0C;QACpD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAC7B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,SAAS,GAAG,IAAI,gCAAc,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;YAChE,IAAI,CAAC,MAAM,GAAG,+BAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACpD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACH,OAAO;QACL,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,SAAiB;QACjD,OAAO,MAAM,IAAA,2BAAiB,EAAC;YAC7B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;SACV,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,UAAU,CACd,MAAc,EACd,SAAiB,EACjB,OAAoB,EACpB,KAAc;QAEd,MAAM,IAAA,0BAAgB,EAAC;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;YACT,OAAO;YACP,KAAK;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,WAAW,CACf,MAAc,EACd,SAAiB,EACjB,QAAuB,EACvB,KAAc;QAEd,MAAM,IAAA,2BAAiB,EAAC;YACtB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;YACT,QAAQ;YACR,KAAK;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,SAAiB;QAC3C,MAAM,IAAA,qBAAW,EAAC;YAChB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;SACV,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,KAAc;QAC/C,OAAO,MAAM,IAAA,4BAAkB,EAAC;YAC9B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,KAAK;SACN,CAAC,CAAC;IACL,CAAC;CACF;AApJD,gEAoJC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/history/index.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAMH,sCAAkD;AAClD,uCAMmB;AACnB,uDAAsE;AAGtE;;;;;;;GAOG;AACH,MAAa,0BAA0B;IACpB,SAAS,CAA6B;IACtC,MAAM,CAAmB;IACzB,SAAS,CAAS;IAClB,OAAO,CAAU;IACjB,UAAU,CAAU;IAErC;;;;;;;;OAQG;IACH,YAAY,OAA0C;QACpD,CAAC;YACC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,GAAG,IAAA,8BAAqB,EAAC,OAAO,CAAC,CAAC,CAAC;QACpC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACH,OAAO;QACL,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,WAAW,CACf,MAAc,EACd,SAAiB,EACjB,UAAoC,EAAE;QAEtC,OAAO,MAAM,IAAA,2BAAiB,EAAC;YAC7B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;YACT,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,UAAU,CACd,MAAc,EACd,SAAiB,EACjB,OAAoB,EACpB,KAAc,EACd,UAAoC,EAAE;QAEtC,MAAM,IAAA,0BAAgB,EAAC;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;YACT,OAAO;YACP,KAAK;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,WAAW,CACf,MAAc,EACd,SAAiB,EACjB,QAAuB,EACvB,KAAc,EACd,UAAoC,EAAE;QAEtC,MAAM,IAAA,2BAAiB,EAAC;YACtB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;YACT,QAAQ;YACR,KAAK;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,KAAK,CACT,MAAc,EACd,SAAiB,EACjB,UAAoC,EAAE;QAEtC,MAAM,IAAA,qBAAW,EAAC;YAChB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;YACT,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,YAAY,CAChB,MAAc,EACd,KAAc,EACd,UAAoC,EAAE;QAEtC,OAAO,MAAM,IAAA,4BAAkB,EAAC;YAC9B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,KAAK;YACL,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,UAAU,CAAC,MAAc,EAAE,SAAiB;QAC1C,OAAO,IAAI,mDAAiC,CAAC;YAC3C,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM;YACN,SAAS;YACT,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;IACL,CAAC;CACF;AA5LD,gEA4LC;AAED,qDAAsE;AAA7D,oIAAA,iCAAiC,OAAA"}

package/dist/history/session-adapter.d.ts

@@ -0,0 +1,39 @@
+/**
+ * Adapter that exposes a single (userId, sessionId) pair through the
+ * LangChain `BaseListChatMessageHistory` contract so the DynamoDB-backed
+ * history can be plugged into `RunnableWithMessageHistory` and anywhere else
+ * that expects the standard interface.
+ *
+ * Use {@link DynamoDBChatMessageHistory.forSession} to obtain one; see that
+ * method's JSDoc for the LangChain integration pattern.
+ */
+import type { DynamoDBDocument } from '@aws-sdk/lib-dynamodb';
+import { BaseListChatMessageHistory } from '@langchain/core/chat_history';
+import type { BaseMessage } from '@langchain/core/messages';
+export interface DynamoDBSessionChatMessageHistoryParams {
+    client: DynamoDBDocument;
+    tableName: string;
+    userId: string;
+    sessionId: string;
+    ttlDays?: number;
+}
+/**
+ * Single-session LangChain `BaseListChatMessageHistory` backed by the same
+ * DynamoDB table used by `DynamoDBChatMessageHistory`. Instances close over a
+ * specific `(userId, sessionId)` pair; pass through
+ * `RunnableWithMessageHistory` via a `GetSessionHistoryCallable`.
+ */
+export declare class DynamoDBSessionChatMessageHistory extends BaseListChatMessageHistory {
+    lc_namespace: string[];
+    private readonly client;
+    private readonly tableName;
+    private readonly userId;
+    private readonly sessionId;
+    private readonly ttlDays?;
+    constructor(params: DynamoDBSessionChatMessageHistoryParams);
+    getMessages(): Promise<BaseMessage[]>;
+    addMessage(message: BaseMessage): Promise<void>;
+    addMessages(messages: BaseMessage[]): Promise<void>;
+    clear(): Promise<void>;
+}
+//# sourceMappingURL=session-adapter.d.ts.map

package/dist/history/session-adapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-adapter.d.ts","sourceRoot":"","sources":["../../src/history/session-adapter.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,EAAE,0BAA0B,EAAE,MAAM,8BAA8B,CAAC;AAC1E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAI5D,MAAM,WAAW,uCAAuC;IACtD,MAAM,EAAE,gBAAgB,CAAC;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;;;;GAKG;AACH,qBAAa,iCAAkC,SAAQ,0BAA0B;IAC/E,YAAY,WAAkD;IAE9D,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAmB;IAC1C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAS;gBAEtB,MAAM,EAAE,uCAAuC;IASrD,WAAW,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IASrC,UAAU,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAW/C,WAAW,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAWnD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAQ7B"}

package/dist/history/session-adapter.js

@@ -0,0 +1,74 @@
+"use strict";
+/**
+ * Adapter that exposes a single (userId, sessionId) pair through the
+ * LangChain `BaseListChatMessageHistory` contract so the DynamoDB-backed
+ * history can be plugged into `RunnableWithMessageHistory` and anywhere else
+ * that expects the standard interface.
+ *
+ * Use {@link DynamoDBChatMessageHistory.forSession} to obtain one; see that
+ * method's JSDoc for the LangChain integration pattern.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DynamoDBSessionChatMessageHistory = void 0;
+const chat_history_1 = require("@langchain/core/chat_history");
+const actions_1 = require("./actions");
+/**
+ * Single-session LangChain `BaseListChatMessageHistory` backed by the same
+ * DynamoDB table used by `DynamoDBChatMessageHistory`. Instances close over a
+ * specific `(userId, sessionId)` pair; pass through
+ * `RunnableWithMessageHistory` via a `GetSessionHistoryCallable`.
+ */
+class DynamoDBSessionChatMessageHistory extends chat_history_1.BaseListChatMessageHistory {
+    lc_namespace = ['langchain', 'stores', 'message', 'dynamodb'];
+    client;
+    tableName;
+    userId;
+    sessionId;
+    ttlDays;
+    constructor(params) {
+        super();
+        this.client = params.client;
+        this.tableName = params.tableName;
+        this.userId = params.userId;
+        this.sessionId = params.sessionId;
+        this.ttlDays = params.ttlDays;
+    }
+    async getMessages() {
+        return await (0, actions_1.getMessagesAction)({
+            client: this.client,
+            tableName: this.tableName,
+            userId: this.userId,
+            sessionId: this.sessionId,
+        });
+    }
+    async addMessage(message) {
+        await (0, actions_1.addMessageAction)({
+            client: this.client,
+            tableName: this.tableName,
+            userId: this.userId,
+            sessionId: this.sessionId,
+            message,
+            ttlDays: this.ttlDays,
+        });
+    }
+    async addMessages(messages) {
+        await (0, actions_1.addMessagesAction)({
+            client: this.client,
+            tableName: this.tableName,
+            userId: this.userId,
+            sessionId: this.sessionId,
+            messages,
+            ttlDays: this.ttlDays,
+        });
+    }
+    async clear() {
+        await (0, actions_1.clearAction)({
+            client: this.client,
+            tableName: this.tableName,
+            userId: this.userId,
+            sessionId: this.sessionId,
+        });
+    }
+}
+exports.DynamoDBSessionChatMessageHistory = DynamoDBSessionChatMessageHistory;
+//# sourceMappingURL=session-adapter.js.map

package/dist/history/session-adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-adapter.js","sourceRoot":"","sources":["../../src/history/session-adapter.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AAGH,+DAA0E;AAG1E,uCAAgG;AAUhG;;;;;GAKG;AACH,MAAa,iCAAkC,SAAQ,yCAA0B;IAC/E,YAAY,GAAG,CAAC,WAAW,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;IAE7C,MAAM,CAAmB;IACzB,SAAS,CAAS;IAClB,MAAM,CAAS;IACf,SAAS,CAAS;IAClB,OAAO,CAAU;IAElC,YAAY,MAA+C;QACzD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,WAAW;QACf,OAAO,MAAM,IAAA,2BAAiB,EAAC;YAC7B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAoB;QACnC,MAAM,IAAA,0BAAgB,EAAC;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO;YACP,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAuB;QACvC,MAAM,IAAA,2BAAiB,EAAC;YACtB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ;YACR,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAA,qBAAW,EAAC;YAChB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;IACL,CAAC;CACF;AAzDD,8EAyDC"}