@farukada/aws-langgraph-dynamodb-ts 0.0.10 → 0.1.0

package/dist/checkpointer/types/index.d.ts

@@ -1,7 +1,8 @@
-import { Checkpoint, CheckpointMetadata, PendingWrite, SerializerProtocol } from '@langchain/langgraph-checkpoint';
 import type { DynamoDBClientConfig } from '@aws-sdk/client-dynamodb';
 import { DynamoDBDocument } from '@aws-sdk/lib-dynamodb';
 import { RunnableConfig } from '@langchain/core/runnables';
+import { Checkpoint, CheckpointMetadata, PendingWrite, SerializerProtocol } from '@langchain/langgraph-checkpoint';
+import type { CompressionConfig, Compressor, S3OffloadConfig, S3Offloader } from '../../shared';
 /**
  * Configuration options for DynamoDBSaver
  *
@@ -15,8 +16,16 @@ export interface DynamoDBSaverOptions {
     checkpointsTableName: string;
     writesTableName: string;
     ttlDays?: number;
+    /** TTL in seconds for automatic item expiration (overrides ttlDays if both set) */
+    ttlSeconds?: number;
+    /** Optional compression configuration for checkpoint data */
+    compression?: CompressionConfig;
+    /** Optional S3 offloading for payloads exceeding DynamoDB's 400KB item limit */
+    s3OffloadConfig?: S3OffloadConfig;
     serde?: SerializerProtocol;
     clientConfig?: DynamoDBClientConfig;
+    /** Optional pre-built DynamoDBDocument client (takes precedence over clientConfig) */
+    client?: DynamoDBDocument;
 }
 /**
  * Properties for creating a Writer instance
@@ -56,17 +65,31 @@ export interface DynamoDBWriteItem {
     channel: string;
     type: string;
     value: Uint8Array;
+    /** Optional TTL timestamp for automatic item expiration */
+    ttl?: number;
+    /** S3 key reference when value was offloaded to S3 */
+    s3_value_key?: string;
 }
 /**
- * DynamoDB representation of a checkpoint
+ * Sort key prefix for payload items in the checkpoints table.
+ * Payload items store the heavy checkpoint blob separately from metadata
+ * to reduce RCU consumption on list() queries.
+ *
+ * UUID checkpoint IDs (hex chars 0-9, a-f) sort before 'PAYLOAD#' in ASCII,
+ * so key condition `checkpoint_id < PAYLOAD#` naturally excludes payload items.
+ */
+export declare const PAYLOAD_SK_PREFIX = "PAYLOAD#";
+/**
+ * DynamoDB representation of a checkpoint metadata item.
+ * Stores lightweight metadata; the heavy checkpoint blob lives in a separate payload item.
  *
  * @property thread_id - Unique identifier for the conversation thread (partition key)
  * @property checkpoint_ns - Namespace for checkpoint isolation
  * @property checkpoint_id - Unique identifier for the checkpoint (sort key)
  * @property parent_checkpoint_id - Optional ID of the parent checkpoint for checkpoint chains
  * @property type - Type identifier for deserialization
- * @property checkpoint - Serialized checkpoint data as binary
  * @property metadata - Serialized metadata as binary
+ * @property checkpoint - Present only for backward compatibility with legacy single-item format
  */
 export interface CheckpointItem {
     thread_id: string;
@@ -74,8 +97,29 @@ export interface CheckpointItem {
     checkpoint_id: string;
     parent_checkpoint_id?: string;
     type: string;
-    checkpoint: Uint8Array;
     metadata: Uint8Array;
+    /** S3 key reference when checkpoint data was offloaded to S3 */
+    s3_checkpoint_key?: string;
+    /** S3 key reference when metadata was offloaded to S3 */
+    s3_metadata_key?: string;
+    /**
+     * Legacy field: present only on old single-item checkpoint items.
+     * New items store the checkpoint blob in a separate payload item.
+     */
+    checkpoint?: Uint8Array;
+}
+/**
+ * DynamoDB representation of a checkpoint payload item.
+ * Contains only the heavy serialized checkpoint blob.
+ *
+ * @property thread_id - Partition key (same as metadata item)
+ * @property checkpoint_id - Sort key with PAYLOAD# prefix
+ * @property checkpoint - Serialized checkpoint data as binary
+ */
+export interface CheckpointPayloadItem {
+    thread_id: string;
+    checkpoint_id: string;
+    checkpoint: Uint8Array;
 }
 /**
  * Validated configurable parameters from RunnableConfig
@@ -102,6 +146,8 @@ export interface DeleteThreadActionParams {
     checkpointsTableName: string;
     writesTableName: string;
     threadId: string;
+    /** Optional S3 offloader for cleaning up offloaded payloads */
+    s3Offloader?: S3Offloader;
 }
 /**
  * Parameters for getTuple action
@@ -118,6 +164,10 @@ export interface GetTupleActionParams {
     writesTableName: string;
     serde: SerializerProtocol;
     config: RunnableConfig;
+    /** Optional compressor for decompressing checkpoint data */
+    compressor?: Compressor;
+    /** Optional S3 offloader for downloading offloaded payloads */
+    s3Offloader?: S3Offloader;
 }
 /**
  * Parameters for put action
@@ -138,6 +188,12 @@ export interface PutActionParams {
     checkpoint: Checkpoint;
     metadata: CheckpointMetadata;
     ttlDays?: number;
+    /** TTL in seconds (overrides ttlDays if both set) */
+    ttlSeconds?: number;
+    /** Optional compressor for compressing checkpoint data */
+    compressor?: Compressor;
+    /** Optional S3 offloader for large payloads */
+    s3Offloader?: S3Offloader;
 }
 /**
  * Parameters for putWrites action
@@ -158,5 +214,11 @@ export interface PutWritesActionParams {
     writes: PendingWrite[];
     taskId: string;
     ttlDays?: number;
+    /** TTL in seconds (overrides ttlDays if both set) */
+    ttlSeconds?: number;
+    /** Optional compressor for compressing write data */
+    compressor?: Compressor;
+    /** Optional S3 offloader for large write payloads */
+    s3Offloader?: S3Offloader;
 }
 //# sourceMappingURL=index.d.ts.map

package/dist/checkpointer/types/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/checkpointer/types/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,EACV,kBAAkB,EAClB,YAAY,EACZ,kBAAkB,EACnB,MAAM,iCAAiC,CAAC;AACzC,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAE3D;;;;;;;;GAQG;AACH,MAAM,WAAW,oBAAoB;IACnC,oBAAoB,EAAE,MAAM,CAAC;IAC7B,eAAe,EAAE,MAAM,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,kBAAkB,CAAC;IAC3B,YAAY,CAAC,EAAE,oBAAoB,CAAC;CACrC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,UAAU,CAAC;CACnB;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW,iBAAiB;IAChC,qCAAqC,EAAE,MAAM,CAAC;IAC9C,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,UAAU,CAAC;CACnB;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,cAAc;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,UAAU,CAAC;IACvB,QAAQ,EAAE,UAAU,CAAC;CACtB;AAED;;;;;;GAMG;AACH,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,GAAG,SAAS,CAAC;CACnC;AAED;;;;;;;GAOG;AACH,MAAM,WAAW,wBAAwB;IACvC,MAAM,EAAE,gBAAgB,CAAC;IACzB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;;;;;;;GAQG;AACH,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,gBAAgB,CAAC;IACzB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,eAAe,EAAE,MAAM,CAAC;IACxB,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,EAAE,cAAc,CAAC;CACxB;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,gBAAgB,CAAC;IACzB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,EAAE,cAAc,CAAC;IACvB,UAAU,EAAE,UAAU,CAAC;IACvB,QAAQ,EAAE,kBAAkB,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,gBAAgB,CAAC;IACzB,eAAe,EAAE,MAAM,CAAC;IACxB,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,EAAE,cAAc,CAAC;IACvB,MAAM,EAAE,YAAY,EAAE,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/checkpointer/types/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EACL,UAAU,EACV,kBAAkB,EAClB,YAAY,EACZ,kBAAkB,EACnB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAEhG;;;;;;;;GAQG;AACH,MAAM,WAAW,oBAAoB;IACnC,oBAAoB,EAAE,MAAM,CAAC;IAC7B,eAAe,EAAE,MAAM,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,mFAAmF;IACnF,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,6DAA6D;IAC7D,WAAW,CAAC,EAAE,iBAAiB,CAAC;IAChC,gFAAgF;IAChF,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,KAAK,CAAC,EAAE,kBAAkB,CAAC;IAC3B,YAAY,CAAC,EAAE,oBAAoB,CAAC;IACpC,sFAAsF;IACtF,MAAM,CAAC,EAAE,gBAAgB,CAAC;CAC3B;AAED;;;;;;;;;;;GAWG;AACH,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,UAAU,CAAC;CACnB;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW,iBAAiB;IAChC,qCAAqC,EAAE,MAAM,CAAC;IAC9C,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,UAAU,CAAC;IAClB,2DAA2D;IAC3D,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,sDAAsD;IACtD,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;;;;;;GAOG;AACH,eAAO,MAAM,iBAAiB,aAAa,CAAC;AAE5C;;;;;;;;;;;GAWG;AACH,MAAM,WAAW,cAAc;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,UAAU,CAAC;IACrB,gEAAgE;IAChE,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,yDAAyD;IACzD,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;;OAGG;IACH,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB;AAED;;;;;;;GAOG;AACH,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,UAAU,CAAC;CACxB;AAED;;;;;;GAMG;AACH,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,GAAG,SAAS,CAAC;CACnC;AAED;;;;;;;GAOG;AACH,MAAM,WAAW,wBAAwB;IACvC,MAAM,EAAE,gBAAgB,CAAC;IACzB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,+DAA+D;IAC/D,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED;;;;;;;;GAQG;AACH,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,gBAAgB,CAAC;IACzB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,eAAe,EAAE,MAAM,CAAC;IACxB,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,EAAE,cAAc,CAAC;IACvB,4DAA4D;IAC5D,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,+DAA+D;IAC/D,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,gBAAgB,CAAC;IACzB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,EAAE,cAAc,CAAC;IACvB,UAAU,EAAE,UAAU,CAAC;IACvB,QAAQ,EAAE,kBAAkB,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,qDAAqD;IACrD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,0DAA0D;IAC1D,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,+CAA+C;IAC/C,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,gBAAgB,CAAC;IACzB,eAAe,EAAE,MAAM,CAAC;IACxB,KAAK,EAAE,kBAAkB,CAAC;IAC1B,MAAM,EAAE,cAAc,CAAC;IACvB,MAAM,EAAE,YAAY,EAAE,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,qDAAqD;IACrD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,qDAAqD;IACrD,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,qDAAqD;IACrD,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B"}

package/dist/checkpointer/types/index.js

@@ -1,3 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.PAYLOAD_SK_PREFIX = void 0;
+/**
+ * Sort key prefix for payload items in the checkpoints table.
+ * Payload items store the heavy checkpoint blob separately from metadata
+ * to reduce RCU consumption on list() queries.
+ *
+ * UUID checkpoint IDs (hex chars 0-9, a-f) sort before 'PAYLOAD#' in ASCII,
+ * so key condition `checkpoint_id < PAYLOAD#` naturally excludes payload items.
+ */
+exports.PAYLOAD_SK_PREFIX = 'PAYLOAD#';
 //# sourceMappingURL=index.js.map

package/dist/checkpointer/types/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/checkpointer/types/index.ts"],"names":[],"mappings":""}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/checkpointer/types/index.ts"],"names":[],"mappings":";;;AAkFA;;;;;;;GAOG;AACU,QAAA,iBAAiB,GAAG,UAAU,CAAC"}

package/dist/checkpointer/utils/deserialization.d.ts

@@ -3,13 +3,18 @@
  * Provides centralized checkpoint tuple conversion
  */
 import type { CheckpointTuple, SerializerProtocol } from '@langchain/langgraph-checkpoint';
+import type { Compressor, S3Offloader } from '../../shared';
 import type { CheckpointItem } from '../types';
 /**
- * Deserialize a DynamoDB checkpoint item into a CheckpointTuple
+ * Deserialize a DynamoDB checkpoint metadata item into a CheckpointTuple.
+ * The heavy checkpoint blob is provided separately (from a payload item or S3).
  *
- * @param item - DynamoDB checkpoint item
+ * @param item - DynamoDB checkpoint metadata item
+ * @param checkpointData - Raw checkpoint blob (from payload item, S3, or legacy inline field)
  * @param serde - Serializer protocol for deserialization
+ * @param compressor - Optional compressor for decompression (auto-detects gzip)
+ * @param s3Offloader - Optional S3 offloader for downloading offloaded metadata
  * @returns CheckpointTuple with deserialized checkpoint and metadata
  */
-export declare function deserializeCheckpointTuple(item: CheckpointItem, serde: SerializerProtocol): Promise<CheckpointTuple>;
+export declare function deserializeCheckpointTuple(item: CheckpointItem, checkpointData: Uint8Array, serde: SerializerProtocol, compressor?: Compressor, s3Offloader?: S3Offloader): Promise<CheckpointTuple>;
 //# sourceMappingURL=deserialization.d.ts.map

package/dist/checkpointer/utils/deserialization.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"deserialization.d.ts","sourceRoot":"","sources":["../../../src/checkpointer/utils/deserialization.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAGV,eAAe,EACf,kBAAkB,EACnB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE/C;;;;;;GAMG;AACH,wBAAsB,0BAA0B,CAC9C,IAAI,EAAE,cAAc,EACpB,KAAK,EAAE,kBAAkB,GACxB,OAAO,CAAC,eAAe,CAAC,CAwB1B"}
+{"version":3,"file":"deserialization.d.ts","sourceRoot":"","sources":["../../../src/checkpointer/utils/deserialization.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAGV,eAAe,EACf,kBAAkB,EACnB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE/C;;;;;;;;;;GAUG;AACH,wBAAsB,0BAA0B,CAC9C,IAAI,EAAE,cAAc,EACpB,cAAc,EAAE,UAAU,EAC1B,KAAK,EAAE,kBAAkB,EACzB,UAAU,CAAC,EAAE,UAAU,EACvB,WAAW,CAAC,EAAE,WAAW,GACxB,OAAO,CAAC,eAAe,CAAC,CAyC1B"}

package/dist/checkpointer/utils/deserialization.js

@@ -6,15 +6,33 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.deserializeCheckpointTuple = deserializeCheckpointTuple;
 /**
- * Deserialize a DynamoDB checkpoint item into a CheckpointTuple
+ * Deserialize a DynamoDB checkpoint metadata item into a CheckpointTuple.
+ * The heavy checkpoint blob is provided separately (from a payload item or S3).
  *
- * @param item - DynamoDB checkpoint item
+ * @param item - DynamoDB checkpoint metadata item
+ * @param checkpointData - Raw checkpoint blob (from payload item, S3, or legacy inline field)
  * @param serde - Serializer protocol for deserialization
+ * @param compressor - Optional compressor for decompression (auto-detects gzip)
+ * @param s3Offloader - Optional S3 offloader for downloading offloaded metadata
  * @returns CheckpointTuple with deserialized checkpoint and metadata
  */
-async function deserializeCheckpointTuple(item, serde) {
-    const checkpoint = (await serde.loadsTyped(item.type, item.checkpoint));
-    const metadata = (await serde.loadsTyped(item.type, item.metadata));
+async function deserializeCheckpointTuple(item, checkpointData, serde, compressor, s3Offloader) {
+    // Download checkpoint from S3 if offloaded (checked first to maintain original S3 call order)
+    let rawCheckpoint = checkpointData;
+    if (item.s3_checkpoint_key && s3Offloader) {
+        rawCheckpoint = await s3Offloader.download(item.s3_checkpoint_key);
+    }
+    // Download metadata from S3 if offloaded, otherwise use DynamoDB data
+    let rawMetadata = item.metadata;
+    if (item.s3_metadata_key && s3Offloader) {
+        rawMetadata = await s3Offloader.download(item.s3_metadata_key);
+    }
+    const decompressedCheckpoint = compressor
+        ? await compressor.decompress(rawCheckpoint)
+        : rawCheckpoint;
+    const decompressedMetadata = compressor ? await compressor.decompress(rawMetadata) : rawMetadata;
+    const checkpoint = (await serde.loadsTyped(item.type, decompressedCheckpoint));
+    const metadata = (await serde.loadsTyped(item.type, decompressedMetadata));
     return {
         config: {
             configurable: {

package/dist/checkpointer/utils/deserialization.js.map

@@ -1 +1 @@
-{"version":3,"file":"deserialization.js","sourceRoot":"","sources":["../../../src/checkpointer/utils/deserialization.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAkBH,gEA2BC;AAlCD;;;;;;GAMG;AACI,KAAK,UAAU,0BAA0B,CAC9C,IAAoB,EACpB,KAAyB;IAEzB,MAAM,UAAU,GAAG,CAAC,MAAM,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,CAAe,CAAC;IACtF,MAAM,QAAQ,GAAG,CAAC,MAAM,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAuB,CAAC;IAE1F,OAAO;QACL,MAAM,EAAE;YACN,YAAY,EAAE;gBACZ,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,aAAa,EAAE,IAAI,CAAC,aAAa;aAClC;SACF;QACD,UAAU;QACV,QAAQ;QACR,YAAY,EAAE,IAAI,CAAC,oBAAoB;YACrC,CAAC,CAAC;gBACE,YAAY,EAAE;oBACZ,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,aAAa,EAAE,IAAI,CAAC,aAAa;oBACjC,aAAa,EAAE,IAAI,CAAC,oBAAoB;iBACzC;aACF;YACH,CAAC,CAAC,SAAS;KACd,CAAC;AACJ,CAAC"}
+{"version":3,"file":"deserialization.js","sourceRoot":"","sources":["../../../src/checkpointer/utils/deserialization.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAuBH,gEA+CC;AA1DD;;;;;;;;;;GAUG;AACI,KAAK,UAAU,0BAA0B,CAC9C,IAAoB,EACpB,cAA0B,EAC1B,KAAyB,EACzB,UAAuB,EACvB,WAAyB;IAEzB,8FAA8F;IAC9F,IAAI,aAAa,GAAe,cAAc,CAAC;IAC/C,IAAI,IAAI,CAAC,iBAAiB,IAAI,WAAW,EAAE,CAAC;QAC1C,aAAa,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACrE,CAAC;IAED,sEAAsE;IACtE,IAAI,WAAW,GAAe,IAAI,CAAC,QAAQ,CAAC;IAC5C,IAAI,IAAI,CAAC,eAAe,IAAI,WAAW,EAAE,CAAC;QACxC,WAAW,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,sBAAsB,GAAG,UAAU;QACvC,CAAC,CAAC,MAAM,UAAU,CAAC,UAAU,CAAC,aAAa,CAAC;QAC5C,CAAC,CAAC,aAAa,CAAC;IAClB,MAAM,oBAAoB,GAAG,UAAU,CAAC,CAAC,CAAC,MAAM,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;IAEjG,MAAM,UAAU,GAAG,CAAC,MAAM,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAe,CAAC;IAC7F,MAAM,QAAQ,GAAG,CAAC,MAAM,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAuB,CAAC;IAEjG,OAAO;QACL,MAAM,EAAE;YACN,YAAY,EAAE;gBACZ,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,aAAa,EAAE,IAAI,CAAC,aAAa;aAClC;SACF;QACD,UAAU;QACV,QAAQ;QACR,YAAY,EAAE,IAAI,CAAC,oBAAoB;YACrC,CAAC,CAAC;gBACE,YAAY,EAAE;oBACZ,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,aAAa,EAAE,IAAI,CAAC,aAAa;oBACjC,aAAa,EAAE,IAAI,CAAC,oBAAoB;iBACzC;aACF;YACH,CAAC,CAAC,SAAS;KACd,CAAC;AACJ,CAAC"}

package/dist/checkpointer/utils/validation.d.ts

@@ -40,7 +40,7 @@ export declare function validateTTLDays(ttlDays: number | undefined): void;
  * Export constants for use in other modules
  */
 export declare const CheckpointerValidationConstants: {
-    readonly MAX_DELETE_BATCH_SIZE: 100;
+    readonly MAX_DELETE_BATCH_SIZE: 1000;
     readonly MAX_WRITES_PER_BATCH: 1000;
     readonly MAX_LIST_LIMIT: 1000;
     readonly SEPARATOR: ":::";

package/dist/checkpointer/utils/validation.js

@@ -20,7 +20,7 @@ const MAX_TASK_ID_LENGTH = 256;
 const MAX_CHANNEL_LENGTH = 256;
 const MAX_WRITES_PER_BATCH = 1000;
 const MAX_LIST_LIMIT = 1000;
-const MAX_DELETE_BATCH_SIZE = 100; // Maximum checkpoints to delete at once
+const MAX_DELETE_BATCH_SIZE = 1000; // Maximum items (metadata + payload) to delete at once
 const SEPARATOR = ':::';
 class CheckpointerValidationError extends Error {
     constructor(message) {

package/dist/checkpointer/utils/validation.js.map

@@ -1 +1 @@
-{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../../src/checkpointer/utils/validation.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAwBH,4CAyBC;AAKD,oDAgCC;AAKD,oDAyBC;AAKD,wCAmBC;AAKD,0CAcC;AAKD,kDAcC;AAKD,8CAgBC;AAKD,0CAOC;AAjND,8CAA0E;AAE1E,MAAM,oBAAoB,GAAG,GAAG,CAAC;AACjC,MAAM,wBAAwB,GAAG,GAAG,CAAC;AACrC,MAAM,wBAAwB,GAAG,GAAG,CAAC;AACrC,MAAM,kBAAkB,GAAG,GAAG,CAAC;AAC/B,MAAM,kBAAkB,GAAG,GAAG,CAAC;AAC/B,MAAM,oBAAoB,GAAG,IAAI,CAAC;AAClC,MAAM,cAAc,GAAG,IAAI,CAAC;AAC5B,MAAM,qBAAqB,GAAG,GAAG,CAAC,CAAC,wCAAwC;AAC3E,MAAM,SAAS,GAAG,KAAK,CAAC;AAExB,MAAa,2BAA4B,SAAQ,KAAK;IACpD,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,6BAA6B,CAAC;IAC5C,CAAC;CACF;AALD,kEAKC;AAED;;GAEG;AACH,SAAgB,gBAAgB,CAAC,QAAa;IAC5C,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,MAAM,IAAI,2BAA2B,CAAC,4BAA4B,CAAC,CAAC;IACtE,CAAC;IAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,2BAA2B,CAAC,2BAA2B,CAAC,CAAC;IACrE,CAAC;IAED,IAAI,QAAQ,CAAC,MAAM,GAAG,oBAAoB,EAAE,CAAC;QAC3C,MAAM,IAAI,2BAA2B,CACnC,uCAAuC,oBAAoB,aAAa,CACzE,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACjC,MAAM,IAAI,2BAA2B,CAAC,uCAAuC,SAAS,GAAG,CAAC,CAAC;IAC7F,CAAC;IAED,4CAA4C;IAC5C,4CAA4C;IAC5C,IAAI,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,6CAA6C,CAAC,CAAC;IACvF,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAAC,YAAiB,EAAE,WAAoB,KAAK;IAC/E,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC/B,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,IAAI,2BAA2B,CAAC,2BAA2B,CAAC,CAAC;QACrE,CAAC;QACD,OAAO;IACT,CAAC;IAED,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,gCAAgC,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,MAAM,IAAI,2BAA2B,CAAC,+BAA+B,CAAC,CAAC;IACzE,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,GAAG,wBAAwB,EAAE,CAAC;QACnD,MAAM,IAAI,2BAA2B,CACnC,2CAA2C,wBAAwB,aAAa,CACjF,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,2CAA2C,SAAS,GAAG,CAAC,CAAC;IACjG,CAAC;IAED,6BAA6B;IAC7B,4CAA4C;IAC5C,IAAI,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,2BAA2B,CAAC,iDAAiD,CAAC,CAAC;IAC3F,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAAC,YAAiB;IACpD,IAAI,YAAY,KAAK,SAAS,IAAI,YAAY,KAAK,EAAE,EAAE,CAAC;QACtD,OAAO,CAAC,6BAA6B;IACvC,CAAC;IAED,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,gCAAgC,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,GAAG,wBAAwB,EAAE,CAAC;QACnD,MAAM,IAAI,2BAA2B,CACnC,2CAA2C,wBAAwB,aAAa,CACjF,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,2CAA2C,SAAS,GAAG,CAAC,CAAC;IACjG,CAAC;IAED,6BAA6B;IAC7B,4CAA4C;IAC5C,IAAI,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,2BAA2B,CAAC,iDAAiD,CAAC,CAAC;IAC3F,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,cAAc,CAAC,MAAW;IACxC,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,MAAM,IAAI,2BAA2B,CAAC,0BAA0B,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,2BAA2B,CAAC,yBAAyB,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,GAAG,kBAAkB,EAAE,CAAC;QACvC,MAAM,IAAI,2BAA2B,CACnC,qCAAqC,kBAAkB,aAAa,CACrE,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,2BAA2B,CAAC,qCAAqC,SAAS,GAAG,CAAC,CAAC;IAC3F,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,eAAe,CAAC,OAAY;IAC1C,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,MAAM,IAAI,2BAA2B,CAAC,0BAA0B,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,2BAA2B,CAAC,yBAAyB,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,GAAG,kBAAkB,EAAE,CAAC;QACxC,MAAM,IAAI,2BAA2B,CACnC,qCAAqC,kBAAkB,aAAa,CACrE,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,mBAAmB,CAAC,WAAgB;IAClD,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,2BAA2B,CAAC,iCAAiC,CAAC,CAAC;IAC3E,CAAC;IAED,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACpB,MAAM,IAAI,2BAA2B,CAAC,iCAAiC,CAAC,CAAC;IAC3E,CAAC;IAED,IAAI,WAAW,GAAG,oBAAoB,EAAE,CAAC;QACvC,MAAM,IAAI,2BAA2B,CACnC,iBAAiB,WAAW,wBAAwB,oBAAoB,EAAE,CAC3E,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,iBAAiB,CAAC,KAAU;IAC1C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,OAAO;IACT,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1D,MAAM,IAAI,2BAA2B,CAAC,0BAA0B,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACf,MAAM,IAAI,2BAA2B,CAAC,wBAAwB,CAAC,CAAC;IAClE,CAAC;IAED,IAAI,KAAK,GAAG,cAAc,EAAE,CAAC;QAC3B,MAAM,IAAI,2BAA2B,CAAC,uBAAuB,cAAc,EAAE,CAAC,CAAC;IACjF,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,eAAe,CAAC,OAA2B;IACzD,IAAI,CAAC;QACH,IAAA,uBAAiB,EAAC,OAAO,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,uDAAuD;QACvD,MAAM,IAAI,2BAA2B,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAChG,CAAC;AACH,CAAC;AAED;;GAEG;AACU,QAAA,+BAA+B,GAAG;IAC7C,qBAAqB;IACrB,oBAAoB;IACpB,cAAc;IACd,SAAS;CACD,CAAC"}
+{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../../src/checkpointer/utils/validation.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAwBH,4CAyBC;AAKD,oDAgCC;AAKD,oDAyBC;AAKD,wCAmBC;AAKD,0CAcC;AAKD,kDAcC;AAKD,8CAgBC;AAKD,0CAOC;AAjND,8CAA0E;AAE1E,MAAM,oBAAoB,GAAG,GAAG,CAAC;AACjC,MAAM,wBAAwB,GAAG,GAAG,CAAC;AACrC,MAAM,wBAAwB,GAAG,GAAG,CAAC;AACrC,MAAM,kBAAkB,GAAG,GAAG,CAAC;AAC/B,MAAM,kBAAkB,GAAG,GAAG,CAAC;AAC/B,MAAM,oBAAoB,GAAG,IAAI,CAAC;AAClC,MAAM,cAAc,GAAG,IAAI,CAAC;AAC5B,MAAM,qBAAqB,GAAG,IAAI,CAAC,CAAC,uDAAuD;AAC3F,MAAM,SAAS,GAAG,KAAK,CAAC;AAExB,MAAa,2BAA4B,SAAQ,KAAK;IACpD,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,6BAA6B,CAAC;IAC5C,CAAC;CACF;AALD,kEAKC;AAED;;GAEG;AACH,SAAgB,gBAAgB,CAAC,QAAa;IAC5C,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,MAAM,IAAI,2BAA2B,CAAC,4BAA4B,CAAC,CAAC;IACtE,CAAC;IAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,2BAA2B,CAAC,2BAA2B,CAAC,CAAC;IACrE,CAAC;IAED,IAAI,QAAQ,CAAC,MAAM,GAAG,oBAAoB,EAAE,CAAC;QAC3C,MAAM,IAAI,2BAA2B,CACnC,uCAAuC,oBAAoB,aAAa,CACzE,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACjC,MAAM,IAAI,2BAA2B,CAAC,uCAAuC,SAAS,GAAG,CAAC,CAAC;IAC7F,CAAC;IAED,4CAA4C;IAC5C,4CAA4C;IAC5C,IAAI,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,6CAA6C,CAAC,CAAC;IACvF,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAAC,YAAiB,EAAE,WAAoB,KAAK;IAC/E,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC/B,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,IAAI,2BAA2B,CAAC,2BAA2B,CAAC,CAAC;QACrE,CAAC;QACD,OAAO;IACT,CAAC;IAED,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,gCAAgC,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,MAAM,IAAI,2BAA2B,CAAC,+BAA+B,CAAC,CAAC;IACzE,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,GAAG,wBAAwB,EAAE,CAAC;QACnD,MAAM,IAAI,2BAA2B,CACnC,2CAA2C,wBAAwB,aAAa,CACjF,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,2CAA2C,SAAS,GAAG,CAAC,CAAC;IACjG,CAAC;IAED,6BAA6B;IAC7B,4CAA4C;IAC5C,IAAI,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,2BAA2B,CAAC,iDAAiD,CAAC,CAAC;IAC3F,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAAC,YAAiB;IACpD,IAAI,YAAY,KAAK,SAAS,IAAI,YAAY,KAAK,EAAE,EAAE,CAAC;QACtD,OAAO,CAAC,6BAA6B;IACvC,CAAC;IAED,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,gCAAgC,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,GAAG,wBAAwB,EAAE,CAAC;QACnD,MAAM,IAAI,2BAA2B,CACnC,2CAA2C,wBAAwB,aAAa,CACjF,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,2BAA2B,CAAC,2CAA2C,SAAS,GAAG,CAAC,CAAC;IACjG,CAAC;IAED,6BAA6B;IAC7B,4CAA4C;IAC5C,IAAI,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,2BAA2B,CAAC,iDAAiD,CAAC,CAAC;IAC3F,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,cAAc,CAAC,MAAW;IACxC,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,MAAM,IAAI,2BAA2B,CAAC,0BAA0B,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,2BAA2B,CAAC,yBAAyB,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,GAAG,kBAAkB,EAAE,CAAC;QACvC,MAAM,IAAI,2BAA2B,CACnC,qCAAqC,kBAAkB,aAAa,CACrE,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,2BAA2B,CAAC,qCAAqC,SAAS,GAAG,CAAC,CAAC;IAC3F,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,eAAe,CAAC,OAAY;IAC1C,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,MAAM,IAAI,2BAA2B,CAAC,0BAA0B,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,2BAA2B,CAAC,yBAAyB,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,GAAG,kBAAkB,EAAE,CAAC;QACxC,MAAM,IAAI,2BAA2B,CACnC,qCAAqC,kBAAkB,aAAa,CACrE,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,mBAAmB,CAAC,WAAgB;IAClD,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,2BAA2B,CAAC,iCAAiC,CAAC,CAAC;IAC3E,CAAC;IAED,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACpB,MAAM,IAAI,2BAA2B,CAAC,iCAAiC,CAAC,CAAC;IAC3E,CAAC;IAED,IAAI,WAAW,GAAG,oBAAoB,EAAE,CAAC;QACvC,MAAM,IAAI,2BAA2B,CACnC,iBAAiB,WAAW,wBAAwB,oBAAoB,EAAE,CAC3E,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,iBAAiB,CAAC,KAAU;IAC1C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,OAAO;IACT,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1D,MAAM,IAAI,2BAA2B,CAAC,0BAA0B,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACf,MAAM,IAAI,2BAA2B,CAAC,wBAAwB,CAAC,CAAC;IAClE,CAAC;IAED,IAAI,KAAK,GAAG,cAAc,EAAE,CAAC;QAC3B,MAAM,IAAI,2BAA2B,CAAC,uBAAuB,cAAc,EAAE,CAAC,CAAC;IACjF,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,eAAe,CAAC,OAA2B;IACzD,IAAI,CAAC;QACH,IAAA,uBAAiB,EAAC,OAAO,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,uDAAuD;QACvD,MAAM,IAAI,2BAA2B,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAChG,CAAC;AACH,CAAC;AAED;;GAEG;AACU,QAAA,+BAA+B,GAAG;IAC7C,qBAAqB;IACrB,oBAAoB;IACpB,cAAc;IACd,SAAS;CACD,CAAC"}

package/dist/factory.d.ts

@@ -2,15 +2,15 @@
  * Factory for creating DynamoDB persistence instances with sensible defaults
  * Provides convenient methods for instantiating checkpointer, store, and chat history
  */
-import type { DynamoDBClientConfig } from '@aws-sdk/client-dynamodb';
-import type { BedrockEmbeddings } from '@langchain/aws';
+import { type DynamoDBClientConfig } from '@aws-sdk/client-dynamodb';
+import type { EmbeddingsInterface } from '@langchain/core/embeddings';
 import type { SerializerProtocol } from '@langchain/langgraph-checkpoint';
 import { DynamoDBSaver } from './checkpointer';
-import { DynamoDBStore } from './store';
-import { DynamoDBChatMessageHistory } from './history';
 import type { DynamoDBSaverOptions } from './checkpointer/types';
-import type { DynamoDBStoreOptions } from './store/types';
+import { DynamoDBChatMessageHistory } from './history';
 import type { DynamoDBChatMessageHistoryOptions } from './history/types';
+import { DynamoDBStore } from './store';
+import type { DynamoDBStoreOptions } from './store/types';
 /**
  * Factory class for creating DynamoDB persistence instances
  */
@@ -126,12 +126,14 @@ export declare class DynamoDBFactory {
         tablePrefix?: string;
         ttlDays?: number;
         clientConfig?: DynamoDBClientConfig;
-        embedding?: BedrockEmbeddings;
+        embedding?: EmbeddingsInterface;
         serde?: SerializerProtocol;
     }): {
         checkpointer: DynamoDBSaver;
         store: DynamoDBStore;
         chatHistory: DynamoDBChatMessageHistory;
+        /** Destroy the shared DynamoDB client created by createAll(). Call when no longer needed. */
+        destroy: () => void;
     };
 }
 //# sourceMappingURL=factory.d.ts.map

package/dist/factory.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"factory.d.ts","sourceRoot":"","sources":["../src/factory.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AAE1E,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACxC,OAAO,EAAE,0BAA0B,EAAE,MAAM,WAAW,CAAC;AACvD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAC1D,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,iBAAiB,CAAC;AAOzE;;GAEG;AACH,qBAAa,eAAe;IAC1B;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACH,MAAM,CAAC,WAAW,CAAC,OAAO,GAAE,OAAO,CAAC,oBAAoB,CAAM,GAAG,aAAa;IAY9E;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2BG;IACH,MAAM,CAAC,WAAW,CAAC,OAAO,GAAE,OAAO,CAAC,oBAAoB,CAAM,GAAG,aAAa;IAW9E;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,MAAM,CAAC,wBAAwB,CAC7B,OAAO,GAAE,OAAO,CAAC,iCAAiC,CAAM,GACvD,0BAA0B;IAU7B;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,MAAM,CAAC,SAAS,CACd,OAAO,GAAE;QACP,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,YAAY,CAAC,EAAE,oBAAoB,CAAC;QACpC,SAAS,CAAC,EAAE,iBAAiB,CAAC;QAC9B,KAAK,CAAC,EAAE,kBAAkB,CAAC;KACvB,GACL;QACD,YAAY,EAAE,aAAa,CAAC;QAC5B,KAAK,EAAE,aAAa,CAAC;QACrB,WAAW,EAAE,0BAA0B,CAAC;KACzC;CAwBF"}
+{"version":3,"file":"factory.d.ts","sourceRoot":"","sources":["../src/factory.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAkB,KAAK,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAErF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AAE1E,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,0BAA0B,EAAE,MAAM,WAAW,CAAC;AACvD,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,iBAAiB,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACxC,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAO1D;;GAEG;AACH,qBAAa,eAAe;IAC1B;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACH,MAAM,CAAC,WAAW,CAAC,OAAO,GAAE,OAAO,CAAC,oBAAoB,CAAM,GAAG,aAAa;IAgB9E;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2BG;IACH,MAAM,CAAC,WAAW,CAAC,OAAO,GAAE,OAAO,CAAC,oBAAoB,CAAM,GAAG,aAAa;IAY9E;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,MAAM,CAAC,wBAAwB,CAC7B,OAAO,GAAE,OAAO,CAAC,iCAAiC,CAAM,GACvD,0BAA0B;IAW7B;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,MAAM,CAAC,SAAS,CACd,OAAO,GAAE;QACP,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,YAAY,CAAC,EAAE,oBAAoB,CAAC;QACpC,SAAS,CAAC,EAAE,mBAAmB,CAAC;QAChC,KAAK,CAAC,EAAE,kBAAkB,CAAC;KACvB,GACL;QACD,YAAY,EAAE,aAAa,CAAC;QAC5B,KAAK,EAAE,aAAa,CAAC;QACrB,WAAW,EAAE,0BAA0B,CAAC;QACxC,6FAA6F;QAC7F,OAAO,EAAE,MAAM,IAAI,CAAC;KACrB;CA6BF"}

package/dist/factory.js

@@ -5,9 +5,11 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DynamoDBFactory = void 0;
+const client_dynamodb_1 = require("@aws-sdk/client-dynamodb");
+const lib_dynamodb_1 = require("@aws-sdk/lib-dynamodb");
 const checkpointer_1 = require("./checkpointer");
-const store_1 = require("./store");
 const history_1 = require("./history");
+const store_1 = require("./store");
 /**
  * Default table names with standard prefix
  */
@@ -47,8 +49,12 @@ class DynamoDBFactory {
             checkpointsTableName: options.checkpointsTableName ?? `${DEFAULT_TABLE_PREFIX}-checkpoints`,
             writesTableName: options.writesTableName ?? `${DEFAULT_TABLE_PREFIX}-writes`,
             ttlDays: options.ttlDays,
+            ttlSeconds: options.ttlSeconds,
+            compression: options.compression,
+            s3OffloadConfig: options.s3OffloadConfig,
             serde: options.serde,
             clientConfig: options.clientConfig,
+            client: options.client,
         };
         return new checkpointer_1.DynamoDBSaver(config);
     }
@@ -86,6 +92,7 @@ class DynamoDBFactory {
             embedding: options.embedding,
             ttlDays: options.ttlDays,
             clientConfig: options.clientConfig,
+            client: options.client,
         };
         return new store_1.DynamoDBStore(config);
     }
@@ -117,6 +124,7 @@ class DynamoDBFactory {
             tableName: options.tableName ?? `${DEFAULT_TABLE_PREFIX}-chat-history`,
             ttlDays: options.ttlDays,
             clientConfig: options.clientConfig,
+            client: options.client,
         };
         return new history_1.DynamoDBChatMessageHistory(config);
     }
@@ -149,25 +157,29 @@ class DynamoDBFactory {
      */
     static createAll(options = {}) {
         const prefix = options.tablePrefix ?? DEFAULT_TABLE_PREFIX;
+        // Create a single shared DynamoDB client for all modules
+        const ddbClient = new client_dynamodb_1.DynamoDBClient(options.clientConfig || {});
+        const sharedClient = lib_dynamodb_1.DynamoDBDocument.from(ddbClient);
         return {
             checkpointer: this.createSaver({
                 checkpointsTableName: `${prefix}-checkpoints`,
                 writesTableName: `${prefix}-writes`,
                 ttlDays: options.ttlDays,
                 serde: options.serde,
-                clientConfig: options.clientConfig,
+                client: sharedClient,
             }),
             store: this.createStore({
                 memoryTableName: `${prefix}-memory`,
                 embedding: options.embedding,
                 ttlDays: options.ttlDays,
-                clientConfig: options.clientConfig,
+                client: sharedClient,
             }),
             chatHistory: this.createChatMessageHistory({
                 tableName: `${prefix}-chat-history`,
                 ttlDays: options.ttlDays,
-                clientConfig: options.clientConfig,
+                client: sharedClient,
             }),
+            destroy: () => ddbClient.destroy(),
         };
     }
 }

package/dist/factory.js.map

@@ -1 +1 @@
-{"version":3,"file":"factory.js","sourceRoot":"","sources":["../src/factory.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAMH,iDAA+C;AAC/C,mCAAwC;AACxC,uCAAuD;AAKvD;;GAEG;AACH,MAAM,oBAAoB,GAAG,WAAW,CAAC;AAEzC;;GAEG;AACH,MAAa,eAAe;IAC1B;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACH,MAAM,CAAC,WAAW,CAAC,UAAyC,EAAE;QAC5D,MAAM,MAAM,GAAyB;YACnC,oBAAoB,EAAE,OAAO,CAAC,oBAAoB,IAAI,GAAG,oBAAoB,cAAc;YAC3F,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,GAAG,oBAAoB,SAAS;YAC5E,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,YAAY,EAAE,OAAO,CAAC,YAAY;SACnC,CAAC;QAEF,OAAO,IAAI,4BAAa,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2BG;IACH,MAAM,CAAC,WAAW,CAAC,UAAyC,EAAE;QAC5D,MAAM,MAAM,GAAyB;YACnC,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,GAAG,oBAAoB,SAAS;YAC5E,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,YAAY,EAAE,OAAO,CAAC,YAAY;SACnC,CAAC;QAEF,OAAO,IAAI,qBAAa,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,MAAM,CAAC,wBAAwB,CAC7B,UAAsD,EAAE;QAExD,MAAM,MAAM,GAAsC;YAChD,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,GAAG,oBAAoB,eAAe;YACtE,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,YAAY,EAAE,OAAO,CAAC,YAAY;SACnC,CAAC;QAEF,OAAO,IAAI,oCAA0B,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,MAAM,CAAC,SAAS,CACd,UAMI,EAAE;QAMN,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,IAAI,oBAAoB,CAAC;QAE3D,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC;gBAC7B,oBAAoB,EAAE,GAAG,MAAM,cAAc;gBAC7C,eAAe,EAAE,GAAG,MAAM,SAAS;gBACnC,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,YAAY,EAAE,OAAO,CAAC,YAAY;aACnC,CAAC;YACF,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC;gBACtB,eAAe,EAAE,GAAG,MAAM,SAAS;gBACnC,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,YAAY,EAAE,OAAO,CAAC,YAAY;aACnC,CAAC;YACF,WAAW,EAAE,IAAI,CAAC,wBAAwB,CAAC;gBACzC,SAAS,EAAE,GAAG,MAAM,eAAe;gBACnC,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,YAAY,EAAE,OAAO,CAAC,YAAY;aACnC,CAAC;SACH,CAAC;IACJ,CAAC;CACF;AAhLD,0CAgLC"}
+{"version":3,"file":"factory.js","sourceRoot":"","sources":["../src/factory.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,8DAAqF;AACrF,wDAAyD;AAIzD,iDAA+C;AAE/C,uCAAuD;AAEvD,mCAAwC;AAGxC;;GAEG;AACH,MAAM,oBAAoB,GAAG,WAAW,CAAC;AAEzC;;GAEG;AACH,MAAa,eAAe;IAC1B;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACH,MAAM,CAAC,WAAW,CAAC,UAAyC,EAAE;QAC5D,MAAM,MAAM,GAAyB;YACnC,oBAAoB,EAAE,OAAO,CAAC,oBAAoB,IAAI,GAAG,oBAAoB,cAAc;YAC3F,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,GAAG,oBAAoB,SAAS;YAC5E,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC;QAEF,OAAO,IAAI,4BAAa,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2BG;IACH,MAAM,CAAC,WAAW,CAAC,UAAyC,EAAE;QAC5D,MAAM,MAAM,GAAyB;YACnC,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,GAAG,oBAAoB,SAAS;YAC5E,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC;QAEF,OAAO,IAAI,qBAAa,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,MAAM,CAAC,wBAAwB,CAC7B,UAAsD,EAAE;QAExD,MAAM,MAAM,GAAsC;YAChD,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,GAAG,oBAAoB,eAAe;YACtE,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC;QAEF,OAAO,IAAI,oCAA0B,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,MAAM,CAAC,SAAS,CACd,UAMI,EAAE;QAQN,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,IAAI,oBAAoB,CAAC;QAE3D,yDAAyD;QACzD,MAAM,SAAS,GAAG,IAAI,gCAAc,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;QACjE,MAAM,YAAY,GAAG,+BAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAEtD,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC;gBAC7B,oBAAoB,EAAE,GAAG,MAAM,cAAc;gBAC7C,eAAe,EAAE,GAAG,MAAM,SAAS;gBACnC,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,MAAM,EAAE,YAAY;aACrB,CAAC;YACF,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC;gBACtB,eAAe,EAAE,GAAG,MAAM,SAAS;gBACnC,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,MAAM,EAAE,YAAY;aACrB,CAAC;YACF,WAAW,EAAE,IAAI,CAAC,wBAAwB,CAAC;gBACzC,SAAS,EAAE,GAAG,MAAM,eAAe;gBACnC,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,MAAM,EAAE,YAAY;aACrB,CAAC;YACF,OAAO,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE;SACnC,CAAC;IACJ,CAAC;CACF;AA7LD,0CA6LC"}

package/dist/history/actions/add-message.d.ts

@@ -1,10 +1,11 @@
 /**
  * Add a single message to a session
+ * Uses an atomic counter to claim a unique message index, preventing race conditions
  */
 import type { AddMessageActionParams } from '../types';
 /**
  * Add a single message to a session
- * Automatically handles creating a new session with a title
+ * Atomically claims a message index via DynamoDB's ADD operation, then writes the message item
  *
  * @param params - Parameters for the add message operation
  * @throws Error if the operation fails or validation fails

package/dist/history/actions/add-message.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"add-message.d.ts","sourceRoot":"","sources":["../../../src/history/actions/add-message.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAYvD;;;;;;GAMG;AACH,eAAO,MAAM,gBAAgB,GAAU,QAAQ,sBAAsB,KAAG,OAAO,CAAC,IAAI,CAkCnF,CAAC"}
+{"version":3,"file":"add-message.d.ts","sourceRoot":"","sources":["../../../src/history/actions/add-message.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAavD;;;;;;GAMG;AACH,eAAO,MAAM,gBAAgB,GAAU,QAAQ,sBAAsB,KAAG,OAAO,CAAC,IAAI,CA4CnF,CAAC"}

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

@@ -1,13 +1,14 @@
 "use strict";
 /**
  * Add a single message to a session
+ * Uses an atomic counter to claim a unique message index, preventing race conditions
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.addMessageAction = void 0;
 const utils_1 = require("../utils");
 /**
  * Add a single message to a session
- * Automatically handles creating a new session with a title
+ * Atomically claims a message index via DynamoDB's ADD operation, then writes the message item
  *
  * @param params - Parameters for the add message operation
  * @throws Error if the operation fails or validation fails
@@ -21,26 +22,29 @@ const addMessageAction = async (params) => {
     (0, utils_1.validateTitle)(title);
     (0, utils_1.validateTTLDays)(ttlDays);
     // Generate a title if this is the first message and no title provided
-    let sessionTitle = title;
-    if (!sessionTitle) {
-        sessionTitle = (0, utils_1.generateTitle)(message);
-    }
-    // Build update expression using utility
-    const { updateExpression, expressionAttributeValues } = (0, utils_1.buildMessageUpdateExpression)({
-        messages: message,
-        title: sessionTitle,
-        ttlDays,
-    });
-    // Execute with retry logic
-    await (0, utils_1.withDynamoDBRetry)(async () => {
-        await client.update({
+    const sessionTitle = title ?? (0, utils_1.generateTitle)(message);
+    // Atomically claim a message index via ADD messageCount
+    const { updateExpression, expressionAttributeValues } = (0, utils_1.buildAtomicMetadataUpdate)(sessionTitle, 1, // Claiming 1 index
+    ttlDays);
+    const updateResult = await (0, utils_1.withDynamoDBRetry)(async () => {
+        return await client.update({
             TableName: tableName,
-            Key: {
-                userId,
-                sessionId,
-            },
+            Key: { userId, sessionId },
             UpdateExpression: updateExpression,
             ExpressionAttributeValues: expressionAttributeValues,
+            ReturnValues: 'ALL_NEW',
+        });
+    });
+    // The new messageCount is the count AFTER incrementing by 1
+    // So our claimed index is newCount - 1
+    const newCount = updateResult.Attributes?.messageCount ?? 1;
+    const startIndex = newCount - 1;
+    // Build and write the message item
+    const [messageItem] = (0, utils_1.buildMessageItems)(userId, sessionId, [message], startIndex, ttlDays);
+    await (0, utils_1.withDynamoDBRetry)(async () => {
+        await client.put({
+            TableName: tableName,
+            Item: messageItem,
         });
     });
 };

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

@@ -1 +1 @@
-{"version":3,"file":"add-message.js","sourceRoot":"","sources":["../../../src/history/actions/add-message.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAGH,oCASkB;AAElB;;;;;;GAMG;AACI,MAAM,gBAAgB,GAAG,KAAK,EAAE,MAA8B,EAAiB,EAAE;IACtF,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC;IAEjF,kBAAkB;IAClB,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,yBAAiB,EAAC,SAAS,CAAC,CAAC;IAC7B,IAAA,uBAAe,EAAC,OAAO,CAAC,CAAC;IACzB,IAAA,qBAAa,EAAC,KAAK,CAAC,CAAC;IACrB,IAAA,uBAAe,EAAC,OAAO,CAAC,CAAC;IAEzB,sEAAsE;IACtE,IAAI,YAAY,GAAG,KAAK,CAAC;IACzB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,IAAA,qBAAa,EAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IACD,wCAAwC;IACxC,MAAM,EAAE,gBAAgB,EAAE,yBAAyB,EAAE,GAAG,IAAA,oCAA4B,EAAC;QACnF,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,YAAY;QACnB,OAAO;KACR,CAAC,CAAC;IAEH,2BAA2B;IAC3B,MAAM,IAAA,yBAAiB,EAAC,KAAK,IAAI,EAAE;QACjC,MAAM,MAAM,CAAC,MAAM,CAAC;YAClB,SAAS,EAAE,SAAS;YACpB,GAAG,EAAE;gBACH,MAAM;gBACN,SAAS;aACV;YACD,gBAAgB,EAAE,gBAAgB;YAClC,yBAAyB,EAAE,yBAAyB;SACrD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAlCW,QAAA,gBAAgB,oBAkC3B"}
+{"version":3,"file":"add-message.js","sourceRoot":"","sources":["../../../src/history/actions/add-message.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAGH,oCAUkB;AAElB;;;;;;GAMG;AACI,MAAM,gBAAgB,GAAG,KAAK,EAAE,MAA8B,EAAiB,EAAE;IACtF,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC;IAEjF,kBAAkB;IAClB,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;IACvB,IAAA,yBAAiB,EAAC,SAAS,CAAC,CAAC;IAC7B,IAAA,uBAAe,EAAC,OAAO,CAAC,CAAC;IACzB,IAAA,qBAAa,EAAC,KAAK,CAAC,CAAC;IACrB,IAAA,uBAAe,EAAC,OAAO,CAAC,CAAC;IAEzB,sEAAsE;IACtE,MAAM,YAAY,GAAG,KAAK,IAAI,IAAA,qBAAa,EAAC,OAAO,CAAC,CAAC;IAErD,wDAAwD;IACxD,MAAM,EAAE,gBAAgB,EAAE,yBAAyB,EAAE,GAAG,IAAA,iCAAyB,EAC/E,YAAY,EACZ,CAAC,EAAE,mBAAmB;IACtB,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,4DAA4D;IAC5D,uCAAuC;IACvC,MAAM,QAAQ,GAAI,YAAY,CAAC,UAAU,EAAE,YAAuB,IAAI,CAAC,CAAC;IACxE,MAAM,UAAU,GAAG,QAAQ,GAAG,CAAC,CAAC;IAEhC,mCAAmC;IACnC,MAAM,CAAC,WAAW,CAAC,GAAG,IAAA,yBAAiB,EAAC,MAAM,EAAE,SAAS,EAAE,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IAE3F,MAAM,IAAA,yBAAiB,EAAC,KAAK,IAAI,EAAE;QACjC,MAAM,MAAM,CAAC,GAAG,CAAC;YACf,SAAS,EAAE,SAAS;YACpB,IAAI,EAAE,WAAW;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AA5CW,QAAA,gBAAgB,oBA4C3B"}

package/dist/history/actions/add-messages.d.ts

@@ -1,11 +1,12 @@
 /**
  * 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)
  */
 import type { AddMessagesActionParams } from '../types';
 /**
  * Add multiple messages to a session
- * Automatically handles creating a new session with title
- * Preferred over calling addMessage multiple times
+ * 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

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

@@ -1 +1 @@
-{"version":3,"file":"add-messages.d.ts","sourceRoot":"","sources":["../../../src/history/actions/add-messages.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAYxD;;;;;;;GAOG;AACH,eAAO,MAAM,iBAAiB,GAAU,QAAQ,uBAAuB,KAAG,OAAO,CAAC,IAAI,CAkCrF,CAAC"}
+{"version":3,"file":"add-messages.d.ts","sourceRoot":"","sources":["../../../src/history/actions/add-messages.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAcxD;;;;;;GAMG;AACH,eAAO,MAAM,iBAAiB,GAAU,QAAQ,uBAAuB,KAAG,OAAO,CAAC,IAAI,CAkDrF,CAAC"}

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

@@ -1,47 +1,57 @@
 "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)
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.addMessagesAction = void 0;
 const utils_1 = require("../utils");
 /**
  * Add multiple messages to a session
- * Automatically handles creating a new session with title
- * Preferred over calling addMessage multiple times
+ * 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
+    // Validate inputs (enforces max 100 messages per batch)
     (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
-    let sessionTitle = title;
-    if (!sessionTitle) {
-        sessionTitle = (0, utils_1.generateTitle)(messages[0]);
-    }
-    // Build update expression using utility
-    const { updateExpression, expressionAttributeValues } = (0, utils_1.buildMessageUpdateExpression)({
-        messages,
-        title: sessionTitle,
-        ttlDays,
-    });
-    // Execute with retry logic
-    await (0, utils_1.withDynamoDBRetry)(async () => {
-        await client.update({
+    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,
-            },
+            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 client.transactWrite({
+            TransactItems: messageItems.map((item) => ({
+                Put: {
+                    TableName: tableName,
+                    Item: item,
+                },
+            })),
         });
     });
 };