@mindstudio-ai/agent 0.1.71 → 0.1.72

package/dist/cli.js

@@ -1375,6 +1375,131 @@ var init_context = __esm({
   }
 });
 
+// src/db-ws.ts
+function getOrOpen(url) {
+  if (socket && socket.readyState === WS_OPEN) {
+    return Promise.resolve(socket);
+  }
+  if (opening) {
+    return opening;
+  }
+  const WebSocketCtor = globalThis.WebSocket;
+  if (!WebSocketCtor) {
+    return Promise.reject(
+      new DbWsTransportError("global WebSocket unavailable")
+    );
+  }
+  opening = new Promise((resolve, reject) => {
+    let ws;
+    const fail = (err) => {
+      clearTimeout(openTimer);
+      if (socket === ws) {
+        socket = null;
+      }
+      opening = null;
+      for (const [, p] of pending) {
+        clearTimeout(p.timer);
+        p.reject(err);
+      }
+      pending.clear();
+      reject(err);
+    };
+    const openTimer = setTimeout(() => {
+      try {
+        ws?.close();
+      } catch {
+      }
+      fail(new DbWsTransportError("ws open timed out"));
+    }, WS_OPEN_TIMEOUT_MS);
+    try {
+      ws = new WebSocketCtor(url);
+    } catch (err) {
+      fail(new DbWsTransportError(err?.message || "ws construct failed"));
+      return;
+    }
+    ws.onopen = () => {
+      clearTimeout(openTimer);
+      socket = ws;
+      opening = null;
+      resolve(ws);
+    };
+    ws.onmessage = (ev) => {
+      let msg;
+      try {
+        msg = JSON.parse(
+          typeof ev.data === "string" ? ev.data : String(ev.data)
+        );
+      } catch {
+        return;
+      }
+      if (!msg || msg.t !== "db-result" || msg.id == null) {
+        return;
+      }
+      const p = pending.get(msg.id);
+      if (!p) {
+        return;
+      }
+      clearTimeout(p.timer);
+      pending.delete(msg.id);
+      if (msg.error) {
+        p.reject(
+          new MindStudioError(
+            `[db] ${msg.error.message || "Database query failed"}`,
+            msg.error.code || "db_query_error",
+            400,
+            msg.error
+          )
+        );
+      } else {
+        p.resolve(msg.results);
+      }
+    };
+    ws.onerror = () => fail(new DbWsTransportError("ws error"));
+    ws.onclose = () => fail(new DbWsTransportError("ws closed"));
+  });
+  return opening;
+}
+async function executeDbBatchOverWs(url, token, databaseId, queries) {
+  const id = ++idCounter;
+  const frame = JSON.stringify({ t: "db", id, token, databaseId, queries });
+  if (frame.length > WS_MAX_FRAME_CHARS) {
+    throw new DbWsTransportError("payload too large for ws");
+  }
+  const ws = await getOrOpen(url);
+  return await new Promise((resolve, reject) => {
+    const timer = setTimeout(() => {
+      pending.delete(id);
+      reject(new DbWsTransportError("ws execute timed out"));
+    }, WS_REQUEST_TIMEOUT_MS);
+    pending.set(id, { resolve, reject, timer });
+    try {
+      ws.send(frame);
+    } catch (err) {
+      clearTimeout(timer);
+      pending.delete(id);
+      reject(new DbWsTransportError(err?.message || "ws send failed"));
+    }
+  });
+}
+var WS_REQUEST_TIMEOUT_MS, WS_OPEN_TIMEOUT_MS, WS_OPEN, WS_MAX_FRAME_CHARS, DbWsTransportError, socket, opening, idCounter, pending;
+var init_db_ws = __esm({
+  "src/db-ws.ts"() {
+    "use strict";
+    init_errors();
+    WS_REQUEST_TIMEOUT_MS = 3e5;
+    WS_OPEN_TIMEOUT_MS = 5e3;
+    WS_OPEN = 1;
+    WS_MAX_FRAME_CHARS = 9e5;
+    DbWsTransportError = class extends Error {
+      name = "DbWsTransportError";
+    };
+    socket = null;
+    opening = null;
+    idCounter = 0;
+    pending = /* @__PURE__ */ new Map();
+  }
+});
+
 // src/rate-limit.ts
 var DEFAULTS, RateLimiter;
 var init_rate_limit = __esm({
@@ -3939,6 +4064,7 @@ var init_client = __esm({
     init_http();
     init_errors();
     await init_context();
+    init_db_ws();
     init_rate_limit();
     init_config();
     init_auth();
@@ -4850,6 +4976,21 @@ var init_client = __esm({
        * enabling RETURNING clauses and multi-statement batches.
        */
       async _executeDbBatch(databaseId, queries) {
+        const dbWsUrl = typeof process !== "undefined" ? process.env?.DB_WS_URL : void 0;
+        if (dbWsUrl) {
+          try {
+            return await executeDbBatchOverWs(
+              dbWsUrl,
+              this._token,
+              databaseId,
+              queries
+            );
+          } catch (err) {
+            if (!(err instanceof DbWsTransportError)) {
+              throw err;
+            }
+          }
+        }
         const url = `${this._currentHttpConfig.baseUrl}/_internal/v2/db/query`;
         const res = await fetch(url, {
           method: "POST",
@@ -6068,7 +6209,7 @@ async function startMcpServer(options) {
           capabilities: { tools: {} },
           serverInfo: {
             name: "mindstudio-agent",
-            version: "0.1.71"
+            version: "0.1.72"
           },
           instructions: 'Welcome to MindStudio \u2014 a platform with 200+ AI models, 850+ third-party integrations, and pre-built agents.\n\nGetting started:\n1. Call `ask` with any question about the SDK \u2014 it knows every action, model, and connector and returns working code with real model IDs and config options. Examples: ask("generate an image with FLUX"), ask("what models support vision?"), ask("how do I send a Slack message?").\n2. Call `changeName` to set your display name \u2014 use your name or whatever your user calls you. This is how you\'ll appear in MindStudio request logs.\n3. If you have a profile picture or icon, call `uploadFile` to upload it, then `changeProfilePicture` with the returned URL.\n4. For manual browsing, call `listActions` to discover all available actions.\n\nThen use the tools to generate text, images, video, audio, search the web, work with data sources, run agents, and more.\n\nImportant:\n- AI-powered actions (text generation, image generation, video, audio, etc.) cost money. Before running these, call `estimateActionCost` and confirm with the user before proceeding \u2014 unless they\'ve explicitly told you to go ahead.\n- Not all agents from `listAgents` are configured for API use. Do not try to run an agent just because it appears in the list \u2014 it will likely fail. Only run agents the user specifically asks you to run.'
         });
@@ -7041,7 +7182,7 @@ function isNewerVersion(current, latest) {
   return false;
 }
 async function checkForUpdate() {
-  const currentVersion = "0.1.71";
+  const currentVersion = "0.1.72";
   if (!currentVersion) return null;
   try {
     const { loadConfig: loadConfig2, saveConfig: saveConfig2 } = await Promise.resolve().then(() => (init_config(), config_exports));
@@ -7070,7 +7211,7 @@ async function checkForUpdate() {
   }
 }
 function printUpdateNotice(latestVersion) {
-  const currentVersion = "0.1.71";
+  const currentVersion = "0.1.72";
   process.stderr.write(
     `
   ${ansi2.cyanBright("Update available")} ${ansi2.gray(currentVersion + " \u2192")} ${ansi2.cyanBold(latestVersion)}
@@ -7083,7 +7224,7 @@ function isStandaloneBinary() {
   return !argv1.includes("node_modules");
 }
 async function cmdUpdate() {
-  const currentVersion = "0.1.71";
+  const currentVersion = "0.1.72";
   process.stderr.write(
     `  ${ansi2.gray("Current version:")} ${currentVersion}
 `
@@ -7218,7 +7359,7 @@ async function cmdLogin(options) {
   process.stderr.write("\n");
   printLogo();
   process.stderr.write("\n");
-  const ver = "0.1.71";
+  const ver = "0.1.72";
   process.stderr.write(
     `  ${ansi2.bold("MindStudio Agent")} ${ver ? " " + ansi2.gray("v" + ver) : ""}
 `
@@ -7545,7 +7686,7 @@ async function main() {
   try {
     if (command === "version" || command === "-v") {
       process.stdout.write(
-        "0.1.71\n"
+        "0.1.72\n"
       );
       return;
     }

package/dist/index.js

@@ -101,6 +101,124 @@ function runWithContext(ctx, fn) {
   return als.run(ctx, fn);
 }
 
+// src/db-ws.ts
+var WS_REQUEST_TIMEOUT_MS = 3e5;
+var WS_OPEN_TIMEOUT_MS = 5e3;
+var WS_OPEN = 1;
+var WS_MAX_FRAME_CHARS = 9e5;
+var DbWsTransportError = class extends Error {
+  name = "DbWsTransportError";
+};
+var socket = null;
+var opening = null;
+var idCounter = 0;
+var pending = /* @__PURE__ */ new Map();
+function getOrOpen(url) {
+  if (socket && socket.readyState === WS_OPEN) {
+    return Promise.resolve(socket);
+  }
+  if (opening) {
+    return opening;
+  }
+  const WebSocketCtor = globalThis.WebSocket;
+  if (!WebSocketCtor) {
+    return Promise.reject(
+      new DbWsTransportError("global WebSocket unavailable")
+    );
+  }
+  opening = new Promise((resolve, reject) => {
+    let ws;
+    const fail = (err) => {
+      clearTimeout(openTimer);
+      if (socket === ws) {
+        socket = null;
+      }
+      opening = null;
+      for (const [, p] of pending) {
+        clearTimeout(p.timer);
+        p.reject(err);
+      }
+      pending.clear();
+      reject(err);
+    };
+    const openTimer = setTimeout(() => {
+      try {
+        ws?.close();
+      } catch {
+      }
+      fail(new DbWsTransportError("ws open timed out"));
+    }, WS_OPEN_TIMEOUT_MS);
+    try {
+      ws = new WebSocketCtor(url);
+    } catch (err) {
+      fail(new DbWsTransportError(err?.message || "ws construct failed"));
+      return;
+    }
+    ws.onopen = () => {
+      clearTimeout(openTimer);
+      socket = ws;
+      opening = null;
+      resolve(ws);
+    };
+    ws.onmessage = (ev) => {
+      let msg;
+      try {
+        msg = JSON.parse(
+          typeof ev.data === "string" ? ev.data : String(ev.data)
+        );
+      } catch {
+        return;
+      }
+      if (!msg || msg.t !== "db-result" || msg.id == null) {
+        return;
+      }
+      const p = pending.get(msg.id);
+      if (!p) {
+        return;
+      }
+      clearTimeout(p.timer);
+      pending.delete(msg.id);
+      if (msg.error) {
+        p.reject(
+          new MindStudioError(
+            `[db] ${msg.error.message || "Database query failed"}`,
+            msg.error.code || "db_query_error",
+            400,
+            msg.error
+          )
+        );
+      } else {
+        p.resolve(msg.results);
+      }
+    };
+    ws.onerror = () => fail(new DbWsTransportError("ws error"));
+    ws.onclose = () => fail(new DbWsTransportError("ws closed"));
+  });
+  return opening;
+}
+async function executeDbBatchOverWs(url, token, databaseId, queries) {
+  const id = ++idCounter;
+  const frame = JSON.stringify({ t: "db", id, token, databaseId, queries });
+  if (frame.length > WS_MAX_FRAME_CHARS) {
+    throw new DbWsTransportError("payload too large for ws");
+  }
+  const ws = await getOrOpen(url);
+  return await new Promise((resolve, reject) => {
+    const timer = setTimeout(() => {
+      pending.delete(id);
+      reject(new DbWsTransportError("ws execute timed out"));
+    }, WS_REQUEST_TIMEOUT_MS);
+    pending.set(id, { resolve, reject, timer });
+    try {
+      ws.send(frame);
+    } catch (err) {
+      clearTimeout(timer);
+      pending.delete(id);
+      reject(new DbWsTransportError(err?.message || "ws send failed"));
+    }
+  });
+}
+
 // src/rate-limit.ts
 var DEFAULTS = {
   internal: { concurrency: 10, callCap: 500 },
@@ -4668,6 +4786,21 @@ var MindStudioAgent = class {
    * enabling RETURNING clauses and multi-statement batches.
    */
   async _executeDbBatch(databaseId, queries) {
+    const dbWsUrl = typeof process !== "undefined" ? process.env?.DB_WS_URL : void 0;
+    if (dbWsUrl) {
+      try {
+        return await executeDbBatchOverWs(
+          dbWsUrl,
+          this._token,
+          databaseId,
+          queries
+        );
+      } catch (err) {
+        if (!(err instanceof DbWsTransportError)) {
+          throw err;
+        }
+      }
+    }
     const url = `${this._currentHttpConfig.baseUrl}/_internal/v2/db/query`;
     const res = await fetch(url, {
       method: "POST",