@langgraph-js/pure-graph 3.1.2 → 3.2.2

package/dist/{sqlite-adapter-BKOLSdoL.js → sqlite-adapter-CJXgit1j.js}

@@ -2,9 +2,26 @@ import { sql } from 'kysely';
 
 class SQLiteAdapter {
   db;
+  pragmaPromise = null;
   constructor(database) {
     this.db = database;
   }
+  /**
+   * 设置 SQLite PRAGMA 配置，解决锁问题
+   * 使用 Promise 缓存确保并发安全且只执行一次
+   */
+  ensurePragma() {
+    if (!this.pragmaPromise) {
+      this.pragmaPromise = this.doEnsurePragma();
+    }
+    return this.pragmaPromise;
+  }
+  async doEnsurePragma() {
+    await sql`PRAGMA busy_timeout = 5000`.execute(this.db);
+    await sql`PRAGMA journal_mode = WAL`.execute(this.db);
+    await sql`PRAGMA synchronous = NORMAL`.execute(this.db);
+    await sql`PRAGMA wal_autocheckpoint = 1000`.execute(this.db);
+  }
   dateToDb(date) {
     return date.toISOString();
   }
@@ -25,8 +42,12 @@ class SQLiteAdapter {
     return dbValue;
   }
   buildJsonQuery(db, field, key, value) {
-    const escapedKey = key.includes("-") || key.includes(".") || key.includes(" ") ? `"${key}"` : key;
-    const jsonPath = `$.${escapedKey}`;
+    const jsonPath = `$.${JSON.stringify(key)}`;
+    if (value === null) {
+      return sql`
+                json_type(${sql.ref(field)}, ${sql.lit(jsonPath)}) = 'null'
+            `;
+    }
     let compareValue;
     if (typeof value === "string") {
       compareValue = value;
@@ -34,24 +55,16 @@ class SQLiteAdapter {
       compareValue = value;
     } else if (typeof value === "boolean") {
       compareValue = value ? 1 : 0;
-    } else if (value === null) {
-      compareValue = null;
     } else {
       compareValue = JSON.stringify(value);
     }
-    if (value === null) {
-      const keyPattern = `"${key}":null`;
-      return sql`
-                json_extract(${sql.ref(field)}, ${sql.lit(jsonPath)}) IS NULL
-                AND ${sql.ref(field)} LIKE ${sql.lit(`%${keyPattern}%`)}
-            `;
-    }
     return sql`json_extract(${sql.ref(field)}, ${sql.lit(jsonPath)}) = ${sql.lit(compareValue)}`;
   }
   now() {
     return (/* @__PURE__ */ new Date()).toISOString();
   }
   async createTables(db) {
+    await this.ensurePragma();
     await sql`
             CREATE TABLE IF NOT EXISTS threads (
                 thread_id TEXT PRIMARY KEY,
@@ -87,4 +100,4 @@ class SQLiteAdapter {
 }
 
 export { SQLiteAdapter };
-//# sourceMappingURL=sqlite-adapter-BKOLSdoL.js.map
+//# sourceMappingURL=sqlite-adapter-CJXgit1j.js.map

package/dist/sqlite-adapter-CJXgit1j.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sqlite-adapter-CJXgit1j.js","sources":["../src/storage/kysely/sqlite-adapter.ts"],"sourcesContent":["import { Kysely, sql, SqlBool, Expression, SqliteDialect } from 'kysely';\nimport { DatabaseAdapter } from './adapter';\nimport { Database } from './types';\n\n/**\n * SQLite 适配器\n * - 时间类型：TEXT (ISO 8601 字符串)\n * - JSON 类型：TEXT\n * - JSON 查询：使用 json_extract 函数\n */\nexport class SQLiteAdapter implements DatabaseAdapter {\n    db: Kysely<Database>;\n    private pragmaPromise: Promise<void> | null = null;\n\n    constructor(database: Kysely<any>) {\n        this.db = database;\n    }\n\n    /**\n     * 设置 SQLite PRAGMA 配置，解决锁问题\n     * 使用 Promise 缓存确保并发安全且只执行一次\n     */\n    private ensurePragma(): Promise<void> {\n        if (!this.pragmaPromise) {\n            this.pragmaPromise = this.doEnsurePragma();\n        }\n        return this.pragmaPromise;\n    }\n\n    private async doEnsurePragma(): Promise<void> {\n        // 锁等待超时 5 秒\n        await sql`PRAGMA busy_timeout = 5000`.execute(this.db);\n        // WAL 模式 - 读写并发\n        await sql`PRAGMA journal_mode = WAL`.execute(this.db);\n        // 平衡安全与性能\n        await sql`PRAGMA synchronous = NORMAL`.execute(this.db);\n        // 自动清理 WAL\n        await sql`PRAGMA wal_autocheckpoint = 1000`.execute(this.db);\n    }\n    dateToDb(date: Date): string {\n        // SQLite 存储为 ISO 8601 字符串\n        return date.toISOString();\n    }\n\n    dbToDate(dbValue: any): Date {\n        // SQLite 返回字符串，需要转换为 Date\n        return new Date(dbValue);\n    }\n\n    jsonToDb(obj: any): string {\n        // SQLite 存储为 JSON 字符串\n        return JSON.stringify(obj);\n    }\n\n    dbToJson(dbValue: any): any {\n        // SQLite 返回字符串，需要解析\n        if (typeof dbValue === 'string') {\n            try {\n                return JSON.parse(dbValue);\n            } catch {\n                return dbValue;\n            }\n        }\n        return dbValue;\n    }\n\n    buildJsonQuery(\n        db: Kysely<Database>,\n        field: 'metadata' | 'interrupts',\n        key: string,\n        value: any,\n    ): Expression<SqlBool> {\n        // SQLite json_extract 的行为:\n        // - 字符串: 返回不带引号的文本\n        // - 数字: 返回数字\n        // - 布尔值: 返回 1/0\n        // - null: 返回 NULL\n        // 所以比较时需要根据类型处理\n\n        // 构建 JSON 路径\n        // 在 SQLite 中，所有键都需要用双引号括起来以确保正确解析\n        // json_extract 语法: $.key 或 $.\"key-with-special\"\n        const jsonPath = `$.${JSON.stringify(key)}`;\n\n        // 构建 NULL 处理条件\n        // 注意：SQLite 中 json_extract 对键不存在和值为 null 都返回 NULL\n        // 使用 json_type 函数区分：\n        // - 键不存在：json_type 返回 NULL\n        // - 值为 null：json_type 返回 'null' 字符串\n        if (value === null) {\n            return sql<boolean>`\n                json_type(${sql.ref(field)}, ${sql.lit(jsonPath)}) = 'null'\n            `;\n        }\n\n        // 处理其他类型的值\n        let compareValue: any;\n        if (typeof value === 'string') {\n            // 字符串: json_extract 返回不带引号的字符串，直接比较\n            compareValue = value;\n        } else if (typeof value === 'number') {\n            // 数字: json_extract 返回数字\n            compareValue = value;\n        } else if (typeof value === 'boolean') {\n            // 布尔值: json_extract 返回 1/0\n            compareValue = value ? 1 : 0;\n        } else {\n            // 其他类型（对象、数组）: 使用 JSON 字符串\n            compareValue = JSON.stringify(value);\n        }\n\n        // 构建普通比较条件\n        return sql<boolean>`json_extract(${sql.ref(field)}, ${sql.lit(jsonPath)}) = ${sql.lit(compareValue)}`;\n    }\n\n    now(): string {\n        return new Date().toISOString();\n    }\n\n    async createTables(db: Kysely<Database>): Promise<void> {\n        // 先设置 PRAGMA\n        await this.ensurePragma();\n\n        // 创建 threads 表\n        await sql`\n            CREATE TABLE IF NOT EXISTS threads (\n                thread_id TEXT PRIMARY KEY,\n                created_at TEXT NOT NULL,\n                updated_at TEXT NOT NULL,\n                metadata TEXT NOT NULL DEFAULT '{}',\n                status TEXT NOT NULL DEFAULT 'idle',\n                \"values\" TEXT,\n                interrupts TEXT NOT NULL DEFAULT '{}'\n            )\n        `.execute(db);\n\n        // 创建 runs 表\n        await sql`\n            CREATE TABLE IF NOT EXISTS runs (\n                run_id TEXT PRIMARY KEY,\n                thread_id TEXT NOT NULL,\n                assistant_id TEXT NOT NULL,\n                created_at TEXT NOT NULL,\n                updated_at TEXT NOT NULL,\n                status TEXT NOT NULL DEFAULT 'pending',\n                metadata TEXT NOT NULL DEFAULT '{}',\n                multitask_strategy TEXT NOT NULL DEFAULT 'reject',\n                FOREIGN KEY (thread_id) REFERENCES threads(thread_id) ON DELETE CASCADE\n            )\n        `.execute(db);\n    }\n\n    async createIndexes(db: Kysely<Database>): Promise<void> {\n        await sql`CREATE INDEX IF NOT EXISTS idx_threads_status ON threads(status)`.execute(db);\n        await sql`CREATE INDEX IF NOT EXISTS idx_threads_created_at ON threads(created_at)`.execute(db);\n        await sql`CREATE INDEX IF NOT EXISTS idx_threads_updated_at ON threads(updated_at)`.execute(db);\n        await sql`CREATE INDEX IF NOT EXISTS idx_runs_thread_id ON runs(thread_id)`.execute(db);\n        await sql`CREATE INDEX IF NOT EXISTS idx_runs_status ON runs(status)`.execute(db);\n    }\n}\n"],"names":[],"mappings":";;AAUO,MAAM,aAAA,CAAyC;AAAA,EAClD,EAAA;AAAA,EACQ,aAAA,GAAsC,IAAA;AAAA,EAE9C,YAAY,QAAA,EAAuB;AAC/B,IAAA,IAAA,CAAK,EAAA,GAAK,QAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,YAAA,GAA8B;AAClC,IAAA,IAAI,CAAC,KAAK,aAAA,EAAe;AACrB,MAAA,IAAA,CAAK,aAAA,GAAgB,KAAK,cAAA,EAAe;AAAA,IAC7C;AACA,IAAA,OAAO,IAAA,CAAK,aAAA;AAAA,EAChB;AAAA,EAEA,MAAc,cAAA,GAAgC;AAE1C,IAAA,MAAM,GAAA,CAAA,0BAAA,CAAA,CAAgC,OAAA,CAAQ,IAAA,CAAK,EAAE,CAAA;AAErD,IAAA,MAAM,GAAA,CAAA,yBAAA,CAAA,CAA+B,OAAA,CAAQ,IAAA,CAAK,EAAE,CAAA;AAEpD,IAAA,MAAM,GAAA,CAAA,2BAAA,CAAA,CAAiC,OAAA,CAAQ,IAAA,CAAK,EAAE,CAAA;AAEtD,IAAA,MAAM,GAAA,CAAA,gCAAA,CAAA,CAAsC,OAAA,CAAQ,IAAA,CAAK,EAAE,CAAA;AAAA,EAC/D;AAAA,EACA,SAAS,IAAA,EAAoB;AAEzB,IAAA,OAAO,KAAK,WAAA,EAAY;AAAA,EAC5B;AAAA,EAEA,SAAS,OAAA,EAAoB;AAEzB,IAAA,OAAO,IAAI,KAAK,OAAO,CAAA;AAAA,EAC3B;AAAA,EAEA,SAAS,GAAA,EAAkB;AAEvB,IAAA,OAAO,IAAA,CAAK,UAAU,GAAG,CAAA;AAAA,EAC7B;AAAA,EAEA,SAAS,OAAA,EAAmB;AAExB,IAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC7B,MAAA,IAAI;AACA,QAAA,OAAO,IAAA,CAAK,MAAM,OAAO,CAAA;AAAA,MAC7B,CAAA,CAAA,MAAQ;AACJ,QAAA,OAAO,OAAA;AAAA,MACX;AAAA,IACJ;AACA,IAAA,OAAO,OAAA;AAAA,EACX;AAAA,EAEA,cAAA,CACI,EAAA,EACA,KAAA,EACA,GAAA,EACA,KAAA,EACmB;AAWnB,IAAA,MAAM,QAAA,GAAW,CAAA,EAAA,EAAK,IAAA,CAAK,SAAA,CAAU,GAAG,CAAC,CAAA,CAAA;AAOzC,IAAA,IAAI,UAAU,IAAA,EAAM;AAChB,MAAA,OAAO,GAAA;AAAA,0BAAA,EACS,GAAA,CAAI,IAAI,KAAK,CAAC,KAAK,GAAA,CAAI,GAAA,CAAI,QAAQ,CAAC,CAAA;AAAA,YAAA,CAAA;AAAA,IAExD;AAGA,IAAA,IAAI,YAAA;AACJ,IAAA,IAAI,OAAO,UAAU,QAAA,EAAU;AAE3B,MAAA,YAAA,GAAe,KAAA;AAAA,IACnB,CAAA,MAAA,IAAW,OAAO,KAAA,KAAU,QAAA,EAAU;AAElC,MAAA,YAAA,GAAe,KAAA;AAAA,IACnB,CAAA,MAAA,IAAW,OAAO,KAAA,KAAU,SAAA,EAAW;AAEnC,MAAA,YAAA,GAAe,QAAQ,CAAA,GAAI,CAAA;AAAA,IAC/B,CAAA,MAAO;AAEH,MAAA,YAAA,GAAe,IAAA,CAAK,UAAU,KAAK,CAAA;AAAA,IACvC;AAGA,IAAA,OAAO,GAAA,CAAA,aAAA,EAA4B,GAAA,CAAI,GAAA,CAAI,KAAK,CAAC,CAAA,EAAA,EAAK,GAAA,CAAI,GAAA,CAAI,QAAQ,CAAC,CAAA,IAAA,EAAO,GAAA,CAAI,GAAA,CAAI,YAAY,CAAC,CAAA,CAAA;AAAA,EACvG;AAAA,EAEA,GAAA,GAAc;AACV,IAAA,OAAA,iBAAO,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,EAClC;AAAA,EAEA,MAAM,aAAa,EAAA,EAAqC;AAEpD,IAAA,MAAM,KAAK,YAAA,EAAa;AAGxB,IAAA,MAAM,GAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QAAA,CAAA,CAUJ,QAAQ,EAAE,CAAA;AAGZ,IAAA,MAAM,GAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QAAA,CAAA,CAYJ,QAAQ,EAAE,CAAA;AAAA,EAChB;AAAA,EAEA,MAAM,cAAc,EAAA,EAAqC;AACrD,IAAA,MAAM,GAAA,CAAA,gEAAA,CAAA,CAAsE,QAAQ,EAAE,CAAA;AACtF,IAAA,MAAM,GAAA,CAAA,wEAAA,CAAA,CAA8E,QAAQ,EAAE,CAAA;AAC9F,IAAA,MAAM,GAAA,CAAA,wEAAA,CAAA,CAA8E,QAAQ,EAAE,CAAA;AAC9F,IAAA,MAAM,GAAA,CAAA,gEAAA,CAAA,CAAsE,QAAQ,EAAE,CAAA;AACtF,IAAA,MAAM,GAAA,CAAA,0DAAA,CAAA,CAAgE,QAAQ,EAAE,CAAA;AAAA,EACpF;AACJ;;;;"}

package/dist/storage/index.d.ts

@@ -3,11 +3,13 @@ import { KyselyThreadsManager } from './kysely/threads';
 import { ShallowMemorySaver } from './memory/shallow-memory';
 import { MemoryThreadsManager } from './memory/threads';
 import type { SqliteSaver as SqliteSaverType } from './sqlite/checkpoint';
+import type { SqliteShallowSaver as SqliteShallowSaverType } from './sqlite/shallow-checkpoint';
 import type { PostgresSaver } from '@langchain/langgraph-checkpoint-postgres';
 export { ShallowMemorySaver } from './memory/shallow-memory';
 export { MemorySaver } from './memory/checkpoint';
-export declare const createCheckPointer: () => Promise<ShallowMemorySaver | SqliteSaverType | PostgresSaver | import("@langchain/langgraph-checkpoint-redis").RedisSaver | import("@langchain/langgraph-checkpoint-redis/shallow").ShallowRedisSaver>;
+export { SqliteShallowSaver } from './sqlite/shallow-checkpoint';
+export declare const createCheckPointer: () => Promise<ShallowMemorySaver | SqliteSaverType | SqliteShallowSaverType | PostgresSaver | import("@langchain/langgraph-checkpoint-redis").RedisSaver | import("@langchain/langgraph-checkpoint-redis/shallow").ShallowRedisSaver>;
 export declare const createMessageQueue: () => Promise<StreamQueueManager<BaseStreamQueueInterface>>;
 export declare const createThreadManager: (config: {
-    checkpointer?: SqliteSaverType | PostgresSaver;
+    checkpointer?: SqliteSaverType | SqliteShallowSaverType | PostgresSaver;
 }) => Promise<import("./kysely").RemoteKyselyThreadsManager<unknown> | KyselyThreadsManager<unknown> | MemoryThreadsManager<unknown>>;

package/dist/storage/kysely/sqlite-adapter.d.ts

@@ -9,7 +9,14 @@ import { Database } from './types';
  */
 export declare class SQLiteAdapter implements DatabaseAdapter {
     db: Kysely<Database>;
+    private pragmaPromise;
     constructor(database: Kysely<any>);
+    /**
+     * 设置 SQLite PRAGMA 配置，解决锁问题
+     * 使用 Promise 缓存确保并发安全且只执行一次
+     */
+    private ensurePragma;
+    private doEnsurePragma;
     dateToDb(date: Date): string;
     dbToDate(dbValue: any): Date;
     jsonToDb(obj: any): string;

package/dist/storage/memory/queue.d.ts

@@ -4,6 +4,8 @@ import { BaseStreamQueueInterface } from '../../queue/stream_queue.js';
 /** 内存实现的消息队列，用于存储消息 */
 export declare class MemoryStreamQueue extends BaseStreamQueue implements BaseStreamQueueInterface {
     private data;
+    private activeGenerators;
+    private isDestroyed;
     push(item: EventMessage): Promise<void>;
     onDataChange(listener: (data: EventMessage) => void): () => void;
     /**
@@ -15,4 +17,8 @@ export declare class MemoryStreamQueue extends BaseStreamQueue implements BaseSt
     cancelSignal: AbortController;
     cancel(): Promise<void>;
     copyToQueue(toId: string, ttl?: number): Promise<MemoryStreamQueue>;
+    /**
+     * 销毁队列，释放所有资源
+     */
+    destroy(): Promise<void>;
 }

package/dist/storage/redis/queue.d.ts

@@ -16,7 +16,16 @@ export declare class RedisStreamQueue extends BaseStreamQueue implements BaseStr
     cancelSignal: AbortController;
     private lastStreamId;
     private pollInterval;
+    private connectionReady;
     constructor(id: string, compressMessages?: boolean, ttl?: number);
+    /**
+     * 初始化 Redis 连接（使用共享连接池）
+     */
+    private initConnection;
+    /**
+     * 确保连接已建立
+     */
+    private ensureConnected;
     /**
      * 推送消息到 Redis Stream 和 List
      * - Stream: 用于实时推送（集群友好）
@@ -34,7 +43,11 @@ export declare class RedisStreamQueue extends BaseStreamQueue implements BaseStr
     /**
      * 清空队列
      */
-    clear(): void;
+    clear(): Promise<void>;
+    /**
+     * 销毁队列实例，释放 Redis 连接引用
+     */
+    destroy(): Promise<void>;
     /**
      * 取消操作
      */

package/dist/storage/sqlite/shallow-checkpoint.d.ts

@@ -0,0 +1,59 @@
+import { Dialect, Kysely } from 'kysely';
+import type { RunnableConfig } from '@langchain/core/runnables';
+import { BaseCheckpointSaver, type Checkpoint, type CheckpointListOptions, type CheckpointTuple, type SerializerProtocol, type PendingWrite, type CheckpointMetadata } from '@langchain/langgraph-checkpoint';
+interface ShallowCheckpointsTable {
+    thread_id: string;
+    checkpoint_ns: string;
+    checkpoint_id: string;
+    parent_checkpoint_id: string | null;
+    type: string | null;
+    checkpoint: Uint8Array;
+    metadata: Uint8Array;
+    checkpoint_ts: number;
+}
+interface WritesTable {
+    thread_id: string;
+    checkpoint_ns: string;
+    checkpoint_id: string;
+    task_id: string;
+    idx: number;
+    channel: string;
+    type: string | null;
+    value: Uint8Array | null;
+}
+interface ShallowCheckpointDatabase {
+    shallow_checkpoints: ShallowCheckpointsTable;
+    writes: WritesTable;
+}
+/**
+ * SqliteShallowSaver - SQLite 浅层检查点存储器
+ *
+ * 特性:
+ * - 每个 thread_id + checkpoint_ns 组合只保留最新的 checkpoint
+ * - 新 checkpoint 写入时自动清理旧 checkpoint 的 writes
+ * - 使用 checkpoint_ts 时间戳排序
+ * - 大幅减少存储数据量
+ */
+export declare class SqliteShallowSaver extends BaseCheckpointSaver {
+    db: Kysely<ShallowCheckpointDatabase>;
+    protected isSetup: boolean;
+    constructor(dialect: Dialect, serde?: SerializerProtocol);
+    static fromConnStringAsync(connStringOrLocalPath: string): Promise<SqliteShallowSaver>;
+    protected setup(): Promise<void>;
+    /**
+     * 获取 checkpoint（便捷方法）
+     */
+    get(config: RunnableConfig): Promise<Checkpoint | undefined>;
+    getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined>;
+    list(config: RunnableConfig, options?: CheckpointListOptions): AsyncGenerator<CheckpointTuple>;
+    /**
+     * Check metadata filter matches (with deep comparison support)
+     * Matches ShallowMemorySaver behavior
+     */
+    private _checkMetadataFilterMatch;
+    put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata, _newVersions?: Record<string, string | number>): Promise<RunnableConfig>;
+    putWrites(config: RunnableConfig, writes: PendingWrite[], taskId: string): Promise<void>;
+    deleteThread(threadId: string): Promise<void>;
+    protected migratePendingSends(checkpoint: Checkpoint, threadId: string, parentCheckpointId: string): Promise<void>;
+}
+export {};