@langgraph-js/pure-graph 2.1.6 → 2.2.7

package/dist/storage/index.d.ts

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

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

@@ -0,0 +1,44 @@
+import { Kysely, SqlBool, Expression } from 'kysely';
+import { Database } from './types';
+/**
+ * 数据库适配器接口
+ * 处理不同数据库间的差异，特别是时间类型和 JSON 查询
+ */
+export interface DatabaseAdapter {
+    db: Kysely<Database>;
+    /**
+     * 将 Date 转换为数据库原生时间类型
+     */
+    dateToDb(date: Date): any;
+    /**
+     * 将数据库时间类型转换为 Date
+     */
+    dbToDate(dbValue: any): Date;
+    /**
+     * 将对象转换为数据库 JSON 类型
+     */
+    jsonToDb(obj: any): any;
+    /**
+     * 将数据库 JSON 类型转换为对象
+     */
+    dbToJson(dbValue: any): any;
+    /**
+     * 构建 JSON 字段查询表达式
+     * @param field - JSON 字段名（如 'metadata'）
+     * @param key - JSON 对象的键
+     * @param value - 要匹配的值
+     */
+    buildJsonQuery(db: Kysely<Database>, field: 'metadata' | 'interrupts', key: string, value: any): Expression<SqlBool>;
+    /**
+     * 获取当前时间的数据库表示
+     */
+    now(): any;
+    /**
+     * 创建数据库表
+     */
+    createTables(db: Kysely<Database>): Promise<void>;
+    /**
+     * 创建索引
+     */
+    createIndexes(db: Kysely<Database>): Promise<void>;
+}

package/dist/storage/kysely/index.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Kysely-based storage implementation
+ * 使用 Kysely 实现的存储层，支持多数据库适配
+ */
+export type { Database, ThreadsTable, RunsTable } from './types';
+export { type DatabaseAdapter } from './adapter';
+export { PostgresAdapter } from './pg-adapter';
+export { SQLiteAdapter } from './sqlite-adapter';
+export { KyselyThreadsManager } from './threads';

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

@@ -0,0 +1,23 @@
+import { Kysely, SqlBool, Expression } from 'kysely';
+import { DatabaseAdapter } from './adapter';
+import { Database } from './types';
+import { Pool } from 'pg';
+/**
+ * PostgreSQL 适配器
+ * - 时间类型：原生 TIMESTAMP
+ * - JSON 类型：JSONB
+ * - JSON 查询：使用 -> 和 ->> 操作符
+ */
+export declare class PostgresAdapter implements DatabaseAdapter {
+    pool: Pool;
+    db: Kysely<Database>;
+    constructor(pool: Pool);
+    dateToDb(date: Date): Date;
+    dbToDate(dbValue: any): Date;
+    jsonToDb(obj: any): any;
+    dbToJson(dbValue: any): any;
+    buildJsonQuery(db: Kysely<Database>, field: 'metadata' | 'interrupts', key: string, value: any): Expression<SqlBool>;
+    now(): Date;
+    createTables(db: Kysely<Database>): Promise<void>;
+    createIndexes(db: Kysely<Database>): Promise<void>;
+}

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

@@ -0,0 +1,22 @@
+import { Kysely, SqlBool, Expression } from 'kysely';
+import { DatabaseAdapter } from './adapter';
+import { Database } from './types';
+/**
+ * SQLite 适配器
+ * - 时间类型：TEXT (ISO 8601 字符串)
+ * - JSON 类型：TEXT
+ * - JSON 查询：使用 json_extract 函数
+ */
+export declare class SQLiteAdapter implements DatabaseAdapter {
+    database: any;
+    db: Kysely<Database>;
+    constructor(database: any);
+    dateToDb(date: Date): string;
+    dbToDate(dbValue: any): Date;
+    jsonToDb(obj: any): string;
+    dbToJson(dbValue: any): any;
+    buildJsonQuery(db: Kysely<Database>, field: 'metadata' | 'interrupts', key: string, value: any): Expression<SqlBool>;
+    now(): string;
+    createTables(db: Kysely<Database>): Promise<void>;
+    createIndexes(db: Kysely<Database>): Promise<void>;
+}

package/dist/storage/{sqlite → kysely}/threads.d.ts

@@ -1,12 +1,15 @@
 import { BaseThreadsManager } from '../../threads/index.js';
 import { Command, Config, Metadata, OnConflictBehavior, Run, Thread, ThreadStatus } from '@langgraph-js/sdk';
-import type { SqliteSaver } from './checkpoint.js';
-import type { DatabaseType } from './type.js';
 import { RunStatus, SortOrder, ThreadSortBy } from '../../types';
-export declare class SQLiteThreadsManager<ValuesType = unknown> implements BaseThreadsManager<ValuesType> {
-    db: DatabaseType;
-    private isSetup;
-    constructor(checkpointer: SqliteSaver);
+import { DatabaseAdapter } from './adapter';
+/**
+ * 使用 Kysely 实现的统一 ThreadsManager
+ * 通过适配器模式处理不同数据库的差异
+ */
+export declare class KyselyThreadsManager<ValuesType = unknown> implements BaseThreadsManager<ValuesType> {
+    private db;
+    private adapter;
+    constructor(adapter: DatabaseAdapter);
     setup(): Promise<void>;
     create(payload?: {
         metadata?: Metadata;

package/dist/storage/kysely/types.d.ts

@@ -0,0 +1,27 @@
+/**
+ * Kysely 数据库类型定义
+ * 这里定义的是应用层的类型，不是数据库原始类型
+ */
+export interface ThreadsTable {
+    thread_id: string;
+    created_at: Date;
+    updated_at: Date;
+    metadata: Record<string, any>;
+    status: string;
+    values: any;
+    interrupts: Record<string, any>;
+}
+export interface RunsTable {
+    run_id: string;
+    thread_id: string;
+    assistant_id: string;
+    created_at: Date;
+    updated_at: Date;
+    status: string;
+    metadata: Record<string, any>;
+    multitask_strategy: string;
+}
+export interface Database {
+    threads: ThreadsTable;
+    runs: RunsTable;
+}