@okx_ai/okx-trade-mcp 1.3.8-beta.3 → 1.3.8-beta.4

package/dist/index.js

@@ -2374,7 +2374,7 @@ function registerIndicatorTools() {
           params: {
             type: "array",
             items: { type: "number" },
-            description: "Indicator-specific parameters as a number array. Examples: MA/EMA [14] (period), MACD [12,26,9] (fast,slow,signal), BB [20,2] (period,stdDev), RSI [14] (period). For period-based indicators a parameter is required; the CLI applies a sensible default when omitted."
+            description: "Indicator-specific parameters as a number array. Examples: MA/EMA [14] (period), MACD [12,26,9] (fast,slow,signal), BB [20,2] (period,stdDev), RSI [14] (period). Omit to use server defaults."
           },
           returnList: {
             type: "boolean",
@@ -2480,6 +2480,203 @@ var MODULES = [
   "skills"
 ];
 var DEFAULT_MODULES = ["spot", "swap", "option", "account", ...BOT_DEFAULT_SUB_MODULES, "skills"];
+var AGGREGATE_ENDPOINT = "/api/v5/aigc/forward/balance-aggregate";
+var TRADING_ENDPOINT = "/api/v5/account/balance";
+var FUNDING_ENDPOINT = "/api/v5/asset/balances";
+var VALUATION_ENDPOINT = "/api/v5/asset/asset-valuation";
+function parseParams(rawArgs) {
+  const args = asRecord(rawArgs);
+  const accounts = readString(args, "accounts") ?? "trading,funding";
+  return {
+    ccy: readString(args, "ccy"),
+    accounts,
+    requestedAccounts: accounts.split(",").map((s) => s.trim().toLowerCase()).filter(Boolean),
+    showValuation: readBoolean(args, "showValuation") ?? true,
+    valuationCcy: readString(args, "valuationCcy") ?? "USDT",
+    preferParallel: readBoolean(args, "preferParallel") ?? false
+  };
+}
+function normalizeSubError(error) {
+  if (!error || typeof error !== "object") {
+    return void 0;
+  }
+  const e = error;
+  return {
+    code: e["code"] === void 0 || e["code"] === null ? "UNKNOWN" : String(e["code"]),
+    msg: typeof e["msg"] === "string" ? e["msg"] : String(e["msg"] ?? "")
+  };
+}
+function toIsoTimestamp(value, fallbackMs) {
+  if (typeof value === "number" && Number.isFinite(value)) {
+    return new Date(value).toISOString();
+  }
+  if (typeof value === "string" && value.trim() !== "") {
+    return value;
+  }
+  return new Date(fallbackMs).toISOString();
+}
+function normalizeAggregate(agg, params, startTime) {
+  const result = {};
+  const trading = agg["trading"];
+  if (trading) {
+    result.trading = compactObject({
+      available: trading["available"] ?? false,
+      totalEq: trading["totalEq"],
+      adjEq: trading["adjEq"],
+      details: Array.isArray(trading["details"]) ? trading["details"] : [],
+      error: normalizeSubError(trading["error"])
+    });
+  }
+  const funding = agg["funding"];
+  if (funding) {
+    result.funding = compactObject({
+      available: funding["available"] ?? false,
+      details: Array.isArray(funding["details"]) ? funding["details"] : [],
+      error: normalizeSubError(funding["error"])
+    });
+  }
+  const valuation = agg["valuation"];
+  if (valuation) {
+    result.valuation = compactObject({
+      available: valuation["available"] ?? false,
+      valuationCcy: valuation["valuationCcy"] ?? params.valuationCcy,
+      totalBal: valuation["totalBal"] ?? "0",
+      details: Array.isArray(valuation["details"]) ? valuation["details"] : [],
+      error: normalizeSubError(valuation["error"])
+    });
+  }
+  const meta = agg["meta"] ?? {};
+  result.meta = compactObject({
+    requestedAt: toIsoTimestamp(meta["requestedAt"], startTime),
+    elapsedMs: typeof meta["elapsedMs"] === "number" ? meta["elapsedMs"] : Date.now() - startTime,
+    partialFailure: Boolean(meta["partialFailure"]),
+    site: typeof meta["site"] === "string" ? meta["site"] : void 0,
+    source: "aggregate"
+  });
+  return result;
+}
+async function queryAggregate(context, params, startTime) {
+  const resp = await context.client.privateGet(
+    AGGREGATE_ENDPOINT,
+    compactObject({
+      ccy: params.ccy,
+      accounts: params.accounts,
+      showValuation: params.showValuation,
+      valuationCcy: params.valuationCcy
+    }),
+    privateRateLimit("account_get_balance_all", 5)
+  );
+  const data = resp.data;
+  const agg = Array.isArray(data) ? data[0] : void 0;
+  if (!agg || agg["trading"] === void 0 && agg["funding"] === void 0) {
+    throw new OkxApiError("Aggregate endpoint returned no usable data", {
+      code: "AGG_EMPTY",
+      endpoint: `GET ${AGGREGATE_ENDPOINT}`
+    });
+  }
+  return normalizeAggregate(agg, params, startTime);
+}
+var REQUESTED_KEYS = /* @__PURE__ */ new Set(["trading", "funding"]);
+function buildBalanceTasks(context, params) {
+  const { ccy, requestedAccounts, showValuation, valuationCcy } = params;
+  const tasks = [];
+  if (requestedAccounts.includes("trading")) {
+    tasks.push({
+      key: "trading",
+      run: () => context.client.privateGet(TRADING_ENDPOINT, compactObject({ ccy }), privateRateLimit("account_get_balance", 10)).then((resp) => resp.data)
+    });
+  }
+  if (requestedAccounts.includes("funding")) {
+    tasks.push({
+      key: "funding",
+      run: () => context.client.privateGet(FUNDING_ENDPOINT, compactObject({ ccy }), privateRateLimit("account_get_asset_balance", 6)).then((resp) => resp.data)
+    });
+  }
+  if (showValuation) {
+    tasks.push({
+      key: "valuation",
+      run: () => context.client.privateGet(VALUATION_ENDPOINT, { ccy: valuationCcy }, privateRateLimit("account_get_asset_valuation", 1)).then((resp) => resp.data)
+    });
+  }
+  return tasks;
+}
+async function settleBalanceTasks(tasks) {
+  return Promise.all(
+    tasks.map(async (task) => {
+      try {
+        return { key: task.key, data: await task.run() };
+      } catch (error) {
+        return { key: task.key, error };
+      }
+    })
+  );
+}
+function buildSuccessSection(key, data, valuationCcy) {
+  const rows = Array.isArray(data) ? data : [];
+  const first = rows[0];
+  if (key === "funding") {
+    return { available: true, details: rows };
+  }
+  if (key === "valuation") {
+    return { available: true, valuationCcy, totalBal: first?.["totalBal"] ?? "0", details: rows };
+  }
+  return {
+    available: true,
+    totalEq: first?.["totalEq"] ?? "0",
+    adjEq: first?.["adjEq"] ?? "0",
+    details: first && Array.isArray(first["details"]) ? first["details"] : []
+  };
+}
+function buildErrorSection(error) {
+  return {
+    available: false,
+    error: {
+      code: error instanceof OkxApiError ? error.code ?? "UNKNOWN" : "UNKNOWN",
+      msg: error.message
+    }
+  };
+}
+async function queryParallel(context, params, startTime) {
+  const tasks = buildBalanceTasks(context, params);
+  const outcomes = await settleBalanceTasks(tasks);
+  const authFailure = outcomes.find((o) => o.error instanceof AuthenticationError);
+  if (authFailure?.error) {
+    throw authFailure.error;
+  }
+  const result = {};
+  for (const outcome of outcomes) {
+    result[outcome.key] = outcome.error ? buildErrorSection(outcome.error) : buildSuccessSection(outcome.key, outcome.data, params.valuationCcy);
+  }
+  const requestedCount = tasks.filter((t) => REQUESTED_KEYS.has(t.key)).length;
+  const requestedFailures = outcomes.filter((o) => o.error && REQUESTED_KEYS.has(o.key)).length;
+  if (requestedCount > 0 && requestedFailures >= requestedCount) {
+    throw new OkxApiError("Both balance queries failed", { code: "-30001" });
+  }
+  const site = typeof context.config?.site === "string" ? context.config.site : void 0;
+  result.meta = compactObject({
+    requestedAt: new Date(startTime).toISOString(),
+    elapsedMs: Date.now() - startTime,
+    partialFailure: requestedFailures > 0,
+    site,
+    source: "fallback"
+  });
+  return result;
+}
+async function buildBalanceAll(rawArgs, context) {
+  const params = parseParams(rawArgs);
+  const startTime = Date.now();
+  if (params.preferParallel) {
+    return queryParallel(context, params, startTime);
+  }
+  try {
+    return await queryAggregate(context, params, startTime);
+  } catch (error) {
+    if (error instanceof AuthenticationError) {
+      throw error;
+    }
+    return queryParallel(context, params, startTime);
+  }
+}
 function registerAccountTools() {
   return [
     {
@@ -3059,6 +3256,39 @@ function registerAccountTools() {
         );
         return normalizeResponse(response);
       }
+    },
+    {
+      name: "account_get_balance_all",
+      title: "Get Aggregate Balance Snapshot",
+      module: "account",
+      description: "One-shot snapshot of all OKX assets: trading-account equity + funding-account balances + optional cross-account valuation. Use when the user asks for total assets / net worth / \u603B\u8D44\u4EA7 / all balances. Prefer this over calling account_get_balance + account_get_asset_balance separately. Returns {trading, funding, valuation, meta} with per-section error so partial failures don't block the rest.",
+      isWrite: false,
+      inputSchema: {
+        type: "object",
+        properties: {
+          ccy: {
+            type: "string",
+            description: "Filter by currency, comma-separated (e.g. BTC,ETH). Omit for all. Applied to trading + funding queries only."
+          },
+          accounts: {
+            type: "string",
+            description: "Comma-separated accounts to query: 'trading','funding'. Default 'trading,funding'."
+          },
+          showValuation: {
+            type: "boolean",
+            description: "Include cross-account asset valuation. Default true."
+          },
+          valuationCcy: {
+            type: "string",
+            description: "Currency for valuation (e.g. USDT, BTC). Default USDT. Only effective when showValuation=true."
+          },
+          preferParallel: {
+            type: "boolean",
+            description: "Skip the server-side aggregate endpoint and query trading/funding/valuation directly in parallel. Default false. Use when you need the per-account valuation breakdown or a non-USD valuation currency."
+          }
+        }
+      },
+      handler: (rawArgs, context) => buildBalanceAll(rawArgs, context)
     }
   ];
 }
@@ -12397,7 +12627,7 @@ var _require = createRequire(import.meta.url);
 var pkg = _require("../package.json");
 var SERVER_NAME = "okx-trade-mcp";
 var SERVER_VERSION = pkg.version;
-var GIT_HASH = true ? "52f6524c" : "dev";
+var GIT_HASH = true ? "9beb3ee8" : "dev";
 
 // src/server.ts
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";