@ztimson/ai-utils 0.8.11 → 0.8.12

package/dist/index.mjs

@@ -1,23 +1,23 @@
 import * as O from "node:os";
 import { tmpdir as U } from "node:os";
-import { objectMap as z, JSONAttemptParse as _, findByProp as R, JSONSanitize as S, clean as W, Http as C, consoleInterceptor as N, fn as J, ASet as F } from "@ztimson/utils";
-import { Anthropic as B } from "@anthropic-ai/sdk";
-import { OpenAI as D } from "openai";
-import { fileURLToPath as H } from "url";
-import { join as I, dirname as G } from "path";
+import { Anthropic as W } from "@anthropic-ai/sdk";
+import { objectMap as z, JSONAttemptParse as S, findByProp as R, JSONSanitize as _, clean as N, Http as C, consoleInterceptor as J, fn as I, ASet as D } from "@ztimson/utils";
+import { OpenAI as F } from "openai";
+import { fileURLToPath as B } from "url";
+import { join as H, dirname as G } from "path";
 import { spawn as w, execSync as K } from "node:child_process";
-import { mkdtempSync as V } from "node:fs";
+import { mkdtempSync as Y } from "node:fs";
 import b from "node:fs/promises";
 import * as M from "node:path";
 import q, { join as A } from "node:path";
-import { createWorker as Y } from "tesseract.js";
+import { createWorker as V } from "tesseract.js";
 import * as Z from "cheerio";
 import { $Sync as j } from "@ztimson/node-utils";
 class L {
 }
 class Q extends L {
   constructor(r, e, t) {
-    super(), this.ai = r, this.apiToken = e, this.model = t, this.client = new B({ apiKey: e });
+    super(), this.ai = r, this.apiToken = e, this.model = t, this.client = new W({ apiKey: e });
   }
   client;
   toStandard(r) {
@@ -33,8 +33,8 @@ class Q extends L {
           if (n.type == "tool_use")
             t.push({ timestamp: e, role: "tool", id: n.id, name: n.name, args: n.input, content: void 0 });
           else if (n.type == "tool_result") {
-            const u = t.findLast((l) => l.id == n.tool_use_id);
-            u && (u[n.is_error ? "error" : "content"] = n.content);
+            const m = t.findLast((l) => l.id == n.tool_use_id);
+            m && (m[n.is_error ? "error" : "content"] = n.content);
           }
         });
       }
@@ -57,33 +57,33 @@ class Q extends L {
     const t = new AbortController();
     return Object.assign(new Promise(async (i) => {
       let o = this.fromStandard([...e.history || [], { role: "user", content: r, timestamp: Date.now() }]);
-      const n = e.tools || this.ai.options.llm?.tools || [], u = {
+      const n = e.tools || this.ai.options.llm?.tools || [], m = {
         model: e.model || this.model,
         max_tokens: e.max_tokens || this.ai.options.llm?.max_tokens || 4096,
         system: e.system || this.ai.options.llm?.system || "",
         temperature: e.temperature || this.ai.options.llm?.temperature || 0.7,
-        tools: n.map((m) => ({
-          name: m.name,
-          description: m.description,
+        tools: n.map((u) => ({
+          name: u.name,
+          description: u.description,
           input_schema: {
             type: "object",
-            properties: m.args ? z(m.args, (s, a) => ({ ...a, required: void 0 })) : {},
-            required: m.args ? Object.entries(m.args).filter((s) => s[1].required).map((s) => s[0]) : []
+            properties: u.args ? z(u.args, (s, c) => ({ ...c, required: void 0 })) : {},
+            required: u.args ? Object.entries(u.args).filter((s) => s[1].required).map((s) => s[0]) : []
           },
           fn: void 0
         })),
         messages: o,
         stream: !!e.stream
       };
-      let l, c = !0;
+      let l, a = !0;
       do {
-        if (l = await this.client.messages.create(u).catch((s) => {
+        if (l = await this.client.messages.create(m).catch((s) => {
           throw s.message += `
 
 Messages:
 ${JSON.stringify(o, null, 2)}`, s;
         }), e.stream) {
-          c ? c = !1 : e.stream({ text: `
+          a ? a = !1 : e.stream({ text: `
 
 ` }), l.content = [];
           for await (const s of l) {
@@ -92,33 +92,33 @@ ${JSON.stringify(o, null, 2)}`, s;
               s.content_block.type === "text" ? l.content.push({ type: "text", text: "" }) : s.content_block.type === "tool_use" && l.content.push({ type: "tool_use", id: s.content_block.id, name: s.content_block.name, input: "" });
             else if (s.type === "content_block_delta")
               if (s.delta.type === "text_delta") {
-                const a = s.delta.text;
-                l.content.at(-1).text += a, e.stream({ text: a });
+                const c = s.delta.text;
+                l.content.at(-1).text += c, e.stream({ text: c });
               } else s.delta.type === "input_json_delta" && (l.content.at(-1).input += s.delta.partial_json);
             else if (s.type === "content_block_stop") {
-              const a = l.content.at(-1);
-              a.input != null && (a.input = a.input ? _(a.input, {}) : {});
+              const c = l.content.at(-1);
+              c.input != null && (c.input = c.input ? S(c.input, {}) : {});
             } else if (s.type === "message_stop")
               break;
           }
         }
-        const m = l.content.filter((s) => s.type === "tool_use");
-        if (m.length && !t.signal.aborted) {
+        const u = l.content.filter((s) => s.type === "tool_use");
+        if (u.length && !t.signal.aborted) {
           o.push({ role: "assistant", content: l.content });
-          const s = await Promise.all(m.map(async (a) => {
-            const d = n.find(R("name", a.name));
-            if (e.stream && e.stream({ tool: a.name }), !d) return { tool_use_id: a.id, is_error: !0, content: "Tool not found" };
+          const s = await Promise.all(u.map(async (c) => {
+            const d = n.find(R("name", c.name));
+            if (e.stream && e.stream({ tool: c.name }), !d) return { tool_use_id: c.id, is_error: !0, content: "Tool not found" };
             try {
-              const p = await d.fn(a.input, e?.stream, this.ai);
-              return { type: "tool_result", tool_use_id: a.id, content: S(p) };
+              const p = await d.fn(c.input, e?.stream, this.ai);
+              return { type: "tool_result", tool_use_id: c.id, content: _(p) };
             } catch (p) {
-              return { type: "tool_result", tool_use_id: a.id, is_error: !0, content: p?.message || p?.toString() || "Unknown" };
+              return { type: "tool_result", tool_use_id: c.id, is_error: !0, content: p?.message || p?.toString() || "Unknown" };
             }
           }));
-          o.push({ role: "user", content: s }), u.messages = o;
+          o.push({ role: "user", content: s }), m.messages = o;
         }
-      } while (!t.signal.aborted && l.content.some((m) => m.type === "tool_use"));
-      o.push({ role: "assistant", content: l.content.filter((m) => m.type == "text").map((m) => m.text).join(`
+      } while (!t.signal.aborted && l.content.some((u) => u.type === "tool_use"));
+      o.push({ role: "assistant", content: l.content.filter((u) => u.type == "text").map((u) => u.text).join(`
 
 `) }), o = this.toStandard(o), e.stream && e.stream({ done: !0 }), e.history && e.history.splice(0, e.history.length, ...o), i(o.at(-1)?.content);
     }), { abort: () => t.abort() });
@@ -126,7 +126,7 @@ ${JSON.stringify(o, null, 2)}`, s;
 }
 class P extends L {
   constructor(r, e, t, i) {
-    super(), this.ai = r, this.host = e, this.token = t, this.model = i, this.client = new D(W({
+    super(), this.ai = r, this.host = e, this.token = t, this.model = i, this.client = new F(N({
       baseURL: e,
       apiKey: t || e ? "ignored" : void 0
     }));
@@ -140,7 +140,7 @@ class P extends L {
           role: "tool",
           id: o.id,
           name: o.function.name,
-          args: _(o.function.arguments, {}),
+          args: S(o.function.arguments, {}),
           timestamp: t.timestamp
         }));
         r.splice(e, 1, ...i), e += i.length - 1;
@@ -178,42 +178,42 @@ class P extends L {
     return Object.assign(new Promise(async (i, o) => {
       e.system && (e.history?.[0]?.role != "system" ? e.history?.splice(0, 0, { role: "system", content: e.system, timestamp: Date.now() }) : e.history[0].content = e.system);
       let n = this.fromStandard([...e.history || [], { role: "user", content: r, timestamp: Date.now() }]);
-      const u = e.tools || this.ai.options.llm?.tools || [], l = {
+      const m = e.tools || this.ai.options.llm?.tools || [], l = {
         model: e.model || this.model,
         messages: n,
         stream: !!e.stream,
         max_tokens: e.max_tokens || this.ai.options.llm?.max_tokens || 4096,
         temperature: e.temperature || this.ai.options.llm?.temperature || 0.7,
-        tools: u.map((s) => ({
+        tools: m.map((s) => ({
           type: "function",
           function: {
             name: s.name,
             description: s.description,
             parameters: {
               type: "object",
-              properties: s.args ? z(s.args, (a, d) => ({ ...d, required: void 0 })) : {},
-              required: s.args ? Object.entries(s.args).filter((a) => a[1].required).map((a) => a[0]) : []
+              properties: s.args ? z(s.args, (c, d) => ({ ...d, required: void 0 })) : {},
+              required: s.args ? Object.entries(s.args).filter((c) => c[1].required).map((c) => c[0]) : []
             }
           }
         }))
       };
-      let c, m = !0;
+      let a, u = !0;
       do {
-        if (c = await this.client.chat.completions.create(l).catch((a) => {
-          throw a.message += `
+        if (a = await this.client.chat.completions.create(l).catch((c) => {
+          throw c.message += `
 
 Messages:
-${JSON.stringify(n, null, 2)}`, a;
+${JSON.stringify(n, null, 2)}`, c;
         }), e.stream) {
-          m ? m = !1 : e.stream({ text: `
+          u ? u = !1 : e.stream({ text: `
 
-` }), c.choices = [{ message: { role: "assistant", content: "", tool_calls: [] } }];
-          for await (const a of c) {
+` }), a.choices = [{ message: { role: "assistant", content: "", tool_calls: [] } }];
+          for await (const c of a) {
             if (t.signal.aborted) break;
-            if (a.choices[0].delta.content && (c.choices[0].message.content += a.choices[0].delta.content, e.stream({ text: a.choices[0].delta.content })), a.choices[0].delta.tool_calls)
-              for (const d of a.choices[0].delta.tool_calls) {
-                const p = c.choices[0].message.tool_calls.find((f) => f.index === d.index);
-                p ? (d.id && (p.id = d.id), d.type && (p.type = d.type), d.function && (p.function || (p.function = {}), d.function.name && (p.function.name = d.function.name), d.function.arguments && (p.function.arguments = (p.function.arguments || "") + d.function.arguments))) : c.choices[0].message.tool_calls.push({
+            if (c.choices[0].delta.content && (a.choices[0].message.content += c.choices[0].delta.content, e.stream({ text: c.choices[0].delta.content })), c.choices[0].delta.tool_calls)
+              for (const d of c.choices[0].delta.tool_calls) {
+                const p = a.choices[0].message.tool_calls.find((f) => f.index === d.index);
+                p ? (d.id && (p.id = d.id), d.type && (p.type = d.type), d.function && (p.function || (p.function = {}), d.function.name && (p.function.name = d.function.name), d.function.arguments && (p.function.arguments = (p.function.arguments || "") + d.function.arguments))) : a.choices[0].message.tool_calls.push({
                   index: d.index,
                   id: d.id || "",
                   type: d.type || "function",
@@ -225,23 +225,23 @@ ${JSON.stringify(n, null, 2)}`, a;
               }
           }
         }
-        const s = c.choices[0].message.tool_calls || [];
+        const s = a.choices[0].message.tool_calls || [];
         if (s.length && !t.signal.aborted) {
-          n.push(c.choices[0].message);
-          const a = await Promise.all(s.map(async (d) => {
-            const p = u?.find(R("name", d.function.name));
+          n.push(a.choices[0].message);
+          const c = await Promise.all(s.map(async (d) => {
+            const p = m?.find(R("name", d.function.name));
             if (e.stream && e.stream({ tool: d.function.name }), !p) return { role: "tool", tool_call_id: d.id, content: '{"error": "Tool not found"}' };
             try {
-              const f = _(d.function.arguments, {}), g = await p.fn(f, e.stream, this.ai);
-              return { role: "tool", tool_call_id: d.id, content: S(g) };
+              const f = S(d.function.arguments, {}), g = await p.fn(f, e.stream, this.ai);
+              return { role: "tool", tool_call_id: d.id, content: _(g) };
             } catch (f) {
-              return { role: "tool", tool_call_id: d.id, content: S({ error: f?.message || f?.toString() || "Unknown" }) };
+              return { role: "tool", tool_call_id: d.id, content: _({ error: f?.message || f?.toString() || "Unknown" }) };
             }
           }));
-          n.push(...a), l.messages = n;
+          n.push(...c), l.messages = n;
         }
-      } while (!t.signal.aborted && c.choices?.[0]?.message?.tool_calls?.length);
-      n.push({ role: "assistant", content: c.choices[0].message.content || "" }), n = this.toStandard(n), e.stream && e.stream({ done: !0 }), e.history && e.history.splice(0, e.history.length, ...n), i(n.at(-1)?.content);
+      } while (!t.signal.aborted && a.choices?.[0]?.message?.tool_calls?.length);
+      n.push({ role: "assistant", content: a.choices[0].message.content || "" }), n = this.toStandard(n), e.stream && e.stream({ done: !0 }), e.history && e.history.splice(0, e.history.length, ...n), i(n.at(-1)?.content);
     }), { abort: () => t.abort() });
   }
 }
@@ -274,13 +274,13 @@ class X {
     };
     return Object.assign(new Promise(async (o) => {
       if (e.history || (e.history = []), e.memory) {
-        const u = async (c, m, s = 10) => {
-          const [a, d] = await Promise.all([
-            m ? this.embedding(m) : Promise.resolve(null),
-            c ? this.embedding(c) : Promise.resolve(null)
+        const m = async (a, u, s = 10) => {
+          const [c, d] = await Promise.all([
+            u ? this.embedding(u) : Promise.resolve(null),
+            a ? this.embedding(a) : Promise.resolve(null)
           ]);
           return (e.memory || []).map((p) => {
-            const f = (a ? this.cosineSimilarity(p.embeddings[0], a[0].embedding) : 0) + (d ? this.cosineSimilarity(p.embeddings[1], d[0].embedding) : 0);
+            const f = (c ? this.cosineSimilarity(p.embeddings[0], c[0].embedding) : 0) + (d ? this.cosineSimilarity(p.embeddings[1], d[0].embedding) : 0);
             return { ...p, score: f };
           }).toSorted((p, f) => p.score - f.score).slice(0, s).map((p) => `- ${p.owner}: ${p.fact}`).join(`
 `);
@@ -288,7 +288,7 @@ class X {
         e.system += `
 You have RAG memory and will be given the top_k closest memories regarding the users query. Save anything new you have learned worth remembering from the user message using the remember tool and feel free to recall memories manually.
 `;
-        const l = await u(r);
+        const l = await m(r);
         l.length && e.history.push({ role: "tool", name: "recall", id: "auto_recall_" + Math.random().toString(), args: {}, content: `Things I remembered:
 ${l}` }), e.tools = [{
           name: "recall",
@@ -298,9 +298,9 @@ ${l}` }), e.tools = [{
             query: { type: "string", description: "Search memory based on a query, can be used with or without subject argument" },
             topK: { type: "number", description: "Result limit, default 5" }
           },
-          fn: (c) => {
-            if (!c.subject && !c.query) throw new Error("Either a subject or query argument is required");
-            return u(c.query, c.subject, c.topK);
+          fn: (a) => {
+            if (!a.subject && !a.query) throw new Error("Either a subject or query argument is required");
+            return m(a.query, a.subject, a.topK);
           }
         }, {
           name: "remember",
@@ -309,20 +309,20 @@ ${l}` }), e.tools = [{
             owner: { type: "string", description: "Subject/person this fact is about" },
             fact: { type: "string", description: "The information to remember" }
           },
-          fn: async (c) => {
+          fn: async (a) => {
             if (!e.memory) return;
-            const m = await Promise.all([
-              this.embedding(c.owner),
-              this.embedding(`${c.owner}: ${c.fact}`)
-            ]), s = { owner: c.owner, fact: c.fact, embeddings: [m[0][0].embedding, m[1][0].embedding] };
-            return e.memory.splice(0, e.memory.length, ...e.memory.filter((a) => !(this.cosineSimilarity(s.embeddings[0], a.embeddings[0]) >= 0.9 && this.cosineSimilarity(s.embeddings[1], a.embeddings[1]) >= 0.8)), s), "Remembered!";
+            const u = await Promise.all([
+              this.embedding(a.owner),
+              this.embedding(`${a.owner}: ${a.fact}`)
+            ]), s = { owner: a.owner, fact: a.fact, embeddings: [u[0][0].embedding, u[1][0].embedding] };
+            return e.memory.splice(0, e.memory.length, ...e.memory.filter((c) => !(this.cosineSimilarity(s.embeddings[0], c.embeddings[0]) >= 0.9 && this.cosineSimilarity(s.embeddings[1], c.embeddings[1]) >= 0.8)), s), "Remembered!";
           }
         }, ...e.tools || []];
       }
       const n = await this.models[t].ask(r, e);
-      if (e.memory && e.history.splice(0, e.history.length, ...e.history.filter((u) => u.role != "tool" || u.name != "recall" && u.name != "remember")), e.compress) {
-        const u = await this.ai.language.compressHistory(e.history, e.compress.max, e.compress.min, e);
-        e.history.splice(0, e.history.length, ...u);
+      if (e.memory && e.history.splice(0, e.history.length, ...e.history.filter((m) => m.role != "tool" || m.name != "recall" && m.name != "remember")), e.compress) {
+        const m = await this.ai.language.compressHistory(e.history, e.compress.max, e.compress.min, e);
+        e.history.splice(0, e.history.length, ...m);
       }
       return o(n);
     }), { abort: i });
@@ -350,10 +350,10 @@ ${l}` }), e.tools = [{
       if (n += this.estimateTokens(d.content), n < t) o++;
       else break;
     if (r.length <= o) return r;
-    const u = r[0].role == "system" ? r[0] : null, l = o == 0 ? [] : r.slice(-o), c = (o == 0 ? r : r.slice(0, -o)).filter((d) => d.role === "assistant" || d.role === "user"), m = await this.summarize(c.map((d) => `[${d.role}]: ${d.content}`).join(`
+    const m = r[0].role == "system" ? r[0] : null, l = o == 0 ? [] : r.slice(-o), a = (o == 0 ? r : r.slice(0, -o)).filter((d) => d.role === "assistant" || d.role === "user"), u = await this.summarize(a.map((d) => `[${d.role}]: ${d.content}`).join(`
 
-`), 500, i), s = Date.now(), a = [{ role: "tool", name: "summary", id: "summary_" + s, args: {}, content: `Conversation Summary: ${m?.summary}`, timestamp: s }, ...l];
-    return u && a.splice(0, 0, u), a;
+`), 500, i), s = Date.now(), c = [{ role: "tool", name: "summary", id: "summary_" + s, args: {}, content: `Conversation Summary: ${u?.summary}`, timestamp: s }, ...l];
+    return m && c.splice(0, 0, m), c;
   }
   /**
    * Compare the difference between embeddings (calculates the angle between two vectors)
@@ -364,8 +364,8 @@ ${l}` }), e.tools = [{
   cosineSimilarity(r, e) {
     if (r.length !== e.length) throw new Error("Vectors must be same length");
     let t = 0, i = 0, o = 0;
-    for (let u = 0; u < r.length; u++)
-      t += r[u] * e[u], i += r[u] * r[u], o += e[u] * e[u];
+    for (let m = 0; m < r.length; m++)
+      t += r[m] * e[m], i += r[m] * r[m], o += e[m] * e[m];
     const n = Math.sqrt(i) * Math.sqrt(o);
     return n === 0 ? 0 : t / n;
   }
@@ -377,25 +377,25 @@ ${l}` }), e.tools = [{
    * @returns {string[]} Chunked strings
    */
   chunk(r, e = 500, t = 50) {
-    const i = (l, c = "") => l ? Object.entries(l).flatMap(([m, s]) => {
-      const a = c ? `${c}${isNaN(+m) ? `.${m}` : `[${m}]`}` : m;
-      return typeof s == "object" && !Array.isArray(s) ? i(s, a) : `${a}: ${Array.isArray(s) ? s.join(", ") : s}`;
+    const i = (l, a = "") => l ? Object.entries(l).flatMap(([u, s]) => {
+      const c = a ? `${a}${isNaN(+u) ? `.${u}` : `[${u}]`}` : u;
+      return typeof s == "object" && !Array.isArray(s) ? i(s, c) : `${c}: ${Array.isArray(s) ? s.join(", ") : s}`;
     }) : [], n = (typeof r == "object" ? i(r) : r.toString().split(`
 `)).flatMap((l) => [...l.split(/\s+/).filter(Boolean), `
-`]), u = [];
+`]), m = [];
     for (let l = 0; l < n.length; ) {
-      let c = "", m = l;
-      for (; m < n.length; ) {
-        const a = c + (c ? " " : "") + n[m];
-        if (this.estimateTokens(a.replace(/\s*\n\s*/g, `
-`)) > e && c) break;
-        c = a, m++;
+      let a = "", u = l;
+      for (; u < n.length; ) {
+        const c = a + (a ? " " : "") + n[u];
+        if (this.estimateTokens(c.replace(/\s*\n\s*/g, `
+`)) > e && a) break;
+        a = c, u++;
       }
-      const s = c.replace(/\s*\n\s*/g, `
+      const s = a.replace(/\s*\n\s*/g, `
 `).trim();
-      s && u.push(s), l = Math.max(m - t, m === l ? l + 1 : m);
+      s && m.push(s), l = Math.max(u - t, u === l ? l + 1 : u);
     }
-    return u;
+    return m;
   }
   /**
    * Create a vector representation of a string
@@ -407,21 +407,21 @@ ${l}` }), e.tools = [{
     let { maxTokens: t = 500, overlapTokens: i = 50 } = e, o = !1;
     const n = () => {
       o = !0;
-    }, u = (c) => new Promise((m, s) => {
+    }, m = (a) => new Promise((u, s) => {
       if (o) return s(new Error("Aborted"));
-      const a = [
-        I(G(H(import.meta.url)), "embedder.js"),
+      const c = [
+        H(G(B(import.meta.url)), "embedder.js"),
         this.ai.options.path,
         this.ai.options?.embedder || "bge-small-en-v1.5"
-      ], d = w("node", a, { stdio: ["pipe", "pipe", "ignore"] });
-      d.stdin.write(c), d.stdin.end();
+      ], d = w("node", c, { stdio: ["pipe", "pipe", "ignore"] });
+      d.stdin.write(a), d.stdin.end();
       let p = "";
       d.stdout.on("data", (f) => p += f.toString()), d.on("close", (f) => {
         if (o) return s(new Error("Aborted"));
         if (f === 0)
           try {
             const g = JSON.parse(p);
-            m(g.embedding);
+            u(g.embedding);
           } catch {
             s(new Error("Failed to parse embedding output"));
           }
@@ -429,12 +429,12 @@ ${l}` }), e.tools = [{
           s(new Error(`Embedder process exited with code ${f}`));
       }), d.on("error", s);
     }), l = (async () => {
-      const c = this.chunk(r, t, i), m = [];
-      for (let s = 0; s < c.length && !o; s++) {
-        const a = c[s], d = await u(a);
-        m.push({ index: s, embedding: d, text: a, tokens: this.estimateTokens(a) });
+      const a = this.chunk(r, t, i), u = [];
+      for (let s = 0; s < a.length && !o; s++) {
+        const c = a[s], d = await m(c);
+        u.push({ index: s, embedding: d, text: c, tokens: this.estimateTokens(c) });
       }
-      return m;
+      return u;
     })();
     return Object.assign(l, { abort: n });
   }
@@ -455,8 +455,8 @@ ${l}` }), e.tools = [{
    */
   fuzzyMatch(r, ...e) {
     if (e.length < 2) throw new Error("Requires at least 2 strings to compare");
-    const t = (n, u = 10) => n.toLowerCase().split("").map((l, c) => l.charCodeAt(0) * (c + 1) % u / u).slice(0, u), i = t(r), o = e.map((n) => t(n)).map((n) => this.cosineSimilarity(i, n));
-    return { avg: o.reduce((n, u) => n + u, 0) / o.length, max: Math.max(...o), similarities: o };
+    const t = (n, m = 10) => n.toLowerCase().split("").map((l, a) => l.charCodeAt(0) * (a + 1) % m / m).slice(0, m), i = t(r), o = e.map((n) => t(n)).map((n) => this.cosineSimilarity(i, n));
+    return { avg: o.reduce((n, m) => n + m, 0) / o.length, max: Math.max(...o), similarities: o };
   }
   /**
    * Ask a question with JSON response
@@ -466,15 +466,35 @@ ${l}` }), e.tools = [{
    * @returns {Promise<{} | {} | RegExpExecArray | null>}
    */
   async json(r, e, t) {
-    const i = await this.code(r, { ...t, system: [
-      t?.system,
-      `Only respond using JSON matching this schema:
+    let i = `Your job is to convert input to JSON. Call \`submit\` exactly once with JSON matching this schema:
 \`\`\`json
 ${e}
-\`\`\``
-    ].filter((o) => !!o).join(`
-`) });
-    return i ? _(i, {}) : null;
+\`\`\``;
+    return t?.system && (i += `
+
+` + t.system), new Promise(async (o, n) => {
+      let m = !1;
+      const l = await this.ask(r, {
+        temperature: 0.3,
+        ...t,
+        system: i,
+        tools: [{
+          name: "submit",
+          description: "Submit JSON",
+          args: { json: { type: "string", description: "Javascript parsable JSON string", required: !0 } },
+          fn: (a) => {
+            try {
+              const u = JSON.parse(a.json);
+              o(u), m = !0;
+            } catch {
+              return "Invalid JSON";
+            }
+            return "Done";
+          }
+        }, ...t?.tools || []]
+      });
+      m || n(`AI failed to create summary: ${l}`);
+    });
   }
   /**
    * Create a summary of some text
@@ -483,8 +503,25 @@ ${e}
    * @param options LLM request options
    * @returns {Promise<string>} Summary
    */
-  summarize(r, e = 500, t) {
-    return this.ask(r, { system: `Generate the shortest summary possible <= ${e} tokens. Output nothing else`, temperature: 0.3, ...t });
+  async summarize(r, e = 500, t) {
+    let i = `Your job is to summarize the users message. Call \`submit\` exactly once with the shortest summary possible that's <= ${e} tokens. Output nothing else`;
+    return t?.system && (i += `
+
+` + t.system), new Promise(async (o, n) => {
+      let m = !1;
+      const l = await this.ask(r, {
+        temperature: 0.3,
+        ...t,
+        system: i,
+        tools: [{
+          name: "submit",
+          description: "Submit summary",
+          args: { summary: { type: "string", description: "Summarization", required: !0 } },
+          fn: (a) => (m = !0, o(a.summary || null), "Done")
+        }, ...t?.tools || []]
+      });
+      m || n(`AI failed to create summary: ${l}`);
+    });
   }
 }
 class ee {
@@ -512,18 +549,18 @@ print(json.dumps(segments))
   async addPunctuation(r, e, t = 150) {
     const i = (n) => {
       if (n = n.toLowerCase().replace(/[^a-z]/g, ""), n.length <= 3) return 1;
-      const u = n.match(/[aeiouy]+/g);
-      let l = u ? u.length : 1;
+      const m = n.match(/[aeiouy]+/g);
+      let l = m ? m.length : 1;
       return n.endsWith("e") && l--, Math.max(1, l);
     };
     let o = "";
-    return r.transcription.filter((n, u) => {
+    return r.transcription.filter((n, m) => {
       let l = !1;
-      const c = r.transcription[u - 1], m = r.transcription[u + 1];
-      return !n.text && m ? (m.offsets.from = n.offsets.from, m.timestamps.from = n.offsets.from) : n.text && n.text[0] != " " && c && (c.offsets.to = n.offsets.to, c.timestamps.to = n.timestamps.to, c.text += n.text, l = !0), !!n.text && !l;
+      const a = r.transcription[m - 1], u = r.transcription[m + 1];
+      return !n.text && u ? (u.offsets.from = n.offsets.from, u.timestamps.from = n.offsets.from) : n.text && n.text[0] != " " && a && (a.offsets.to = n.offsets.to, a.timestamps.to = n.timestamps.to, a.text += n.text, l = !0), !!n.text && !l;
     }).forEach((n) => {
-      const u = /^[A-Z]/.test(n.text.trim()), l = n.offsets.to - n.offsets.from, m = i(n.text.trim()) * t;
-      u && l > m * 2 && n.text[0] == " " && (o += "."), o += n.text;
+      const m = /^[A-Z]/.test(n.text.trim()), l = n.offsets.to - n.offsets.from, u = i(n.text.trim()) * t;
+      m && l > u * 2 && n.text[0] == " " && (o += "."), o += n.text;
     }), e ? this.ai.language.ask(o, {
       system: "Remove any misplaced punctuation from the following ASR transcript using the replace tool. Avoid modifying words unless there is an obvious typo",
       temperature: 0.1,
@@ -544,15 +581,15 @@ print(json.dumps(segments))
     e.forEach((p) => {
       i.has(p.speaker) || i.set(p.speaker, ++o);
     });
-    const n = await this.addPunctuation(r, t), u = n.match(/[^.!?]+[.!?]+/g) || [n], l = r.transcription.filter((p) => p.text.trim()), c = u.map((p) => {
+    const n = await this.addPunctuation(r, t), m = n.match(/[^.!?]+[.!?]+/g) || [n], l = r.transcription.filter((p) => p.text.trim()), a = m.map((p) => {
       if (p = p.trim(), !p) return null;
       const f = p.toLowerCase().replace(/[^\w\s]/g, "").split(/\s+/), g = /* @__PURE__ */ new Map();
       f.forEach((x) => {
         const k = l.find((y) => x === y.text.trim().toLowerCase().replace(/[^\w]/g, ""));
         if (!k) return;
-        const $ = k.offsets.from / 1e3, v = e.find((y) => $ >= y.start && $ <= y.end);
-        if (v) {
-          const y = i.get(v.speaker);
+        const v = k.offsets.from / 1e3, $ = e.find((y) => v >= y.start && v <= y.end);
+        if ($) {
+          const y = i.get($.speaker);
           g.set(y, (g.get(y) || 0) + 1);
         }
       });
@@ -560,17 +597,17 @@ print(json.dumps(segments))
       return g.forEach((x, k) => {
         x > E && (E = x, T = k);
       }), { speaker: T, text: p };
-    }).filter((p) => p !== null), m = [];
-    c.forEach((p) => {
-      const f = m[m.length - 1];
-      f && f.speaker === p.speaker ? f.text += " " + p.text : m.push({ ...p });
+    }).filter((p) => p !== null), u = [];
+    a.forEach((p) => {
+      const f = u[u.length - 1];
+      f && f.speaker === p.speaker ? f.text += " " + p.text : u.push({ ...p });
     });
-    let s = m.map((p) => `[Speaker ${p.speaker}]: ${p.text}`).join(`
+    let s = u.map((p) => `[Speaker ${p.speaker}]: ${p.text}`).join(`
 `).trim();
     if (!t) return s;
-    let a = this.ai.language.chunk(s, 500, 0);
-    a.length > 4 && (a = [...a.slice(0, 3), a.at(-1)]);
-    const d = await this.ai.language.json(a.join(`
+    let c = this.ai.language.chunk(s, 500, 0);
+    c.length > 4 && (c = [...c.slice(0, 3), c.at(-1)]);
+    const d = await this.ai.language.json(c.join(`
 `), '{1: "Detected Name", 2: "Second Name"}', {
       system: "Use the following transcript to identify speakers. Only identify speakers you are positive about, dont mention speakers you are unsure about in your response",
       temperature: 0.1
@@ -580,15 +617,15 @@ print(json.dumps(segments))
   runAsr(r, e = {}) {
     let t;
     const i = new Promise((o, n) => {
-      this.downloadAsrModel(e.model).then((u) => {
+      this.downloadAsrModel(e.model).then((m) => {
         if (e.diarization) {
           let l = M.join(M.dirname(r), "transcript");
           t = w(
             this.ai.options.whisper,
-            ["-m", u, "-f", r, "-np", "-ml", "1", "-oj", "-of", l],
+            ["-m", m, "-f", r, "-np", "-ml", "1", "-oj", "-of", l],
             { stdio: ["ignore", "ignore", "pipe"] }
-          ), t.on("error", (c) => n(c)), t.on("close", async (c) => {
-            if (c === 0) {
+          ), t.on("error", (a) => n(a)), t.on("close", async (a) => {
+            if (a === 0) {
               l = await b.readFile(l + ".json", "utf-8"), b.rm(l + ".json").catch(() => {
               });
               try {
@@ -597,12 +634,12 @@ print(json.dumps(segments))
                 n(new Error("Failed to parse whisper JSON"));
               }
             } else
-              n(new Error(`Exit code ${c}`));
+              n(new Error(`Exit code ${a}`));
           });
         } else {
           let l = "";
-          t = w(this.ai.options.whisper, ["-m", u, "-f", r, "-np", "-nt"]), t.on("error", (c) => n(c)), t.stdout.on("data", (c) => l += c.toString()), t.on("close", async (c) => {
-            c === 0 ? o(l.trim() || null) : n(new Error(`Exit code ${c}`));
+          t = w(this.ai.options.whisper, ["-m", m, "-f", r, "-np", "-nt"]), t.on("error", (a) => n(a)), t.stdout.on("data", (a) => l += a.toString()), t.on("close", async (a) => {
+            a === 0 ? o(l.trim() || null) : n(new Error(`Exit code ${a}`));
           });
         }
       });
@@ -613,53 +650,53 @@ print(json.dumps(segments))
     let e = !1, t = () => {
       e = !0;
     };
-    const i = (n) => new Promise((u) => {
+    const i = (n) => new Promise((m) => {
       const l = w(n, ["-W", "ignore", "-c", "import pyannote.audio"]);
-      l.on("close", (c) => u(c === 0)), l.on("error", () => u(!1));
+      l.on("close", (a) => m(a === 0)), l.on("error", () => m(!1));
     }), o = Promise.all([
       i("python"),
       i("python3")
-    ]).then((async ([n, u]) => {
+    ]).then((async ([n, m]) => {
       if (e) return;
-      if (!n && !u) throw new Error("Pyannote is not installed: pip install pyannote.audio");
-      const l = u ? "python3" : "python";
-      return new Promise((c, m) => {
+      if (!n && !m) throw new Error("Pyannote is not installed: pip install pyannote.audio");
+      const l = m ? "python3" : "python";
+      return new Promise((a, u) => {
         if (e) return;
         let s = "";
-        const a = w(l, ["-W", "ignore", "-c", this.pyannote, r]);
-        a.stdout.on("data", (d) => s += d.toString()), a.stderr.on("data", (d) => console.error(d.toString())), a.on("close", (d) => {
+        const c = w(l, ["-W", "ignore", "-c", this.pyannote, r]);
+        c.stdout.on("data", (d) => s += d.toString()), c.stderr.on("data", (d) => console.error(d.toString())), c.on("close", (d) => {
           if (d === 0)
             try {
-              c(JSON.parse(s));
+              a(JSON.parse(s));
             } catch {
-              m(new Error("Failed to parse diarization output"));
+              u(new Error("Failed to parse diarization output"));
             }
           else
-            m(new Error(`Python process exited with code ${d}`));
-        }), a.on("error", m), t = () => a.kill("SIGTERM");
+            u(new Error(`Python process exited with code ${d}`));
+        }), c.on("error", u), t = () => c.kill("SIGTERM");
       });
     }));
     return Object.assign(o, { abort: t });
   }
   asr(r, e = {}) {
     if (!this.ai.options.whisper) throw new Error("Whisper not configured");
-    const t = A(V(A(U(), "audio-")), "converted.wav");
+    const t = A(Y(A(U(), "audio-")), "converted.wav");
     K(`ffmpeg -i "${r}" -ar 16000 -ac 1 -f wav "${t}"`, { stdio: "ignore" });
     const i = () => b.rm(q.dirname(t), { recursive: !0, force: !0 }).catch(() => {
     });
     if (!e.diarization) return this.runAsr(t, { model: e.model });
     const o = this.runAsr(t, { model: e.model, diarization: !0 }), n = this.runDiarization(t);
-    let u = !1, l = () => {
-      u = !0, o.abort(), n.abort(), i();
+    let m = !1, l = () => {
+      m = !0, o.abort(), n.abort(), i();
     };
-    const c = Promise.allSettled([o, n]).then(async ([m, s]) => {
-      if (m.status == "rejected") throw new Error(`Whisper.cpp timestamps:
-` + m.reason);
+    const a = Promise.allSettled([o, n]).then(async ([u, s]) => {
+      if (u.status == "rejected") throw new Error(`Whisper.cpp timestamps:
+` + u.reason);
       if (s.status == "rejected") throw new Error(`Pyannote:
 ` + s.reason);
-      return u || !e.diarization ? m.value : this.diarizeTranscript(m.value, s.value, e.diarization == "llm");
+      return m || !e.diarization ? u.value : this.diarizeTranscript(u.value, s.value, e.diarization == "llm");
     }).finally(() => i());
-    return Object.assign(c, { abort: l });
+    return Object.assign(a, { abort: l });
   }
   async downloadAsrModel(r = this.whisperModel) {
     if (!this.ai.options.whisper) throw new Error("Whisper not configured");
@@ -680,7 +717,7 @@ class te {
   ocr(r) {
     let e;
     const t = new Promise(async (i) => {
-      e = await Y(this.ai.options.ocr || "eng", 2, { cachePath: this.ai.options.path });
+      e = await V(this.ai.options.ocr || "eng", 2, { cachePath: this.ai.options.path });
       const { data: o } = await e.recognize(r);
       await e.terminate(), i(o.text.trim() || null);
     });
@@ -736,7 +773,7 @@ const re = () => O.platform() == "win32" ? "cmd" : j`echo $SHELL`?.split("/").po
       return { error: t?.message || t.toString() };
     }
   }
-}, _e = {
+}, Se = {
   name: "fetch",
   description: "Make HTTP request to URL",
   args: {
@@ -753,7 +790,7 @@ const re = () => O.platform() == "win32" ? "cmd" : j`echo $SHELL`?.split("/").po
     code: { type: "string", description: "CommonJS javascript", required: !0 }
   },
   fn: async (h) => {
-    const r = N(null), e = await J({ console: r }, h.code, !0).catch((t) => r.output.error.push(t));
+    const r = J(null), e = await I({ console: r }, h.code, !0).catch((t) => r.output.error.push(t));
     return { ...r.output, return: e, stdout: void 0, stderr: void 0 };
   }
 }, oe = {
@@ -763,7 +800,7 @@ const re = () => O.platform() == "win32" ? "cmd" : j`echo $SHELL`?.split("/").po
     code: { type: "string", description: "CommonJS javascript", required: !0 }
   },
   fn: async (h) => ({ result: j`python -c "${h.code}"` })
-}, Se = {
+}, _e = {
   name: "read_webpage",
   description: "Extract clean, structured content from a webpage or convert media/documents to accessible formats",
   args: {
@@ -788,8 +825,8 @@ const re = () => O.platform() == "win32" ? "cmd" : j`echo $SHELL`?.split("/").po
       const s = await e.arrayBuffer();
       if (s.byteLength > 10485760)
         return { url: h.url, type: "media", mimeType: i, error: "File too large", size: s.byteLength, maxSize: 10485760 };
-      const a = Buffer.from(s).toString("base64");
-      return { url: h.url, type: "media", mimeType: i, dataUrl: `data:${i};base64,${a}`, size: s.byteLength };
+      const c = Buffer.from(s).toString("base64");
+      return { url: h.url, type: "media", mimeType: i, dataUrl: `data:${i};base64,${c}`, size: s.byteLength };
     }
     if (i.match(/^(text\/(plain|csv|xml)|application\/(json|xml|csv|x-yaml))/) || h.url.match(/\.(txt|json|xml|csv|yaml|yml|md)$/i)) {
       const s = await e.text();
@@ -799,12 +836,12 @@ const re = () => O.platform() == "win32" ? "cmd" : j`echo $SHELL`?.split("/").po
       const s = await e.arrayBuffer();
       if (s.byteLength > 10485760)
         return { url: h.url, type: "binary", mimeType: i, error: "File too large", size: s.byteLength, maxSize: 10485760 };
-      const a = Buffer.from(s).toString("base64");
-      return { url: h.url, type: "binary", mimeType: i, dataUrl: `data:${i};base64,${a}`, size: s.byteLength };
+      const c = Buffer.from(s).toString("base64");
+      return { url: h.url, type: "binary", mimeType: i, dataUrl: `data:${i};base64,${c}`, size: s.byteLength };
     }
     const o = await e.text(), n = Z.load(o);
     n('script, style, nav, footer, header, aside, iframe, noscript, svg, [role="navigation"], [role="banner"], [role="complementary"], .ad, .ads, .advertisement, .cookie, .popup, .modal, .sidebar, .related, .comments, .social-share').remove();
-    const u = {
+    const m = {
       title: n('meta[property="og:title"]').attr("content") || n("title").text() || "",
       description: n('meta[name="description"]').attr("content") || n('meta[property="og:description"]').attr("content") || "",
       author: n('meta[name="author"]').attr("content") || "",
@@ -812,30 +849,30 @@ const re = () => O.platform() == "win32" ? "cmd" : j`echo $SHELL`?.split("/").po
       image: n('meta[property="og:image"]').attr("content") || ""
     };
     let l = "";
-    const c = ["article", "main", '[role="main"]', ".content", ".post-content", ".entry-content", ".article-content", "body"];
-    for (const s of c) {
-      const a = n(s).first();
-      if (a.length && a.text().trim().length > 200) {
-        l = a.text();
+    const a = ["article", "main", '[role="main"]', ".content", ".post-content", ".entry-content", ".article-content", "body"];
+    for (const s of a) {
+      const c = n(s).first();
+      if (c.length && c.text().trim().length > 200) {
+        l = c.text();
         break;
       }
     }
     l || (l = n("body").text()), l = l.replace(/\n\s*\n\s*\n/g, `
 
 `).replace(/[ \t]+/g, " ").trim().slice(0, 5e4);
-    let m = [];
-    return l.length < 500 && (n("a[href]").each((s, a) => {
-      const d = n(a).attr("href"), p = n(a).text().trim();
-      d && p && !d.startsWith("#") && m.push({ text: p, href: d });
-    }), m = m.slice(0, 50)), {
+    let u = [];
+    return l.length < 500 && (n("a[href]").each((s, c) => {
+      const d = n(c).attr("href"), p = n(c).text().trim();
+      d && p && !d.startsWith("#") && u.push({ text: p, href: d });
+    }), u = u.slice(0, 50)), {
       url: h.url,
       type: "html",
-      title: u.title.trim(),
-      description: u.description.trim(),
-      author: u.author.trim(),
-      published: u.published,
+      title: m.title.trim(),
+      description: m.description.trim(),
+      author: m.author.trim(),
+      published: m.published,
       content: l,
-      links: m.length ? m : void 0
+      links: u.length ? u : void 0
     };
   }
 }, je = {
@@ -850,7 +887,7 @@ const re = () => O.platform() == "win32" ? "cmd" : j`echo $SHELL`?.split("/").po
       headers: { "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)", "Accept-Language": "en-US,en;q=0.9" }
     }).then((o) => o.text());
     let e, t = /<a .*?href="(.+?)".+?<\/a>/g;
-    const i = new F();
+    const i = new D();
     for (; (e = t.exec(r)) !== null; ) {
       let o = /uddg=(.+)&amp?/.exec(decodeURIComponent(e[1]))?.[1];
       if (o && (o = decodeURIComponent(o)), o && i.add(o), i.size >= (h.length || 5)) break;
@@ -866,12 +903,12 @@ export {
   we as DateTimeTool,
   xe as DateTimeUTCTool,
   ke as ExecTool,
-  _e as FetchTool,
+  Se as FetchTool,
   se as JSTool,
   L as LLMProvider,
   P as OpenAi,
   oe as PythonTool,
-  Se as ReadWebpageTool,
+  _e as ReadWebpageTool,
   te as Vision,
   je as WebSearchTool
 };