@okx_ai/okx-trade-mcp 1.3.0-beta.4 → 1.3.0

package/dist/index.js

@@ -5,32 +5,20 @@ import { parseArgs } from "util";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 
 // ../core/dist/index.js
-import { Agent, ProxyAgent } from "undici";
-import { execFile } from "child_process";
-import { homedir } from "os";
-import { join } from "path";
-import {
-  readFileSync,
-  writeFileSync,
-  mkdirSync,
-  unlinkSync,
-  renameSync
-} from "fs";
-import { homedir as homedir2 } from "os";
-import { join as join2, dirname } from "path";
+import { ProxyAgent } from "undici";
 import { createHmac } from "crypto";
 import fs from "fs";
 import path from "path";
 import os from "os";
-import { writeFileSync as writeFileSync2, renameSync as renameSync2, unlinkSync as unlinkSync2, mkdirSync as mkdirSync2 } from "fs";
-import { join as join3, resolve, basename, sep } from "path";
+import { writeFileSync, renameSync, unlinkSync, mkdirSync } from "fs";
+import { join, resolve, basename, sep } from "path";
 import { randomUUID } from "crypto";
 import yauzl from "yauzl";
-import { join as join5, dirname as dirname3 } from "path";
-import { homedir as homedir3 } from "os";
-import { readFileSync as readFileSync4, writeFileSync as writeFileSync4, mkdirSync as mkdirSync5, existsSync as existsSync3 } from "fs";
-import { join as join6, dirname as dirname4 } from "path";
-import { homedir as homedir4 } from "os";
+import { join as join3, dirname as dirname2 } from "path";
+import { homedir } from "os";
+import { readFileSync as readFileSync3, writeFileSync as writeFileSync3, mkdirSync as mkdirSync4, existsSync as existsSync3 } from "fs";
+import { join as join4, dirname as dirname3 } from "path";
+import { homedir as homedir2 } from "os";
 
 // ../../node_modules/.pnpm/smol-toml@1.6.0/node_modules/smol-toml/dist/error.js
 function getLineColFromPtr(string, ptr) {
@@ -718,9 +706,9 @@ function parse(toml, { maxDepth = 1e3, integersAsBigInt } = {}) {
 }
 
 // ../core/dist/index.js
-import { readFileSync as readFileSync5, writeFileSync as writeFileSync5, mkdirSync as mkdirSync6, existsSync as existsSync4 } from "fs";
-import { join as join7 } from "path";
-import { homedir as homedir5 } from "os";
+import { readFileSync as readFileSync4, writeFileSync as writeFileSync4, mkdirSync as mkdirSync5, existsSync as existsSync4 } from "fs";
+import { join as join5 } from "path";
+import { homedir as homedir3 } from "os";
 import fs2 from "fs";
 import path2 from "path";
 import os2 from "os";
@@ -728,123 +716,6 @@ import * as fs3 from "fs";
 import * as path3 from "path";
 import * as os3 from "os";
 import { execFileSync } from "child_process";
-var EXEC_TIMEOUT_MS = 3e4;
-var DOH_BIN_DIR = join(homedir(), ".okx", "bin");
-function getDohBinaryPath() {
-  if (process.env.OKX_DOH_BINARY_PATH) {
-    return process.env.OKX_DOH_BINARY_PATH;
-  }
-  const ext = process.platform === "win32" ? ".exe" : "";
-  return join(DOH_BIN_DIR, `okx-doh-resolver${ext}`);
-}
-function execDohBinary(domain, exclude = [], userAgent) {
-  const binPath = getDohBinaryPath();
-  const args = ["--domain", domain];
-  if (exclude.length > 0) {
-    args.push("--exclude", exclude.join(","));
-  }
-  if (userAgent) {
-    args.push("--user-agent", userAgent);
-  }
-  return new Promise((resolve3) => {
-    execFile(
-      binPath,
-      args,
-      { timeout: EXEC_TIMEOUT_MS, encoding: "utf-8" },
-      (error, stdout) => {
-        if (error) {
-          resolve3(null);
-          return;
-        }
-        try {
-          const result = JSON.parse(stdout);
-          if (result.code === 0 && result.data) {
-            resolve3(result.data);
-          } else {
-            resolve3(null);
-          }
-        } catch {
-          resolve3(null);
-        }
-      }
-    );
-  });
-}
-var DOH_CACHE_PATH = join2(homedir2(), ".okx", "doh-node-cache.json");
-function readCache(hostname, cachePath = DOH_CACHE_PATH) {
-  try {
-    const raw = readFileSync(cachePath, "utf-8");
-    const file = JSON.parse(raw);
-    return file[hostname] ?? null;
-  } catch {
-    return null;
-  }
-}
-function writeCache(hostname, entry, cachePath = DOH_CACHE_PATH) {
-  try {
-    const dir = dirname(cachePath);
-    mkdirSync(dir, { recursive: true });
-    let file = {};
-    try {
-      file = JSON.parse(readFileSync(cachePath, "utf-8"));
-    } catch {
-    }
-    file[hostname] = entry;
-    const tmpPath = `${cachePath}.tmp`;
-    writeFileSync(tmpPath, JSON.stringify(file));
-    renameSync(tmpPath, cachePath);
-  } catch {
-  }
-}
-var FAILED_NODE_TTL_MS = 60 * 60 * 1e3;
-function classifyAndCache(node, hostname, failedNodes, cachePath) {
-  if (!node) {
-    return { mode: null, node: null };
-  }
-  if (node.ip === hostname || node.host === hostname) {
-    writeCache(hostname, {
-      mode: "direct",
-      node: null,
-      failedNodes,
-      updatedAt: Date.now()
-    }, cachePath);
-    return { mode: "direct", node: null };
-  }
-  writeCache(hostname, {
-    mode: "proxy",
-    node,
-    failedNodes,
-    updatedAt: Date.now()
-  }, cachePath);
-  return { mode: "proxy", node };
-}
-function getActiveFailedNodes(nodes) {
-  if (!nodes || nodes.length === 0) return [];
-  const now = Date.now();
-  return nodes.filter((n) => now - n.failedAt < FAILED_NODE_TTL_MS);
-}
-function resolveDoh(hostname, cachePath) {
-  const entry = readCache(hostname, cachePath);
-  if (entry) {
-    if (entry.mode === "direct") {
-      return { mode: "direct", node: null };
-    }
-    if (entry.mode === "proxy" && entry.node) {
-      return { mode: "proxy", node: entry.node };
-    }
-  }
-  return { mode: null, node: null };
-}
-async function reResolveDoh(hostname, failedIp, userAgent, cachePath) {
-  const entry = readCache(hostname, cachePath);
-  const active = getActiveFailedNodes(entry?.failedNodes);
-  const now = Date.now();
-  const alreadyFailed = failedIp && active.some((n) => n.ip === failedIp);
-  const failedNodes = failedIp && !alreadyFailed ? [...active, { ip: failedIp, failedAt: now }] : active;
-  const excludeIps = failedNodes.map((n) => n.ip);
-  const node = await execDohBinary(hostname, excludeIps, userAgent);
-  return classifyAndCache(node, hostname, failedNodes, cachePath);
-}
 function getNow() {
   return (/* @__PURE__ */ new Date()).toISOString();
 }
@@ -1063,14 +934,6 @@ var OkxRestClient = class _OkxRestClient {
   config;
   rateLimiter;
   dispatcher;
-  // DoH proxy state (lazy-resolved on first request)
-  dohResolved = false;
-  dohRetried = false;
-  directUnverified = false;
-  // The first direct connection has not yet been verified
-  dohNode = null;
-  dohAgent = null;
-  dohBaseUrl = null;
   constructor(config) {
     this.config = config;
     this.rateLimiter = new RateLimiter(3e4, config.verbose);
@@ -1078,98 +941,6 @@ var OkxRestClient = class _OkxRestClient {
       this.dispatcher = new ProxyAgent(config.proxyUrl);
     }
   }
-  /**
-   * Lazily resolve the DoH proxy node on the first request.
-   * Uses cache-first strategy via the resolver.
-   */
-  ensureDoh() {
-    if (this.dohResolved || this.dispatcher) return;
-    this.dohResolved = true;
-    try {
-      const { hostname, protocol } = new URL(this.config.baseUrl);
-      const result = resolveDoh(hostname);
-      if (!result.mode) {
-        this.directUnverified = true;
-        if (this.config.verbose) {
-          vlog("DoH: no cache, trying direct connection first");
-        }
-        return;
-      }
-      if (result.mode === "direct") {
-        if (this.config.verbose) {
-          vlog("DoH: mode=direct (overseas or cached), using direct connection");
-        }
-        return;
-      }
-      if (result.node) {
-        this.applyDohNode(result.node, protocol);
-      }
-    } catch (err) {
-      if (this.config.verbose) {
-        const cause = err instanceof Error ? err.message : String(err);
-        vlog(`DoH resolution failed, falling back to direct: ${cause}`);
-      }
-    }
-  }
-  /** Apply a DoH node: set up the custom Agent + base URL. */
-  applyDohNode(node, protocol) {
-    this.dohNode = node;
-    this.dohBaseUrl = `${protocol}//${node.host}`;
-    this.dohAgent = new Agent({
-      connect: {
-        lookup: (_hostname, options, callback) => {
-          if (options?.all) {
-            callback(null, [{ address: node.ip, family: 4 }]);
-          } else {
-            callback(null, node.ip, 4);
-          }
-        }
-      }
-    });
-    if (this.config.verbose) {
-      vlog(`DoH proxy active: \u2192 ${node.host} (${node.ip}), ttl=${node.ttl}s`);
-    }
-  }
-  /**
-   * Handle network failure: re-resolve with --exclude and retry once.
-   * Returns true if retry should proceed, false if already retried.
-   */
-  async handleDohNetworkFailure() {
-    if (this.dohRetried) return false;
-    this.dohRetried = true;
-    const failedIp = this.dohNode?.ip ?? "";
-    const { hostname, protocol } = new URL(this.config.baseUrl);
-    this.dohNode = null;
-    this.dohAgent = null;
-    this.dohBaseUrl = null;
-    if (!failedIp) this.directUnverified = false;
-    if (this.config.verbose) {
-      vlog(failedIp ? `DoH: proxy node ${failedIp} failed, re-resolving with --exclude` : "DoH: direct connection failed, calling binary for DoH resolution");
-    }
-    try {
-      const result = await reResolveDoh(hostname, failedIp, this.dohUserAgent);
-      if (result.mode === "proxy" && result.node) {
-        this.applyDohNode(result.node, protocol);
-        this.dohRetried = false;
-        return true;
-      }
-    } catch {
-    }
-    if (this.config.verbose) {
-      vlog("DoH: re-resolution failed or switched to direct, retrying with direct connection");
-    }
-    return true;
-  }
-  get activeBaseUrl() {
-    return this.dohNode ? this.dohBaseUrl : this.config.baseUrl;
-  }
-  get activeDispatcher() {
-    return this.dispatcher ?? this.dohAgent ?? void 0;
-  }
-  /** User-Agent for DoH proxy requests: OKX/@okx_ai/{packageName}/{version} */
-  get dohUserAgent() {
-    return `OKX/@okx_ai/${this.config.userAgent ?? "unknown"}`;
-  }
   logRequest(method, url, auth) {
     if (!this.config.verbose) return;
     vlog(`\u2192 ${method} ${url}`);
@@ -1184,13 +955,14 @@ var OkxRestClient = class _OkxRestClient {
       vlog(`\u2190 ${status} | code=${code ?? "0"} | ${rawLen}B | ${elapsed}ms | trace=${traceId ?? "-"}`);
     }
   }
-  async publicGet(path4, query, rateLimit) {
+  async publicGet(path4, query, rateLimit, simulatedTrading) {
     return this.request({
       method: "GET",
       path: path4,
       auth: "public",
       query,
-      rateLimit
+      rateLimit,
+      simulatedTrading
     });
   }
   async privateGet(path4, query, rateLimit) {
@@ -1339,17 +1111,13 @@ var OkxRestClient = class _OkxRestClient {
    * Security: validates Content-Type and enforces maxBytes limit.
    */
   async privatePostBinary(path4, body, opts) {
-    this.ensureDoh();
     const maxBytes = opts?.maxBytes ?? _OkxRestClient.DEFAULT_MAX_BYTES;
     const expectedCT = opts?.expectedContentType ?? "application/octet-stream";
     const bodyJson = body ? JSON.stringify(body) : "";
     const endpoint = `POST ${path4}`;
-    this.logRequest("POST", `${this.activeBaseUrl}${path4}`, "private");
+    this.logRequest("POST", `${this.config.baseUrl}${path4}`, "private");
     const reqConfig = { method: "POST", path: path4, auth: "private" };
     const headers = this.buildHeaders(reqConfig, path4, bodyJson, getNow());
-    if (this.dohNode) {
-      headers.set("User-Agent", this.dohUserAgent);
-    }
     const t0 = Date.now();
     const response = await this.fetchBinary(path4, endpoint, headers, bodyJson, t0);
     const elapsed = Date.now() - t0;
@@ -1383,10 +1151,10 @@ var OkxRestClient = class _OkxRestClient {
         method: "POST",
         headers,
         body: bodyJson || void 0,
-        signal: AbortSignal.timeout(this.config.timeoutMs),
-        dispatcher: this.activeDispatcher
+        signal: AbortSignal.timeout(this.config.timeoutMs)
       };
-      return await fetch(`${this.activeBaseUrl}${path4}`, fetchOptions);
+      if (this.dispatcher) fetchOptions.dispatcher = this.dispatcher;
+      return await fetch(`${this.config.baseUrl}${path4}`, fetchOptions);
     } catch (error) {
       if (this.config.verbose) {
         vlog(`\u2717 NetworkError after ${Date.now() - t0}ms: ${error instanceof Error ? error.message : String(error)}`);
@@ -1408,7 +1176,8 @@ var OkxRestClient = class _OkxRestClient {
     if (reqConfig.auth === "private") {
       this.setAuthHeaders(headers, reqConfig.method, requestPath, bodyJson, timestamp);
     }
-    if (this.config.demo) {
+    const useSimulated = reqConfig.simulatedTrading !== void 0 ? reqConfig.simulatedTrading : this.config.demo;
+    if (useSimulated) {
       headers.set("x-simulated-trading", "1");
     }
     return headers;
@@ -1416,55 +1185,10 @@ var OkxRestClient = class _OkxRestClient {
   // ---------------------------------------------------------------------------
   // JSON request
   // ---------------------------------------------------------------------------
-  /**
-   * Handle network error during a JSON request: refresh DoH and maybe retry.
-   * Always either returns a retry result or throws NetworkError.
-   */
-  async handleRequestNetworkError(error, reqConfig, requestPath, t0) {
-    if (!this.dohRetried) {
-      if (this.config.verbose) {
-        const cause = error instanceof Error ? error.message : String(error);
-        vlog(`Network failure, refreshing DoH: ${cause}`);
-      }
-      const shouldRetry = await this.handleDohNetworkFailure();
-      if (shouldRetry && reqConfig.method === "GET") {
-        return this.request(reqConfig);
-      }
-    }
-    if (this.config.verbose) {
-      const elapsed = Date.now() - t0;
-      const cause = error instanceof Error ? error.message : String(error);
-      vlog(`\u2717 NetworkError after ${elapsed}ms: ${cause}`);
-    }
-    throw new NetworkError(
-      `Failed to call OKX endpoint ${reqConfig.method} ${requestPath}.`,
-      `${reqConfig.method} ${requestPath}`,
-      error
-    );
-  }
-  /**
-   * After a successful HTTP response on direct connection, cache mode=direct.
-   * (Even if the business response is an error, the network path is valid.)
-   */
-  cacheDirectConnectionIfNeeded() {
-    if (!this.directUnverified || this.dohNode) return;
-    this.directUnverified = false;
-    const { hostname } = new URL(this.config.baseUrl);
-    writeCache(hostname, {
-      mode: "direct",
-      node: null,
-      failedNodes: [],
-      updatedAt: Date.now()
-    });
-    if (this.config.verbose) {
-      vlog("DoH: direct connection succeeded, cached mode=direct");
-    }
-  }
   async request(reqConfig) {
-    this.ensureDoh();
     const queryString = buildQueryString(reqConfig.query);
     const requestPath = queryString.length > 0 ? `${reqConfig.path}?${queryString}` : reqConfig.path;
-    const url = `${this.activeBaseUrl}${requestPath}`;
+    const url = `${this.config.baseUrl}${requestPath}`;
     const bodyJson = reqConfig.body ? JSON.stringify(reqConfig.body) : "";
     const timestamp = getNow();
     this.logRequest(reqConfig.method, url, reqConfig.auth);
@@ -1472,9 +1196,6 @@ var OkxRestClient = class _OkxRestClient {
       await this.rateLimiter.consume(reqConfig.rateLimit);
     }
     const headers = this.buildHeaders(reqConfig, requestPath, bodyJson, timestamp);
-    if (this.dohNode) {
-      headers.set("User-Agent", this.dohUserAgent);
-    }
     const t0 = Date.now();
     let response;
     try {
@@ -1482,17 +1203,27 @@ var OkxRestClient = class _OkxRestClient {
         method: reqConfig.method,
         headers,
         body: reqConfig.method === "POST" ? bodyJson : void 0,
-        signal: AbortSignal.timeout(this.config.timeoutMs),
-        dispatcher: this.activeDispatcher
+        signal: AbortSignal.timeout(this.config.timeoutMs)
       };
+      if (this.dispatcher) {
+        fetchOptions.dispatcher = this.dispatcher;
+      }
       response = await fetch(url, fetchOptions);
     } catch (error) {
-      return await this.handleRequestNetworkError(error, reqConfig, requestPath, t0);
+      if (this.config.verbose) {
+        const elapsed2 = Date.now() - t0;
+        const cause = error instanceof Error ? error.message : String(error);
+        vlog(`\u2717 NetworkError after ${elapsed2}ms: ${cause}`);
+      }
+      throw new NetworkError(
+        `Failed to call OKX endpoint ${reqConfig.method} ${requestPath}.`,
+        `${reqConfig.method} ${requestPath}`,
+        error
+      );
     }
     const rawText = await response.text();
     const elapsed = Date.now() - t0;
     const traceId = extractTraceId(response.headers);
-    this.cacheDirectConnectionIfNeeded();
     return this.processResponse(rawText, response, elapsed, traceId, reqConfig, requestPath);
   }
 };
@@ -3371,19 +3102,19 @@ function safeWriteFile(targetDir, fileName, data) {
     throw new Error(`Invalid file name: "${fileName}"`);
   }
   const resolvedDir = resolve(targetDir);
-  const filePath = join3(resolvedDir, safeName);
+  const filePath = join(resolvedDir, safeName);
   const resolvedPath = resolve(filePath);
   if (!resolvedPath.startsWith(resolvedDir + sep)) {
     throw new Error(`Path traversal detected: "${fileName}" resolves outside target directory`);
   }
-  mkdirSync2(resolvedDir, { recursive: true });
+  mkdirSync(resolvedDir, { recursive: true });
   const tmpPath = `${resolvedPath}.${randomUUID()}.tmp`;
   try {
-    writeFileSync2(tmpPath, data);
-    renameSync2(tmpPath, resolvedPath);
+    writeFileSync(tmpPath, data);
+    renameSync(tmpPath, resolvedPath);
   } catch (err) {
     try {
-      unlinkSync2(tmpPath);
+      unlinkSync(tmpPath);
     } catch {
     }
     throw err;
@@ -3391,18 +3122,19 @@ function safeWriteFile(targetDir, fileName, data) {
   return resolvedPath;
 }
 var MAX_DOWNLOAD_BYTES = 50 * 1024 * 1024;
-async function downloadSkillZip(client, name, targetDir) {
+async function downloadSkillZip(client, name, targetDir, format = "zip") {
   const result = await client.privatePostBinary(
     "/api/v5/skill/download",
     { name },
     { maxBytes: MAX_DOWNLOAD_BYTES }
   );
-  const fileName = `${name}.zip`;
+  const ext = format === "skill" ? "skill" : "zip";
+  const fileName = `${name}.${ext}`;
   const filePath = safeWriteFile(targetDir, fileName, result.data);
   return filePath;
 }
 var DEFAULT_MAX_TOTAL_BYTES = 100 * 1024 * 1024;
-var DEFAULT_REGISTRY_PATH = join5(homedir3(), ".okx", "skills", "registry.json");
+var DEFAULT_REGISTRY_PATH = join3(homedir(), ".okx", "skills", "registry.json");
 function registerSkillsTools() {
   return [
     {
@@ -3449,7 +3181,7 @@ function registerSkillsTools() {
     {
       name: "skills_download",
       module: "skills",
-      description: "Download a skill zip file from OKX Skills Marketplace to a local directory. Always call skills_search first to confirm the skill name exists. Downloads the latest approved version. NOTE: Downloads third-party developer content as a zip \u2014 does NOT install to agents. For full installation use CLI: okx skill add <name>. Use when the user wants to inspect or manually install a skill package.",
+      description: "Download a skill package from OKX Skills Marketplace to a local directory. Always call skills_search first to confirm the skill name exists. Downloads the latest approved version. NOTE: Downloads third-party developer content \u2014 does NOT install to agents. For full installation use CLI: okx skill add <name>. Use when the user wants to inspect or manually install a skill package.",
       inputSchema: {
         type: "object",
         properties: {
@@ -3459,7 +3191,12 @@ function registerSkillsTools() {
           },
           targetDir: {
             type: "string",
-            description: "Directory path where the zip file will be saved"
+            description: "Directory path where the file will be saved"
+          },
+          format: {
+            type: "string",
+            description: "Output file format: 'zip' or 'skill' (default: 'skill')",
+            enum: ["zip", "skill"]
           }
         },
         required: ["name", "targetDir"],
@@ -3492,7 +3229,8 @@ async function handleSearch(args, ctx) {
 async function handleDownload(args, ctx) {
   const name = String(args.name);
   const targetDir = String(args.targetDir);
-  const filePath = await downloadSkillZip(ctx.client, name, targetDir);
+  const format = args.format === "zip" ? "zip" : "skill";
+  const filePath = await downloadSkillZip(ctx.client, name, targetDir, format);
   return {
     endpoint: "POST /api/v5/skill/download",
     requestTime: (/* @__PURE__ */ new Date()).toISOString(),
@@ -5639,6 +5377,12 @@ function registerFuturesTools() {
   ];
 }
 var TWO_DAYS_MS = 2 * 24 * 60 * 60 * 1e3;
+var DEMO_PROPERTY = {
+  demo: {
+    type: "boolean",
+    description: "Query simulated trading (demo) market data. Default: false (live market data)."
+  }
+};
 function registerMarketTools() {
   return [
     {
@@ -5652,7 +5396,8 @@ function registerMarketTools() {
           instId: {
             type: "string",
             description: "e.g. BTC-USDT, BTC-USDT-SWAP"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instId"]
       },
@@ -5661,7 +5406,8 @@ function registerMarketTools() {
         const response = await context.client.publicGet(
           "/api/v5/market/ticker",
           { instId: requireString(args, "instId") },
-          publicRateLimit("market_get_ticker", 20)
+          publicRateLimit("market_get_ticker", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -5685,7 +5431,8 @@ function registerMarketTools() {
           instFamily: {
             type: "string",
             description: "e.g. BTC-USD"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instType"]
       },
@@ -5698,7 +5445,8 @@ function registerMarketTools() {
             uly: readString(args, "uly"),
             instFamily: readString(args, "instFamily")
           }),
-          publicRateLimit("market_get_tickers", 20)
+          publicRateLimit("market_get_tickers", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -5718,7 +5466,8 @@ function registerMarketTools() {
           sz: {
             type: "number",
             description: "Depth per side, default 1, max 400"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instId"]
       },
@@ -5730,7 +5479,8 @@ function registerMarketTools() {
             instId: requireString(args, "instId"),
             sz: readNumber(args, "sz")
           }),
-          publicRateLimit("market_get_orderbook", 20)
+          publicRateLimit("market_get_orderbook", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -5763,7 +5513,8 @@ function registerMarketTools() {
           limit: {
             type: "number",
             description: "Max results (default 100)"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instId"]
       },
@@ -5771,6 +5522,7 @@ function registerMarketTools() {
         const args = asRecord(rawArgs);
         const afterTs = readString(args, "after");
         const beforeTs = readString(args, "before");
+        const demo = readBoolean(args, "demo") ?? false;
         const query = compactObject({
           instId: requireString(args, "instId"),
           bar: readString(args, "bar"),
@@ -5782,9 +5534,9 @@ function registerMarketTools() {
         const hasTimestamp = afterTs !== void 0 || beforeTs !== void 0;
         const useHistory = afterTs !== void 0 && Number(afterTs) < Date.now() - TWO_DAYS_MS;
         const path4 = useHistory ? "/api/v5/market/history-candles" : "/api/v5/market/candles";
-        const response = await context.client.publicGet(path4, query, rateLimit);
+        const response = await context.client.publicGet(path4, query, rateLimit, demo);
         if (!useHistory && hasTimestamp && Array.isArray(response.data) && response.data.length === 0) {
-          return normalizeResponse(await context.client.publicGet("/api/v5/market/history-candles", query, rateLimit));
+          return normalizeResponse(await context.client.publicGet("/api/v5/market/history-candles", query, rateLimit, demo));
         }
         return normalizeResponse(response);
       }
@@ -5812,7 +5564,8 @@ function registerMarketTools() {
           instFamily: {
             type: "string",
             description: "e.g. BTC-USD"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instType"]
       },
@@ -5826,7 +5579,8 @@ function registerMarketTools() {
             uly: readString(args, "uly"),
             instFamily: readString(args, "instFamily")
           }),
-          publicRateLimit("market_get_instruments", 20)
+          publicRateLimit("market_get_instruments", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -5858,13 +5612,15 @@ function registerMarketTools() {
           limit: {
             type: "number",
             description: "History records (default 20, max 100)"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instId"]
       },
       handler: async (rawArgs, context) => {
         const args = asRecord(rawArgs);
         const isHistory = readBoolean(args, "history") ?? false;
+        const demo = readBoolean(args, "demo") ?? false;
         if (isHistory) {
           const response2 = await context.client.publicGet(
             "/api/v5/public/funding-rate-history",
@@ -5874,14 +5630,16 @@ function registerMarketTools() {
               before: readString(args, "before"),
               limit: readNumber(args, "limit") ?? 20
             }),
-            publicRateLimit("market_get_funding_rate", 20)
+            publicRateLimit("market_get_funding_rate", 20),
+            demo
           );
           return normalizeResponse(response2);
         }
         const response = await context.client.publicGet(
           "/api/v5/public/funding-rate",
           { instId: requireString(args, "instId") },
-          publicRateLimit("market_get_funding_rate", 20)
+          publicRateLimit("market_get_funding_rate", 20),
+          demo
         );
         return normalizeResponse(response);
       }
@@ -5908,7 +5666,8 @@ function registerMarketTools() {
           },
           instFamily: {
             type: "string"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instType"]
       },
@@ -5922,7 +5681,8 @@ function registerMarketTools() {
             uly: readString(args, "uly"),
             instFamily: readString(args, "instFamily")
           }),
-          publicRateLimit("market_get_mark_price", 10)
+          publicRateLimit("market_get_mark_price", 10),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -5942,7 +5702,8 @@ function registerMarketTools() {
           limit: {
             type: "number",
             description: "Default 20, max 500"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instId"]
       },
@@ -5954,7 +5715,8 @@ function registerMarketTools() {
             instId: requireString(args, "instId"),
             limit: readNumber(args, "limit") ?? 20
           }),
-          publicRateLimit("market_get_trades", 20)
+          publicRateLimit("market_get_trades", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -5974,7 +5736,8 @@ function registerMarketTools() {
           quoteCcy: {
             type: "string",
             description: "e.g. USD or USDT"
-          }
+          },
+          ...DEMO_PROPERTY
         }
       },
       handler: async (rawArgs, context) => {
@@ -5985,7 +5748,8 @@ function registerMarketTools() {
             instId: readString(args, "instId"),
             quoteCcy: readString(args, "quoteCcy")
           }),
-          publicRateLimit("market_get_index_ticker", 20)
+          publicRateLimit("market_get_index_ticker", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -6022,7 +5786,8 @@ function registerMarketTools() {
           history: {
             type: "boolean",
             description: "true=older historical data"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instId"]
       },
@@ -6039,7 +5804,8 @@ function registerMarketTools() {
             before: readString(args, "before"),
             limit: readNumber(args, "limit")
           }),
-          publicRateLimit("market_get_index_candles", 20)
+          publicRateLimit("market_get_index_candles", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -6055,7 +5821,8 @@ function registerMarketTools() {
           instId: {
             type: "string",
             description: "SWAP or FUTURES ID, e.g. BTC-USDT-SWAP"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instId"]
       },
@@ -6064,7 +5831,8 @@ function registerMarketTools() {
         const response = await context.client.publicGet(
           "/api/v5/public/price-limit",
           { instId: requireString(args, "instId") },
-          publicRateLimit("market_get_price_limit", 20)
+          publicRateLimit("market_get_price_limit", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -6091,7 +5859,8 @@ function registerMarketTools() {
           },
           instFamily: {
             type: "string"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instType"]
       },
@@ -6105,7 +5874,8 @@ function registerMarketTools() {
             uly: readString(args, "uly"),
             instFamily: readString(args, "instFamily")
           }),
-          publicRateLimit("market_get_open_interest", 20)
+          publicRateLimit("market_get_open_interest", 20),
+          readBoolean(args, "demo") ?? false
         );
         return normalizeResponse(response);
       }
@@ -6126,7 +5896,8 @@ function registerMarketTools() {
           instId: {
             type: "string",
             description: "Optional: filter by specific instrument ID, e.g. AAPL-USDT-SWAP"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: []
       },
@@ -6137,7 +5908,8 @@ function registerMarketTools() {
         const response = await context.client.publicGet(
           "/api/v5/public/instruments",
           compactObject({ instType, instId }),
-          publicRateLimit("market_get_stock_tokens", 20)
+          publicRateLimit("market_get_stock_tokens", 20),
+          readBoolean(args, "demo") ?? false
         );
         const data = response.data;
         const filtered = Array.isArray(data) ? data.filter((item) => item.instCategory === "3") : data;
@@ -6147,7 +5919,7 @@ function registerMarketTools() {
     {
       name: "market_get_instruments_by_category",
       module: "market",
-      description: "Discover tradeable instruments by asset category. Stock tokens (instCategory=3, e.g. AAPL-USDT-SWAP, TSLA-USDT-SWAP), Metals (4, e.g. XAUUSDT-USDT-SWAP for gold), Commodities (5, e.g. OIL-USDT-SWAP for crude oil), Forex (6, e.g. EURUSDT-USDT-SWAP for EUR/USD), Bonds (7, e.g. US30Y-USDT-SWAP). Use this to find instIds before querying prices or placing orders. Filters client-side by instCategory.",
+      description: "Discover tradeable instruments by asset category. Stock tokens (instCategory=3, e.g. AAPL-USDT-SWAP, TSLA-USDT-SWAP), Metals (4, e.g. XAUUSDT-USDT-SWAP for gold), Commodities (5, e.g. OIL-USDT-SWAP for crude oil), Forex (6, e.g. EURUSDT-USDT-SWAP for EUR/USD), Bonds (7, e.g. US30Y-USDT-SWAP for crude oil). Use this to find instIds before querying prices or placing orders. Filters client-side by instCategory.",
       isWrite: false,
       inputSchema: {
         type: "object",
@@ -6165,7 +5937,8 @@ function registerMarketTools() {
           instId: {
             type: "string",
             description: "Optional: filter by specific instrument ID"
-          }
+          },
+          ...DEMO_PROPERTY
         },
         required: ["instCategory"]
       },
@@ -6177,7 +5950,8 @@ function registerMarketTools() {
         const response = await context.client.publicGet(
           "/api/v5/public/instruments",
           compactObject({ instType, instId }),
-          publicRateLimit("market_get_instruments_by_category", 20)
+          publicRateLimit("market_get_instruments_by_category", 20),
+          readBoolean(args, "demo") ?? false
         );
         const data = response.data;
         const filtered = Array.isArray(data) ? data.filter((item) => item.instCategory === instCategory) : data;
@@ -7709,12 +7483,12 @@ function toMcpTool(tool) {
   };
 }
 function configFilePath() {
-  return join6(homedir4(), ".okx", "config.toml");
+  return join4(homedir2(), ".okx", "config.toml");
 }
 function readFullConfig() {
   const path4 = configFilePath();
   if (!existsSync3(path4)) return { profiles: {} };
-  const raw = readFileSync4(path4, "utf-8");
+  const raw = readFileSync3(path4, "utf-8");
   try {
     return parse(raw);
   } catch (err) {
@@ -7845,21 +7619,21 @@ function loadConfig(cli) {
     verbose: cli.verbose ?? false
   };
 }
-var CACHE_FILE = join7(homedir5(), ".okx", "update-check.json");
+var CACHE_FILE = join5(homedir3(), ".okx", "update-check.json");
 var CHECK_INTERVAL_MS = 24 * 60 * 60 * 1e3;
-function readCache2() {
+function readCache() {
   try {
     if (existsSync4(CACHE_FILE)) {
-      return JSON.parse(readFileSync5(CACHE_FILE, "utf-8"));
+      return JSON.parse(readFileSync4(CACHE_FILE, "utf-8"));
     }
   } catch {
   }
   return {};
 }
-function writeCache2(cache) {
+function writeCache(cache) {
   try {
-    mkdirSync6(join7(homedir5(), ".okx"), { recursive: true });
-    writeFileSync5(CACHE_FILE, JSON.stringify(cache, null, 2), "utf-8");
+    mkdirSync5(join5(homedir3(), ".okx"), { recursive: true });
+    writeFileSync4(CACHE_FILE, JSON.stringify(cache, null, 2), "utf-8");
   } catch {
   }
 }
@@ -7890,14 +7664,14 @@ async function fetchLatestVersion(packageName) {
 function refreshCacheInBackground(packageName) {
   fetchLatestVersion(packageName).then((latest) => {
     if (!latest) return;
-    const cache = readCache2();
+    const cache = readCache();
     cache[packageName] = { latestVersion: latest, checkedAt: Date.now() };
-    writeCache2(cache);
+    writeCache(cache);
   }).catch(() => {
   });
 }
 function checkForUpdates(packageName, currentVersion) {
-  const cache = readCache2();
+  const cache = readCache();
   const entry = cache[packageName];
   if (entry && isNewerVersion(currentVersion, entry.latestVersion)) {
     process.stderr.write(
@@ -8127,7 +7901,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 ? "caa6dae" : "dev";
+var GIT_HASH = true ? "6d4d559" : "dev";
 
 // src/server.ts
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";