zen-code 4.7.3 → 4.8.0

package/dist/checkpoint-C5AFBYE--CczNxXBB.mjs

@@ -1,355 +0,0 @@
-import { Kysely as m, sql as d } from "kysely";
-import { B as b, T as l, c as E, m as g } from "./app-CK3nD8Gj.mjs";
-import "./graphBuilder-DJoMXf6J.mjs";
-const u = {
-  maxRetries: 3,
-  baseDelayMs: 100,
-  isRetryableError: (w) => {
-    const e = w?.message?.toLowerCase() || "";
-    return e.includes("sqlite_busy") || e.includes("database is locked") || e.includes("database disk image is malformed") || e === "sqlite_busy" || e === "database is locked";
-  }
-};
-async function T(w, e) {
-  let i = null;
-  for (let n = 0; n < u.maxRetries; n++)
-    try {
-      return await w();
-    } catch (a) {
-      if (i = a, !u.isRetryableError(a))
-        throw a;
-      if (n < u.maxRetries - 1) {
-        const c = u.baseDelayMs * Math.pow(2, n);
-        console.warn(
-          `SQLite lock detected${e ? ` (${e})` : ""}, retrying in ${c}ms (attempt ${n + 1}/${u.maxRetries})`
-        ), await new Promise((t) => setTimeout(t, c));
-      }
-    }
-  throw i;
-}
-const A = ["source", "step", "parents"];
-const N = A;
-class f extends b {
-  db;
-  isSetup;
-  constructor(e, i) {
-    super(i), this.db = new m({
-      dialect: e
-    }), this.isSetup = !1;
-  }
-  static async fromConnStringAsync(e) {
-    let i;
-    if (globalThis.Bun) {
-      console.log("LG | Using BunWorkerDialect " + e);
-      const { BunWorkerDialect: n } = await import("kysely-bun-worker");
-      i = new f(new n({ url: e }));
-    } else {
-      console.log("LG | Using NodeWasmDialect");
-      const { default: n } = await import("node-sqlite3-wasm"), { NodeWasmDialect: a } = await import("kysely-wasm");
-      console.log(e);
-      const c = new a({
-        database: new n.Database(e)
-      });
-      i = new f(c);
-    }
-    return await i.setup(), i;
-  }
-  async setup() {
-    this.isSetup || (await d`PRAGMA busy_timeout = 5000`.execute(this.db), await d`PRAGMA journal_mode = WAL`.execute(this.db), await d`PRAGMA synchronous = NORMAL`.execute(this.db), await d`PRAGMA wal_autocheckpoint = 1000`.execute(this.db), await d`
-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 d`
-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 = !0);
-  }
-  async getTuple(e) {
-    await this.setup();
-    const { thread_id: i, checkpoint_ns: n = "", checkpoint_id: a } = e.configurable ?? {};
-    let c = this.db.selectFrom("checkpoints").select([
-      "thread_id",
-      "checkpoint_ns",
-      "checkpoint_id",
-      "parent_checkpoint_id",
-      "type",
-      "checkpoint",
-      "metadata",
-      d`(
-                    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"),
-      d`(
-                    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 = ${l}
-                    ORDER BY ps.idx
-                )`.as("pending_sends")
-    ]).where("thread_id", "=", i).where("checkpoint_ns", "=", n);
-    a ? c = c.where("checkpoint_id", "=", a) : c = c.orderBy("checkpoint_id", "desc").limit(1);
-    const t = await c.executeTakeFirst();
-    if (!t) return;
-    let r = e;
-    if (a || (r = {
-      configurable: {
-        thread_id: t.thread_id,
-        checkpoint_ns: n,
-        checkpoint_id: t.checkpoint_id
-      }
-    }), r.configurable?.thread_id === void 0 || r.configurable?.checkpoint_id === void 0)
-      throw new Error("Missing thread_id or checkpoint_id");
-    const s = await Promise.all(
-      JSON.parse(t.pending_writes).map(async (h) => [
-        h.task_id,
-        h.channel,
-        await this.serde.loadsTyped(h.type ?? "json", h.value ?? "")
-      ])
-    ), p = await this.serde.loadsTyped(
-      t.type ?? "json",
-      new TextDecoder().decode(t.checkpoint)
-    );
-    return p.v < 4 && t.parent_checkpoint_id != null && await this.migratePendingSends(p, t.thread_id, t.parent_checkpoint_id), {
-      checkpoint: p,
-      config: r,
-      metadata: await this.serde.loadsTyped(
-        t.type ?? "json",
-        new TextDecoder().decode(t.metadata)
-      ),
-      parentConfig: t.parent_checkpoint_id ? {
-        configurable: {
-          thread_id: t.thread_id,
-          checkpoint_ns: n,
-          checkpoint_id: t.parent_checkpoint_id
-        }
-      } : void 0,
-      pendingWrites: s
-    };
-  }
-  async *list(e, i) {
-    const { limit: n, before: a, filter: c } = i ?? {};
-    await this.setup();
-    const t = e.configurable?.thread_id, r = e.configurable?.checkpoint_ns;
-    let s = this.db.selectFrom("checkpoints").select([
-      "thread_id",
-      "checkpoint_ns",
-      "checkpoint_id",
-      "parent_checkpoint_id",
-      "type",
-      "checkpoint",
-      "metadata",
-      d`(
-                    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"),
-      d`(
-                    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 = ${l}
-                    ORDER BY ps.idx
-                )`.as("pending_sends")
-    ]);
-    t && (s = s.where("thread_id", "=", t)), r != null && (s = s.where("checkpoint_ns", "=", r)), a?.configurable?.checkpoint_id !== void 0 && (s = s.where("checkpoint_id", "<", a.configurable.checkpoint_id));
-    const p = Object.fromEntries(
-      Object.entries(c ?? {}).filter(
-        ([o, _]) => _ !== void 0 && N.includes(o)
-      )
-    );
-    for (const [o, _] of Object.entries(p))
-      s = s.where(
-        d`json_extract(CAST(metadata AS TEXT), ${d.lit("$." + o)})`,
-        "=",
-        d.lit(JSON.stringify(_))
-      );
-    s = s.orderBy("checkpoint_id", "desc"), n && (s = s.limit(parseInt(n, 10)));
-    const h = await s.execute();
-    for (const o of h) {
-      const _ = await Promise.all(
-        JSON.parse(o.pending_writes).map(async (k) => [
-          k.task_id,
-          k.channel,
-          await this.serde.loadsTyped(k.type ?? "json", k.value ?? "")
-        ])
-      ), y = await this.serde.loadsTyped(
-        o.type ?? "json",
-        new TextDecoder().decode(o.checkpoint)
-      );
-      y.v < 4 && o.parent_checkpoint_id != null && await this.migratePendingSends(y, o.thread_id, o.parent_checkpoint_id), yield {
-        config: {
-          configurable: {
-            thread_id: o.thread_id,
-            checkpoint_ns: o.checkpoint_ns,
-            checkpoint_id: o.checkpoint_id
-          }
-        },
-        checkpoint: y,
-        metadata: await this.serde.loadsTyped(
-          o.type ?? "json",
-          new TextDecoder().decode(o.metadata)
-        ),
-        parentConfig: o.parent_checkpoint_id ? {
-          configurable: {
-            thread_id: o.thread_id,
-            checkpoint_ns: o.checkpoint_ns,
-            checkpoint_id: o.parent_checkpoint_id
-          }
-        } : void 0,
-        pendingWrites: _
-      };
-    }
-  }
-  async put(e, i, n) {
-    if (await this.setup(), !e.configurable)
-      throw new Error("Empty configuration supplied.");
-    const a = e.configurable?.thread_id, c = e.configurable?.checkpoint_ns ?? "", t = e.configurable?.checkpoint_id;
-    if (!a)
-      throw new Error('Missing "thread_id" field in passed "config.configurable".');
-    const r = E(i), [[s, p], [h, o]] = await Promise.all([
-      this.serde.dumpsTyped(r),
-      this.serde.dumpsTyped(n)
-    ]);
-    if (s !== h)
-      throw new Error("Failed to serialized checkpoint and metadata to the same type.");
-    return await T(
-      async () => {
-        await this.db.insertInto("checkpoints").values({
-          thread_id: a,
-          checkpoint_ns: c,
-          checkpoint_id: i.id,
-          parent_checkpoint_id: t ?? null,
-          type: s,
-          checkpoint: new Uint8Array(Buffer.from(p)),
-          metadata: new Uint8Array(Buffer.from(o))
-        }).onConflict(
-          (_) => _.columns(["thread_id", "checkpoint_ns", "checkpoint_id"]).doUpdateSet({
-            parent_checkpoint_id: t ?? null,
-            type: s,
-            checkpoint: new Uint8Array(Buffer.from(p)),
-            metadata: new Uint8Array(Buffer.from(o))
-          })
-        ).execute();
-      },
-      `put(${a}/${i.id})`
-    ), {
-      configurable: {
-        thread_id: a,
-        checkpoint_ns: c,
-        checkpoint_id: i.id
-      }
-    };
-  }
-  async putWrites(e, i, n) {
-    if (await this.setup(), !e.configurable)
-      throw new Error("Empty configuration supplied.");
-    if (!e.configurable?.thread_id)
-      throw new Error("Missing thread_id field in config.configurable.");
-    if (!e.configurable?.checkpoint_id)
-      throw new Error("Missing checkpoint_id field in config.configurable.");
-    const a = await Promise.all(
-      i.map(async (r, s) => {
-        const [p, h] = await this.serde.dumpsTyped(r[1]);
-        return {
-          thread_id: e.configurable.thread_id,
-          checkpoint_ns: e.configurable.checkpoint_ns ?? "",
-          checkpoint_id: e.configurable.checkpoint_id,
-          task_id: n,
-          idx: s,
-          channel: r[0],
-          type: p,
-          value: new Uint8Array(Buffer.from(h))
-        };
-      })
-    );
-    if (a.length === 0) return;
-    const c = e.configurable.thread_id, t = e.configurable.checkpoint_id;
-    await T(
-      async () => {
-        await this.db.transaction().execute(async (r) => {
-          await r.deleteFrom("writes").where("thread_id", "=", c).where("checkpoint_ns", "=", a[0].checkpoint_ns).where("checkpoint_id", "=", t).where("task_id", "=", n).execute();
-          for (const s of a)
-            await r.insertInto("writes").values(s).execute();
-        });
-      },
-      `putWrites(${c}/${t}/${n})`
-    );
-  }
-  async deleteThread(e) {
-    await T(
-      async () => {
-        await this.db.transaction().execute(async (i) => {
-          await i.deleteFrom("checkpoints").where("thread_id", "=", e).execute(), await i.deleteFrom("writes").where("thread_id", "=", e).execute();
-        });
-      },
-      `deleteThread(${e})`
-    );
-  }
-  async migratePendingSends(e, i, n) {
-    const a = await this.db.selectFrom("writes as ps").select([
-      "ps.checkpoint_id",
-      d`json_group_array(
-                    json_object(
-                        'type', ps.type,
-                        'value', CAST(ps.value AS TEXT)
-                    )
-                )`.as("pending_sends")
-    ]).where("ps.thread_id", "=", i).where("ps.checkpoint_id", "=", n).where("ps.channel", "=", l).orderBy("ps.idx").executeTakeFirst();
-    if (!a) return;
-    const c = e;
-    c.channel_values ??= {}, c.channel_values[l] = await Promise.all(
-      JSON.parse(a.pending_sends).map(
-        ({ type: t, value: r }) => this.serde.loadsTyped(t, r)
-      )
-    ), c.channel_versions[l] = Object.keys(e.channel_versions).length > 0 ? g(...Object.values(e.channel_versions)) : this.getNextVersion(void 0);
-  }
-}
-export {
-  f as SqliteSaver
-};