@thinkingai/ae-cli 1.0.3 → 1.0.18

package/dist/{chunk-53FNNCZW.js → chunk-7QBNU44L.js}

@@ -1,12 +1,12 @@
 import {
   getToken
-} from "./chunk-3CDLNXTD.js";
+} from "./chunk-4KQ7H7DY.js";
 import {
   getActiveHost,
   getConfigDir,
   safeJsonParse,
   safeReadJsonFile
-} from "./chunk-EBFUVWIZ.js";
+} from "./chunk-MR57UIKC.js";
 
 // src/core/mcp.ts
 import fs from "fs";
@@ -54,6 +54,7 @@ function genRequestId() {
 }
 var MCP_TOKEN_FILE = path.join(getConfigDir(), "mcp-tokens.json");
 var MCP_TOKEN_GENERATE_PATH = "/v1/ta/mcp/token/generate";
+var FALLBACK_MCP_TOKEN_FILE = "/data/app/te_agent_ta/.ae-config/mcp-token.json";
 function loadMcpTokenStore() {
   try {
     if (fs.existsSync(MCP_TOKEN_FILE)) {
@@ -68,6 +69,23 @@ function saveMcpTokenStore(store) {
   if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
   fs.writeFileSync(MCP_TOKEN_FILE, JSON.stringify(store, null, 2));
 }
+function getFallbackMcpToken(hostUrl) {
+  try {
+    if (!fs.existsSync(FALLBACK_MCP_TOKEN_FILE)) {
+      return null;
+    }
+    const fallbackData = safeReadJsonFile(FALLBACK_MCP_TOKEN_FILE);
+    if (fallbackData && typeof fallbackData === "object" && !Array.isArray(fallbackData)) {
+      const url = fallbackData.url;
+      const token = fallbackData["mcp-token"];
+      if (url === hostUrl && token) {
+        return token;
+      }
+    }
+  } catch {
+  }
+  return null;
+}
 async function generateMcpToken(hostUrl) {
   const teToken = await getToken(hostUrl);
   const base = hostUrl.replace(/\/+$/, "");
@@ -102,8 +120,60 @@ async function getMcpToken(hostOverride) {
   saveMcpTokenStore(updatedStore);
   return mcpToken;
 }
+function clearMcpToken(hostUrl) {
+  const store = loadMcpTokenStore();
+  if (hostUrl) {
+    delete store[hostUrl];
+  } else {
+    for (const key of Object.keys(store)) {
+      delete store[key];
+    }
+  }
+  saveMcpTokenStore(store);
+}
+function setMcpTokenManual(token, hostUrl) {
+  const store = loadMcpTokenStore();
+  store[hostUrl] = token;
+  saveMcpTokenStore(store);
+}
+async function validateMcpToken(token, hostUrl) {
+  const mapping = getMcpMapping("analysis");
+  const base = hostUrl.replace(/\/+$/, "");
+  const url = `${base}/mcp/${mapping.componentName}/http/${mapping.mappingPath}`;
+  const requestId = 1;
+  const body = {
+    jsonrpc: JSONRPC_VERSION,
+    id: requestId,
+    method: "tools/list",
+    params: { _meta: { progressToken: requestId } }
+  };
+  const headers = {
+    "Content-Type": "application/json",
+    "Accept": "application/json, text/event-stream",
+    "mcp-protocol-version": MCP_PROTOCOL_VERSION,
+    "mcp-token": token
+  };
+  try {
+    const resp = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(body)
+    });
+    if (!resp.ok) {
+      return false;
+    }
+    const data = safeJsonParse(await resp.text());
+    if (data.error) {
+      return false;
+    }
+    return data.result?.tools !== void 0;
+  } catch {
+    return false;
+  }
+}
 async function mcpRequest(url, method, params = {}, hostOverride) {
-  const token = await getMcpToken(hostOverride);
+  const hostUrl = hostOverride || getActiveHost();
+  const token = await getMcpToken(hostUrl);
   const requestId = genRequestId();
   const paramsWithMeta = {
     ...params,
@@ -126,6 +196,31 @@ async function mcpRequest(url, method, params = {}, hostOverride) {
     headers,
     body: JSON.stringify(body)
   });
+  if (resp.status === 403) {
+    const fallbackToken = getFallbackMcpToken(hostUrl);
+    if (fallbackToken) {
+      const store = loadMcpTokenStore();
+      store[hostUrl] = fallbackToken;
+      saveMcpTokenStore(store);
+      process.stderr.write(`[ae-cli] MCP token refreshed from fallback for ${hostUrl}
+`);
+      headers["mcp-token"] = fallbackToken;
+      const retryResp = await fetch(url, {
+        method: "POST",
+        headers,
+        body: JSON.stringify(body)
+      });
+      if (!retryResp.ok) {
+        throw new Error(`MCP HTTP error: ${retryResp.status} ${retryResp.statusText}`);
+      }
+      const retryData = safeJsonParse(await retryResp.text());
+      if (retryData.error) {
+        throw new Error(`MCP error: ${retryData.error.message || JSON.stringify(retryData.error)}`);
+      }
+      return retryData.result;
+    }
+    throw new Error(`MCP token expired or invalid. Please set a new MCP token: ae-cli auth set-mcp-token <token> --host ${hostUrl}`);
+  }
   if (!resp.ok) {
     throw new Error(`MCP HTTP error: ${resp.status} ${resp.statusText}`);
   }
@@ -169,6 +264,9 @@ function parseMcpResult(result) {
 export {
   registerMcpMappings,
   resolveMcpUrl,
+  clearMcpToken,
+  setMcpTokenManual,
+  validateMcpToken,
   callMcpTool,
   parseMcpResult
 };

package/dist/{chunk-ZYXIGPG7.js → chunk-LAAS6ITR.js}

@@ -2,10 +2,10 @@ import {
   clearToken,
   getToken,
   resolveHost
-} from "./chunk-3CDLNXTD.js";
+} from "./chunk-4KQ7H7DY.js";
 import {
   safeJsonParse
-} from "./chunk-EBFUVWIZ.js";
+} from "./chunk-MR57UIKC.js";
 
 // src/core/client.ts
 import WebSocket from "ws";
@@ -49,7 +49,7 @@ async function request(method, modulePath, params = {}, body = null, retry = tru
     return request(method, modulePath, params, body, false, resolvedHost);
   }
   if (data.return_code !== 0 && data.return_code !== void 0) {
-    throw new Error(`TE API error: ${data.return_message || "unknown"} (code: ${data.return_code})`);
+    throw new Error(`AE API error: ${data.return_message || "unknown"} (code: ${data.return_code})`);
   }
   return data.data !== void 0 ? data.data : data;
 }

package/dist/{chunk-EBFUVWIZ.js → chunk-MR57UIKC.js}

@@ -17,11 +17,37 @@ import fs2 from "fs";
 import path from "path";
 var CONFIG_DIR = path.join(process.env.HOME || "", ".ae-cli");
 var CONFIG_FILE = path.join(CONFIG_DIR, "config.json");
+var MCP_TOKENS_FILE = path.join(CONFIG_DIR, "mcp-tokens.json");
+var FALLBACK_MCP_TOKEN_FILE = "/data/app/te_agent_ta/.ae-config/mcp-token.json";
 function ensureDir() {
   if (!fs2.existsSync(CONFIG_DIR)) fs2.mkdirSync(CONFIG_DIR, { recursive: true });
 }
+function migrateFromFallback() {
+  if (!fs2.existsSync(FALLBACK_MCP_TOKEN_FILE) || fs2.existsSync(MCP_TOKENS_FILE)) {
+    return null;
+  }
+  try {
+    const fallbackData = safeReadJsonFile(FALLBACK_MCP_TOKEN_FILE);
+    const tokenStore = {};
+    if (fallbackData && typeof fallbackData === "object" && !Array.isArray(fallbackData)) {
+      const url = fallbackData.url;
+      const token = fallbackData["mcp-token"];
+      if (url && token) {
+        tokenStore[url] = token;
+      }
+    }
+    if (Object.keys(tokenStore).length > 0) {
+      ensureDir();
+      fs2.writeFileSync(MCP_TOKENS_FILE, JSON.stringify(tokenStore, null, 2));
+      return tokenStore;
+    }
+  } catch {
+  }
+  return null;
+}
 function loadConfig() {
   try {
+    const migratedTokens = migrateFromFallback();
     if (fs2.existsSync(CONFIG_FILE)) {
       const raw = safeReadJsonFile(CONFIG_FILE);
       if (raw.defaultHost && !raw.activeHost) {
@@ -31,6 +57,22 @@ function loadConfig() {
       }
       return raw;
     }
+    if (migratedTokens && Object.keys(migratedTokens).length > 0) {
+      const hosts = {};
+      for (const url of Object.keys(migratedTokens)) {
+        hosts[url] = { label: url };
+        process.stderr.write(`[ae-cli] Added host: ${url}
+`);
+      }
+      const activeHost = Object.keys(migratedTokens)[0];
+      process.stderr.write(`[ae-cli] Active host set to: ${activeHost}
+`);
+      process.stderr.write(`[ae-cli] MCP tokens migrated from ${FALLBACK_MCP_TOKEN_FILE}
+`);
+      const config = { activeHost, hosts };
+      saveConfig(config);
+      return config;
+    }
   } catch (err) {
     console.error(`Error loading config: ${err.message}`);
   }

package/dist/{client-5QSQBBDB.js → client-O56G2SNJ.js}

@@ -4,9 +4,9 @@ import {
   queryReportData,
   querySql,
   wsQuery
-} from "./chunk-3GM5LJZR.js";
-import "./chunk-RJNLN3KQ.js";
-import "./chunk-OVQOK35G.js";
+} from "./chunk-LAAS6ITR.js";
+import "./chunk-4KQ7H7DY.js";
+import "./chunk-MR57UIKC.js";
 export {
   httpGet,
   httpPost,

package/dist/{config-Q7J3Z7Z2.js → config-BU5FHGGE.js}

@@ -1,26 +1,29 @@
 import {
+  clearToken,
   getAuthStatus,
-  getToken
-} from "./chunk-RJNLN3KQ.js";
+  getToken,
+  loadToken,
+  validateToken
+} from "./chunk-4KQ7H7DY.js";
 import {
   addHost,
   listHosts,
   removeHost,
   setActiveHost,
   updateHostLabel
-} from "./chunk-OVQOK35G.js";
+} from "./chunk-MR57UIKC.js";
 
 // src/commands/config.ts
 import * as readline from "readline";
 function registerConfig(program) {
-  program.command("config").description("Interactive TE host configuration manager").action(async () => {
+  program.command("config").description("Interactive AE host configuration manager").action(async () => {
     await runConfigUI(program);
   });
 }
 async function runConfigUI(program) {
   const hosts = listHosts();
   if (hosts.length === 0) {
-    process.stderr.write("\x1B[1m[ae-cli] No TE hosts configured. Let's add one.\x1B[0m\n\n");
+    process.stderr.write("\x1B[1m[ae-cli] No AE hosts configured. Let's add one.\x1B[0m\n\n");
     const newHost = await promptNewHost();
     if (!newHost) {
       process.stderr.write("Cancelled.\n");
@@ -136,9 +139,21 @@ Delete \x1B[1m${label}\x1B[0m (${url})? [y/N]: `, (a) => {
 async function ensureAuth(hostUrl) {
   const status = getAuthStatus(hostUrl);
   if (status.authenticated) {
-    process.stderr.write(`\x1B[32m\u2713\x1B[0m Token valid for ${hostUrl}
+    const cached = loadToken(hostUrl);
+    if (cached && cached.token) {
+      process.stderr.write(`\x1B[2m[ae-cli] Validating token...\x1B[0m
 `);
-    return;
+      const isValid = await validateToken(cached.token, hostUrl);
+      if (isValid) {
+        process.stderr.write(`\x1B[32m\u2713\x1B[0m Token valid for ${hostUrl}
+`);
+        return;
+      } else {
+        clearToken(hostUrl);
+        process.stderr.write(`\x1B[33m!\x1B[0m Token expired or invalid. Re-authenticating...
+`);
+      }
+    }
   }
   process.stderr.write(`
 \x1B[33m!\x1B[0m No valid token for ${hostUrl}. Starting login...
@@ -157,7 +172,7 @@ async function ensureAuth(hostUrl) {
 async function promptNewHost() {
   const rl = readline.createInterface({ input: process.stdin, output: process.stderr });
   const ask = (q) => new Promise((resolve) => rl.question(q, (a) => resolve(a.trim())));
-  const url = await ask("TE host URL (e.g., https://ta.thinkingdata.cn): ");
+  const url = await ask("AE host URL (e.g., https://ta.thinkingdata.cn): ");
   if (!url) {
     rl.close();
     return null;

package/dist/index.js

@@ -5,7 +5,7 @@ import {
 import {
   getActiveHost,
   safeJsonParse
-} from "./chunk-EBFUVWIZ.js";
+} from "./chunk-MR57UIKC.js";
 
 // src/index.ts
 import { Command as CommanderCommand } from "commander";
@@ -50,7 +50,7 @@ async function runCommand(cmd, opts, globalOpts) {
     const message = err.message || String(err);
     if (message.includes("token") || message.includes("auth") || message.includes("401") || message.includes("403")) {
       printError("auth", message, "Run: ae-cli auth login");
-    } else if (message.includes("TE API error")) {
+    } else if (message.includes("AE API error")) {
       printError("api", message);
     } else {
       printError("api", message);
@@ -62,7 +62,7 @@ function createRuntimeContext(cmd, opts, globalOpts) {
   let _clientModule = null;
   async function getClient() {
     if (!_clientModule) {
-      _clientModule = await import("./client-T4VODLA6.js");
+      _clientModule = await import("./client-O56G2SNJ.js");
     }
     return _clientModule;
   }
@@ -105,7 +105,7 @@ function createRuntimeContext(cmd, opts, globalOpts) {
       return client.queryReportData(projectId, reportId, qp, eventModel, options, ctx.host());
     },
     async token() {
-      const { getToken } = await import("./auth-BQC5DBE7.js");
+      const { getToken } = await import("./auth-ECD46NBA.js");
       return getToken(ctx.host());
     },
     host() {
@@ -206,41 +206,41 @@ function extractGlobalOptions(program2) {
 var require2 = createRequire(import.meta.url);
 var { version } = require2("../package.json");
 var program = new CommanderCommand();
-program.name("ae-cli").version(version).description("CLI tool for ThinkingEngine (TE) analytics platform").option("--host <url>", "Override active TE host URL (e.g., https://ta.thinkingdata.cn)").option("--mcp-url <url>", "Override MCP server URL (e.g., http://localhost/mcp/http/example)").option("--format <format>", "Output format: json | table", "json").option("--jq <expr>", "jq filter expression").option("--dry-run", "Show request details without executing", false).option("--yes", "Skip confirmation for write operations", false);
+program.name("ae-cli").version(version).description("CLI tool for ThinkingAI (AE) analytics platform").option("--host <url>", "Override active AE host URL (e.g., https://ta.thinkingdata.cn)").option("--mcp-url <url>", "Override MCP server URL (e.g., http://localhost/mcp/http/example)").option("--format <format>", "Output format: json | table", "json").option("--jq <expr>", "jq filter expression").option("--dry-run", "Show request details without executing", false).option("--yes", "Skip confirmation for write operations", false);
 async function loadCommands() {
   const commands = [];
   try {
-    const teAnalysis = await import("./te-analysis-CHFTDN55.js");
+    const teAnalysis = await import("./te-analysis-UGKESPBE.js");
     commands.push(...teAnalysis.default);
   } catch {
   }
   try {
-    const teAudience = await import("./te-audience-M6LQY6M7.js");
+    const teAudience = await import("./te-audience-THZNVHYB.js");
     commands.push(...teAudience.default);
   } catch {
   }
   try {
-    const teMeta = await import("./te-meta-FMM7KCVI.js");
+    const teMeta = await import("./te-meta-MGO5BCMG.js");
     commands.push(...teMeta.default);
   } catch {
   }
   try {
-    const teCommon = await import("./te-common-BLQ7DVGC.js");
+    const teCommon = await import("./te-common-QZOUME3S.js");
     commands.push(...teCommon.default);
   } catch {
   }
   try {
-    const engage = await import("./te-engage-DXE4R5EW.js");
+    const engage = await import("./te-engage-RYGEQTYK.js");
     commands.push(...engage.default);
   } catch {
   }
   try {
-    const community = await import("./community-FVDAWGS4.js");
+    const community = await import("./te-community-AZVW3NFX.js");
     commands.push(...community.default);
   } catch {
   }
   try {
-    const dataops = await import("./te-dataops-V4OTKY35.js");
+    const dataops = await import("./te-dataops-OE5PX2K6.js");
     commands.push(...dataops.default);
   } catch {
   }
@@ -248,21 +248,21 @@ async function loadCommands() {
 }
 async function registerAuthCommands() {
   try {
-    const { registerAuth } = await import("./auth-DXKYLMO7.js");
+    const { registerAuth } = await import("./auth-2UTBG5U3.js");
     registerAuth(program);
   } catch {
   }
 }
 async function registerConfigCommands() {
   try {
-    const { registerConfig } = await import("./config-V7AXKDMB.js");
+    const { registerConfig } = await import("./config-BU5FHGGE.js");
     registerConfig(program);
   } catch {
   }
 }
 async function registerApiCommand() {
   try {
-    const { registerApi } = await import("./raw-G3GKHAPZ.js");
+    const { registerApi } = await import("./raw-A3VN2B6I.js");
     registerApi(program);
   } catch {
   }

package/dist/{raw-G3GKHAPZ.js → raw-A3VN2B6I.js}

@@ -5,13 +5,13 @@ import {
 import {
   httpGet,
   httpPost
-} from "./chunk-ZYXIGPG7.js";
+} from "./chunk-LAAS6ITR.js";
 import {
   resolveHost
-} from "./chunk-3CDLNXTD.js";
+} from "./chunk-4KQ7H7DY.js";
 import {
   safeJsonParse
-} from "./chunk-EBFUVWIZ.js";
+} from "./chunk-MR57UIKC.js";
 
 // src/api/raw.ts
 function registerApi(program) {
@@ -21,7 +21,7 @@ function registerApi(program) {
     const format = globalOpts.format || "json";
     const jq = globalOpts.jq;
     if (!host) {
-      printError("config", "No TE host configured.", "Run: ae-cli config set-host");
+      printError("config", "No AE host configured.", "Run: ae-cli config set-host");
       process.exit(1);
     }
     try {