@ztimson/ai-utils 0.8.5 → 0.8.7

package/dist/index.mjs

@@ -1,23 +1,23 @@
-import * as z from "node:os";
-import { tmpdir as U } from "node:os";
-import { objectMap as A, JSONAttemptParse as _, findByProp as O, JSONSanitize as S, clean as W, Http as N, consoleInterceptor as L, fn as C, ASet as J } from "@ztimson/utils";
-import { Anthropic as F } from "@anthropic-ai/sdk";
-import { OpenAI as H } from "openai";
-import { fileURLToPath as D } from "url";
-import { join as G, dirname as I } from "path";
-import { spawn as b, execSync as B } from "node:child_process";
-import { mkdtempSync as K } from "node:fs";
+import * as O from "node:os";
+import { tmpdir as L } from "node:os";
+import { objectMap as R, JSONAttemptParse as _, findByProp as U, JSONSanitize as S, clean as W, Http as N, consoleInterceptor as C, fn as J, ASet as F } from "@ztimson/utils";
+import { Anthropic as H } from "@anthropic-ai/sdk";
+import { OpenAI as D } from "openai";
+import { fileURLToPath as G } from "url";
+import { join as I, dirname as B } from "path";
+import { spawn as b, execSync as K } from "node:child_process";
+import { mkdtempSync as V } from "node:fs";
 import w from "node:fs/promises";
-import * as v from "node:path";
-import M, { join as P } from "node:path";
-import { createWorker as V } from "tesseract.js";
-import * as Y from "cheerio";
-import { $ as Z, $Sync as Q } from "@ztimson/node-utils";
-class R {
+import * as M from "node:path";
+import P, { join as q } from "node:path";
+import { createWorker as Y } from "tesseract.js";
+import * as Z from "cheerio";
+import { $Sync as j } from "@ztimson/node-utils";
+class z {
 }
-class X extends R {
+class Q extends z {
   constructor(r, e, t) {
-    super(), this.ai = r, this.apiToken = e, this.model = t, this.client = new F({ apiKey: e });
+    super(), this.ai = r, this.apiToken = e, this.model = t, this.client = new H({ apiKey: e });
   }
   client;
   toStandard(r) {
@@ -67,7 +67,7 @@ class X extends R {
           description: m.description,
           input_schema: {
             type: "object",
-            properties: m.args ? A(m.args, (o, l) => ({ ...l, required: void 0 })) : {},
+            properties: m.args ? R(m.args, (o, l) => ({ ...l, required: void 0 })) : {},
             required: m.args ? Object.entries(m.args).filter((o) => o[1].required).map((o) => o[0]) : []
           },
           fn: void 0
@@ -106,7 +106,7 @@ ${JSON.stringify(s, null, 2)}`, o;
         if (m.length && !t.signal.aborted) {
           s.push({ role: "assistant", content: u.content });
           const o = await Promise.all(m.map(async (l) => {
-            const d = n.find(O("name", l.name));
+            const d = n.find(U("name", l.name));
             if (e.stream && e.stream({ tool: l.name }), !d) return { tool_use_id: l.id, is_error: !0, content: "Tool not found" };
             try {
               const p = await d.fn(l.input, e?.stream, this.ai);
@@ -124,9 +124,9 @@ ${JSON.stringify(s, null, 2)}`, o;
     }), { abort: () => t.abort() });
   }
 }
-class q extends R {
+class A extends z {
   constructor(r, e, t, a) {
-    super(), this.ai = r, this.host = e, this.token = t, this.model = a, this.client = new H(W({
+    super(), this.ai = r, this.host = e, this.token = t, this.model = a, this.client = new D(W({
       baseURL: e,
       apiKey: t || e ? "ignored" : void 0
     }));
@@ -191,7 +191,7 @@ class q extends R {
             description: o.description,
             parameters: {
               type: "object",
-              properties: o.args ? A(o.args, (l, d) => ({ ...d, required: void 0 })) : {},
+              properties: o.args ? R(o.args, (l, d) => ({ ...d, required: void 0 })) : {},
               required: o.args ? Object.entries(o.args).filter((l) => l[1].required).map((l) => l[0]) : []
             }
           }
@@ -229,11 +229,11 @@ ${JSON.stringify(n, null, 2)}`, l;
         if (o.length && !t.signal.aborted) {
           n.push(i.choices[0].message);
           const l = await Promise.all(o.map(async (d) => {
-            const p = c?.find(O("name", d.function.name));
+            const p = c?.find(U("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) };
+              return console.log(g), { role: "tool", tool_call_id: d.id, content: S(g) };
             } catch (f) {
               return { role: "tool", tool_call_id: d.id, content: S({ error: f?.message || f?.toString() || "Unknown" }) };
             }
@@ -245,10 +245,10 @@ ${JSON.stringify(n, null, 2)}`, l;
     }), { abort: () => t.abort() });
   }
 }
-class ee {
+class X {
   constructor(r) {
     this.ai = r, r.options.llm?.models && Object.entries(r.options.llm.models).forEach(([e, t]) => {
-      this.defaultModel || (this.defaultModel = e), t.proto == "anthropic" ? this.models[e] = new X(this.ai, t.token, e) : t.proto == "ollama" ? this.models[e] = new q(this.ai, t.host, "not-needed", e) : t.proto == "openai" && (this.models[e] = new q(this.ai, t.host || null, t.token, e));
+      this.defaultModel || (this.defaultModel = e), t.proto == "anthropic" ? this.models[e] = new Q(this.ai, t.token, e) : t.proto == "ollama" ? this.models[e] = new A(this.ai, t.host, "not-needed", e) : t.proto == "openai" && (this.models[e] = new A(this.ai, t.host || null, t.token, e));
     });
   }
   defaultModel;
@@ -410,7 +410,7 @@ You have RAG memory and will be given the top_k closest memories regarding the u
     }, c = (i) => new Promise((m, o) => {
       if (s) return o(new Error("Aborted"));
       const l = [
-        G(I(D(import.meta.url)), "embedder.js"),
+        I(B(G(import.meta.url)), "embedder.js"),
         this.ai.options.path,
         this.ai.options?.embedder || "bge-small-en-v1.5"
       ], d = b("node", l, { stdio: ["pipe", "pipe", "ignore"] });
@@ -487,7 +487,7 @@ ${e}
     return this.ask(r, { system: `Generate the shortest summary possible <= ${e} tokens. Output nothing else`, temperature: 0.3, ...t });
   }
 }
-class te {
+class ee {
   constructor(r) {
     this.ai = r, r.options.whisper && (this.whisperModel = r.options.asr || "ggml-base.en.bin", this.downloadAsrModel()), this.pyannote = `
 import sys
@@ -550,16 +550,16 @@ print(json.dumps(segments))
       f.forEach((k) => {
         const x = u.find((y) => k === y.text.trim().toLowerCase().replace(/[^\w]/g, ""));
         if (!x) return;
-        const E = x.offsets.from / 1e3, $ = e.find((y) => E >= y.start && E <= y.end);
-        if ($) {
-          const y = a.get($.speaker);
+        const $ = x.offsets.from / 1e3, v = e.find((y) => $ >= y.start && $ <= y.end);
+        if (v) {
+          const y = a.get(v.speaker);
           g.set(y, (g.get(y) || 0) + 1);
         }
       });
-      let j = 1, T = 0;
+      let T = 1, E = 0;
       return g.forEach((k, x) => {
-        k > T && (T = k, j = x);
-      }), { speaker: j, text: p };
+        k > E && (E = k, T = x);
+      }), { speaker: T, text: p };
     }).filter((p) => p !== null), m = [];
     i.forEach((p) => {
       const f = m[m.length - 1];
@@ -582,7 +582,7 @@ print(json.dumps(segments))
     const a = new Promise((s, n) => {
       this.downloadAsrModel(e.model).then((c) => {
         if (e.diarization) {
-          let u = v.join(v.dirname(r), "transcript");
+          let u = M.join(M.dirname(r), "transcript");
           t = b(
             this.ai.options.whisper,
             ["-m", c, "-f", r, "-np", "-ml", "1", "-oj", "-of", u],
@@ -643,9 +643,9 @@ print(json.dumps(segments))
   }
   asr(r, e = {}) {
     if (!this.ai.options.whisper) throw new Error("Whisper not configured");
-    const t = P(K(P(U(), "audio-")), "converted.wav");
-    B(`ffmpeg -i "${r}" -ar 16000 -ac 1 -f wav "${t}"`, { stdio: "ignore" });
-    const a = () => w.rm(M.dirname(t), { recursive: !0, force: !0 }).catch(() => {
+    const t = q(V(q(L(), "audio-")), "converted.wav");
+    K(`ffmpeg -i "${r}" -ar 16000 -ac 1 -f wav "${t}"`, { stdio: "ignore" });
+    const a = () => w.rm(P.dirname(t), { recursive: !0, force: !0 }).catch(() => {
     });
     if (!e.diarization) return this.runAsr(t, { model: e.model });
     const s = this.runAsr(t, { model: e.model, diarization: !0 }), n = this.runDiarization(t);
@@ -664,11 +664,11 @@ print(json.dumps(segments))
   async downloadAsrModel(r = this.whisperModel) {
     if (!this.ai.options.whisper) throw new Error("Whisper not configured");
     r.endsWith(".bin") || (r += ".bin");
-    const e = M.join(this.ai.options.path, r);
+    const e = P.join(this.ai.options.path, r);
     return await w.stat(e).then(() => !0).catch(() => !1) ? e : this.downloads[r] ? this.downloads[r] : (this.downloads[r] = fetch(`https://huggingface.co/ggerganov/whisper.cpp/resolve/main/${r}`).then((t) => t.arrayBuffer()).then((t) => Buffer.from(t)).then(async (t) => (await w.writeFile(e, t), delete this.downloads[r], e)), this.downloads[r]);
   }
 }
-class re {
+class te {
   constructor(r) {
     this.ai = r;
   }
@@ -680,7 +680,7 @@ class re {
   ocr(r) {
     let e;
     const t = new Promise(async (a) => {
-      e = await V(this.ai.options.ocr || "eng", 2, { cachePath: this.ai.options.path });
+      e = await Y(this.ai.options.ocr || "eng", 2, { cachePath: this.ai.options.path });
       const { data: s } = await e.recognize(r);
       await e.terminate(), a(s.text.trim() || null);
     });
@@ -689,7 +689,7 @@ class re {
 }
 class we {
   constructor(r) {
-    this.options = r, r.path || (r.path = z.tmpdir()), process.env.TRANSFORMERS_CACHE = r.path, this.audio = new te(this), this.language = new ee(this), this.vision = new re(this);
+    this.options = r, r.path || (r.path = O.tmpdir()), process.env.TRANSFORMERS_CACHE = r.path, this.audio = new ee(this), this.language = new X(this), this.vision = new te(this);
   }
   /** Audio processing AI */
   audio;
@@ -698,11 +698,11 @@ class we {
   /** Vision processing AI */
   vision;
 }
-const ne = {
+const re = () => O.platform() == "win32" ? "cmd" : j`echo $SHELL`?.split("/").pop() || "bash", ne = {
   name: "cli",
   description: "Use the command line interface, returns any output",
   args: { command: { type: "string", description: "Command to run", required: !0 } },
-  fn: (h) => Z`${h.command}`
+  fn: (h) => j`${h.command}`
 }, be = {
   name: "get_datetime",
   description: "Get current UTC date / time",
@@ -712,18 +712,20 @@ const ne = {
   name: "exec",
   description: "Run code/scripts",
   args: {
-    language: { type: "string", description: "Execution language", enum: ["cli", "node", "python"], required: !0 },
+    language: { type: "string", description: `Execution language (CLI: ${re()})`, enum: ["cli", "node", "python"], required: !0 },
     code: { type: "string", description: "Code to execute", required: !0 }
   },
   fn: async (h, r, e) => {
     try {
-      switch (h.type) {
-        case "bash":
+      switch (h.language) {
+        case "cli":
           return await ne.fn({ command: h.code }, r, e);
         case "node":
           return await se.fn({ code: h.code }, r, e);
         case "python":
           return await oe.fn({ code: h.code }, r, e);
+        default:
+          throw new Error(`Unsupported language: ${h.language}`);
       }
     } catch (t) {
       return { error: t?.message || t.toString() };
@@ -746,7 +748,8 @@ const ne = {
     code: { type: "string", description: "CommonJS javascript", required: !0 }
   },
   fn: async (h) => {
-    const r = L(null), e = await C({ console: r }, h.code, !0).catch((t) => r.output.error.push(t));
+    console.log("executing js");
+    const r = C(null), e = await J({ console: r }, h.code, !0).catch((t) => r.output.error.push(t));
     return { ...r.output, return: e, stdout: void 0, stderr: void 0 };
   }
 }, oe = {
@@ -755,7 +758,7 @@ const ne = {
   args: {
     code: { type: "string", description: "CommonJS javascript", required: !0 }
   },
-  fn: async (h) => ({ result: Q`python -c "${h.code}"` })
+  fn: async (h) => ({ result: j`python -c "${h.code}"` })
 }, _e = {
   name: "read_webpage",
   description: "Extract clean, structured content from a webpage. Use after web_search to read specific URLs",
@@ -766,7 +769,7 @@ const ne = {
   fn: async (h) => {
     const r = await fetch(h.url, { headers: { "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)" } }).then((n) => n.text()).catch((n) => {
       throw new Error(`Failed to fetch: ${n.message}`);
-    }), e = Y.load(r);
+    }), e = Z.load(r);
     e('script, style, nav, footer, header, aside, iframe, noscript, [role="navigation"], [role="banner"], .ad, .ads, .cookie, .popup').remove();
     const t = {
       title: e('meta[property="og:title"]').attr("content") || e("title").text() || "",
@@ -795,7 +798,7 @@ const ne = {
       headers: { "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)", "Accept-Language": "en-US,en;q=0.9" }
     }).then((s) => s.text());
     let e, t = /<a .*?href="(.+?)".+?<\/a>/g;
-    const a = new J();
+    const a = new F();
     for (; (e = t.exec(r)) !== null; ) {
       let s = /uddg=(.+)&amp?/.exec(decodeURIComponent(e[1]))?.[1];
       if (s && (s = decodeURIComponent(s)), s && a.add(s), a.size >= (h.length || 5)) break;
@@ -805,18 +808,18 @@ const ne = {
 };
 export {
   we as Ai,
-  X as Anthropic,
-  te as Audio,
+  Q as Anthropic,
+  ee as Audio,
   ne as CliTool,
   be as DateTimeTool,
   ke as ExecTool,
   xe as FetchTool,
   se as JSTool,
-  R as LLMProvider,
-  q as OpenAi,
+  z as LLMProvider,
+  A as OpenAi,
   oe as PythonTool,
   _e as ReadWebpageTool,
-  re as Vision,
+  te as Vision,
   Se as WebSearchTool
 };
 //# sourceMappingURL=index.mjs.map