@langgraph-js/pure-graph 2.4.2 → 2.5.0

package/dist/adapter/hono/index.js

@@ -1,10 +1,10 @@
 import { Hono } from 'hono';
 import { zValidator } from '@hono/zod-validator';
-import { c as createEndpoint } from '../../createEndpoint-Bxzbc61h.js';
+import { c as createEndpoint } from '../../createEndpoint-pEWA0kt2.js';
 import { A as AssistantsSearchSchema, a as AssistantGraphQuerySchema, T as ThreadIdParamSchema, R as RunStreamPayloadSchema, b as RunIdParamSchema, c as RunJoinStreamQuerySchema, d as RunListQuerySchema, e as RunCancelQuerySchema, f as ThreadStateUpdate, g as ThreadCreatePayloadSchema, h as ThreadSearchPayloadSchema } from '../../zod-B6xyK6pu.js';
 import camelcaseKeys from 'camelcase-keys';
 import { streamSSE } from 'hono/streaming';
-import { s as serialiseAsDict, L as LangGraphGlobal } from '../../stream-CAZ9j0yn.js';
+import { s as serialiseAsDict, L as LangGraphGlobal } from '../../stream-Bwuk4b6F.js';
 import z from 'zod';
 
 const client = createEndpoint();

package/dist/adapter/nextjs/index.js

@@ -1,11 +1,11 @@
-import { L as LangGraphGlobal } from '../../stream-CAZ9j0yn.js';
+import { L as LangGraphGlobal } from '../../stream-Bwuk4b6F.js';
 
 async function ensureInitialized(attachGraphPromise) {
   if (globalThis.LG_INIT_PROMISE === void 0) {
     globalThis.LG_INIT_PROMISE = (async () => {
       await LangGraphGlobal.initGlobal();
       await attachGraphPromise();
-      const { GET, POST, DELETE } = await import('../../router-DOVN2of2.js');
+      const { GET, POST, DELETE } = await import('../../router-DIyq7O-Q.js');
       return {
         GET,
         POST,

package/dist/checkpoint-D1UR3Ika.js

@@ -0,0 +1,374 @@
+import { Kysely, SqliteDialect, sql } from 'kysely';
+import { BaseCheckpointSaver, TASKS, copyCheckpoint, maxChannelVersion } from '@langchain/langgraph-checkpoint';
+
+const checkpointMetadataKeys = ["source", "step", "parents"];
+function validateKeys(keys) {
+  return keys;
+}
+const validCheckpointMetadataKeys = validateKeys(
+  checkpointMetadataKeys
+);
+class SqliteSaver extends BaseCheckpointSaver {
+  db;
+  isSetup;
+  constructor(dialect, serde) {
+    super(serde);
+    this.db = new Kysely({
+      dialect
+    });
+    this.isSetup = false;
+  }
+  static async fromConnStringAsync(connStringOrLocalPath) {
+    let saver;
+    if (globalThis.Bun) {
+      console.log("LG | Using BunWorkerDialect " + connStringOrLocalPath);
+      const { BunWorkerDialect } = await import('kysely-bun-worker');
+      saver = new SqliteSaver(new BunWorkerDialect({ url: connStringOrLocalPath }));
+    } else {
+      const { default: Database } = await import('better-sqlite3');
+      console.log("LG | Using BetterSQLite3Dialect");
+      const database = new Database(connStringOrLocalPath);
+      saver = new SqliteSaver(new SqliteDialect({ database }));
+    }
+    await saver.setup();
+    return saver;
+  }
+  async setup() {
+    if (this.isSetup) {
+      return;
+    }
+    await sql`PRAGMA journal_mode = WAL`.execute(this.db);
+    await sql`
+CREATE TABLE IF NOT EXISTS checkpoints (
+  thread_id TEXT NOT NULL,
+  checkpoint_ns TEXT NOT NULL DEFAULT '',
+  checkpoint_id TEXT NOT NULL,
+  parent_checkpoint_id TEXT,
+  type TEXT,
+  checkpoint BLOB,
+  metadata BLOB,
+  PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id)
+)`.execute(this.db);
+    await sql`
+CREATE TABLE IF NOT EXISTS writes (
+  thread_id TEXT NOT NULL,
+  checkpoint_ns TEXT NOT NULL DEFAULT '',
+  checkpoint_id TEXT NOT NULL,
+  task_id TEXT NOT NULL,
+  idx INTEGER NOT NULL,
+  channel TEXT NOT NULL,
+  type TEXT,
+  value BLOB,
+  PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id, task_id, idx)
+)`.execute(this.db);
+    this.isSetup = true;
+  }
+  async getTuple(config) {
+    await this.setup();
+    const { thread_id, checkpoint_ns = "", checkpoint_id } = config.configurable ?? {};
+    let query = this.db.selectFrom("checkpoints").select([
+      "thread_id",
+      "checkpoint_ns",
+      "checkpoint_id",
+      "parent_checkpoint_id",
+      "type",
+      "checkpoint",
+      "metadata",
+      sql`(
+                    SELECT json_group_array(
+                        json_object(
+                            'task_id', pw.task_id,
+                            'channel', pw.channel,
+                            'type', pw.type,
+                            'value', CAST(pw.value AS TEXT)
+                        )
+                    )
+                    FROM writes as pw
+                    WHERE pw.thread_id = checkpoints.thread_id
+                        AND pw.checkpoint_ns = checkpoints.checkpoint_ns
+                        AND pw.checkpoint_id = checkpoints.checkpoint_id
+                )`.as("pending_writes"),
+      sql`(
+                    SELECT json_group_array(
+                        json_object(
+                            'type', ps.type,
+                            'value', CAST(ps.value AS TEXT)
+                        )
+                    )
+                    FROM writes as ps
+                    WHERE ps.thread_id = checkpoints.thread_id
+                        AND ps.checkpoint_ns = checkpoints.checkpoint_ns
+                        AND ps.checkpoint_id = checkpoints.parent_checkpoint_id
+                        AND ps.channel = ${TASKS}
+                    ORDER BY ps.idx
+                )`.as("pending_sends")
+    ]).where("thread_id", "=", thread_id).where("checkpoint_ns", "=", checkpoint_ns);
+    if (checkpoint_id) {
+      query = query.where("checkpoint_id", "=", checkpoint_id);
+    } else {
+      query = query.orderBy("checkpoint_id", "desc").limit(1);
+    }
+    const row = await query.executeTakeFirst();
+    if (!row) return void 0;
+    let finalConfig = config;
+    if (!checkpoint_id) {
+      finalConfig = {
+        configurable: {
+          thread_id: row.thread_id,
+          checkpoint_ns,
+          checkpoint_id: row.checkpoint_id
+        }
+      };
+    }
+    if (finalConfig.configurable?.thread_id === void 0 || finalConfig.configurable?.checkpoint_id === void 0) {
+      throw new Error("Missing thread_id or checkpoint_id");
+    }
+    const pendingWrites = await Promise.all(
+      JSON.parse(row.pending_writes).map(async (write) => {
+        return [
+          write.task_id,
+          write.channel,
+          await this.serde.loadsTyped(write.type ?? "json", write.value ?? "")
+        ];
+      })
+    );
+    const checkpoint = await this.serde.loadsTyped(row.type ?? "json", row.checkpoint.toString());
+    if (checkpoint.v < 4 && row.parent_checkpoint_id != null) {
+      await this.migratePendingSends(checkpoint, row.thread_id, row.parent_checkpoint_id);
+    }
+    return {
+      checkpoint,
+      config: finalConfig,
+      metadata: await this.serde.loadsTyped(row.type ?? "json", row.metadata.toString()),
+      parentConfig: row.parent_checkpoint_id ? {
+        configurable: {
+          thread_id: row.thread_id,
+          checkpoint_ns,
+          checkpoint_id: row.parent_checkpoint_id
+        }
+      } : void 0,
+      pendingWrites
+    };
+  }
+  async *list(config, options) {
+    const { limit, before, filter } = options ?? {};
+    await this.setup();
+    const thread_id = config.configurable?.thread_id;
+    const checkpoint_ns = config.configurable?.checkpoint_ns;
+    let query = this.db.selectFrom("checkpoints").select([
+      "thread_id",
+      "checkpoint_ns",
+      "checkpoint_id",
+      "parent_checkpoint_id",
+      "type",
+      "checkpoint",
+      "metadata",
+      sql`(
+                    SELECT json_group_array(
+                        json_object(
+                            'task_id', pw.task_id,
+                            'channel', pw.channel,
+                            'type', pw.type,
+                            'value', CAST(pw.value AS TEXT)
+                        )
+                    )
+                    FROM writes as pw
+                    WHERE pw.thread_id = checkpoints.thread_id
+                        AND pw.checkpoint_ns = checkpoints.checkpoint_ns
+                        AND pw.checkpoint_id = checkpoints.checkpoint_id
+                )`.as("pending_writes"),
+      sql`(
+                    SELECT json_group_array(
+                        json_object(
+                            'type', ps.type,
+                            'value', CAST(ps.value AS TEXT)
+                        )
+                    )
+                    FROM writes as ps
+                    WHERE ps.thread_id = checkpoints.thread_id
+                        AND ps.checkpoint_ns = checkpoints.checkpoint_ns
+                        AND ps.checkpoint_id = checkpoints.parent_checkpoint_id
+                        AND ps.channel = ${TASKS}
+                    ORDER BY ps.idx
+                )`.as("pending_sends")
+    ]);
+    if (thread_id) {
+      query = query.where("thread_id", "=", thread_id);
+    }
+    if (checkpoint_ns !== void 0 && checkpoint_ns !== null) {
+      query = query.where("checkpoint_ns", "=", checkpoint_ns);
+    }
+    if (before?.configurable?.checkpoint_id !== void 0) {
+      query = query.where("checkpoint_id", "<", before.configurable.checkpoint_id);
+    }
+    const sanitizedFilter = Object.fromEntries(
+      Object.entries(filter ?? {}).filter(
+        ([key, value]) => value !== void 0 && validCheckpointMetadataKeys.includes(key)
+      )
+    );
+    for (const [key, value] of Object.entries(sanitizedFilter)) {
+      query = query.where(
+        sql`json_extract(CAST(metadata AS TEXT), ${sql.lit("$." + key)})`,
+        "=",
+        sql.lit(JSON.stringify(value))
+      );
+    }
+    query = query.orderBy("checkpoint_id", "desc");
+    if (limit) {
+      query = query.limit(parseInt(limit, 10));
+    }
+    const rows = await query.execute();
+    for (const row of rows) {
+      const pendingWrites = await Promise.all(
+        JSON.parse(row.pending_writes).map(async (write) => {
+          return [
+            write.task_id,
+            write.channel,
+            await this.serde.loadsTyped(write.type ?? "json", write.value ?? "")
+          ];
+        })
+      );
+      const checkpoint = await this.serde.loadsTyped(
+        row.type ?? "json",
+        row.checkpoint.toString()
+      );
+      if (checkpoint.v < 4 && row.parent_checkpoint_id != null) {
+        await this.migratePendingSends(checkpoint, row.thread_id, row.parent_checkpoint_id);
+      }
+      yield {
+        config: {
+          configurable: {
+            thread_id: row.thread_id,
+            checkpoint_ns: row.checkpoint_ns,
+            checkpoint_id: row.checkpoint_id
+          }
+        },
+        checkpoint,
+        metadata: await this.serde.loadsTyped(
+          row.type ?? "json",
+          row.metadata.toString()
+        ),
+        parentConfig: row.parent_checkpoint_id ? {
+          configurable: {
+            thread_id: row.thread_id,
+            checkpoint_ns: row.checkpoint_ns,
+            checkpoint_id: row.parent_checkpoint_id
+          }
+        } : void 0,
+        pendingWrites
+      };
+    }
+  }
+  async put(config, checkpoint, metadata) {
+    await this.setup();
+    if (!config.configurable) {
+      throw new Error("Empty configuration supplied.");
+    }
+    const thread_id = config.configurable?.thread_id;
+    const checkpoint_ns = config.configurable?.checkpoint_ns ?? "";
+    const parent_checkpoint_id = config.configurable?.checkpoint_id;
+    if (!thread_id) {
+      throw new Error(`Missing "thread_id" field in passed "config.configurable".`);
+    }
+    const preparedCheckpoint = copyCheckpoint(checkpoint);
+    const [[type1, serializedCheckpoint], [type2, serializedMetadata]] = await Promise.all([
+      this.serde.dumpsTyped(preparedCheckpoint),
+      this.serde.dumpsTyped(metadata)
+    ]);
+    if (type1 !== type2) {
+      throw new Error("Failed to serialized checkpoint and metadata to the same type.");
+    }
+    await this.db.insertInto("checkpoints").values({
+      thread_id,
+      checkpoint_ns,
+      checkpoint_id: checkpoint.id,
+      parent_checkpoint_id: parent_checkpoint_id ?? null,
+      type: type1,
+      checkpoint: new Uint8Array(Buffer.from(serializedCheckpoint)),
+      metadata: new Uint8Array(Buffer.from(serializedMetadata))
+    }).onConflict(
+      (oc) => oc.columns(["thread_id", "checkpoint_ns", "checkpoint_id"]).doUpdateSet({
+        parent_checkpoint_id: parent_checkpoint_id ?? null,
+        type: type1,
+        checkpoint: new Uint8Array(Buffer.from(serializedCheckpoint)),
+        metadata: new Uint8Array(Buffer.from(serializedMetadata))
+      })
+    ).execute();
+    return {
+      configurable: {
+        thread_id,
+        checkpoint_ns,
+        checkpoint_id: checkpoint.id
+      }
+    };
+  }
+  async putWrites(config, writes, taskId) {
+    await this.setup();
+    if (!config.configurable) {
+      throw new Error("Empty configuration supplied.");
+    }
+    if (!config.configurable?.thread_id) {
+      throw new Error("Missing thread_id field in config.configurable.");
+    }
+    if (!config.configurable?.checkpoint_id) {
+      throw new Error("Missing checkpoint_id field in config.configurable.");
+    }
+    const values = await Promise.all(
+      writes.map(async (write, idx) => {
+        const [type, serializedWrite] = await this.serde.dumpsTyped(write[1]);
+        return {
+          thread_id: config.configurable.thread_id,
+          checkpoint_ns: config.configurable.checkpoint_ns ?? "",
+          checkpoint_id: config.configurable.checkpoint_id,
+          task_id: taskId,
+          idx,
+          channel: write[0],
+          type,
+          value: new Uint8Array(Buffer.from(serializedWrite))
+        };
+      })
+    );
+    if (values.length > 0) {
+      await this.db.transaction().execute(async (trx) => {
+        for (const value of values) {
+          await trx.insertInto("writes").values(value).onConflict(
+            (oc) => oc.columns(["thread_id", "checkpoint_ns", "checkpoint_id", "task_id", "idx"]).doUpdateSet({
+              channel: value.channel,
+              type: value.type,
+              value: value.value
+            })
+          ).execute();
+        }
+      });
+    }
+  }
+  async deleteThread(threadId) {
+    await this.db.transaction().execute(async (trx) => {
+      await trx.deleteFrom("checkpoints").where("thread_id", "=", threadId).execute();
+      await trx.deleteFrom("writes").where("thread_id", "=", threadId).execute();
+    });
+  }
+  async migratePendingSends(checkpoint, threadId, parentCheckpointId) {
+    const result = await this.db.selectFrom("writes as ps").select([
+      "ps.checkpoint_id",
+      sql`json_group_array(
+                    json_object(
+                        'type', ps.type,
+                        'value', CAST(ps.value AS TEXT)
+                    )
+                )`.as("pending_sends")
+    ]).where("ps.thread_id", "=", threadId).where("ps.checkpoint_id", "=", parentCheckpointId).where("ps.channel", "=", TASKS).orderBy("ps.idx").executeTakeFirst();
+    if (!result) return;
+    const mutableCheckpoint = checkpoint;
+    mutableCheckpoint.channel_values ??= {};
+    mutableCheckpoint.channel_values[TASKS] = await Promise.all(
+      JSON.parse(result.pending_sends).map(
+        ({ type, value }) => this.serde.loadsTyped(type, value)
+      )
+    );
+    mutableCheckpoint.channel_versions[TASKS] = Object.keys(checkpoint.channel_versions).length > 0 ? maxChannelVersion(...Object.values(checkpoint.channel_versions)) : this.getNextVersion(void 0);
+  }
+}
+
+export { SqliteSaver };
+//# sourceMappingURL=checkpoint-D1UR3Ika.js.map

package/dist/checkpoint-D1UR3Ika.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoint-D1UR3Ika.js","sources":["../src/storage/sqlite/checkpoint.ts"],"sourcesContent":["import { Dialect, Kysely, SqliteDialect, sql } from 'kysely';\nimport type { RunnableConfig } from '@langchain/core/runnables';\n\nimport {\n    BaseCheckpointSaver,\n    type Checkpoint,\n    type CheckpointListOptions,\n    type CheckpointTuple,\n    type SerializerProtocol,\n    type PendingWrite,\n    type CheckpointMetadata,\n    TASKS,\n    copyCheckpoint,\n    maxChannelVersion,\n} from '@langchain/langgraph-checkpoint';\n\n// Kysely 数据库表类型定义\ninterface CheckpointsTable {\n    thread_id: string;\n    checkpoint_ns: string;\n    checkpoint_id: string;\n    parent_checkpoint_id: string | null;\n    type: string | null;\n    checkpoint: Uint8Array;\n    metadata: Uint8Array;\n}\n\ninterface WritesTable {\n    thread_id: string;\n    checkpoint_ns: string;\n    checkpoint_id: string;\n    task_id: string;\n    idx: number;\n    channel: string;\n    type: string | null;\n    value: Uint8Array | null;\n}\n\ninterface CheckpointDatabase {\n    checkpoints: CheckpointsTable;\n    writes: WritesTable;\n}\n\ninterface CheckpointRow {\n    checkpoint: string;\n    metadata: string;\n    parent_checkpoint_id?: string;\n    thread_id: string;\n    checkpoint_id: string;\n    checkpoint_ns?: string;\n    type?: string;\n    pending_writes: string;\n}\n\ninterface PendingWriteColumn {\n    task_id: string;\n    channel: string;\n    type: string;\n    value: string;\n}\n\ninterface PendingSendColumn {\n    type: string;\n    value: string;\n}\n\n// In the `SqliteSaver.list` method, we need to sanitize the `options.filter` argument to ensure it only contains keys\n// that are part of the `CheckpointMetadata` type. The lines below ensure that we get compile-time errors if the list\n// of keys that we use is out of sync with the `CheckpointMetadata` type.\nconst checkpointMetadataKeys = ['source', 'step', 'parents'] as const;\n\ntype CheckKeys<T, K extends readonly (keyof T)[]> = [K[number]] extends [keyof T]\n    ? [keyof T] extends [K[number]]\n        ? K\n        : never\n    : never;\n\nfunction validateKeys<T, K extends readonly (keyof T)[]>(keys: CheckKeys<T, K>): K {\n    return keys;\n}\n\n// If this line fails to compile, the list of keys that we use in the `SqliteSaver.list` method is out of sync with the\n// `CheckpointMetadata` type. In that case, just update `checkpointMetadataKeys` to contain all the keys in\n// `CheckpointMetadata`\nconst validCheckpointMetadataKeys = validateKeys<CheckpointMetadata, typeof checkpointMetadataKeys>(\n    checkpointMetadataKeys,\n);\n\nexport class SqliteSaver extends BaseCheckpointSaver {\n    db: Kysely<CheckpointDatabase>;\n\n    protected isSetup: boolean;\n\n    constructor(dialect: Dialect, serde?: SerializerProtocol) {\n        super(serde);\n        this.db = new Kysely<CheckpointDatabase>({\n            dialect,\n        });\n        this.isSetup = false;\n    }\n\n    static async fromConnStringAsync(connStringOrLocalPath: string): Promise<SqliteSaver> {\n        let saver: SqliteSaver;\n        /** @ts-ignore */\n        if (globalThis.Bun) {\n            console.log('LG | Using BunWorkerDialect ' + connStringOrLocalPath);\n            const { BunWorkerDialect } = await import('kysely-bun-worker');\n            saver = new SqliteSaver(new BunWorkerDialect({ url: connStringOrLocalPath }));\n        } else {\n            /** @ts-ignore */\n            const { default: Database } = await import('better-sqlite3');\n            console.log('LG | Using BetterSQLite3Dialect');\n            const database = new Database(connStringOrLocalPath);\n            saver = new SqliteSaver(new SqliteDialect({ database }));\n        }\n        await saver.setup();\n        return saver;\n    }\n\n    protected async setup(): Promise<void> {\n        if (this.isSetup) {\n            return;\n        }\n\n        await sql`PRAGMA journal_mode = WAL`.execute(this.db);\n\n        await sql`\nCREATE TABLE IF NOT EXISTS checkpoints (\n  thread_id TEXT NOT NULL,\n  checkpoint_ns TEXT NOT NULL DEFAULT '',\n  checkpoint_id TEXT NOT NULL,\n  parent_checkpoint_id TEXT,\n  type TEXT,\n  checkpoint BLOB,\n  metadata BLOB,\n  PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id)\n)`.execute(this.db);\n\n        await sql`\nCREATE TABLE IF NOT EXISTS writes (\n  thread_id TEXT NOT NULL,\n  checkpoint_ns TEXT NOT NULL DEFAULT '',\n  checkpoint_id TEXT NOT NULL,\n  task_id TEXT NOT NULL,\n  idx INTEGER NOT NULL,\n  channel TEXT NOT NULL,\n  type TEXT,\n  value BLOB,\n  PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id, task_id, idx)\n)`.execute(this.db);\n        this.isSetup = true;\n    }\n\n    async getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined> {\n        await this.setup();\n        const { thread_id, checkpoint_ns = '', checkpoint_id } = config.configurable ?? {};\n\n        let query = this.db\n            .selectFrom('checkpoints')\n            .select([\n                'thread_id',\n                'checkpoint_ns',\n                'checkpoint_id',\n                'parent_checkpoint_id',\n                'type',\n                'checkpoint',\n                'metadata',\n                sql<string>`(\n                    SELECT json_group_array(\n                        json_object(\n                            'task_id', pw.task_id,\n                            'channel', pw.channel,\n                            'type', pw.type,\n                            'value', CAST(pw.value AS TEXT)\n                        )\n                    )\n                    FROM writes as pw\n                    WHERE pw.thread_id = checkpoints.thread_id\n                        AND pw.checkpoint_ns = checkpoints.checkpoint_ns\n                        AND pw.checkpoint_id = checkpoints.checkpoint_id\n                )`.as('pending_writes'),\n                sql<string>`(\n                    SELECT json_group_array(\n                        json_object(\n                            'type', ps.type,\n                            'value', CAST(ps.value AS TEXT)\n                        )\n                    )\n                    FROM writes as ps\n                    WHERE ps.thread_id = checkpoints.thread_id\n                        AND ps.checkpoint_ns = checkpoints.checkpoint_ns\n                        AND ps.checkpoint_id = checkpoints.parent_checkpoint_id\n                        AND ps.channel = ${TASKS}\n                    ORDER BY ps.idx\n                )`.as('pending_sends'),\n            ])\n            .where('thread_id', '=', thread_id)\n            .where('checkpoint_ns', '=', checkpoint_ns);\n\n        if (checkpoint_id) {\n            query = query.where('checkpoint_id', '=', checkpoint_id);\n        } else {\n            query = query.orderBy('checkpoint_id', 'desc').limit(1);\n        }\n\n        const row = await query.executeTakeFirst();\n        if (!row) return undefined;\n\n        let finalConfig = config;\n\n        if (!checkpoint_id) {\n            finalConfig = {\n                configurable: {\n                    thread_id: row.thread_id,\n                    checkpoint_ns,\n                    checkpoint_id: row.checkpoint_id,\n                },\n            };\n        }\n\n        if (\n            finalConfig.configurable?.thread_id === undefined ||\n            finalConfig.configurable?.checkpoint_id === undefined\n        ) {\n            throw new Error('Missing thread_id or checkpoint_id');\n        }\n\n        const pendingWrites = await Promise.all(\n            (JSON.parse(row.pending_writes) as PendingWriteColumn[]).map(async (write) => {\n                return [\n                    write.task_id,\n                    write.channel,\n                    await this.serde.loadsTyped(write.type ?? 'json', write.value ?? ''),\n                ] as [string, string, unknown];\n            }),\n        );\n\n        const checkpoint = (await this.serde.loadsTyped(row.type ?? 'json', row.checkpoint.toString())) as Checkpoint;\n\n        if (checkpoint.v < 4 && row.parent_checkpoint_id != null) {\n            await this.migratePendingSends(checkpoint, row.thread_id, row.parent_checkpoint_id);\n        }\n\n        return {\n            checkpoint,\n            config: finalConfig,\n            metadata: (await this.serde.loadsTyped(row.type ?? 'json', row.metadata.toString())) as CheckpointMetadata,\n            parentConfig: row.parent_checkpoint_id\n                ? {\n                      configurable: {\n                          thread_id: row.thread_id,\n                          checkpoint_ns,\n                          checkpoint_id: row.parent_checkpoint_id,\n                      },\n                  }\n                : undefined,\n            pendingWrites,\n        };\n    }\n\n    async *list(config: RunnableConfig, options?: CheckpointListOptions): AsyncGenerator<CheckpointTuple> {\n        const { limit, before, filter } = options ?? {};\n        await this.setup();\n        const thread_id = config.configurable?.thread_id;\n        const checkpoint_ns = config.configurable?.checkpoint_ns;\n\n        let query = this.db.selectFrom('checkpoints').select([\n            'thread_id',\n            'checkpoint_ns',\n            'checkpoint_id',\n            'parent_checkpoint_id',\n            'type',\n            'checkpoint',\n            'metadata',\n            sql<string>`(\n                    SELECT json_group_array(\n                        json_object(\n                            'task_id', pw.task_id,\n                            'channel', pw.channel,\n                            'type', pw.type,\n                            'value', CAST(pw.value AS TEXT)\n                        )\n                    )\n                    FROM writes as pw\n                    WHERE pw.thread_id = checkpoints.thread_id\n                        AND pw.checkpoint_ns = checkpoints.checkpoint_ns\n                        AND pw.checkpoint_id = checkpoints.checkpoint_id\n                )`.as('pending_writes'),\n            sql<string>`(\n                    SELECT json_group_array(\n                        json_object(\n                            'type', ps.type,\n                            'value', CAST(ps.value AS TEXT)\n                        )\n                    )\n                    FROM writes as ps\n                    WHERE ps.thread_id = checkpoints.thread_id\n                        AND ps.checkpoint_ns = checkpoints.checkpoint_ns\n                        AND ps.checkpoint_id = checkpoints.parent_checkpoint_id\n                        AND ps.channel = ${TASKS}\n                    ORDER BY ps.idx\n                )`.as('pending_sends'),\n        ]);\n\n        if (thread_id) {\n            query = query.where('thread_id', '=', thread_id);\n        }\n\n        if (checkpoint_ns !== undefined && checkpoint_ns !== null) {\n            query = query.where('checkpoint_ns', '=', checkpoint_ns);\n        }\n\n        if (before?.configurable?.checkpoint_id !== undefined) {\n            query = query.where('checkpoint_id', '<', before.configurable.checkpoint_id);\n        }\n\n        const sanitizedFilter = Object.fromEntries(\n            Object.entries(filter ?? {}).filter(\n                ([key, value]) =>\n                    value !== undefined && validCheckpointMetadataKeys.includes(key as keyof CheckpointMetadata),\n            ),\n        );\n\n        for (const [key, value] of Object.entries(sanitizedFilter)) {\n            query = query.where(\n                sql`json_extract(CAST(metadata AS TEXT), ${sql.lit('$.' + key)})`,\n                '=',\n                sql.lit(JSON.stringify(value)),\n            );\n        }\n\n        query = query.orderBy('checkpoint_id', 'desc');\n\n        if (limit) {\n            // eslint-disable-next-line @typescript-eslint/no-explicit-any\n            query = query.limit(parseInt(limit as any, 10));\n        }\n\n        const rows = await query.execute();\n\n        for (const row of rows) {\n            const pendingWrites = await Promise.all(\n                (JSON.parse(row.pending_writes) as PendingWriteColumn[]).map(async (write) => {\n                    return [\n                        write.task_id,\n                        write.channel,\n                        await this.serde.loadsTyped(write.type ?? 'json', write.value ?? ''),\n                    ] as [string, string, unknown];\n                }),\n            );\n\n            const checkpoint = (await this.serde.loadsTyped(\n                row.type ?? 'json',\n                row.checkpoint.toString(),\n            )) as Checkpoint;\n\n            if (checkpoint.v < 4 && row.parent_checkpoint_id != null) {\n                await this.migratePendingSends(checkpoint, row.thread_id, row.parent_checkpoint_id);\n            }\n\n            yield {\n                config: {\n                    configurable: {\n                        thread_id: row.thread_id,\n                        checkpoint_ns: row.checkpoint_ns,\n                        checkpoint_id: row.checkpoint_id,\n                    },\n                },\n                checkpoint,\n                metadata: (await this.serde.loadsTyped(\n                    row.type ?? 'json',\n                    row.metadata.toString(),\n                )) as CheckpointMetadata,\n                parentConfig: row.parent_checkpoint_id\n                    ? {\n                          configurable: {\n                              thread_id: row.thread_id,\n                              checkpoint_ns: row.checkpoint_ns,\n                              checkpoint_id: row.parent_checkpoint_id,\n                          },\n                      }\n                    : undefined,\n                pendingWrites,\n            };\n        }\n    }\n\n    async put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata): Promise<RunnableConfig> {\n        await this.setup();\n\n        if (!config.configurable) {\n            throw new Error('Empty configuration supplied.');\n        }\n\n        const thread_id = config.configurable?.thread_id;\n        const checkpoint_ns = config.configurable?.checkpoint_ns ?? '';\n        const parent_checkpoint_id = config.configurable?.checkpoint_id;\n\n        if (!thread_id) {\n            throw new Error(`Missing \"thread_id\" field in passed \"config.configurable\".`);\n        }\n\n        const preparedCheckpoint: Partial<Checkpoint> = copyCheckpoint(checkpoint);\n\n        const [[type1, serializedCheckpoint], [type2, serializedMetadata]] = await Promise.all([\n            this.serde.dumpsTyped(preparedCheckpoint),\n            this.serde.dumpsTyped(metadata),\n        ]);\n\n        if (type1 !== type2) {\n            throw new Error('Failed to serialized checkpoint and metadata to the same type.');\n        }\n\n        await this.db\n            .insertInto('checkpoints')\n            .values({\n                thread_id,\n                checkpoint_ns,\n                checkpoint_id: checkpoint.id,\n                parent_checkpoint_id: parent_checkpoint_id ?? null,\n                type: type1,\n                checkpoint: new Uint8Array(Buffer.from(serializedCheckpoint)),\n                metadata: new Uint8Array(Buffer.from(serializedMetadata)),\n            })\n            .onConflict((oc) =>\n                oc.columns(['thread_id', 'checkpoint_ns', 'checkpoint_id']).doUpdateSet({\n                    parent_checkpoint_id: parent_checkpoint_id ?? null,\n                    type: type1,\n                    checkpoint: new Uint8Array(Buffer.from(serializedCheckpoint)),\n                    metadata: new Uint8Array(Buffer.from(serializedMetadata)),\n                }),\n            )\n            .execute();\n\n        return {\n            configurable: {\n                thread_id,\n                checkpoint_ns,\n                checkpoint_id: checkpoint.id,\n            },\n        };\n    }\n\n    async putWrites(config: RunnableConfig, writes: PendingWrite[], taskId: string): Promise<void> {\n        await this.setup();\n\n        if (!config.configurable) {\n            throw new Error('Empty configuration supplied.');\n        }\n\n        if (!config.configurable?.thread_id) {\n            throw new Error('Missing thread_id field in config.configurable.');\n        }\n\n        if (!config.configurable?.checkpoint_id) {\n            throw new Error('Missing checkpoint_id field in config.configurable.');\n        }\n\n        const values = await Promise.all(\n            writes.map(async (write, idx) => {\n                const [type, serializedWrite] = await this.serde.dumpsTyped(write[1]);\n                return {\n                    thread_id: config.configurable!.thread_id,\n                    checkpoint_ns: config.configurable!.checkpoint_ns ?? '',\n                    checkpoint_id: config.configurable!.checkpoint_id,\n                    task_id: taskId,\n                    idx,\n                    channel: write[0],\n                    type,\n                    value: new Uint8Array(Buffer.from(serializedWrite)),\n                };\n            }),\n        );\n\n        if (values.length > 0) {\n            await this.db.transaction().execute(async (trx) => {\n                for (const value of values) {\n                    await trx\n                        .insertInto('writes')\n                        .values(value)\n                        .onConflict((oc) =>\n                            oc.columns(['thread_id', 'checkpoint_ns', 'checkpoint_id', 'task_id', 'idx']).doUpdateSet({\n                                channel: value.channel,\n                                type: value.type,\n                                value: value.value,\n                            }),\n                        )\n                        .execute();\n                }\n            });\n        }\n    }\n\n    async deleteThread(threadId: string) {\n        await this.db.transaction().execute(async (trx) => {\n            await trx.deleteFrom('checkpoints').where('thread_id', '=', threadId).execute();\n            await trx.deleteFrom('writes').where('thread_id', '=', threadId).execute();\n        });\n    }\n\n    protected async migratePendingSends(checkpoint: Checkpoint, threadId: string, parentCheckpointId: string) {\n        const result = await this.db\n            .selectFrom('writes as ps')\n            .select([\n                'ps.checkpoint_id',\n                sql<string>`json_group_array(\n                    json_object(\n                        'type', ps.type,\n                        'value', CAST(ps.value AS TEXT)\n                    )\n                )`.as('pending_sends'),\n            ])\n            .where('ps.thread_id', '=', threadId)\n            .where('ps.checkpoint_id', '=', parentCheckpointId)\n            .where('ps.channel', '=', TASKS)\n            .orderBy('ps.idx')\n            .executeTakeFirst();\n\n        if (!result) return;\n\n        const mutableCheckpoint = checkpoint;\n\n        // add pending sends to checkpoint\n        mutableCheckpoint.channel_values ??= {};\n        mutableCheckpoint.channel_values[TASKS] = await Promise.all(\n            JSON.parse(result.pending_sends).map(({ type, value }: PendingSendColumn) =>\n                this.serde.loadsTyped(type, value),\n            ),\n        );\n\n        // add to versions\n        mutableCheckpoint.channel_versions[TASKS] =\n            Object.keys(checkpoint.channel_versions).length > 0\n                ? maxChannelVersion(...Object.values(checkpoint.channel_versions))\n                : this.getNextVersion(undefined);\n    }\n}\n"],"names":[],"mappings":";;;AAqEA,MAAM,sBAAA,GAAyB,CAAC,QAAA,EAAU,MAAA,EAAQ,SAAS,CAAA;AAQ3D,SAAS,aAAgD,IAAA,EAA0B;AAC/E,EAAA,OAAO,IAAA;AACX;AAKA,MAAM,2BAAA,GAA8B,YAAA;AAAA,EAChC;AACJ,CAAA;AAEO,MAAM,oBAAoB,mBAAA,CAAoB;AAAA,EACjD,EAAA;AAAA,EAEU,OAAA;AAAA,EAEV,WAAA,CAAY,SAAkB,KAAA,EAA4B;AACtD,IAAA,KAAA,CAAM,KAAK,CAAA;AACX,IAAA,IAAA,CAAK,EAAA,GAAK,IAAI,MAAA,CAA2B;AAAA,MACrC;AAAA,KACH,CAAA;AACD,IAAA,IAAA,CAAK,OAAA,GAAU,KAAA;AAAA,EACnB;AAAA,EAEA,aAAa,oBAAoB,qBAAA,EAAqD;AAClF,IAAA,IAAI,KAAA;AAEJ,IAAA,IAAI,WAAW,GAAA,EAAK;AAChB,MAAA,OAAA,CAAQ,GAAA,CAAI,iCAAiC,qBAAqB,CAAA;AAClE,MAAA,MAAM,EAAE,gBAAA,EAAiB,GAAI,MAAM,OAAO,mBAAmB,CAAA;AAC7D,MAAA,KAAA,GAAQ,IAAI,YAAY,IAAI,gBAAA,CAAiB,EAAE,GAAA,EAAK,qBAAA,EAAuB,CAAC,CAAA;AAAA,IAChF,CAAA,MAAO;AAEH,MAAA,MAAM,EAAE,OAAA,EAAS,QAAA,EAAS,GAAI,MAAM,OAAO,gBAAgB,CAAA;AAC3D,MAAA,OAAA,CAAQ,IAAI,iCAAiC,CAAA;AAC7C,MAAA,MAAM,QAAA,GAAW,IAAI,QAAA,CAAS,qBAAqB,CAAA;AACnD,MAAA,KAAA,GAAQ,IAAI,WAAA,CAAY,IAAI,cAAc,EAAE,QAAA,EAAU,CAAC,CAAA;AAAA,IAC3D;AACA,IAAA,MAAM,MAAM,KAAA,EAAM;AAClB,IAAA,OAAO,KAAA;AAAA,EACX;AAAA,EAEA,MAAgB,KAAA,GAAuB;AACnC,IAAA,IAAI,KAAK,OAAA,EAAS;AACd,MAAA;AAAA,IACJ;AAEA,IAAA,MAAM,GAAA,CAAA,yBAAA,CAAA,CAA+B,OAAA,CAAQ,IAAA,CAAK,EAAE,CAAA;AAEpD,IAAA,MAAM,GAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,CAAA,CAAA,CAUX,OAAA,CAAQ,KAAK,EAAE,CAAA;AAEV,IAAA,MAAM,GAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,CAAA,CAAA,CAWX,OAAA,CAAQ,KAAK,EAAE,CAAA;AACV,IAAA,IAAA,CAAK,OAAA,GAAU,IAAA;AAAA,EACnB;AAAA,EAEA,MAAM,SAAS,MAAA,EAA8D;AACzE,IAAA,MAAM,KAAK,KAAA,EAAM;AACjB,IAAA,MAAM,EAAE,WAAW,aAAA,GAAgB,EAAA,EAAI,eAAc,GAAI,MAAA,CAAO,gBAAgB,EAAC;AAEjF,IAAA,IAAI,QAAQ,IAAA,CAAK,EAAA,CACZ,UAAA,CAAW,aAAa,EACxB,MAAA,CAAO;AAAA,MACJ,WAAA;AAAA,MACA,eAAA;AAAA,MACA,eAAA;AAAA,MACA,sBAAA;AAAA,MACA,MAAA;AAAA,MACA,YAAA;AAAA,MACA,UAAA;AAAA,MACA,GAAA,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,iBAAA,CAAA,CAaG,GAAG,gBAAgB,CAAA;AAAA,MACtB,GAAA,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,yCAAA,EAW2B,KAAK;AAAA;AAAA,iBAAA,CAAA,CAE7B,GAAG,eAAe;AAAA,KACxB,CAAA,CACA,KAAA,CAAM,WAAA,EAAa,GAAA,EAAK,SAAS,CAAA,CACjC,KAAA,CAAM,eAAA,EAAiB,GAAA,EAAK,aAAa,CAAA;AAE9C,IAAA,IAAI,aAAA,EAAe;AACf,MAAA,KAAA,GAAQ,KAAA,CAAM,KAAA,CAAM,eAAA,EAAiB,GAAA,EAAK,aAAa,CAAA;AAAA,IAC3D,CAAA,MAAO;AACH,MAAA,KAAA,GAAQ,MAAM,OAAA,CAAQ,eAAA,EAAiB,MAAM,CAAA,CAAE,MAAM,CAAC,CAAA;AAAA,IAC1D;AAEA,IAAA,MAAM,GAAA,GAAM,MAAM,KAAA,CAAM,gBAAA,EAAiB;AACzC,IAAA,IAAI,CAAC,KAAK,OAAO,MAAA;AAEjB,IAAA,IAAI,WAAA,GAAc,MAAA;AAElB,IAAA,IAAI,CAAC,aAAA,EAAe;AAChB,MAAA,WAAA,GAAc;AAAA,QACV,YAAA,EAAc;AAAA,UACV,WAAW,GAAA,CAAI,SAAA;AAAA,UACf,aAAA;AAAA,UACA,eAAe,GAAA,CAAI;AAAA;AACvB,OACJ;AAAA,IACJ;AAEA,IAAA,IACI,YAAY,YAAA,EAAc,SAAA,KAAc,UACxC,WAAA,CAAY,YAAA,EAAc,kBAAkB,MAAA,EAC9C;AACE,MAAA,MAAM,IAAI,MAAM,oCAAoC,CAAA;AAAA,IACxD;AAEA,IAAA,MAAM,aAAA,GAAgB,MAAM,OAAA,CAAQ,GAAA;AAAA,MAC/B,KAAK,KAAA,CAAM,GAAA,CAAI,cAAc,CAAA,CAA2B,GAAA,CAAI,OAAO,KAAA,KAAU;AAC1E,QAAA,OAAO;AAAA,UACH,KAAA,CAAM,OAAA;AAAA,UACN,KAAA,CAAM,OAAA;AAAA,UACN,MAAM,KAAK,KAAA,CAAM,UAAA,CAAW,MAAM,IAAA,IAAQ,MAAA,EAAQ,KAAA,CAAM,KAAA,IAAS,EAAE;AAAA,SACvE;AAAA,MACJ,CAAC;AAAA,KACL;AAEA,IAAA,MAAM,UAAA,GAAc,MAAM,IAAA,CAAK,KAAA,CAAM,UAAA,CAAW,GAAA,CAAI,IAAA,IAAQ,MAAA,EAAQ,GAAA,CAAI,UAAA,CAAW,QAAA,EAAU,CAAA;AAE7F,IAAA,IAAI,UAAA,CAAW,CAAA,GAAI,CAAA,IAAK,GAAA,CAAI,wBAAwB,IAAA,EAAM;AACtD,MAAA,MAAM,KAAK,mBAAA,CAAoB,UAAA,EAAY,GAAA,CAAI,SAAA,EAAW,IAAI,oBAAoB,CAAA;AAAA,IACtF;AAEA,IAAA,OAAO;AAAA,MACH,UAAA;AAAA,MACA,MAAA,EAAQ,WAAA;AAAA,MACR,QAAA,EAAW,MAAM,IAAA,CAAK,KAAA,CAAM,UAAA,CAAW,GAAA,CAAI,IAAA,IAAQ,MAAA,EAAQ,GAAA,CAAI,QAAA,CAAS,QAAA,EAAU,CAAA;AAAA,MAClF,YAAA,EAAc,IAAI,oBAAA,GACZ;AAAA,QACI,YAAA,EAAc;AAAA,UACV,WAAW,GAAA,CAAI,SAAA;AAAA,UACf,aAAA;AAAA,UACA,eAAe,GAAA,CAAI;AAAA;AACvB,OACJ,GACA,MAAA;AAAA,MACN;AAAA,KACJ;AAAA,EACJ;AAAA,EAEA,OAAO,IAAA,CAAK,MAAA,EAAwB,OAAA,EAAkE;AAClG,IAAA,MAAM,EAAE,KAAA,EAAO,MAAA,EAAQ,MAAA,EAAO,GAAI,WAAW,EAAC;AAC9C,IAAA,MAAM,KAAK,KAAA,EAAM;AACjB,IAAA,MAAM,SAAA,GAAY,OAAO,YAAA,EAAc,SAAA;AACvC,IAAA,MAAM,aAAA,GAAgB,OAAO,YAAA,EAAc,aAAA;AAE3C,IAAA,IAAI,QAAQ,IAAA,CAAK,EAAA,CAAG,UAAA,CAAW,aAAa,EAAE,MAAA,CAAO;AAAA,MACjD,WAAA;AAAA,MACA,eAAA;AAAA,MACA,eAAA;AAAA,MACA,sBAAA;AAAA,MACA,MAAA;AAAA,MACA,YAAA;AAAA,MACA,UAAA;AAAA,MACA,GAAA,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,iBAAA,CAAA,CAaO,GAAG,gBAAgB,CAAA;AAAA,MAC1B,GAAA,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,yCAAA,EAW+B,KAAK;AAAA;AAAA,iBAAA,CAAA,CAE7B,GAAG,eAAe;AAAA,KAC5B,CAAA;AAED,IAAA,IAAI,SAAA,EAAW;AACX,MAAA,KAAA,GAAQ,KAAA,CAAM,KAAA,CAAM,WAAA,EAAa,GAAA,EAAK,SAAS,CAAA;AAAA,IACnD;AAEA,IAAA,IAAI,aAAA,KAAkB,MAAA,IAAa,aAAA,KAAkB,IAAA,EAAM;AACvD,MAAA,KAAA,GAAQ,KAAA,CAAM,KAAA,CAAM,eAAA,EAAiB,GAAA,EAAK,aAAa,CAAA;AAAA,IAC3D;AAEA,IAAA,IAAI,MAAA,EAAQ,YAAA,EAAc,aAAA,KAAkB,MAAA,EAAW;AACnD,MAAA,KAAA,GAAQ,MAAM,KAAA,CAAM,eAAA,EAAiB,GAAA,EAAK,MAAA,CAAO,aAAa,aAAa,CAAA;AAAA,IAC/E;AAEA,IAAA,MAAM,kBAAkB,MAAA,CAAO,WAAA;AAAA,MAC3B,MAAA,CAAO,OAAA,CAAQ,MAAA,IAAU,EAAE,CAAA,CAAE,MAAA;AAAA,QACzB,CAAC,CAAC,GAAA,EAAK,KAAK,MACR,KAAA,KAAU,MAAA,IAAa,2BAAA,CAA4B,QAAA,CAAS,GAA+B;AAAA;AACnG,KACJ;AAEA,IAAA,KAAA,MAAW,CAAC,GAAA,EAAK,KAAK,KAAK,MAAA,CAAO,OAAA,CAAQ,eAAe,CAAA,EAAG;AACxD,MAAA,KAAA,GAAQ,KAAA,CAAM,KAAA;AAAA,QACV,GAAA,CAAA,qCAAA,EAA2C,GAAA,CAAI,GAAA,CAAI,IAAA,GAAO,GAAG,CAAC,CAAA,CAAA,CAAA;AAAA,QAC9D,GAAA;AAAA,QACA,GAAA,CAAI,GAAA,CAAI,IAAA,CAAK,SAAA,CAAU,KAAK,CAAC;AAAA,OACjC;AAAA,IACJ;AAEA,IAAA,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,eAAA,EAAiB,MAAM,CAAA;AAE7C,IAAA,IAAI,KAAA,EAAO;AAEP,MAAA,KAAA,GAAQ,KAAA,CAAM,KAAA,CAAM,QAAA,CAAS,KAAA,EAAc,EAAE,CAAC,CAAA;AAAA,IAClD;AAEA,IAAA,MAAM,IAAA,GAAO,MAAM,KAAA,CAAM,OAAA,EAAQ;AAEjC,IAAA,KAAA,MAAW,OAAO,IAAA,EAAM;AACpB,MAAA,MAAM,aAAA,GAAgB,MAAM,OAAA,CAAQ,GAAA;AAAA,QAC/B,KAAK,KAAA,CAAM,GAAA,CAAI,cAAc,CAAA,CAA2B,GAAA,CAAI,OAAO,KAAA,KAAU;AAC1E,UAAA,OAAO;AAAA,YACH,KAAA,CAAM,OAAA;AAAA,YACN,KAAA,CAAM,OAAA;AAAA,YACN,MAAM,KAAK,KAAA,CAAM,UAAA,CAAW,MAAM,IAAA,IAAQ,MAAA,EAAQ,KAAA,CAAM,KAAA,IAAS,EAAE;AAAA,WACvE;AAAA,QACJ,CAAC;AAAA,OACL;AAEA,MAAA,MAAM,UAAA,GAAc,MAAM,IAAA,CAAK,KAAA,CAAM,UAAA;AAAA,QACjC,IAAI,IAAA,IAAQ,MAAA;AAAA,QACZ,GAAA,CAAI,WAAW,QAAA;AAAS,OAC5B;AAEA,MAAA,IAAI,UAAA,CAAW,CAAA,GAAI,CAAA,IAAK,GAAA,CAAI,wBAAwB,IAAA,EAAM;AACtD,QAAA,MAAM,KAAK,mBAAA,CAAoB,UAAA,EAAY,GAAA,CAAI,SAAA,EAAW,IAAI,oBAAoB,CAAA;AAAA,MACtF;AAEA,MAAA,MAAM;AAAA,QACF,MAAA,EAAQ;AAAA,UACJ,YAAA,EAAc;AAAA,YACV,WAAW,GAAA,CAAI,SAAA;AAAA,YACf,eAAe,GAAA,CAAI,aAAA;AAAA,YACnB,eAAe,GAAA,CAAI;AAAA;AACvB,SACJ;AAAA,QACA,UAAA;AAAA,QACA,QAAA,EAAW,MAAM,IAAA,CAAK,KAAA,CAAM,UAAA;AAAA,UACxB,IAAI,IAAA,IAAQ,MAAA;AAAA,UACZ,GAAA,CAAI,SAAS,QAAA;AAAS,SAC1B;AAAA,QACA,YAAA,EAAc,IAAI,oBAAA,GACZ;AAAA,UACI,YAAA,EAAc;AAAA,YACV,WAAW,GAAA,CAAI,SAAA;AAAA,YACf,eAAe,GAAA,CAAI,aAAA;AAAA,YACnB,eAAe,GAAA,CAAI;AAAA;AACvB,SACJ,GACA,MAAA;AAAA,QACN;AAAA,OACJ;AAAA,IACJ;AAAA,EACJ;AAAA,EAEA,MAAM,GAAA,CAAI,MAAA,EAAwB,UAAA,EAAwB,QAAA,EAAuD;AAC7G,IAAA,MAAM,KAAK,KAAA,EAAM;AAEjB,IAAA,IAAI,CAAC,OAAO,YAAA,EAAc;AACtB,MAAA,MAAM,IAAI,MAAM,+BAA+B,CAAA;AAAA,IACnD;AAEA,IAAA,MAAM,SAAA,GAAY,OAAO,YAAA,EAAc,SAAA;AACvC,IAAA,MAAM,aAAA,GAAgB,MAAA,CAAO,YAAA,EAAc,aAAA,IAAiB,EAAA;AAC5D,IAAA,MAAM,oBAAA,GAAuB,OAAO,YAAA,EAAc,aAAA;AAElD,IAAA,IAAI,CAAC,SAAA,EAAW;AACZ,MAAA,MAAM,IAAI,MAAM,CAAA,0DAAA,CAA4D,CAAA;AAAA,IAChF;AAEA,IAAA,MAAM,kBAAA,GAA0C,eAAe,UAAU,CAAA;AAEzE,IAAA,MAAM,CAAC,CAAC,KAAA,EAAO,oBAAoB,CAAA,EAAG,CAAC,KAAA,EAAO,kBAAkB,CAAC,CAAA,GAAI,MAAM,OAAA,CAAQ,GAAA,CAAI;AAAA,MACnF,IAAA,CAAK,KAAA,CAAM,UAAA,CAAW,kBAAkB,CAAA;AAAA,MACxC,IAAA,CAAK,KAAA,CAAM,UAAA,CAAW,QAAQ;AAAA,KACjC,CAAA;AAED,IAAA,IAAI,UAAU,KAAA,EAAO;AACjB,MAAA,MAAM,IAAI,MAAM,gEAAgE,CAAA;AAAA,IACpF;AAEA,IAAA,MAAM,IAAA,CAAK,EAAA,CACN,UAAA,CAAW,aAAa,EACxB,MAAA,CAAO;AAAA,MACJ,SAAA;AAAA,MACA,aAAA;AAAA,MACA,eAAe,UAAA,CAAW,EAAA;AAAA,MAC1B,sBAAsB,oBAAA,IAAwB,IAAA;AAAA,MAC9C,IAAA,EAAM,KAAA;AAAA,MACN,YAAY,IAAI,UAAA,CAAW,MAAA,CAAO,IAAA,CAAK,oBAAoB,CAAC,CAAA;AAAA,MAC5D,UAAU,IAAI,UAAA,CAAW,MAAA,CAAO,IAAA,CAAK,kBAAkB,CAAC;AAAA,KAC3D,CAAA,CACA,UAAA;AAAA,MAAW,CAAC,EAAA,KACT,EAAA,CAAG,OAAA,CAAQ,CAAC,aAAa,eAAA,EAAiB,eAAe,CAAC,CAAA,CAAE,WAAA,CAAY;AAAA,QACpE,sBAAsB,oBAAA,IAAwB,IAAA;AAAA,QAC9C,IAAA,EAAM,KAAA;AAAA,QACN,YAAY,IAAI,UAAA,CAAW,MAAA,CAAO,IAAA,CAAK,oBAAoB,CAAC,CAAA;AAAA,QAC5D,UAAU,IAAI,UAAA,CAAW,MAAA,CAAO,IAAA,CAAK,kBAAkB,CAAC;AAAA,OAC3D;AAAA,MAEJ,OAAA,EAAQ;AAEb,IAAA,OAAO;AAAA,MACH,YAAA,EAAc;AAAA,QACV,SAAA;AAAA,QACA,aAAA;AAAA,QACA,eAAe,UAAA,CAAW;AAAA;AAC9B,KACJ;AAAA,EACJ;AAAA,EAEA,MAAM,SAAA,CAAU,MAAA,EAAwB,MAAA,EAAwB,MAAA,EAA+B;AAC3F,IAAA,MAAM,KAAK,KAAA,EAAM;AAEjB,IAAA,IAAI,CAAC,OAAO,YAAA,EAAc;AACtB,MAAA,MAAM,IAAI,MAAM,+BAA+B,CAAA;AAAA,IACnD;AAEA,IAAA,IAAI,CAAC,MAAA,CAAO,YAAA,EAAc,SAAA,EAAW;AACjC,MAAA,MAAM,IAAI,MAAM,iDAAiD,CAAA;AAAA,IACrE;AAEA,IAAA,IAAI,CAAC,MAAA,CAAO,YAAA,EAAc,aAAA,EAAe;AACrC,MAAA,MAAM,IAAI,MAAM,qDAAqD,CAAA;AAAA,IACzE;AAEA,IAAA,MAAM,MAAA,GAAS,MAAM,OAAA,CAAQ,GAAA;AAAA,MACzB,MAAA,CAAO,GAAA,CAAI,OAAO,KAAA,EAAO,GAAA,KAAQ;AAC7B,QAAA,MAAM,CAAC,IAAA,EAAM,eAAe,CAAA,GAAI,MAAM,KAAK,KAAA,CAAM,UAAA,CAAW,KAAA,CAAM,CAAC,CAAC,CAAA;AACpE,QAAA,OAAO;AAAA,UACH,SAAA,EAAW,OAAO,YAAA,CAAc,SAAA;AAAA,UAChC,aAAA,EAAe,MAAA,CAAO,YAAA,CAAc,aAAA,IAAiB,EAAA;AAAA,UACrD,aAAA,EAAe,OAAO,YAAA,CAAc,aAAA;AAAA,UACpC,OAAA,EAAS,MAAA;AAAA,UACT,GAAA;AAAA,UACA,OAAA,EAAS,MAAM,CAAC,CAAA;AAAA,UAChB,IAAA;AAAA,UACA,OAAO,IAAI,UAAA,CAAW,MAAA,CAAO,IAAA,CAAK,eAAe,CAAC;AAAA,SACtD;AAAA,MACJ,CAAC;AAAA,KACL;AAEA,IAAA,IAAI,MAAA,CAAO,SAAS,CAAA,EAAG;AACnB,MAAA,MAAM,KAAK,EAAA,CAAG,WAAA,EAAY,CAAE,OAAA,CAAQ,OAAO,GAAA,KAAQ;AAC/C,QAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AACxB,UAAA,MAAM,IACD,UAAA,CAAW,QAAQ,CAAA,CACnB,MAAA,CAAO,KAAK,CAAA,CACZ,UAAA;AAAA,YAAW,CAAC,EAAA,KACT,EAAA,CAAG,OAAA,CAAQ,CAAC,WAAA,EAAa,eAAA,EAAiB,eAAA,EAAiB,SAAA,EAAW,KAAK,CAAC,CAAA,CAAE,WAAA,CAAY;AAAA,cACtF,SAAS,KAAA,CAAM,OAAA;AAAA,cACf,MAAM,KAAA,CAAM,IAAA;AAAA,cACZ,OAAO,KAAA,CAAM;AAAA,aAChB;AAAA,YAEJ,OAAA,EAAQ;AAAA,QACjB;AAAA,MACJ,CAAC,CAAA;AAAA,IACL;AAAA,EACJ;AAAA,EAEA,MAAM,aAAa,QAAA,EAAkB;AACjC,IAAA,MAAM,KAAK,EAAA,CAAG,WAAA,EAAY,CAAE,OAAA,CAAQ,OAAO,GAAA,KAAQ;AAC/C,MAAA,MAAM,GAAA,CAAI,WAAW,aAAa,CAAA,CAAE,MAAM,WAAA,EAAa,GAAA,EAAK,QAAQ,CAAA,CAAE,OAAA,EAAQ;AAC9E,MAAA,MAAM,GAAA,CAAI,WAAW,QAAQ,CAAA,CAAE,MAAM,WAAA,EAAa,GAAA,EAAK,QAAQ,CAAA,CAAE,OAAA,EAAQ;AAAA,IAC7E,CAAC,CAAA;AAAA,EACL;AAAA,EAEA,MAAgB,mBAAA,CAAoB,UAAA,EAAwB,QAAA,EAAkB,kBAAA,EAA4B;AACtG,IAAA,MAAM,SAAS,MAAM,IAAA,CAAK,GACrB,UAAA,CAAW,cAAc,EACzB,MAAA,CAAO;AAAA,MACJ,kBAAA;AAAA,MACA,GAAA,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA,iBAAA,CAAA,CAKG,GAAG,eAAe;AAAA,KACxB,EACA,KAAA,CAAM,cAAA,EAAgB,KAAK,QAAQ,CAAA,CACnC,MAAM,kBAAA,EAAoB,GAAA,EAAK,kBAAkB,CAAA,CACjD,KAAA,CAAM,cAAc,GAAA,EAAK,KAAK,EAC9B,OAAA,CAAQ,QAAQ,EAChB,gBAAA,EAAiB;AAEtB,IAAA,IAAI,CAAC,MAAA,EAAQ;AAEb,IAAA,MAAM,iBAAA,GAAoB,UAAA;AAG1B,IAAA,iBAAA,CAAkB,mBAAmB,EAAC;AACtC,IAAA,iBAAA,CAAkB,cAAA,CAAe,KAAK,CAAA,GAAI,MAAM,OAAA,CAAQ,GAAA;AAAA,MACpD,IAAA,CAAK,KAAA,CAAM,MAAA,CAAO,aAAa,CAAA,CAAE,GAAA;AAAA,QAAI,CAAC,EAAE,IAAA,EAAM,KAAA,OAC1C,IAAA,CAAK,KAAA,CAAM,UAAA,CAAW,IAAA,EAAM,KAAK;AAAA;AACrC,KACJ;AAGA,IAAA,iBAAA,CAAkB,gBAAA,CAAiB,KAAK,CAAA,GACpC,MAAA,CAAO,KAAK,UAAA,CAAW,gBAAgB,EAAE,MAAA,GAAS,CAAA,GAC5C,kBAAkB,GAAG,MAAA,CAAO,OAAO,UAAA,CAAW,gBAAgB,CAAC,CAAA,GAC/D,IAAA,CAAK,eAAe,MAAS,CAAA;AAAA,EAC3C;AACJ;;;;"}

package/dist/{createEndpoint-Bxzbc61h.js → createEndpoint-pEWA0kt2.js}

@@ -1,4 +1,4 @@
-import { L as LangGraphGlobal, a as streamState, g as getGraph, G as GRAPHS } from './stream-CAZ9j0yn.js';
+import { L as LangGraphGlobal, a as streamState, g as getGraph, G as GRAPHS } from './stream-Bwuk4b6F.js';
 
 const AssistantEndpoint = {
   async search(query) {
@@ -108,4 +108,4 @@ const createEndpoint = () => {
 };
 
 export { AssistantEndpoint as A, createEndpoint as c };
-//# sourceMappingURL=createEndpoint-Bxzbc61h.js.map
+//# sourceMappingURL=createEndpoint-pEWA0kt2.js.map

package/dist/{createEndpoint-Bxzbc61h.js.map → createEndpoint-pEWA0kt2.js.map}

@@ -1 +1 @@
-{"version":3,"file":"createEndpoint-Bxzbc61h.js","sources":["../src/createEndpoint.ts"],"sourcesContent":["import { StreamEvent } from '@langchain/core/tracers/log_stream';\nimport { streamState } from './graph/stream.js';\nimport { Assistant, Run, StreamMode, Metadata, AssistantGraph } from '@langchain/langgraph-sdk';\nimport { getGraph, GRAPHS } from './utils/getGraph.js';\nimport { LangGraphGlobal } from './global.js';\nimport { AssistantSortBy, CancelAction, ILangGraphClient, RunStatus, SortOrder, StreamInputData } from './types.js';\nexport { registerGraph } from './utils/getGraph.js';\n\nexport const AssistantEndpoint: ILangGraphClient['assistants'] = {\n    async search(query?: {\n        graphId?: string;\n        metadata?: Metadata;\n        limit?: number;\n        offset?: number;\n        sortBy?: AssistantSortBy;\n        sortOrder?: SortOrder;\n    }): Promise<Assistant[]> {\n        if (query?.graphId) {\n            return [\n                {\n                    assistant_id: query.graphId,\n                    graph_id: query.graphId,\n                    config: {},\n                    created_at: new Date().toISOString(),\n                    updated_at: new Date().toISOString(),\n                    metadata: {},\n                    version: 1,\n                    name: query.graphId,\n                    description: '',\n                } as Assistant,\n            ];\n        }\n        return Object.entries(GRAPHS).map(\n            ([graphId, _]) =>\n                ({\n                    assistant_id: graphId,\n                    graph_id: graphId,\n                    config: {},\n                    metadata: {},\n                    version: 1,\n                    name: graphId,\n                    description: '',\n                    created_at: new Date().toISOString(),\n                    updated_at: new Date().toISOString(),\n                } as Assistant),\n        );\n    },\n    async getGraph(assistantId: string, options?: { xray?: boolean | number }): Promise<AssistantGraph> {\n        const config = {};\n        const graph = await getGraph(assistantId, config);\n        const drawable = await graph.getGraphAsync({\n            ...config,\n            xray: options?.xray ?? undefined,\n        });\n        return drawable.toJSON() as AssistantGraph;\n    },\n};\n\nexport const createEndpoint = () => {\n    const getThreads = () => {\n        return LangGraphGlobal.globalThreadsManager;\n    };\n    return {\n        assistants: AssistantEndpoint,\n        get threads() {\n            return LangGraphGlobal.globalThreadsManager;\n        },\n        runs: {\n            list(\n                threadId: string,\n                options?: {\n                    limit?: number;\n                    offset?: number;\n                    status?: RunStatus;\n                },\n            ): Promise<Run[]> {\n                return getThreads().listRuns(threadId, options);\n            },\n            async cancel(threadId: string, runId: string, wait?: boolean, action?: CancelAction): Promise<void> {\n                return LangGraphGlobal.globalMessageQueue.cancelQueue(runId);\n            },\n            async *stream(threadId: string, assistantId: string, payload: StreamInputData) {\n                payload.config = {\n                    ...(payload.config ?? {}),\n                    configurable: {\n                        ...(payload.config?.configurable ?? {}),\n                        graph_id: assistantId,\n                        thread_id: threadId,\n                    },\n                };\n                const threads = getThreads();\n                for await (const data of streamState(\n                    threads,\n                    threads.createRun(threadId, assistantId, payload),\n                    payload,\n                    {\n                        attempt: 0,\n                        getGraph,\n                    },\n                )) {\n                    yield data;\n                }\n            },\n            async *joinStream(\n                threadId: string,\n                runId: string,\n                options?:\n                    | {\n                          signal?: AbortSignal;\n                          cancelOnDisconnect?: boolean;\n                          lastEventId?: string;\n                          streamMode?: StreamMode | StreamMode[];\n                      }\n                    | AbortSignal,\n            ): AsyncGenerator<{ id?: string; event: StreamEvent; data: any }> {\n                // 处理参数兼容性\n                const config = options && typeof options === 'object' && 'signal' in options ? options : {};\n                const signal =\n                    (options instanceof AbortSignal ? options : config.signal) || new AbortController().signal;\n\n                try {\n                    // 获取 Redis 队列实例\n                    const queue = LangGraphGlobal.globalMessageQueue.getQueue(runId);\n\n                    // 监听队列数据并转换格式\n                    for await (const eventMessage of queue.onDataReceive()) {\n                        // 检查是否被取消\n                        if (signal.aborted) {\n                            break;\n                        }\n\n                        // 转换 EventMessage 为期望的格式\n                        const event = eventMessage.event as unknown as StreamEvent;\n                        const data = eventMessage.data;\n\n                        yield {\n                            id: eventMessage.id,\n                            event,\n                            data,\n                        };\n\n                        // 如果是流结束信号，停止监听\n                        if (\n                            eventMessage.event === '__stream_end__' ||\n                            eventMessage.event === '__stream_error__' ||\n                            eventMessage.event === '__stream_cancel__'\n                        ) {\n                            break;\n                        }\n                    }\n                } catch (error) {\n                    // 如果队列不存在或其他错误，记录警告但不抛出错误\n                    console.warn('Join stream failed:', error);\n                }\n            },\n        },\n    };\n};\n"],"names":[],"mappings":";;AAQO,MAAM,iBAAA,GAAoD;AAAA,EAC7D,MAAM,OAAO,KAAA,EAOY;AACrB,IAAA,IAAI,OAAO,OAAA,EAAS;AAChB,MAAA,OAAO;AAAA,QACH;AAAA,UACI,cAAc,KAAA,CAAM,OAAA;AAAA,UACpB,UAAU,KAAA,CAAM,OAAA;AAAA,UAChB,QAAQ,EAAC;AAAA,UACT,UAAA,EAAA,iBAAY,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,UACnC,UAAA,EAAA,iBAAY,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,UACnC,UAAU,EAAC;AAAA,UACX,OAAA,EAAS,CAAA;AAAA,UACT,MAAM,KAAA,CAAM,OAAA;AAAA,UACZ,WAAA,EAAa;AAAA;AACjB,OACJ;AAAA,IACJ;AACA,IAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,MAAM,CAAA,CAAE,GAAA;AAAA,MAC1B,CAAC,CAAC,OAAA,EAAS,CAAC,CAAA,MACP;AAAA,QACG,YAAA,EAAc,OAAA;AAAA,QACd,QAAA,EAAU,OAAA;AAAA,QACV,QAAQ,EAAC;AAAA,QACT,UAAU,EAAC;AAAA,QACX,OAAA,EAAS,CAAA;AAAA,QACT,IAAA,EAAM,OAAA;AAAA,QACN,WAAA,EAAa,EAAA;AAAA,QACb,UAAA,EAAA,iBAAY,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,QACnC,UAAA,EAAA,iBAAY,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,OACvC;AAAA,KACR;AAAA,EACJ,CAAA;AAAA,EACA,MAAM,QAAA,CAAS,WAAA,EAAqB,OAAA,EAAgE;AAChG,IAAA,MAAM,SAAS,EAAC;AAChB,IAAA,MAAM,KAAA,GAAQ,MAAM,QAAA,CAAS,WAAA,EAAa,MAAM,CAAA;AAChD,IAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,aAAA,CAAc;AAAA,MACvC,GAAG,MAAA;AAAA,MACH,IAAA,EAAM,SAAS,IAAA,IAAQ;AAAA,KAC1B,CAAA;AACD,IAAA,OAAO,SAAS,MAAA,EAAO;AAAA,EAC3B;AACJ;AAEO,MAAM,iBAAiB,MAAM;AAChC,EAAA,MAAM,aAAa,MAAM;AACrB,IAAA,OAAO,eAAA,CAAgB,oBAAA;AAAA,EAC3B,CAAA;AACA,EAAA,OAAO;AAAA,IACH,UAAA,EAAY,iBAAA;AAAA,IACZ,IAAI,OAAA,GAAU;AACV,MAAA,OAAO,eAAA,CAAgB,oBAAA;AAAA,IAC3B,CAAA;AAAA,IACA,IAAA,EAAM;AAAA,MACF,IAAA,CACI,UACA,OAAA,EAKc;AACd,QAAA,OAAO,UAAA,EAAW,CAAE,QAAA,CAAS,QAAA,EAAU,OAAO,CAAA;AAAA,MAClD,CAAA;AAAA,MACA,MAAM,MAAA,CAAO,QAAA,EAAkB,KAAA,EAAe,MAAgB,MAAA,EAAsC;AAChG,QAAA,OAAO,eAAA,CAAgB,kBAAA,CAAmB,WAAA,CAAY,KAAK,CAAA;AAAA,MAC/D,CAAA;AAAA,MACA,OAAO,MAAA,CAAO,QAAA,EAAkB,WAAA,EAAqB,OAAA,EAA0B;AAC3E,QAAA,OAAA,CAAQ,MAAA,GAAS;AAAA,UACb,GAAI,OAAA,CAAQ,MAAA,IAAU,EAAC;AAAA,UACvB,YAAA,EAAc;AAAA,YACV,GAAI,OAAA,CAAQ,MAAA,EAAQ,YAAA,IAAgB,EAAC;AAAA,YACrC,QAAA,EAAU,WAAA;AAAA,YACV,SAAA,EAAW;AAAA;AACf,SACJ;AACA,QAAA,MAAM,UAAU,UAAA,EAAW;AAC3B,QAAA,WAAA,MAAiB,IAAA,IAAQ,WAAA;AAAA,UACrB,OAAA;AAAA,UACA,OAAA,CAAQ,SAAA,CAAU,QAAA,EAAU,WAAA,EAAa,OAAO,CAAA;AAAA,UAChD,OAAA;AAAA,UACA;AAAA,YACI,OAAA,EAAS,CAAA;AAAA,YACT;AAAA;AACJ,SACJ,EAAG;AACC,UAAA,MAAM,IAAA;AAAA,QACV;AAAA,MACJ,CAAA;AAAA,MACA,OAAO,UAAA,CACH,QAAA,EACA,KAAA,EACA,OAAA,EAQ8D;AAE9D,QAAA,MAAM,MAAA,GAAS,WAAW,OAAO,OAAA,KAAY,YAAY,QAAA,IAAY,OAAA,GAAU,UAAU,EAAC;AAC1F,QAAA,MAAM,MAAA,GAAA,CACD,mBAAmB,WAAA,GAAc,OAAA,GAAU,OAAO,MAAA,KAAW,IAAI,iBAAgB,CAAE,MAAA;AAExF,QAAA,IAAI;AAEA,UAAA,MAAM,KAAA,GAAQ,eAAA,CAAgB,kBAAA,CAAmB,QAAA,CAAS,KAAK,CAAA;AAG/D,UAAA,WAAA,MAAiB,YAAA,IAAgB,KAAA,CAAM,aAAA,EAAc,EAAG;AAEpD,YAAA,IAAI,OAAO,OAAA,EAAS;AAChB,cAAA;AAAA,YACJ;AAGA,YAAA,MAAM,QAAQ,YAAA,CAAa,KAAA;AAC3B,YAAA,MAAM,OAAO,YAAA,CAAa,IAAA;AAE1B,YAAA,MAAM;AAAA,cACF,IAAI,YAAA,CAAa,EAAA;AAAA,cACjB,KAAA;AAAA,cACA;AAAA,aACJ;AAGA,YAAA,IACI,YAAA,CAAa,UAAU,gBAAA,IACvB,YAAA,CAAa,UAAU,kBAAA,IACvB,YAAA,CAAa,UAAU,mBAAA,EACzB;AACE,cAAA;AAAA,YACJ;AAAA,UACJ;AAAA,QACJ,SAAS,KAAA,EAAO;AAEZ,UAAA,OAAA,CAAQ,IAAA,CAAK,uBAAuB,KAAK,CAAA;AAAA,QAC7C;AAAA,MACJ;AAAA;AACJ,GACJ;AACJ;;;;"}
+{"version":3,"file":"createEndpoint-pEWA0kt2.js","sources":["../src/createEndpoint.ts"],"sourcesContent":["import { StreamEvent } from '@langchain/core/tracers/log_stream';\nimport { streamState } from './graph/stream.js';\nimport { Assistant, Run, StreamMode, Metadata, AssistantGraph } from '@langchain/langgraph-sdk';\nimport { getGraph, GRAPHS } from './utils/getGraph.js';\nimport { LangGraphGlobal } from './global.js';\nimport { AssistantSortBy, CancelAction, ILangGraphClient, RunStatus, SortOrder, StreamInputData } from './types.js';\nexport { registerGraph } from './utils/getGraph.js';\n\nexport const AssistantEndpoint: ILangGraphClient['assistants'] = {\n    async search(query?: {\n        graphId?: string;\n        metadata?: Metadata;\n        limit?: number;\n        offset?: number;\n        sortBy?: AssistantSortBy;\n        sortOrder?: SortOrder;\n    }): Promise<Assistant[]> {\n        if (query?.graphId) {\n            return [\n                {\n                    assistant_id: query.graphId,\n                    graph_id: query.graphId,\n                    config: {},\n                    created_at: new Date().toISOString(),\n                    updated_at: new Date().toISOString(),\n                    metadata: {},\n                    version: 1,\n                    name: query.graphId,\n                    description: '',\n                } as Assistant,\n            ];\n        }\n        return Object.entries(GRAPHS).map(\n            ([graphId, _]) =>\n                ({\n                    assistant_id: graphId,\n                    graph_id: graphId,\n                    config: {},\n                    metadata: {},\n                    version: 1,\n                    name: graphId,\n                    description: '',\n                    created_at: new Date().toISOString(),\n                    updated_at: new Date().toISOString(),\n                } as Assistant),\n        );\n    },\n    async getGraph(assistantId: string, options?: { xray?: boolean | number }): Promise<AssistantGraph> {\n        const config = {};\n        const graph = await getGraph(assistantId, config);\n        const drawable = await graph.getGraphAsync({\n            ...config,\n            xray: options?.xray ?? undefined,\n        });\n        return drawable.toJSON() as AssistantGraph;\n    },\n};\n\nexport const createEndpoint = () => {\n    const getThreads = () => {\n        return LangGraphGlobal.globalThreadsManager;\n    };\n    return {\n        assistants: AssistantEndpoint,\n        get threads() {\n            return LangGraphGlobal.globalThreadsManager;\n        },\n        runs: {\n            list(\n                threadId: string,\n                options?: {\n                    limit?: number;\n                    offset?: number;\n                    status?: RunStatus;\n                },\n            ): Promise<Run[]> {\n                return getThreads().listRuns(threadId, options);\n            },\n            async cancel(threadId: string, runId: string, wait?: boolean, action?: CancelAction): Promise<void> {\n                return LangGraphGlobal.globalMessageQueue.cancelQueue(runId);\n            },\n            async *stream(threadId: string, assistantId: string, payload: StreamInputData) {\n                payload.config = {\n                    ...(payload.config ?? {}),\n                    configurable: {\n                        ...(payload.config?.configurable ?? {}),\n                        graph_id: assistantId,\n                        thread_id: threadId,\n                    },\n                };\n                const threads = getThreads();\n                for await (const data of streamState(\n                    threads,\n                    threads.createRun(threadId, assistantId, payload),\n                    payload,\n                    {\n                        attempt: 0,\n                        getGraph,\n                    },\n                )) {\n                    yield data;\n                }\n            },\n            async *joinStream(\n                threadId: string,\n                runId: string,\n                options?:\n                    | {\n                          signal?: AbortSignal;\n                          cancelOnDisconnect?: boolean;\n                          lastEventId?: string;\n                          streamMode?: StreamMode | StreamMode[];\n                      }\n                    | AbortSignal,\n            ): AsyncGenerator<{ id?: string; event: StreamEvent; data: any }> {\n                // 处理参数兼容性\n                const config = options && typeof options === 'object' && 'signal' in options ? options : {};\n                const signal =\n                    (options instanceof AbortSignal ? options : config.signal) || new AbortController().signal;\n\n                try {\n                    // 获取 Redis 队列实例\n                    const queue = LangGraphGlobal.globalMessageQueue.getQueue(runId);\n\n                    // 监听队列数据并转换格式\n                    for await (const eventMessage of queue.onDataReceive()) {\n                        // 检查是否被取消\n                        if (signal.aborted) {\n                            break;\n                        }\n\n                        // 转换 EventMessage 为期望的格式\n                        const event = eventMessage.event as unknown as StreamEvent;\n                        const data = eventMessage.data;\n\n                        yield {\n                            id: eventMessage.id,\n                            event,\n                            data,\n                        };\n\n                        // 如果是流结束信号，停止监听\n                        if (\n                            eventMessage.event === '__stream_end__' ||\n                            eventMessage.event === '__stream_error__' ||\n                            eventMessage.event === '__stream_cancel__'\n                        ) {\n                            break;\n                        }\n                    }\n                } catch (error) {\n                    // 如果队列不存在或其他错误，记录警告但不抛出错误\n                    console.warn('Join stream failed:', error);\n                }\n            },\n        },\n    };\n};\n"],"names":[],"mappings":";;AAQO,MAAM,iBAAA,GAAoD;AAAA,EAC7D,MAAM,OAAO,KAAA,EAOY;AACrB,IAAA,IAAI,OAAO,OAAA,EAAS;AAChB,MAAA,OAAO;AAAA,QACH;AAAA,UACI,cAAc,KAAA,CAAM,OAAA;AAAA,UACpB,UAAU,KAAA,CAAM,OAAA;AAAA,UAChB,QAAQ,EAAC;AAAA,UACT,UAAA,EAAA,iBAAY,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,UACnC,UAAA,EAAA,iBAAY,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,UACnC,UAAU,EAAC;AAAA,UACX,OAAA,EAAS,CAAA;AAAA,UACT,MAAM,KAAA,CAAM,OAAA;AAAA,UACZ,WAAA,EAAa;AAAA;AACjB,OACJ;AAAA,IACJ;AACA,IAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,MAAM,CAAA,CAAE,GAAA;AAAA,MAC1B,CAAC,CAAC,OAAA,EAAS,CAAC,CAAA,MACP;AAAA,QACG,YAAA,EAAc,OAAA;AAAA,QACd,QAAA,EAAU,OAAA;AAAA,QACV,QAAQ,EAAC;AAAA,QACT,UAAU,EAAC;AAAA,QACX,OAAA,EAAS,CAAA;AAAA,QACT,IAAA,EAAM,OAAA;AAAA,QACN,WAAA,EAAa,EAAA;AAAA,QACb,UAAA,EAAA,iBAAY,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,QACnC,UAAA,EAAA,iBAAY,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,OACvC;AAAA,KACR;AAAA,EACJ,CAAA;AAAA,EACA,MAAM,QAAA,CAAS,WAAA,EAAqB,OAAA,EAAgE;AAChG,IAAA,MAAM,SAAS,EAAC;AAChB,IAAA,MAAM,KAAA,GAAQ,MAAM,QAAA,CAAS,WAAA,EAAa,MAAM,CAAA;AAChD,IAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,aAAA,CAAc;AAAA,MACvC,GAAG,MAAA;AAAA,MACH,IAAA,EAAM,SAAS,IAAA,IAAQ;AAAA,KAC1B,CAAA;AACD,IAAA,OAAO,SAAS,MAAA,EAAO;AAAA,EAC3B;AACJ;AAEO,MAAM,iBAAiB,MAAM;AAChC,EAAA,MAAM,aAAa,MAAM;AACrB,IAAA,OAAO,eAAA,CAAgB,oBAAA;AAAA,EAC3B,CAAA;AACA,EAAA,OAAO;AAAA,IACH,UAAA,EAAY,iBAAA;AAAA,IACZ,IAAI,OAAA,GAAU;AACV,MAAA,OAAO,eAAA,CAAgB,oBAAA;AAAA,IAC3B,CAAA;AAAA,IACA,IAAA,EAAM;AAAA,MACF,IAAA,CACI,UACA,OAAA,EAKc;AACd,QAAA,OAAO,UAAA,EAAW,CAAE,QAAA,CAAS,QAAA,EAAU,OAAO,CAAA;AAAA,MAClD,CAAA;AAAA,MACA,MAAM,MAAA,CAAO,QAAA,EAAkB,KAAA,EAAe,MAAgB,MAAA,EAAsC;AAChG,QAAA,OAAO,eAAA,CAAgB,kBAAA,CAAmB,WAAA,CAAY,KAAK,CAAA;AAAA,MAC/D,CAAA;AAAA,MACA,OAAO,MAAA,CAAO,QAAA,EAAkB,WAAA,EAAqB,OAAA,EAA0B;AAC3E,QAAA,OAAA,CAAQ,MAAA,GAAS;AAAA,UACb,GAAI,OAAA,CAAQ,MAAA,IAAU,EAAC;AAAA,UACvB,YAAA,EAAc;AAAA,YACV,GAAI,OAAA,CAAQ,MAAA,EAAQ,YAAA,IAAgB,EAAC;AAAA,YACrC,QAAA,EAAU,WAAA;AAAA,YACV,SAAA,EAAW;AAAA;AACf,SACJ;AACA,QAAA,MAAM,UAAU,UAAA,EAAW;AAC3B,QAAA,WAAA,MAAiB,IAAA,IAAQ,WAAA;AAAA,UACrB,OAAA;AAAA,UACA,OAAA,CAAQ,SAAA,CAAU,QAAA,EAAU,WAAA,EAAa,OAAO,CAAA;AAAA,UAChD,OAAA;AAAA,UACA;AAAA,YACI,OAAA,EAAS,CAAA;AAAA,YACT;AAAA;AACJ,SACJ,EAAG;AACC,UAAA,MAAM,IAAA;AAAA,QACV;AAAA,MACJ,CAAA;AAAA,MACA,OAAO,UAAA,CACH,QAAA,EACA,KAAA,EACA,OAAA,EAQ8D;AAE9D,QAAA,MAAM,MAAA,GAAS,WAAW,OAAO,OAAA,KAAY,YAAY,QAAA,IAAY,OAAA,GAAU,UAAU,EAAC;AAC1F,QAAA,MAAM,MAAA,GAAA,CACD,mBAAmB,WAAA,GAAc,OAAA,GAAU,OAAO,MAAA,KAAW,IAAI,iBAAgB,CAAE,MAAA;AAExF,QAAA,IAAI;AAEA,UAAA,MAAM,KAAA,GAAQ,eAAA,CAAgB,kBAAA,CAAmB,QAAA,CAAS,KAAK,CAAA;AAG/D,UAAA,WAAA,MAAiB,YAAA,IAAgB,KAAA,CAAM,aAAA,EAAc,EAAG;AAEpD,YAAA,IAAI,OAAO,OAAA,EAAS;AAChB,cAAA;AAAA,YACJ;AAGA,YAAA,MAAM,QAAQ,YAAA,CAAa,KAAA;AAC3B,YAAA,MAAM,OAAO,YAAA,CAAa,IAAA;AAE1B,YAAA,MAAM;AAAA,cACF,IAAI,YAAA,CAAa,EAAA;AAAA,cACjB,KAAA;AAAA,cACA;AAAA,aACJ;AAGA,YAAA,IACI,YAAA,CAAa,UAAU,gBAAA,IACvB,YAAA,CAAa,UAAU,kBAAA,IACvB,YAAA,CAAa,UAAU,mBAAA,EACzB;AACE,cAAA;AAAA,YACJ;AAAA,UACJ;AAAA,QACJ,SAAS,KAAA,EAAO;AAEZ,UAAA,OAAA,CAAQ,IAAA,CAAK,uBAAuB,KAAK,CAAA;AAAA,QAC7C;AAAA,MACJ;AAAA;AACJ,GACJ;AACJ;;;;"}

package/dist/index.js

@@ -1,5 +1,5 @@
-export { A as AssistantEndpoint, c as createEndpoint } from './createEndpoint-Bxzbc61h.js';
-export { L as LangGraphGlobal, r as registerGraph } from './stream-CAZ9j0yn.js';
+export { A as AssistantEndpoint, c as createEndpoint } from './createEndpoint-pEWA0kt2.js';
+export { L as LangGraphGlobal, r as registerGraph } from './stream-Bwuk4b6F.js';
 import { StateGraph } from '@langchain/langgraph';
 
 const createEntrypointGraph = ({

package/dist/{queue-D3gPPSDe.js → queue-5gRqr4RI.js}

@@ -1,4 +1,4 @@
-import { B as BaseStreamQueue, C as CancelEventMessage } from './stream-CAZ9j0yn.js';
+import { B as BaseStreamQueue, C as CancelEventMessage } from './stream-Bwuk4b6F.js';
 import { createClient } from 'redis';
 
 class RedisStreamQueue extends BaseStreamQueue {
@@ -121,4 +121,4 @@ class RedisStreamQueue extends BaseStreamQueue {
 }
 
 export { RedisStreamQueue };
-//# sourceMappingURL=queue-D3gPPSDe.js.map
+//# sourceMappingURL=queue-5gRqr4RI.js.map

package/dist/{queue-D3gPPSDe.js.map → queue-5gRqr4RI.js.map}

@@ -1 +1 @@
-{"version":3,"file":"queue-D3gPPSDe.js","sources":["../src/storage/redis/queue.ts"],"sourcesContent":["import { CancelEventMessage, EventMessage } from '../../queue/event_message.js';\nimport { BaseStreamQueue } from '../../queue/stream_queue.js';\nimport { BaseStreamQueueInterface } from '../../queue/stream_queue.js';\nimport { createClient, RedisClientType } from 'redis';\n\n/**\n * Redis 实现的消息队列，用于存储消息\n */\nexport class RedisStreamQueue extends BaseStreamQueue implements BaseStreamQueueInterface {\n    static redis: RedisClientType = createClient({ url: process.env.REDIS_URL! });\n    static subscriberRedis: RedisClientType = createClient({ url: process.env.REDIS_URL! });\n    private redis: RedisClientType;\n    private subscriberRedis: RedisClientType;\n    private queueKey: string;\n    private channelKey: string;\n    private isConnected = false;\n    public cancelSignal: AbortController;\n\n    constructor(readonly id: string = 'default') {\n        super(id, true);\n        this.queueKey = `queue:${this.id}`;\n        this.channelKey = `channel:${this.id}`;\n        this.redis = RedisStreamQueue.redis;\n        this.subscriberRedis = RedisStreamQueue.subscriberRedis;\n        this.cancelSignal = new AbortController();\n\n        // 连接 Redis 客户端（检查是否已经连接）\n        if (!this.redis.isOpen) {\n            this.redis.connect();\n        }\n        if (!this.subscriberRedis.isOpen) {\n            this.subscriberRedis.connect();\n        }\n        this.isConnected = true;\n    }\n\n    /**\n     * 推送消息到 Redis 队列\n     */\n    async push(item: EventMessage): Promise<void> {\n        const data = await this.encodeData(item);\n        const serializedData = Buffer.from(data);\n\n        // 推送到队列\n        await this.redis.lPush(this.queueKey, serializedData);\n\n        // 设置队列 TTL 为 300 秒\n        await this.redis.expire(this.queueKey, 300);\n\n        // 发布到频道通知有新数据\n        await this.redis.publish(this.channelKey, serializedData);\n\n        this.emit('dataChange', data);\n    }\n\n    /**\n     * 异步生成器：支持 for await...of 方式消费队列数据\n     */\n    async *onDataReceive(): AsyncGenerator<EventMessage, void, unknown> {\n        let queue: EventMessage[] = [];\n        let pendingResolve: (() => void) | null = null;\n        let isStreamEnded = false;\n        const handleMessage = async (message: string) => {\n            const data = (await this.decodeData(message)) as EventMessage;\n            queue.push(data);\n            // 检查是否为流结束或错误信号\n            if (\n                data.event === '__stream_end__' ||\n                data.event === '__stream_error__' ||\n                data.event === '__stream_cancel__'\n            ) {\n                setTimeout(() => {\n                    isStreamEnded = true;\n                    if (pendingResolve) {\n                        pendingResolve();\n                        pendingResolve = null;\n                    }\n                }, 300);\n\n                if (data.event === '__stream_cancel__') {\n                    this.cancel();\n                }\n            }\n\n            if (pendingResolve) {\n                pendingResolve();\n                pendingResolve = null;\n            }\n        };\n\n        // 订阅 Redis 频道\n        await this.subscriberRedis.subscribe(this.channelKey, (message) => {\n            handleMessage(message);\n        });\n\n        try {\n            while (!isStreamEnded) {\n                if (queue.length > 0) {\n                    for (const item of queue) {\n                        yield item;\n                    }\n                    queue = [];\n                } else {\n                    await new Promise((resolve) => {\n                        pendingResolve = resolve as () => void;\n                    });\n                }\n            }\n        } finally {\n            await this.subscriberRedis.unsubscribe(this.channelKey);\n        }\n    }\n\n    /**\n     * 获取队列中的所有数据\n     */\n    async getAll(): Promise<EventMessage[]> {\n        const data = await this.redis.lRange(this.queueKey, 0, -1);\n\n        if (!data || data.length === 0) {\n            return [];\n        }\n\n        if (this.compressMessages) {\n            return (await Promise.all(\n                data.map(async (item: string) => {\n                    const parsed = JSON.parse(item) as EventMessage;\n                    return (await this.decodeData(parsed as any)) as EventMessage;\n                }),\n            )) as EventMessage[];\n        } else {\n            return data.map((item: string) => JSON.parse(item) as EventMessage);\n        }\n    }\n\n    /**\n     * 清空队列\n     */\n    clear(): void {\n        if (this.isConnected) {\n            this.redis.del(this.queueKey);\n        }\n    }\n\n    /**\n     * 取消操作\n     */\n    cancel(): void {\n        this.push(new CancelEventMessage());\n        this.cancelSignal.abort('user cancel this run');\n    }\n}\n"],"names":[],"mappings":";;;AAQO,MAAM,yBAAyB,eAAA,CAAoD;AAAA,EAUtF,WAAA,CAAqB,KAAa,SAAA,EAAW;AACzC,IAAA,KAAA,CAAM,IAAI,IAAI,CAAA;AADG,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA;AAEjB,IAAA,IAAA,CAAK,QAAA,GAAW,CAAA,MAAA,EAAS,IAAA,CAAK,EAAE,CAAA,CAAA;AAChC,IAAA,IAAA,CAAK,UAAA,GAAa,CAAA,QAAA,EAAW,IAAA,CAAK,EAAE,CAAA,CAAA;AACpC,IAAA,IAAA,CAAK,QAAQ,gBAAA,CAAiB,KAAA;AAC9B,IAAA,IAAA,CAAK,kBAAkB,gBAAA,CAAiB,eAAA;AACxC,IAAA,IAAA,CAAK,YAAA,GAAe,IAAI,eAAA,EAAgB;AAGxC,IAAA,IAAI,CAAC,IAAA,CAAK,KAAA,CAAM,MAAA,EAAQ;AACpB,MAAA,IAAA,CAAK,MAAM,OAAA,EAAQ;AAAA,IACvB;AACA,IAAA,IAAI,CAAC,IAAA,CAAK,eAAA,CAAgB,MAAA,EAAQ;AAC9B,MAAA,IAAA,CAAK,gBAAgB,OAAA,EAAQ;AAAA,IACjC;AACA,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAAA,EACvB;AAAA,EAzBA,OAAO,QAAyB,YAAA,CAAa,EAAE,KAAK,OAAA,CAAQ,GAAA,CAAI,WAAY,CAAA;AAAA,EAC5E,OAAO,kBAAmC,YAAA,CAAa,EAAE,KAAK,OAAA,CAAQ,GAAA,CAAI,WAAY,CAAA;AAAA,EAC9E,KAAA;AAAA,EACA,eAAA;AAAA,EACA,QAAA;AAAA,EACA,UAAA;AAAA,EACA,WAAA,GAAc,KAAA;AAAA,EACf,YAAA;AAAA;AAAA;AAAA;AAAA,EAuBP,MAAM,KAAK,IAAA,EAAmC;AAC1C,IAAA,MAAM,IAAA,GAAO,MAAM,IAAA,CAAK,UAAA,CAAW,IAAI,CAAA;AACvC,IAAA,MAAM,cAAA,GAAiB,MAAA,CAAO,IAAA,CAAK,IAAI,CAAA;AAGvC,IAAA,MAAM,IAAA,CAAK,KAAA,CAAM,KAAA,CAAM,IAAA,CAAK,UAAU,cAAc,CAAA;AAGpD,IAAA,MAAM,IAAA,CAAK,KAAA,CAAM,MAAA,CAAO,IAAA,CAAK,UAAU,GAAG,CAAA;AAG1C,IAAA,MAAM,IAAA,CAAK,KAAA,CAAM,OAAA,CAAQ,IAAA,CAAK,YAAY,cAAc,CAAA;AAExD,IAAA,IAAA,CAAK,IAAA,CAAK,cAAc,IAAI,CAAA;AAAA,EAChC;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,aAAA,GAA6D;AAChE,IAAA,IAAI,QAAwB,EAAC;AAC7B,IAAA,IAAI,cAAA,GAAsC,IAAA;AAC1C,IAAA,IAAI,aAAA,GAAgB,KAAA;AACpB,IAAA,MAAM,aAAA,GAAgB,OAAO,OAAA,KAAoB;AAC7C,MAAA,MAAM,IAAA,GAAQ,MAAM,IAAA,CAAK,UAAA,CAAW,OAAO,CAAA;AAC3C,MAAA,KAAA,CAAM,KAAK,IAAI,CAAA;AAEf,MAAA,IACI,IAAA,CAAK,UAAU,gBAAA,IACf,IAAA,CAAK,UAAU,kBAAA,IACf,IAAA,CAAK,UAAU,mBAAA,EACjB;AACE,QAAA,UAAA,CAAW,MAAM;AACb,UAAA,aAAA,GAAgB,IAAA;AAChB,UAAA,IAAI,cAAA,EAAgB;AAChB,YAAA,cAAA,EAAe;AACf,YAAA,cAAA,GAAiB,IAAA;AAAA,UACrB;AAAA,QACJ,GAAG,GAAG,CAAA;AAEN,QAAA,IAAI,IAAA,CAAK,UAAU,mBAAA,EAAqB;AACpC,UAAA,IAAA,CAAK,MAAA,EAAO;AAAA,QAChB;AAAA,MACJ;AAEA,MAAA,IAAI,cAAA,EAAgB;AAChB,QAAA,cAAA,EAAe;AACf,QAAA,cAAA,GAAiB,IAAA;AAAA,MACrB;AAAA,IACJ,CAAA;AAGA,IAAA,MAAM,KAAK,eAAA,CAAgB,SAAA,CAAU,IAAA,CAAK,UAAA,EAAY,CAAC,OAAA,KAAY;AAC/D,MAAA,aAAA,CAAc,OAAO,CAAA;AAAA,IACzB,CAAC,CAAA;AAED,IAAA,IAAI;AACA,MAAA,OAAO,CAAC,aAAA,EAAe;AACnB,QAAA,IAAI,KAAA,CAAM,SAAS,CAAA,EAAG;AAClB,UAAA,KAAA,MAAW,QAAQ,KAAA,EAAO;AACtB,YAAA,MAAM,IAAA;AAAA,UACV;AACA,UAAA,KAAA,GAAQ,EAAC;AAAA,QACb,CAAA,MAAO;AACH,UAAA,MAAM,IAAI,OAAA,CAAQ,CAAC,OAAA,KAAY;AAC3B,YAAA,cAAA,GAAiB,OAAA;AAAA,UACrB,CAAC,CAAA;AAAA,QACL;AAAA,MACJ;AAAA,IACJ,CAAA,SAAE;AACE,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,WAAA,CAAY,IAAA,CAAK,UAAU,CAAA;AAAA,IAC1D;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,MAAA,GAAkC;AACpC,IAAA,MAAM,IAAA,GAAO,MAAM,IAAA,CAAK,KAAA,CAAM,OAAO,IAAA,CAAK,QAAA,EAAU,GAAG,EAAE,CAAA;AAEzD,IAAA,IAAI,CAAC,IAAA,IAAQ,IAAA,CAAK,MAAA,KAAW,CAAA,EAAG;AAC5B,MAAA,OAAO,EAAC;AAAA,IACZ;AAEA,IAAA,IAAI,KAAK,gBAAA,EAAkB;AACvB,MAAA,OAAQ,MAAM,OAAA,CAAQ,GAAA;AAAA,QAClB,IAAA,CAAK,GAAA,CAAI,OAAO,IAAA,KAAiB;AAC7B,UAAA,MAAM,MAAA,GAAS,IAAA,CAAK,KAAA,CAAM,IAAI,CAAA;AAC9B,UAAA,OAAQ,MAAM,IAAA,CAAK,UAAA,CAAW,MAAa,CAAA;AAAA,QAC/C,CAAC;AAAA,OACL;AAAA,IACJ,CAAA,MAAO;AACH,MAAA,OAAO,KAAK,GAAA,CAAI,CAAC,SAAiB,IAAA,CAAK,KAAA,CAAM,IAAI,CAAiB,CAAA;AAAA,IACtE;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,KAAA,GAAc;AACV,IAAA,IAAI,KAAK,WAAA,EAAa;AAClB,MAAA,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,QAAQ,CAAA;AAAA,IAChC;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,MAAA,GAAe;AACX,IAAA,IAAA,CAAK,IAAA,CAAK,IAAI,kBAAA,EAAoB,CAAA;AAClC,IAAA,IAAA,CAAK,YAAA,CAAa,MAAM,sBAAsB,CAAA;AAAA,EAClD;AACJ;;;;"}
+{"version":3,"file":"queue-5gRqr4RI.js","sources":["../src/storage/redis/queue.ts"],"sourcesContent":["import { CancelEventMessage, EventMessage } from '../../queue/event_message.js';\nimport { BaseStreamQueue } from '../../queue/stream_queue.js';\nimport { BaseStreamQueueInterface } from '../../queue/stream_queue.js';\nimport { createClient, RedisClientType } from 'redis';\n\n/**\n * Redis 实现的消息队列，用于存储消息\n */\nexport class RedisStreamQueue extends BaseStreamQueue implements BaseStreamQueueInterface {\n    static redis: RedisClientType = createClient({ url: process.env.REDIS_URL! });\n    static subscriberRedis: RedisClientType = createClient({ url: process.env.REDIS_URL! });\n    private redis: RedisClientType;\n    private subscriberRedis: RedisClientType;\n    private queueKey: string;\n    private channelKey: string;\n    private isConnected = false;\n    public cancelSignal: AbortController;\n\n    constructor(readonly id: string = 'default') {\n        super(id, true);\n        this.queueKey = `queue:${this.id}`;\n        this.channelKey = `channel:${this.id}`;\n        this.redis = RedisStreamQueue.redis;\n        this.subscriberRedis = RedisStreamQueue.subscriberRedis;\n        this.cancelSignal = new AbortController();\n\n        // 连接 Redis 客户端（检查是否已经连接）\n        if (!this.redis.isOpen) {\n            this.redis.connect();\n        }\n        if (!this.subscriberRedis.isOpen) {\n            this.subscriberRedis.connect();\n        }\n        this.isConnected = true;\n    }\n\n    /**\n     * 推送消息到 Redis 队列\n     */\n    async push(item: EventMessage): Promise<void> {\n        const data = await this.encodeData(item);\n        const serializedData = Buffer.from(data);\n\n        // 推送到队列\n        await this.redis.lPush(this.queueKey, serializedData);\n\n        // 设置队列 TTL 为 300 秒\n        await this.redis.expire(this.queueKey, 300);\n\n        // 发布到频道通知有新数据\n        await this.redis.publish(this.channelKey, serializedData);\n\n        this.emit('dataChange', data);\n    }\n\n    /**\n     * 异步生成器：支持 for await...of 方式消费队列数据\n     */\n    async *onDataReceive(): AsyncGenerator<EventMessage, void, unknown> {\n        let queue: EventMessage[] = [];\n        let pendingResolve: (() => void) | null = null;\n        let isStreamEnded = false;\n        const handleMessage = async (message: string) => {\n            const data = (await this.decodeData(message)) as EventMessage;\n            queue.push(data);\n            // 检查是否为流结束或错误信号\n            if (\n                data.event === '__stream_end__' ||\n                data.event === '__stream_error__' ||\n                data.event === '__stream_cancel__'\n            ) {\n                setTimeout(() => {\n                    isStreamEnded = true;\n                    if (pendingResolve) {\n                        pendingResolve();\n                        pendingResolve = null;\n                    }\n                }, 300);\n\n                if (data.event === '__stream_cancel__') {\n                    this.cancel();\n                }\n            }\n\n            if (pendingResolve) {\n                pendingResolve();\n                pendingResolve = null;\n            }\n        };\n\n        // 订阅 Redis 频道\n        await this.subscriberRedis.subscribe(this.channelKey, (message) => {\n            handleMessage(message);\n        });\n\n        try {\n            while (!isStreamEnded) {\n                if (queue.length > 0) {\n                    for (const item of queue) {\n                        yield item;\n                    }\n                    queue = [];\n                } else {\n                    await new Promise((resolve) => {\n                        pendingResolve = resolve as () => void;\n                    });\n                }\n            }\n        } finally {\n            await this.subscriberRedis.unsubscribe(this.channelKey);\n        }\n    }\n\n    /**\n     * 获取队列中的所有数据\n     */\n    async getAll(): Promise<EventMessage[]> {\n        const data = await this.redis.lRange(this.queueKey, 0, -1);\n\n        if (!data || data.length === 0) {\n            return [];\n        }\n\n        if (this.compressMessages) {\n            return (await Promise.all(\n                data.map(async (item: string) => {\n                    const parsed = JSON.parse(item) as EventMessage;\n                    return (await this.decodeData(parsed as any)) as EventMessage;\n                }),\n            )) as EventMessage[];\n        } else {\n            return data.map((item: string) => JSON.parse(item) as EventMessage);\n        }\n    }\n\n    /**\n     * 清空队列\n     */\n    clear(): void {\n        if (this.isConnected) {\n            this.redis.del(this.queueKey);\n        }\n    }\n\n    /**\n     * 取消操作\n     */\n    cancel(): void {\n        this.push(new CancelEventMessage());\n        this.cancelSignal.abort('user cancel this run');\n    }\n}\n"],"names":[],"mappings":";;;AAQO,MAAM,yBAAyB,eAAA,CAAoD;AAAA,EAUtF,WAAA,CAAqB,KAAa,SAAA,EAAW;AACzC,IAAA,KAAA,CAAM,IAAI,IAAI,CAAA;AADG,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA;AAEjB,IAAA,IAAA,CAAK,QAAA,GAAW,CAAA,MAAA,EAAS,IAAA,CAAK,EAAE,CAAA,CAAA;AAChC,IAAA,IAAA,CAAK,UAAA,GAAa,CAAA,QAAA,EAAW,IAAA,CAAK,EAAE,CAAA,CAAA;AACpC,IAAA,IAAA,CAAK,QAAQ,gBAAA,CAAiB,KAAA;AAC9B,IAAA,IAAA,CAAK,kBAAkB,gBAAA,CAAiB,eAAA;AACxC,IAAA,IAAA,CAAK,YAAA,GAAe,IAAI,eAAA,EAAgB;AAGxC,IAAA,IAAI,CAAC,IAAA,CAAK,KAAA,CAAM,MAAA,EAAQ;AACpB,MAAA,IAAA,CAAK,MAAM,OAAA,EAAQ;AAAA,IACvB;AACA,IAAA,IAAI,CAAC,IAAA,CAAK,eAAA,CAAgB,MAAA,EAAQ;AAC9B,MAAA,IAAA,CAAK,gBAAgB,OAAA,EAAQ;AAAA,IACjC;AACA,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAAA,EACvB;AAAA,EAzBA,OAAO,QAAyB,YAAA,CAAa,EAAE,KAAK,OAAA,CAAQ,GAAA,CAAI,WAAY,CAAA;AAAA,EAC5E,OAAO,kBAAmC,YAAA,CAAa,EAAE,KAAK,OAAA,CAAQ,GAAA,CAAI,WAAY,CAAA;AAAA,EAC9E,KAAA;AAAA,EACA,eAAA;AAAA,EACA,QAAA;AAAA,EACA,UAAA;AAAA,EACA,WAAA,GAAc,KAAA;AAAA,EACf,YAAA;AAAA;AAAA;AAAA;AAAA,EAuBP,MAAM,KAAK,IAAA,EAAmC;AAC1C,IAAA,MAAM,IAAA,GAAO,MAAM,IAAA,CAAK,UAAA,CAAW,IAAI,CAAA;AACvC,IAAA,MAAM,cAAA,GAAiB,MAAA,CAAO,IAAA,CAAK,IAAI,CAAA;AAGvC,IAAA,MAAM,IAAA,CAAK,KAAA,CAAM,KAAA,CAAM,IAAA,CAAK,UAAU,cAAc,CAAA;AAGpD,IAAA,MAAM,IAAA,CAAK,KAAA,CAAM,MAAA,CAAO,IAAA,CAAK,UAAU,GAAG,CAAA;AAG1C,IAAA,MAAM,IAAA,CAAK,KAAA,CAAM,OAAA,CAAQ,IAAA,CAAK,YAAY,cAAc,CAAA;AAExD,IAAA,IAAA,CAAK,IAAA,CAAK,cAAc,IAAI,CAAA;AAAA,EAChC;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,aAAA,GAA6D;AAChE,IAAA,IAAI,QAAwB,EAAC;AAC7B,IAAA,IAAI,cAAA,GAAsC,IAAA;AAC1C,IAAA,IAAI,aAAA,GAAgB,KAAA;AACpB,IAAA,MAAM,aAAA,GAAgB,OAAO,OAAA,KAAoB;AAC7C,MAAA,MAAM,IAAA,GAAQ,MAAM,IAAA,CAAK,UAAA,CAAW,OAAO,CAAA;AAC3C,MAAA,KAAA,CAAM,KAAK,IAAI,CAAA;AAEf,MAAA,IACI,IAAA,CAAK,UAAU,gBAAA,IACf,IAAA,CAAK,UAAU,kBAAA,IACf,IAAA,CAAK,UAAU,mBAAA,EACjB;AACE,QAAA,UAAA,CAAW,MAAM;AACb,UAAA,aAAA,GAAgB,IAAA;AAChB,UAAA,IAAI,cAAA,EAAgB;AAChB,YAAA,cAAA,EAAe;AACf,YAAA,cAAA,GAAiB,IAAA;AAAA,UACrB;AAAA,QACJ,GAAG,GAAG,CAAA;AAEN,QAAA,IAAI,IAAA,CAAK,UAAU,mBAAA,EAAqB;AACpC,UAAA,IAAA,CAAK,MAAA,EAAO;AAAA,QAChB;AAAA,MACJ;AAEA,MAAA,IAAI,cAAA,EAAgB;AAChB,QAAA,cAAA,EAAe;AACf,QAAA,cAAA,GAAiB,IAAA;AAAA,MACrB;AAAA,IACJ,CAAA;AAGA,IAAA,MAAM,KAAK,eAAA,CAAgB,SAAA,CAAU,IAAA,CAAK,UAAA,EAAY,CAAC,OAAA,KAAY;AAC/D,MAAA,aAAA,CAAc,OAAO,CAAA;AAAA,IACzB,CAAC,CAAA;AAED,IAAA,IAAI;AACA,MAAA,OAAO,CAAC,aAAA,EAAe;AACnB,QAAA,IAAI,KAAA,CAAM,SAAS,CAAA,EAAG;AAClB,UAAA,KAAA,MAAW,QAAQ,KAAA,EAAO;AACtB,YAAA,MAAM,IAAA;AAAA,UACV;AACA,UAAA,KAAA,GAAQ,EAAC;AAAA,QACb,CAAA,MAAO;AACH,UAAA,MAAM,IAAI,OAAA,CAAQ,CAAC,OAAA,KAAY;AAC3B,YAAA,cAAA,GAAiB,OAAA;AAAA,UACrB,CAAC,CAAA;AAAA,QACL;AAAA,MACJ;AAAA,IACJ,CAAA,SAAE;AACE,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,WAAA,CAAY,IAAA,CAAK,UAAU,CAAA;AAAA,IAC1D;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,MAAA,GAAkC;AACpC,IAAA,MAAM,IAAA,GAAO,MAAM,IAAA,CAAK,KAAA,CAAM,OAAO,IAAA,CAAK,QAAA,EAAU,GAAG,EAAE,CAAA;AAEzD,IAAA,IAAI,CAAC,IAAA,IAAQ,IAAA,CAAK,MAAA,KAAW,CAAA,EAAG;AAC5B,MAAA,OAAO,EAAC;AAAA,IACZ;AAEA,IAAA,IAAI,KAAK,gBAAA,EAAkB;AACvB,MAAA,OAAQ,MAAM,OAAA,CAAQ,GAAA;AAAA,QAClB,IAAA,CAAK,GAAA,CAAI,OAAO,IAAA,KAAiB;AAC7B,UAAA,MAAM,MAAA,GAAS,IAAA,CAAK,KAAA,CAAM,IAAI,CAAA;AAC9B,UAAA,OAAQ,MAAM,IAAA,CAAK,UAAA,CAAW,MAAa,CAAA;AAAA,QAC/C,CAAC;AAAA,OACL;AAAA,IACJ,CAAA,MAAO;AACH,MAAA,OAAO,KAAK,GAAA,CAAI,CAAC,SAAiB,IAAA,CAAK,KAAA,CAAM,IAAI,CAAiB,CAAA;AAAA,IACtE;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,KAAA,GAAc;AACV,IAAA,IAAI,KAAK,WAAA,EAAa;AAClB,MAAA,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,QAAQ,CAAA;AAAA,IAChC;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,MAAA,GAAe;AACX,IAAA,IAAA,CAAK,IAAA,CAAK,IAAI,kBAAA,EAAoB,CAAA;AAClC,IAAA,IAAA,CAAK,YAAA,CAAa,MAAM,sBAAsB,CAAA;AAAA,EAClD;AACJ;;;;"}

package/dist/{router-DOVN2of2.js → router-DIyq7O-Q.js}

@@ -1,7 +1,7 @@
 import { NextResponse } from 'next/server';
-import { c as createEndpoint } from './createEndpoint-Bxzbc61h.js';
+import { c as createEndpoint } from './createEndpoint-pEWA0kt2.js';
 import { a as AssistantGraphQuerySchema, d as RunListQuerySchema, c as RunJoinStreamQuerySchema, A as AssistantsSearchSchema, g as ThreadCreatePayloadSchema, h as ThreadSearchPayloadSchema, f as ThreadStateUpdate, R as RunStreamPayloadSchema, e as RunCancelQuerySchema } from './zod-B6xyK6pu.js';
-import { s as serialiseAsDict } from './stream-CAZ9j0yn.js';
+import { s as serialiseAsDict } from './stream-Bwuk4b6F.js';
 import camelcaseKeys from 'camelcase-keys';
 
 const client = createEndpoint();
@@ -215,4 +215,4 @@ async function DELETE(req) {
 }
 
 export { DELETE, GET, POST };
-//# sourceMappingURL=router-DOVN2of2.js.map
+//# sourceMappingURL=router-DIyq7O-Q.js.map