@ztimson/ai-utils 0.1.22 → 0.2.0

package/dist/index.mjs

@@ -1,18 +1,19 @@
-import { createWorker as j } from "tesseract.js";
-import { deepCopy as T, objectMap as b, JSONAttemptParse as _, findByProp as k, JSONSanitize as y, Http as q, consoleInterceptor as v, fn as P, ASet as A } from "@ztimson/utils";
-import { Anthropic as E } from "@anthropic-ai/sdk";
-import { Ollama as M } from "ollama";
-import { OpenAI as O } from "openai";
+import { pipeline as T } from "@xenova/transformers";
+import { deepCopy as j, objectMap as b, JSONAttemptParse as _, findByProp as k, JSONSanitize as y, Http as M, consoleInterceptor as A, fn as q, ASet as v } from "@ztimson/utils";
+import { Anthropic as P } from "@anthropic-ai/sdk";
+import { Ollama as E } from "ollama";
+import { OpenAI as $ } from "openai";
+import * as w from "@tensorflow/tfjs";
+import { spawn as O } from "node:child_process";
 import S from "node:fs/promises";
 import U from "node:path";
-import * as w from "@tensorflow/tfjs";
-import { spawn as $ } from "node:child_process";
-import { $ as L, $Sync as R } from "@ztimson/node-utils";
+import { createWorker as L } from "tesseract.js";
+import { $ as R, $Sync as D } from "@ztimson/node-utils";
 class x {
 }
-class D extends x {
+class z extends x {
   constructor(t, e, n) {
-    super(), this.ai = t, this.apiToken = e, this.model = n, this.client = new E({ apiKey: e });
+    super(), this.ai = t, this.apiToken = e, this.model = n, this.client = new P({ apiKey: e });
   }
   client;
   toStandard(t) {
@@ -45,52 +46,52 @@ class D extends x {
   ask(t, e = {}) {
     const n = new AbortController(), a = new Promise(async (c, f) => {
       let i = this.fromStandard([...e.history || [], { role: "user", content: t, timestamp: Date.now() }]);
-      const l = T(i);
-      e.compress && (i = await this.ai.llm.compress(i, e.compress.max, e.compress.min, e));
+      const l = j(i);
+      e.compress && (i = await this.ai.language.compressHistory(i, e.compress.max, e.compress.min, e));
       const m = {
         model: e.model || this.model,
         max_tokens: e.max_tokens || this.ai.options.max_tokens || 4096,
         system: e.system || this.ai.options.system || "",
         temperature: e.temperature || this.ai.options.temperature || 0.7,
-        tools: (e.tools || this.ai.options.tools || []).map((o) => ({
-          name: o.name,
-          description: o.description,
+        tools: (e.tools || this.ai.options.tools || []).map((s) => ({
+          name: s.name,
+          description: s.description,
           input_schema: {
             type: "object",
-            properties: o.args ? b(o.args, (r, u) => ({ ...u, required: void 0 })) : {},
-            required: o.args ? Object.entries(o.args).filter((r) => r[1].required).map((r) => r[0]) : []
+            properties: s.args ? b(s.args, (r, u) => ({ ...u, required: void 0 })) : {},
+            required: s.args ? Object.entries(s.args).filter((r) => r[1].required).map((r) => r[0]) : []
           },
           fn: void 0
         })),
         messages: i,
         stream: !!e.stream
       };
-      let s, p = !0;
+      let o, d = !0;
       do {
-        if (s = await this.client.messages.create(m), e.stream) {
-          p ? p = !1 : e.stream({ text: `
+        if (o = await this.client.messages.create(m), e.stream) {
+          d ? d = !1 : e.stream({ text: `
 
-` }), s.content = [];
-          for await (const r of s) {
+` }), o.content = [];
+          for await (const r of o) {
             if (n.signal.aborted) break;
             if (r.type === "content_block_start")
-              r.content_block.type === "text" ? s.content.push({ type: "text", text: "" }) : r.content_block.type === "tool_use" && s.content.push({ type: "tool_use", id: r.content_block.id, name: r.content_block.name, input: "" });
+              r.content_block.type === "text" ? o.content.push({ type: "text", text: "" }) : r.content_block.type === "tool_use" && o.content.push({ type: "tool_use", id: r.content_block.id, name: r.content_block.name, input: "" });
             else if (r.type === "content_block_delta")
               if (r.delta.type === "text_delta") {
                 const u = r.delta.text;
-                s.content.at(-1).text += u, e.stream({ text: u });
-              } else r.delta.type === "input_json_delta" && (s.content.at(-1).input += r.delta.partial_json);
+                o.content.at(-1).text += u, e.stream({ text: u });
+              } else r.delta.type === "input_json_delta" && (o.content.at(-1).input += r.delta.partial_json);
             else if (r.type === "content_block_stop") {
-              const u = s.content.at(-1);
+              const u = o.content.at(-1);
               u.input != null && (u.input = u.input ? _(u.input, {}) : {});
             } else if (r.type === "message_stop")
               break;
           }
         }
-        const o = s.content.filter((r) => r.type === "tool_use");
-        if (o.length && !n.signal.aborted) {
-          i.push({ role: "assistant", content: s.content }), l.push({ role: "assistant", content: s.content });
-          const r = await Promise.all(o.map(async (u) => {
+        const s = o.content.filter((r) => r.type === "tool_use");
+        if (s.length && !n.signal.aborted) {
+          i.push({ role: "assistant", content: o.content }), l.push({ role: "assistant", content: o.content });
+          const r = await Promise.all(s.map(async (u) => {
             const h = e.tools?.find(k("name", u.name));
             if (!h) return { tool_use_id: u.id, is_error: !0, content: "Tool not found" };
             try {
@@ -102,17 +103,17 @@ class D extends x {
           }));
           i.push({ role: "user", content: r }), m.messages = i;
         }
-      } while (!n.signal.aborted && s.content.some((o) => o.type === "tool_use"));
-      e.stream && e.stream({ done: !0 }), c(this.toStandard([...i, { role: "assistant", content: s.content.filter((o) => o.type == "text").map((o) => o.text).join(`
+      } while (!n.signal.aborted && o.content.some((s) => s.type === "tool_use"));
+      e.stream && e.stream({ done: !0 }), c(this.toStandard([...i, { role: "assistant", content: o.content.filter((s) => s.type == "text").map((s) => s.text).join(`
 
 `) }]));
     });
     return Object.assign(a, { abort: () => n.abort() });
   }
 }
-class I extends x {
+class H extends x {
   constructor(t, e, n) {
-    super(), this.ai = t, this.host = e, this.model = n, this.client = new M({ host: e });
+    super(), this.ai = t, this.host = e, this.model = n, this.client = new E({ host: e });
   }
   client;
   toStandard(t) {
@@ -136,7 +137,7 @@ class I extends x {
   ask(t, e = {}) {
     const n = new AbortController(), a = new Promise(async (c, f) => {
       let i = e.system || this.ai.options.system, l = this.fromStandard([...e.history || [], { role: "user", content: t, timestamp: Date.now() }]);
-      l[0].roll == "system" && (i ? l.shift() : i = l.shift()), e.compress && (l = await this.ai.llm.compress(l, e.compress.max, e.compress.min)), e.system && l.unshift({ role: "system", content: i });
+      l[0].roll == "system" && (i ? l.shift() : i = l.shift()), e.compress && (l = await this.ai.language.compressHistory(l, e.compress.max, e.compress.min)), e.system && l.unshift({ role: "system", content: i });
       const m = {
         model: e.model || this.model,
         messages: l,
@@ -146,31 +147,31 @@ class I extends x {
           temperature: e.temperature || this.ai.options.temperature || 0.7,
           num_predict: e.max_tokens || this.ai.options.max_tokens || 4096
         },
-        tools: (e.tools || this.ai.options.tools || []).map((o) => ({
+        tools: (e.tools || this.ai.options.tools || []).map((s) => ({
           type: "function",
           function: {
-            name: o.name,
-            description: o.description,
+            name: s.name,
+            description: s.description,
             parameters: {
               type: "object",
-              properties: o.args ? b(o.args, (r, u) => ({ ...u, required: void 0 })) : {},
-              required: o.args ? Object.entries(o.args).filter((r) => r[1].required).map((r) => r[0]) : []
+              properties: s.args ? b(s.args, (r, u) => ({ ...u, required: void 0 })) : {},
+              required: s.args ? Object.entries(s.args).filter((r) => r[1].required).map((r) => r[0]) : []
             }
           }
         }))
       };
-      let s, p = !0;
+      let o, d = !0;
       do {
-        if (s = await this.client.chat(m), e.stream) {
-          p ? p = !1 : e.stream({ text: `
+        if (o = await this.client.chat(m), e.stream) {
+          d ? d = !1 : e.stream({ text: `
 
-` }), s.message = { role: "assistant", content: "", tool_calls: [] };
-          for await (const o of s)
-            if (n.signal.aborted || (o.message?.content && (s.message.content += o.message.content, e.stream({ text: o.message.content })), o.message?.tool_calls && (s.message.tool_calls = o.message.tool_calls), o.done)) break;
+` }), o.message = { role: "assistant", content: "", tool_calls: [] };
+          for await (const s of o)
+            if (n.signal.aborted || (s.message?.content && (o.message.content += s.message.content, e.stream({ text: s.message.content })), s.message?.tool_calls && (o.message.tool_calls = s.message.tool_calls), s.done)) break;
         }
-        if (s.message?.tool_calls?.length && !n.signal.aborted) {
-          l.push(s.message);
-          const o = await Promise.all(s.message.tool_calls.map(async (r) => {
+        if (o.message?.tool_calls?.length && !n.signal.aborted) {
+          l.push(o.message);
+          const s = await Promise.all(o.message.tool_calls.map(async (r) => {
             const u = (e.tools || this.ai.options.tools)?.find(k("name", r.function.name));
             if (!u) return { role: "tool", tool_name: r.function.name, content: '{"error": "Tool not found"}' };
             const h = typeof r.function.arguments == "string" ? _(r.function.arguments, {}) : r.function.arguments;
@@ -181,17 +182,17 @@ class I extends x {
               return { role: "tool", tool_name: r.function.name, args: h, content: y({ error: g?.message || g?.toString() || "Unknown" }) };
             }
           }));
-          l.push(...o), m.messages = l;
+          l.push(...s), m.messages = l;
         }
-      } while (!n.signal.aborted && s.message?.tool_calls?.length);
-      e.stream && e.stream({ done: !0 }), c(this.toStandard([...l, { role: "assistant", content: s.message?.content }]));
+      } while (!n.signal.aborted && o.message?.tool_calls?.length);
+      e.stream && e.stream({ done: !0 }), c(this.toStandard([...l, { role: "assistant", content: o.message?.content }]));
     });
     return Object.assign(a, { abort: () => n.abort() });
   }
 }
-class J extends x {
+class N extends x {
   constructor(t, e, n) {
-    super(), this.ai = t, this.apiToken = e, this.model = n, this.client = new O({ apiKey: e });
+    super(), this.ai = t, this.apiToken = e, this.model = n, this.client = new $({ apiKey: e });
   }
   client;
   toStandard(t) {
@@ -238,41 +239,41 @@ class J extends x {
   ask(t, e = {}) {
     const n = new AbortController(), a = new Promise(async (c, f) => {
       let i = this.fromStandard([...e.history || [], { role: "user", content: t, timestamp: Date.now() }]);
-      e.compress && (i = await this.ai.llm.compress(i, e.compress.max, e.compress.min, e));
+      e.compress && (i = await this.ai.language.compressHistory(i, e.compress.max, e.compress.min, e));
       const l = {
         model: e.model || this.model,
         messages: i,
         stream: !!e.stream,
         max_tokens: e.max_tokens || this.ai.options.max_tokens || 4096,
         temperature: e.temperature || this.ai.options.temperature || 0.7,
-        tools: (e.tools || this.ai.options.tools || []).map((p) => ({
+        tools: (e.tools || this.ai.options.tools || []).map((d) => ({
           type: "function",
           function: {
-            name: p.name,
-            description: p.description,
+            name: d.name,
+            description: d.description,
             parameters: {
               type: "object",
-              properties: p.args ? b(p.args, (o, r) => ({ ...r, required: void 0 })) : {},
-              required: p.args ? Object.entries(p.args).filter((o) => o[1].required).map((o) => o[0]) : []
+              properties: d.args ? b(d.args, (s, r) => ({ ...r, required: void 0 })) : {},
+              required: d.args ? Object.entries(d.args).filter((s) => s[1].required).map((s) => s[0]) : []
             }
           }
         }))
       };
-      let m, s = !0;
+      let m, o = !0;
       do {
         if (m = await this.client.chat.completions.create(l), e.stream) {
-          s ? s = !1 : e.stream({ text: `
+          o ? o = !1 : e.stream({ text: `
 
 ` }), m.choices = [{ message: { content: "", tool_calls: [] } }];
-          for await (const o of m) {
+          for await (const s of m) {
             if (n.signal.aborted) break;
-            o.choices[0].delta.content && (m.choices[0].message.content += o.choices[0].delta.content, e.stream({ text: o.choices[0].delta.content })), o.choices[0].delta.tool_calls && (m.choices[0].message.tool_calls = o.choices[0].delta.tool_calls);
+            s.choices[0].delta.content && (m.choices[0].message.content += s.choices[0].delta.content, e.stream({ text: s.choices[0].delta.content })), s.choices[0].delta.tool_calls && (m.choices[0].message.tool_calls = s.choices[0].delta.tool_calls);
           }
         }
-        const p = m.choices[0].message.tool_calls || [];
-        if (p.length && !n.signal.aborted) {
+        const d = m.choices[0].message.tool_calls || [];
+        if (d.length && !n.signal.aborted) {
           i.push(m.choices[0].message);
-          const o = await Promise.all(p.map(async (r) => {
+          const s = await Promise.all(d.map(async (r) => {
             const u = e.tools?.find(k("name", r.function.name));
             if (!u) return { role: "tool", tool_call_id: r.id, content: '{"error": "Tool not found"}' };
             try {
@@ -282,7 +283,7 @@ class J extends x {
               return { role: "tool", tool_call_id: r.id, content: y({ error: h?.message || h?.toString() || "Unknown" }) };
             }
           }));
-          i.push(...o), l.messages = i;
+          i.push(...s), l.messages = i;
         }
       } while (!n.signal.aborted && m.choices?.[0]?.message?.tool_calls?.length);
       e.stream && e.stream({ done: !0 }), c(this.toStandard([...i, { role: "assistant", content: m.choices[0].message.content || "" }]));
@@ -290,10 +291,11 @@ class J extends x {
     return Object.assign(a, { abort: () => n.abort() });
   }
 }
-class W {
-  constructor(t, e) {
-    this.ai = t, this.options = e, e.anthropic?.token && (this.providers.anthropic = new D(this.ai, e.anthropic.token, e.anthropic.model)), e.ollama?.host && (this.providers.ollama = new I(this.ai, e.ollama.host, e.ollama.model)), e.openAi?.token && (this.providers.openAi = new J(this.ai, e.openAi.token, e.openAi.model));
+class I {
+  constructor(t) {
+    this.ai = t, this.embedModel = T("feature-extraction", "Xenova/all-MiniLM-L6-v2"), t.options.anthropic?.token && (this.providers.anthropic = new z(this.ai, t.options.anthropic.token, t.options.anthropic.model)), t.options.ollama?.host && (this.providers.ollama = new H(this.ai, t.options.ollama.host, t.options.ollama.model)), t.options.openAi?.token && (this.providers.openAi = new N(this.ai, t.options.openAi.token, t.options.openAi.model));
   }
+  embedModel;
   providers = {};
   /**
    * Chat with LLM
@@ -303,7 +305,7 @@ class W {
    */
   ask(t, e = {}) {
     let n = [null, null];
-    if (e.model && (typeof e.model == "object" ? n = e.model : n = [e.model, this.options[e.model]?.model]), (!e.model || n[1] == null) && (typeof this.options.model == "object" ? n = this.options.model : n = [this.options.model, this.options[this.options.model]?.model]), !n[0] || !n[1]) throw new Error(`Unknown LLM provider or model: ${n[0]} / ${n[1]}`);
+    if (e.model && (typeof e.model == "object" ? n = e.model : n = [e.model, this.ai.options[e.model]?.model]), (!e.model || n[1] == null) && (typeof this.ai.options.model == "object" ? n = this.ai.options.model : n = [this.ai.options.model, this.ai.options[this.ai.options.model]?.model]), !n[0] || !n[1]) throw new Error(`Unknown LLM provider or model: ${n[0]} / ${n[1]}`);
     return this.providers[n[0]].ask(t, { ...e, model: n[1] });
   }
   /**
@@ -314,18 +316,50 @@ class W {
    * @param {LLMRequest} options LLM options
    * @returns {Promise<LLMMessage[]>} New chat history will summary at index 0
    */
-  async compress(t, e, n, a) {
+  async compressHistory(t, e, n, a) {
     if (this.estimateTokens(t) < e) return t;
     let c = 0, f = 0;
-    for (let s of t.toReversed())
-      if (f += this.estimateTokens(s.content), f < n) c++;
+    for (let o of t.toReversed())
+      if (f += this.estimateTokens(o.content), f < n) c++;
       else break;
     if (t.length <= c) return t;
-    const i = c == 0 ? [] : t.slice(-c), l = (c == 0 ? t : t.slice(0, -c)).filter((s) => s.role === "assistant" || s.role === "user");
-    return [{ role: "assistant", content: `Conversation Summary: ${await this.summarize(l.map((s) => `${s.role}: ${s.content}`).join(`
+    const i = c == 0 ? [] : t.slice(-c), l = (c == 0 ? t : t.slice(0, -c)).filter((o) => o.role === "assistant" || o.role === "user");
+    return [{ role: "assistant", content: `Conversation Summary: ${await this.summarize(l.map((o) => `${o.role}: ${o.content}`).join(`
 
 `), 250, a)}`, timestamp: Date.now() }, ...i];
   }
+  embedding(t, e = 500, n = 50) {
+    const a = (o, d = "") => o == null ? [] : Object.entries(o).flatMap(([s, r]) => {
+      const u = d ? `${d}${isNaN(+s) ? `.${s}` : `[${s}]`}` : s;
+      if (typeof r == "object" && r !== null && !Array.isArray(r)) return a(r, u);
+      const h = Array.isArray(r) ? r.join(", ") : String(r);
+      return `${u}: ${h}`;
+    }), c = async (o) => {
+      const s = await (await this.embedModel)(o, { pooling: "mean", normalize: !0 });
+      return Array.from(s.data);
+    }, i = (typeof t == "object" ? a(t) : t.split(`
+`)).flatMap((o) => [...o.split(/\s+/).filter((d) => d.trim()), `
+`]), l = [];
+    let m = 0;
+    for (; m < i.length; ) {
+      let o = m, d = "";
+      for (; o < i.length; ) {
+        const r = i[o], u = d + (d ? " " : "") + r;
+        if (this.estimateTokens(u.replace(/\s*\n\s*/g, `
+`)) > e && d) break;
+        d = u, o++;
+      }
+      const s = d.replace(/\s*\n\s*/g, `
+`).trim();
+      s && l.push(s), m = o - n, m <= o - i.length + o && (m = o);
+    }
+    return Promise.all(l.map(async (o, d) => ({
+      index: d,
+      embedding: await c(o),
+      text: o,
+      tokens: this.estimateTokens(o)
+    })));
+  }
   /**
    * Estimate variable as tokens
    * @param history Object to size
@@ -335,6 +369,21 @@ class W {
     const e = JSON.stringify(t);
     return Math.ceil(e.length / 4 * 1.2);
   }
+  /**
+   * Compare the difference between two strings using tensor math
+   * @param target Text that will checked
+   * @param {string} searchTerms Multiple search terms to check against target
+   * @returns {{avg: number, max: number, similarities: number[]}} Similarity values 0-1: 0 = unique, 1 = identical
+   */
+  fuzzyMatch(t, ...e) {
+    if (e.length < 2) throw new Error("Requires at least 2 strings to compare");
+    const n = (i, l = 10) => i.toLowerCase().split("").map((m, o) => m.charCodeAt(0) * (o + 1) % l / l).slice(0, l), a = (i, l) => {
+      if (i.length !== l.length) throw new Error("Vectors must be same length");
+      const m = w.tensor1d(i), o = w.tensor1d(l), d = w.dot(m, o), s = w.norm(m), r = w.norm(o);
+      return s.dataSync()[0] === 0 || r.dataSync()[0] === 0 ? 0 : d.dataSync()[0] / (s.dataSync()[0] * r.dataSync()[0]);
+    }, c = n(t), f = e.map((i) => n(i)).map((i) => a(c, i));
+    return { avg: f.reduce((i, l) => i + l, 0) / f.length, max: Math.max(...f), similarities: f };
+  }
   /**
    * Ask a question with JSON response
    * @param {string} message Question
@@ -359,14 +408,12 @@ class W {
     return this.ask(t, { system: `Generate a brief summary <= ${e} tokens. Output nothing else`, temperature: 0.3, ...n }).then((a) => a.pop()?.content || null);
   }
 }
-class Z {
+class J {
   constructor(t) {
-    this.options = t, this.llm = new W(this, t), this.options.whisper?.binary && (this.whisperModel = this.options.whisper?.model.endsWith(".bin") ? this.options.whisper?.model : this.options.whisper?.model + ".bin", this.downloadAsrModel());
+    this.ai = t, t.options.whisper?.binary && (this.whisperModel = t.options.whisper?.model.endsWith(".bin") ? t.options.whisper?.model : t.options.whisper?.model + ".bin", this.downloadAsrModel());
   }
   downloads = {};
   whisperModel;
-  /** Large Language Models */
-  llm;
   /**
    * Convert audio to text using Auditory Speech Recognition
    * @param {string} path Path to audio
@@ -374,15 +421,15 @@ class Z {
    * @returns {Promise<any>} Extracted text
    */
   asr(t, e = this.whisperModel) {
-    if (!this.options.whisper?.binary) throw new Error("Whisper not configured");
+    if (!this.ai.options.whisper?.binary) throw new Error("Whisper not configured");
     let n = () => {
     };
     return { response: new Promise((c, f) => {
       this.downloadAsrModel(e).then((i) => {
         let l = "";
-        const m = $(this.options.whisper?.binary, ["-nt", "-np", "-m", i, "-f", t], { stdio: ["ignore", "pipe", "ignore"] });
-        n = () => m.kill("SIGTERM"), m.on("error", (s) => f(s)), m.stdout.on("data", (s) => l += s.toString()), m.on("close", (s) => {
-          s === 0 ? c(l.trim() || null) : f(new Error(`Exit code ${s}`));
+        const m = O(this.ai.options.whisper?.binary, ["-nt", "-np", "-m", i, "-f", t], { stdio: ["ignore", "pipe", "ignore"] });
+        n = () => m.kill("SIGTERM"), m.on("error", (o) => f(o)), m.stdout.on("data", (o) => l += o.toString()), m.on("close", (o) => {
+          o === 0 ? c(l.trim() || null) : f(new Error(`Exit code ${o}`));
         });
       });
     }), abort: n };
@@ -394,11 +441,16 @@ class Z {
    * @return {Promise<string>} Absolute path to model file, resolves once downloaded
    */
   async downloadAsrModel(t = this.whisperModel) {
-    if (!this.options.whisper?.binary) throw new Error("Whisper not configured");
+    if (!this.ai.options.whisper?.binary) throw new Error("Whisper not configured");
     t.endsWith(".bin") || (t += ".bin");
-    const e = U.join(this.options.whisper.path, t);
+    const e = U.join(this.ai.options.whisper.path, t);
     return await S.stat(e).then(() => !0).catch(() => !1) ? e : this.downloads[t] ? this.downloads[t] : (this.downloads[t] = fetch(`https://huggingface.co/ggerganov/whisper.cpp/resolve/main/${t}`).then((n) => n.arrayBuffer()).then((n) => Buffer.from(n)).then(async (n) => (await S.writeFile(e, n), delete this.downloads[t], e)), this.downloads[t]);
   }
+}
+class W {
+  constructor(t) {
+    this.ai = t;
+  }
   /**
    * Convert image to text using Optical Character Recognition
    * @param {string} path Path to image
@@ -411,60 +463,58 @@ class Z {
         e?.terminate();
       },
       response: new Promise(async (n) => {
-        e = await j("eng");
+        e = await L("eng");
         const { data: a } = await e.recognize(t);
         await e.terminate(), n(a.text.trim() || null);
       })
     };
   }
-  /**
-   * Compare the difference between two strings using tensor math
-   * @param target Text that will checked
-   * @param {string} searchTerms Multiple search terms to check against target
-   * @returns {{avg: number, max: number, similarities: number[]}} Similarity values 0-1: 0 = unique, 1 = identical
-   */
-  semanticSimilarity(t, ...e) {
-    if (e.length < 2) throw new Error("Requires at least 2 strings to compare");
-    const n = (i, l = 10) => i.toLowerCase().split("").map((m, s) => m.charCodeAt(0) * (s + 1) % l / l).slice(0, l), a = (i, l) => {
-      if (i.length !== l.length) throw new Error("Vectors must be same length");
-      const m = w.tensor1d(i), s = w.tensor1d(l), p = w.dot(m, s), o = w.norm(m), r = w.norm(s);
-      return o.dataSync()[0] === 0 || r.dataSync()[0] === 0 ? 0 : p.dataSync()[0] / (o.dataSync()[0] * r.dataSync()[0]);
-    }, c = n(t), f = e.map((i) => n(i)).map((i) => a(c, i));
-    return { avg: f.reduce((i, l) => i + l, 0) / f.length, max: Math.max(...f), similarities: f };
+}
+class oe {
+  constructor(t) {
+    this.options = t, this.audio = new J(this), this.language = new I(this), this.vision = new W(this);
   }
+  downloads = {};
+  whisperModel;
+  /** Audio processing AI */
+  audio;
+  /** Language processing AI */
+  language;
+  /** Vision processing AI */
+  vision;
 }
-const N = {
+const G = {
   name: "cli",
   description: "Use the command line interface, returns any output",
   args: { command: { type: "string", description: "Command to run", required: !0 } },
-  fn: (d) => L`${d.command}`
-}, ee = {
+  fn: (p) => R`${p.command}`
+}, se = {
   name: "get_datetime",
   description: "Get current date and time",
   args: {},
   fn: async () => (/* @__PURE__ */ new Date()).toISOString()
-}, te = {
+}, re = {
   name: "exec",
   description: "Run code/scripts",
   args: {
     language: { type: "string", description: "Execution language", enum: ["cli", "node", "python"], required: !0 },
     code: { type: "string", description: "Code to execute", required: !0 }
   },
-  fn: async (d, t) => {
+  fn: async (p, t) => {
     try {
-      switch (d.type) {
+      switch (p.type) {
         case "bash":
-          return await N.fn({ command: d.code }, t);
+          return await G.fn({ command: p.code }, t);
         case "node":
-          return await z.fn({ code: d.code }, t);
+          return await F.fn({ code: p.code }, t);
         case "python":
-          return await G.fn({ code: d.code }, t);
+          return await B.fn({ code: p.code }, t);
       }
     } catch (e) {
       return { error: e?.message || e.toString() };
     }
   }
-}, ne = {
+}, ae = {
   name: "fetch",
   description: "Make HTTP request to URL",
   args: {
@@ -473,54 +523,54 @@ const N = {
     headers: { type: "object", description: "HTTP headers to send", default: {} },
     body: { type: "object", description: "HTTP body to send" }
   },
-  fn: (d) => new q({ url: d.url, headers: d.headers }).request({ method: d.method || "GET", body: d.body })
-}, z = {
+  fn: (p) => new M({ url: p.url, headers: p.headers }).request({ method: p.method || "GET", body: p.body })
+}, F = {
   name: "exec_javascript",
   description: "Execute commonjs javascript",
   args: {
     code: { type: "string", description: "CommonJS javascript", required: !0 }
   },
-  fn: async (d) => {
-    const t = v(null), e = await P({ console: t }, d.code, !0).catch((n) => t.output.error.push(n));
+  fn: async (p) => {
+    const t = A(null), e = await q({ console: t }, p.code, !0).catch((n) => t.output.error.push(n));
     return { ...t.output, return: e, stdout: void 0, stderr: void 0 };
   }
-}, G = {
+}, B = {
   name: "exec_javascript",
   description: "Execute commonjs javascript",
   args: {
     code: { type: "string", description: "CommonJS javascript", required: !0 }
   },
-  fn: async (d) => ({ result: R`python -c "${d.code}"` })
-}, se = {
+  fn: async (p) => ({ result: D`python -c "${p.code}"` })
+}, ie = {
   name: "search",
   description: "Use a search engine to find relevant URLs, should be changed with fetch to scrape sources",
   args: {
     query: { type: "string", description: "Search string", required: !0 },
     length: { type: "string", description: "Number of results to return", default: 5 }
   },
-  fn: async (d) => {
-    const t = await fetch(`https://html.duckduckgo.com/html/?q=${encodeURIComponent(d.query)}`, {
+  fn: async (p) => {
+    const t = await fetch(`https://html.duckduckgo.com/html/?q=${encodeURIComponent(p.query)}`, {
       headers: { "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)", "Accept-Language": "en-US,en;q=0.9" }
     }).then((c) => c.text());
     let e, n = /<a .*?href="(.+?)".+?<\/a>/g;
-    const a = new A();
+    const a = new v();
     for (; (e = n.exec(t)) !== null; ) {
       let c = /uddg=(.+)&amp?/.exec(decodeURIComponent(e[1]))?.[1];
-      if (c && (c = decodeURIComponent(c)), c && a.add(c), a.size >= (d.length || 5)) break;
+      if (c && (c = decodeURIComponent(c)), c && a.add(c), a.size >= (p.length || 5)) break;
     }
     return a;
   }
 };
 export {
-  Z as Ai,
-  D as Anthropic,
-  N as CliTool,
-  ee as DateTimeTool,
-  te as ExecTool,
-  ne as FetchTool,
-  z as JSTool,
-  W as LLM,
-  G as PythonTool,
-  se as SearchTool
+  oe as Ai,
+  z as Anthropic,
+  G as CliTool,
+  se as DateTimeTool,
+  re as ExecTool,
+  ae as FetchTool,
+  F as JSTool,
+  I as LLM,
+  B as PythonTool,
+  ie as SearchTool
 };
 //# sourceMappingURL=index.mjs.map