zen-code 4.6.1 → 4.7.0

package/dist/index-BeLj9JYN.mjs

@@ -0,0 +1,463 @@
+import { B as C, u as v, c as x, T as $, m as I } from "./app-CcnStH9S.mjs";
+import "./graphBuilder-skVEY6RJ.mjs";
+import { createClient as O, createCluster as N } from "redis";
+const W = [
+  {
+    index: "checkpoints",
+    prefix: "checkpoint:",
+    schema: {
+      "$.thread_id": {
+        type: "TAG",
+        AS: "thread_id"
+      },
+      "$.checkpoint_ns": {
+        type: "TAG",
+        AS: "checkpoint_ns"
+      },
+      "$.checkpoint_id": {
+        type: "TAG",
+        AS: "checkpoint_id"
+      },
+      "$.parent_checkpoint_id": {
+        type: "TAG",
+        AS: "parent_checkpoint_id"
+      },
+      "$.checkpoint_ts": {
+        type: "NUMERIC",
+        AS: "checkpoint_ts"
+      },
+      "$.has_writes": {
+        type: "TAG",
+        AS: "has_writes"
+      },
+      "$.source": {
+        type: "TAG",
+        AS: "source"
+      },
+      "$.step": {
+        type: "NUMERIC",
+        AS: "step"
+      }
+    }
+  },
+  {
+    index: "checkpoint_blobs",
+    prefix: "checkpoint_blob:",
+    schema: {
+      "$.thread_id": {
+        type: "TAG",
+        AS: "thread_id"
+      },
+      "$.checkpoint_ns": {
+        type: "TAG",
+        AS: "checkpoint_ns"
+      },
+      "$.checkpoint_id": {
+        type: "TAG",
+        AS: "checkpoint_id"
+      },
+      "$.channel": {
+        type: "TAG",
+        AS: "channel"
+      },
+      "$.version": {
+        type: "TAG",
+        AS: "version"
+      },
+      "$.type": {
+        type: "TAG",
+        AS: "type"
+      }
+    }
+  },
+  {
+    index: "checkpoint_writes",
+    prefix: "checkpoint_write:",
+    schema: {
+      "$.thread_id": {
+        type: "TAG",
+        AS: "thread_id"
+      },
+      "$.checkpoint_ns": {
+        type: "TAG",
+        AS: "checkpoint_ns"
+      },
+      "$.checkpoint_id": {
+        type: "TAG",
+        AS: "checkpoint_id"
+      },
+      "$.task_id": {
+        type: "TAG",
+        AS: "task_id"
+      },
+      "$.idx": {
+        type: "NUMERIC",
+        AS: "idx"
+      },
+      "$.channel": {
+        type: "TAG",
+        AS: "channel"
+      },
+      "$.type": {
+        type: "TAG",
+        AS: "type"
+      }
+    }
+  }
+];
+var E = class T extends C {
+  client;
+  ttlConfig;
+  constructor(e, t) {
+    super(), this.client = e, this.ttlConfig = t;
+  }
+  static async fromUrl(e, t) {
+    const s = O({ url: e });
+    await s.connect();
+    const i = new T(s, t);
+    return await i.ensureIndexes(), i;
+  }
+  static async fromCluster(e, t) {
+    const s = N({ rootNodes: e });
+    await s.connect();
+    const i = new T(s, t);
+    return await i.ensureIndexes(), i;
+  }
+  async get(e) {
+    return (await this.getTuple(e))?.checkpoint;
+  }
+  async getTuple(e) {
+    const t = e.configurable?.thread_id, s = e.configurable?.checkpoint_ns ?? "", i = e.configurable?.checkpoint_id;
+    if (!t) return;
+    let c, o;
+    if (i)
+      c = `checkpoint:${t}:${s}:${i}`, o = await this.client.json.get(c);
+    else {
+      const r = `checkpoint:${t}:${s}:*`, p = await this.client.keys(r);
+      if (p.length === 0) return;
+      p.sort(), c = p[p.length - 1], o = await this.client.json.get(c);
+    }
+    if (!o) return;
+    this.ttlConfig?.refreshOnRead && this.ttlConfig?.defaultTTL && await this.applyTTL(c);
+    const { checkpoint: f, pendingWrites: n } = await this.loadCheckpointWithWrites(o);
+    return await this.createCheckpointTuple(o, f, n);
+  }
+  async put(e, t, s, i) {
+    await this.ensureIndexes();
+    const c = e.configurable?.thread_id, o = e.configurable?.checkpoint_ns ?? "", f = e.configurable?.checkpoint_id;
+    if (!c) throw new Error("thread_id is required");
+    const n = t.id || v(0), r = `checkpoint:${c}:${o}:${n}`, p = x(t);
+    if (p.channel_values && i !== void 0) if (Object.keys(i).length === 0) p.channel_values = {};
+    else {
+      const u = {};
+      for (const a of Object.keys(i)) a in p.channel_values && (u[a] = p.channel_values[a]);
+      p.channel_values = u;
+    }
+    const l = {
+      thread_id: c,
+      checkpoint_ns: o === "" ? "__empty__" : o,
+      checkpoint_id: n,
+      parent_checkpoint_id: f || null,
+      checkpoint: p,
+      metadata: s,
+      checkpoint_ts: Date.now(),
+      has_writes: "false"
+    };
+    return this.addSearchableMetadataFields(l, s), await this.client.json.set(r, "$", l), this.ttlConfig?.defaultTTL && await this.applyTTL(r), { configurable: {
+      thread_id: c,
+      checkpoint_ns: o,
+      checkpoint_id: n
+    } };
+  }
+  async *list(e, t) {
+    if (await this.ensureIndexes(), t?.filter !== void 0) {
+      const i = Object.values(t.filter).some((n) => n === null), c = [];
+      if (e?.configurable?.thread_id) {
+        const n = e.configurable.thread_id.replace(/[-.@]/g, "\\$&");
+        c.push(`(@thread_id:{${n}})`);
+      }
+      if (e?.configurable?.checkpoint_ns !== void 0) {
+        const n = e.configurable.checkpoint_ns;
+        if (n === "") c.push("(@checkpoint_ns:{__empty__})");
+        else {
+          const r = n.replace(/[-.@]/g, "\\$&");
+          c.push(`(@checkpoint_ns:{${r}})`);
+        }
+      }
+      if (!t?.before && t?.filter)
+        for (const [n, r] of Object.entries(t.filter)) r === void 0 || r === null || (typeof r == "string" ? c.push(`(@${n}:{${r}})`) : typeof r == "number" ? c.push(`(@${n}:[${r} ${r}])`) : typeof r == "object" && Object.keys(r).length);
+      c.length === 0 && c.push("*");
+      const o = c.join(" "), f = t?.limit ?? 10;
+      try {
+        const n = t?.before && !e?.configurable?.thread_id ? 1e3 : t?.before ? f * 10 : f;
+        let p = (await this.client.ft.search("checkpoints", o, {
+          LIMIT: {
+            from: 0,
+            size: n
+          },
+          SORTBY: {
+            BY: "checkpoint_ts",
+            DIRECTION: "DESC"
+          }
+        })).documents, l = 0;
+        for (const u of p) {
+          if (l >= f) break;
+          if (t?.before?.configurable?.checkpoint_id) {
+            const k = u.value.checkpoint_id, d = t.before.configurable.checkpoint_id;
+            if (k >= d) continue;
+          }
+          const a = u.value;
+          let h = !0;
+          if ((i || t?.before) && t?.filter) {
+            for (const [k, d] of Object.entries(t.filter)) if (d === null) {
+              if (a.metadata?.[k] !== null) {
+                h = !1;
+                break;
+              }
+            } else if (d !== void 0) {
+              const b = a.metadata?.[k];
+              if (typeof d == "object" && d !== null) {
+                if (g(b) !== g(d)) {
+                  h = !1;
+                  break;
+                }
+              } else if (b !== d) {
+                h = !1;
+                break;
+              }
+            }
+            if (!h) continue;
+          }
+          const { checkpoint: y, pendingWrites: _ } = await this.loadCheckpointWithWrites(a);
+          yield await this.createCheckpointTuple(a, y, _), l++;
+        }
+        return;
+      } catch (n) {
+        if (!n.message?.includes("no such index")) throw n;
+      }
+      if (e?.configurable?.thread_id) {
+        const n = e.configurable.thread_id, r = e.configurable.checkpoint_ns ?? "", p = `checkpoint:${n}:${r}:*`, l = await this.client.keys(p);
+        l.sort().reverse();
+        let u = l;
+        if (t?.before?.configurable?.checkpoint_id) {
+          const y = t.before.configurable.thread_id || n, _ = t.before.configurable.checkpoint_ns ?? r, k = `checkpoint:${y}:${_}:${t.before.configurable.checkpoint_id}`, d = l.indexOf(k);
+          d > 0 ? u = l.slice(d + 1) : d === 0 && (u = []);
+        }
+        const a = t?.limit ?? 10, h = u.slice(0, a);
+        for (const y of h) {
+          const _ = await this.client.json.get(y);
+          if (_) {
+            let k = !0;
+            for (const [S, m] of Object.entries(t.filter)) {
+              const A = _.metadata?.[S];
+              if (m === null) {
+                if (A !== null) {
+                  k = !1;
+                  break;
+                }
+              } else if (A !== m) {
+                k = !1;
+                break;
+              }
+            }
+            if (!k) continue;
+            const { checkpoint: d, pendingWrites: b } = await this.loadCheckpointWithWrites(_);
+            yield await this.createCheckpointTuple(_, d, b);
+          }
+        }
+      } else {
+        const n = e?.configurable?.checkpoint_ns !== void 0 ? `checkpoint:*:${e.configurable.checkpoint_ns === "" ? "__empty__" : e.configurable.checkpoint_ns}:*` : "checkpoint:*", r = await this.client.keys(n), p = [];
+        for (const a of r) {
+          const h = await this.client.json.get(a);
+          h && p.push({
+            key: a,
+            doc: h
+          });
+        }
+        p.sort((a, h) => h.doc.checkpoint_ts - a.doc.checkpoint_ts);
+        let l = 0;
+        const u = t?.limit ?? 10;
+        for (const { doc: a } of p) {
+          if (l >= u) break;
+          if (t?.before?.configurable?.checkpoint_id) {
+            const k = a.checkpoint_id, d = t.before.configurable.checkpoint_id;
+            if (k >= d) continue;
+          }
+          let h = !0;
+          if (t?.filter) {
+            for (const [k, d] of Object.entries(t.filter)) if (d === null) {
+              if (a.metadata?.[k] !== null) {
+                h = !1;
+                break;
+              }
+            } else if (d !== void 0) {
+              const b = a.metadata?.[k];
+              if (typeof d == "object" && d !== null) {
+                if (g(b) !== g(d)) {
+                  h = !1;
+                  break;
+                }
+              } else if (b !== d) {
+                h = !1;
+                break;
+              }
+            }
+            if (!h) continue;
+          }
+          const { checkpoint: y, pendingWrites: _ } = await this.loadCheckpointWithWrites(a);
+          yield await this.createCheckpointTuple(a, y, _), l++;
+        }
+      }
+      return;
+    }
+    const s = {
+      ...t,
+      filter: {}
+    };
+    yield* this.list(e, s);
+  }
+  async putWrites(e, t, s) {
+    await this.ensureIndexes();
+    const i = e.configurable?.thread_id, c = e.configurable?.checkpoint_ns ?? "", o = e.configurable?.checkpoint_id;
+    if (!i || !o) throw new Error("thread_id and checkpoint_id are required");
+    const f = [], n = performance.now() * 1e3;
+    for (let l = 0; l < t.length; l++) {
+      const [u, a] = t[l], h = `checkpoint_write:${i}:${c}:${o}:${s}:${l}`;
+      f.push(h);
+      const y = {
+        thread_id: i,
+        checkpoint_ns: c,
+        checkpoint_id: o,
+        task_id: s,
+        idx: l,
+        channel: u,
+        type: typeof a == "object" ? "json" : "string",
+        value: a,
+        timestamp: n,
+        global_idx: n + l
+      };
+      await this.client.json.set(h, "$", y);
+    }
+    if (f.length > 0) {
+      const l = `write_keys_zset:${i}:${c}:${o}`, u = {};
+      f.forEach((a, h) => {
+        u[a] = n + h;
+      }), await this.client.zAdd(l, Object.entries(u).map(([a, h]) => ({
+        score: h,
+        value: a
+      }))), this.ttlConfig?.defaultTTL && await this.applyTTL(...f, l);
+    }
+    const r = `checkpoint:${i}:${c}:${o}`;
+    if (await this.client.exists(r)) {
+      const l = await this.client.json.get(r);
+      l && (l.has_writes = "true", await this.client.json.set(r, "$", l));
+    }
+  }
+  async deleteThread(e) {
+    const t = `checkpoint:${e}:*`, s = await this.client.keys(t);
+    s.length > 0 && await this.client.del(s);
+    const i = `writes:${e}:*`, c = await this.client.keys(i);
+    c.length > 0 && await this.client.del(c);
+  }
+  async end() {
+    await this.client.quit();
+  }
+  async loadPendingWrites(e, t, s) {
+    const i = `checkpoint_write:${e}:${t}:${s}:*`, c = await this.client.keys(i);
+    if (c.length === 0) return;
+    const o = [];
+    for (const n of c) {
+      const r = await this.client.json.get(n);
+      r && o.push(r);
+    }
+    o.sort((n, r) => (n.global_idx || 0) - (r.global_idx || 0));
+    const f = [];
+    for (const n of o) {
+      const r = await this.serde.loadsTyped("json", JSON.stringify(n.value));
+      f.push([
+        n.task_id,
+        n.channel,
+        r
+      ]);
+    }
+    return f;
+  }
+  async loadCheckpointWithWrites(e) {
+    const t = await this.serde.loadsTyped("json", JSON.stringify(e.checkpoint));
+    if (t.v < 4 && e.parent_checkpoint_id != null) {
+      const i = e.checkpoint_ns === "__empty__" ? "" : e.checkpoint_ns;
+      await this.migratePendingSends(t, e.thread_id, i, e.parent_checkpoint_id);
+    }
+    let s;
+    if (e.has_writes === "true") {
+      const i = e.checkpoint_ns === "__empty__" ? "" : e.checkpoint_ns;
+      s = await this.loadPendingWrites(e.thread_id, i, e.checkpoint_id);
+    }
+    return {
+      checkpoint: t,
+      pendingWrites: s
+    };
+  }
+  async migratePendingSends(e, t, s, i) {
+    const c = await this.loadPendingWrites(t, s, i);
+    if (!c || c.length === 0) return;
+    const o = c.filter(([, n]) => n === $);
+    if (o.length === 0) return;
+    const f = [];
+    for (const [, , n] of o) f.push(n);
+    e.channel_values ??= {}, e.channel_values[$] = f, e.channel_versions[$] = Object.keys(e.channel_versions).length > 0 ? I(...Object.values(e.channel_versions)) : 1;
+  }
+  async createCheckpointTuple(e, t, s) {
+    const i = e.checkpoint_ns === "__empty__" ? "" : e.checkpoint_ns, c = await this.serde.loadsTyped("json", JSON.stringify(e.metadata));
+    return {
+      config: { configurable: {
+        thread_id: e.thread_id,
+        checkpoint_ns: i,
+        checkpoint_id: e.checkpoint_id
+      } },
+      checkpoint: t,
+      metadata: c,
+      parentConfig: e.parent_checkpoint_id ? { configurable: {
+        thread_id: e.thread_id,
+        checkpoint_ns: i,
+        checkpoint_id: e.parent_checkpoint_id
+      } } : void 0,
+      pendingWrites: s
+    };
+  }
+  addSearchableMetadataFields(e, t) {
+    t && ("source" in t && (e.source = t.source), "step" in t && (e.step = t.step), "writes" in t && (e.writes = typeof t.writes == "object" ? JSON.stringify(t.writes) : t.writes), "score" in t && (e.score = t.score));
+  }
+  async applyTTL(...e) {
+    if (!this.ttlConfig?.defaultTTL) return;
+    const t = Math.floor(this.ttlConfig.defaultTTL * 60), s = await Promise.allSettled(e.map((i) => this.client.expire(i, t)));
+    for (let i = 0; i < s.length; i++) s[i].status === "rejected" && console.warn(`Failed to set TTL for key ${e[i]}:`, s[i].reason);
+  }
+  async ensureIndexes() {
+    for (const e of W) try {
+      await this.client.ft.create(e.index, e.schema, {
+        ON: "JSON",
+        PREFIX: e.prefix
+      });
+    } catch (t) {
+      t.message?.includes("Index already exists") || console.error(`Failed to create index ${e.index}:`, t.message);
+    }
+  }
+};
+function g(w) {
+  if (w === null || typeof w != "object") return JSON.stringify(w);
+  if (Array.isArray(w)) return JSON.stringify(w.map((s) => g(s)));
+  const e = {}, t = Object.keys(w).sort();
+  for (const s of t) e[s] = w[s];
+  return JSON.stringify(e, (s, i) => {
+    if (i !== null && typeof i == "object" && !Array.isArray(i)) {
+      const c = {}, o = Object.keys(i).sort();
+      for (const f of o) c[f] = i[f];
+      return c;
+    }
+    return i;
+  });
+}
+export {
+  E as RedisSaver
+};

package/dist/{index-DEcMgCTj.mjs → index-D_dnZWV3.mjs}

@@ -1,5 +1,5 @@
-import { A as s, C as m, F as o, e as c, S as i, a as p, d as h, b as k, h as n, f as g, g as T, T as f, c as l } from "./graphBuilder-B3IJ7dB2.mjs";
-import { FileSystemSkillStore as y } from "./FileSystemSkillStore-yvEvcRxB.mjs";
+import { A as s, C as m, F as o, e as c, S as i, a as p, d as h, b as k, h as n, f as g, g as T, T as f, c as l } from "./graphBuilder-skVEY6RJ.mjs";
+import { FileSystemSkillStore as y } from "./FileSystemSkillStore-Dl5briBv.mjs";
 import { FileSystemPluginStore as F } from "./FileSystemPluginStore-ChortK7z.mjs";
 import "lowdb";
 import "lowdb/node";

package/dist/{load-Bh8eo6qE.mjs → load-iYJqYrjn.mjs}

@@ -22,7 +22,7 @@ function I(t, o) {
     const n = k(t);
     if (n.size > Y)
       return console.warn(`Skipping ${t}: file too large (${n.size} bytes)`), null;
-    const i = x(t, "utf-8"), e = /^---\s*\n(.*?)\n---\s*\n/s, a = i.match(e);
+    const i = x(t, "utf-8"), e = /^---\s*\n([\s\S]*?)\n---\s*\n/, a = i.match(e);
     if (!a)
       return console.warn(`Skipping ${t}: no valid YAML frontmatter found`), null;
     const c = a[1];

package/dist/{memories-rSGKIPk5.mjs → memories-CSOPZKuJ.mjs}

@@ -1,5 +1,5 @@
-import { l } from "./load-Bh8eo6qE.mjs";
-import { SystemMessage as h } from "@langchain/core/messages";
+import { l } from "./load-iYJqYrjn.mjs";
+import { n as h } from "./graphBuilder-skVEY6RJ.mjs";
 const M = `## Memory System
 
 你可以访问一个包含之前任务和经验知识的记忆库。

package/dist/nonInteractive.mjs

@@ -1,46 +1,49 @@
 import { HumanMessage as c } from "langchain";
-import { i as d, j as m, k as l } from "./graphBuilder-B3IJ7dB2.mjs";
-import "@langgraph-js/pro";
+import { i as m, j as d, k as p } from "./graphBuilder-skVEY6RJ.mjs";
+import "@langchain/langgraph";
+import "zod";
+import "@langchain/openai";
 import "@langchain/anthropic";
 import "@langchain/google-genai";
 import "yaml";
-import "@langchain/core/messages";
 import "@langgraph-js/standard-agent";
-import "@langgraph-js/auk";
-async function p() {
+import "micromatch";
+import "lowdb";
+import "lowdb/node";
+async function l() {
   return new Promise((t, n) => {
-    let r = "";
-    process.stdin.setEncoding("utf8"), process.stdin.on("data", (e) => {
-      r += e;
+    let e = "";
+    process.stdin.setEncoding("utf8"), process.stdin.on("data", (r) => {
+      e += r;
     }), process.stdin.on("end", () => {
-      t(r);
-    }), process.stdin.on("error", (e) => {
-      n(e);
+      t(e);
+    }), process.stdin.on("error", (r) => {
+      n(r);
     });
   });
 }
-async function b(t, n = !1) {
-  await d();
-  const r = await m();
-  let e = t || "";
+async function S(t, n = !1) {
+  await m();
+  const e = await d();
+  let r = t || "";
   if (n)
     try {
-      e = (await p()).trim();
+      r = (await l()).trim();
     } catch (o) {
       throw console.error("❌ 读取 stdin 失败:", o), o;
     }
-  e || (console.error("❌ 错误: 未提供输入内容"), console.error(`
+  r || (console.error("❌ 错误: 未提供输入内容"), console.error(`
 用法:`), console.error('  zen-code -p "你的任务描述"'), console.error('  echo "内容" | zen-code'), process.exit(1));
   try {
     const o = {
-      messages: [new c(e)],
-      provider_id: r.provider_id,
-      model_id: r.model_id,
-      enable_thinking: r.enable_thinking
-    }, s = await l.invoke(o, {
+      messages: [new c(r)],
+      provider_id: e.provider_id,
+      model_id: e.model_id,
+      enable_thinking: e.enable_thinking
+    }, i = await p.invoke(o, {
       recursionLimit: 500
-    }), i = s.messages || [], a = i[i.length - 1];
-    return a && console.log(a.text), s;
+    }), s = i.messages || [], a = s[s.length - 1];
+    return a && console.log(a.text), i;
   } catch (o) {
     throw console.error(`
 ❌ 执行失败: ${o instanceof Error ? o.message : String(o)}
@@ -48,5 +51,5 @@ async function b(t, n = !1) {
   }
 }
 export {
-  b as runNonInteractive
+  S as runNonInteractive
 };

package/dist/pg-adapter-BFtir1GE-BbUXBpaX.mjs

@@ -0,0 +1,60 @@
+import { Kysely as r, PostgresDialect as s, sql as e } from "kysely";
+class i {
+  constructor(t) {
+    this.pool = t, this.db = new r({
+      dialect: new s({
+        pool: t
+      })
+    });
+  }
+  db;
+  dateToDb(t) {
+    return t;
+  }
+  dbToDate(t) {
+    return t instanceof Date ? t : new Date(t);
+  }
+  jsonToDb(t) {
+    return t;
+  }
+  dbToJson(t) {
+    return t;
+  }
+  buildJsonQuery(t, a, T, E) {
+    return e`${e.ref(a)}->>${e.lit(T)} = ${e.lit(JSON.stringify(E))}`;
+  }
+  now() {
+    return /* @__PURE__ */ new Date();
+  }
+  async createTables(t) {
+    await e`
+            CREATE TABLE IF NOT EXISTS threads (
+                thread_id TEXT PRIMARY KEY,
+                created_at TIMESTAMP NOT NULL,
+                updated_at TIMESTAMP NOT NULL,
+                metadata JSONB NOT NULL DEFAULT '{}',
+                status TEXT NOT NULL DEFAULT 'idle',
+                values JSONB,
+                interrupts JSONB NOT NULL DEFAULT '{}'
+            )
+        `.execute(t), await e`
+            CREATE TABLE IF NOT EXISTS runs (
+                run_id TEXT PRIMARY KEY,
+                thread_id TEXT NOT NULL,
+                assistant_id TEXT NOT NULL,
+                created_at TIMESTAMP NOT NULL,
+                updated_at TIMESTAMP NOT NULL,
+                status TEXT NOT NULL DEFAULT 'pending',
+                metadata JSONB NOT NULL DEFAULT '{}',
+                multitask_strategy TEXT NOT NULL DEFAULT 'reject',
+                FOREIGN KEY (thread_id) REFERENCES threads(thread_id) ON DELETE CASCADE
+            )
+        `.execute(t);
+  }
+  async createIndexes(t) {
+    await e`CREATE INDEX IF NOT EXISTS idx_threads_status ON threads(status)`.execute(t), await e`CREATE INDEX IF NOT EXISTS idx_threads_created_at ON threads(created_at)`.execute(t), await e`CREATE INDEX IF NOT EXISTS idx_threads_updated_at ON threads(updated_at)`.execute(t), await e`CREATE INDEX IF NOT EXISTS idx_runs_thread_id ON runs(thread_id)`.execute(t), await e`CREATE INDEX IF NOT EXISTS idx_runs_status ON runs(status)`.execute(t);
+  }
+}
+export {
+  i as PostgresAdapter
+};