@thinkingai/ae-cli 1.0.22 → 1.0.27

package/dist/{te-kb-GD7SJRYB.js → te-kb-XHVBWP5E.js}

@@ -1,22 +1,19 @@
 import {
-  httpDelete,
-  httpUpload
-} from "./chunk-NBFTPIAH.js";
-import "./chunk-24ZKQWG2.js";
-import "./chunk-TMMUBSKW.js";
+  kbApi,
+  kbUpload
+} from "./chunk-MWWYGZ76.js";
+import "./chunk-2NGWYMLB.js";
+import "./chunk-SRRUNQIM.js";
+import "./chunk-BI5ZVD6T.js";
+import "./chunk-MNHE6SSI.js";
 
 // src/commands/te-kb/query.ts
 var API_PATH = "/agent/api/external/knowledge-bases/query";
 function buildBody(ctx) {
-  const body = {
+  return {
     query: ctx.str("query"),
     sources: ctx.json("sources")
   };
-  const modelId = ctx.str("model-id");
-  if (modelId) body.modelId = modelId;
-  const maxTurnsRaw = ctx.str("max-turns");
-  if (maxTurnsRaw !== "") body.maxTurns = ctx.num("max-turns");
-  return body;
 }
 var query = {
   service: "kb",
@@ -24,9 +21,7 @@ var query = {
   description: "Query knowledge.",
   flags: [
     { name: "query", type: "string", required: true, alias: "q", desc: "Natural language question to query against the knowledge bases" },
-    { name: "sources", type: "json", required: true, desc: 'JSON array of knowledge base refs, e.g. [{"scope":"company","name":"engineering-handbook"}]' },
-    { name: "model-id", type: "string", required: false, default: "AE-Auto", desc: "Model identifier (default: AE-Auto)" },
-    { name: "max-turns", type: "number", required: false, default: 6, desc: "Maximum reasoning turns (default: 6)" }
+    { name: "sources", type: "json", required: true, desc: 'JSON array of knowledge base refs, e.g. [{"scope":"company","name":"engineering-handbook"}]' }
   ],
   risk: "read",
   dryRun: (ctx) => ({
@@ -34,7 +29,7 @@ var query = {
     url: `${ctx.host().replace(/\/$/, "")}${API_PATH}`,
     body: buildBody(ctx)
   }),
-  execute: async (ctx) => ctx.api("POST", API_PATH, {}, buildBody(ctx))
+  execute: async (ctx) => kbApi(ctx, "POST", API_PATH, {}, buildBody(ctx))
 };
 
 // src/commands/te-kb/add.ts
@@ -42,7 +37,26 @@ import { promises as fs } from "fs";
 import * as path from "path";
 import TurndownService from "turndown";
 var API_PATH2 = "/agent/api/external/knowledge-bases/sources/upload";
-var MD_EXT = /* @__PURE__ */ new Set([".md", ".markdown"]);
+var SUPPORTED_EXTENSIONS = /* @__PURE__ */ new Set([
+  ".md",
+  ".markdown",
+  ".txt",
+  ".csv",
+  ".pdf",
+  ".doc",
+  ".docx",
+  ".xls",
+  ".xlsx",
+  ".ppt",
+  ".pptx",
+  ".jpg",
+  ".jpeg",
+  ".png",
+  ".gif",
+  ".webp",
+  ".bmp",
+  ".svg"
+]);
 function normalizeFilesInput(raw) {
   if (!Array.isArray(raw)) {
     throw new Error(`--files must be a JSON array of strings, e.g. '["./a.md","./docs","https://example.com/page"]'`);
@@ -67,13 +81,55 @@ function sanitizeFilename(name) {
   const cleaned = name.replace(/[\\/:*?"<>|\s]+/g, "-").replace(/^-+|-+$/g, "");
   return cleaned || "document";
 }
+function getMimeType(filename) {
+  const ext = path.extname(filename).toLowerCase();
+  switch (ext) {
+    case ".md":
+    case ".markdown":
+      return "text/markdown";
+    case ".txt":
+      return "text/plain";
+    case ".csv":
+      return "text/csv";
+    case ".pdf":
+      return "application/pdf";
+    case ".doc":
+      return "application/msword";
+    case ".docx":
+      return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
+    case ".xls":
+      return "application/vnd.ms-excel";
+    case ".xlsx":
+      return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
+    case ".ppt":
+      return "application/vnd.ms-powerpoint";
+    case ".pptx":
+      return "application/vnd.openxmlformats-officedocument.presentationml.presentation";
+    case ".jpg":
+    case ".jpeg":
+      return "image/jpeg";
+    case ".png":
+      return "image/png";
+    case ".gif":
+      return "image/gif";
+    case ".webp":
+      return "image/webp";
+    case ".bmp":
+      return "image/bmp";
+    case ".svg":
+      return "image/svg+xml";
+    default:
+      return "application/octet-stream";
+  }
+}
 async function readLocalFile(filePath) {
   const ext = path.extname(filePath).toLowerCase();
-  if (!MD_EXT.has(ext)) {
-    throw new Error(`Not a markdown file (only .md / .markdown allowed): ${filePath}`);
+  if (!SUPPORTED_EXTENSIONS.has(ext)) {
+    throw new Error(`Unsupported file extension (${ext || "none"}): ${filePath}`);
   }
-  const content = await fs.readFile(filePath, "utf8");
-  return { filename: path.basename(filePath), content, origin: "file", source: filePath };
+  const content = await fs.readFile(filePath);
+  const filename = path.basename(filePath);
+  return { filename, content, mimeType: getMimeType(filename), origin: "file", source: filePath };
 }
 async function readDirectory(dir) {
   const entries = await fs.readdir(dir, { withFileTypes: true });
@@ -81,13 +137,13 @@ async function readDirectory(dir) {
   for (const entry of entries) {
     if (!entry.isFile()) continue;
     const ext = path.extname(entry.name).toLowerCase();
-    if (!MD_EXT.has(ext)) continue;
+    if (!SUPPORTED_EXTENSIONS.has(ext)) continue;
     const full = path.join(dir, entry.name);
-    const content = await fs.readFile(full, "utf8");
-    files.push({ filename: entry.name, content, origin: "dir", source: full });
+    const content = await fs.readFile(full);
+    files.push({ filename: entry.name, content, mimeType: getMimeType(entry.name), origin: "dir", source: full });
   }
   if (files.length === 0) {
-    throw new Error(`No .md / .markdown files found in directory: ${dir}`);
+    throw new Error(`No supported files found in directory: ${dir}`);
   }
   return files;
 }
@@ -112,6 +168,7 @@ async function fetchAsMarkdown(rawUrl) {
   return {
     filename: deriveFilenameFromUrl(rawUrl),
     content: markdown,
+    mimeType: "text/markdown",
     origin: "url",
     source: rawUrl
   };
@@ -146,7 +203,7 @@ async function collectFiles(ctx) {
     }
   }
   if (collected.length === 0) {
-    throw new Error("No markdown files collected from the given inputs");
+    throw new Error("No supported files collected from the given inputs");
   }
   return dedupeByFilename(collected);
 }
@@ -165,21 +222,21 @@ function buildForm(name, files) {
   const form = new FormData();
   form.append("name", name);
   for (const f of files) {
-    form.append("files", new Blob([f.content], { type: "text/markdown" }), f.filename);
+    form.append("files", new Blob([f.content], { type: f.mimeType }), f.filename);
   }
   return form;
 }
 var add = {
   service: "kb",
   command: "+add",
-  description: "Upload markdown sources to a knowledge. --files accepts a JSON array; each entry can be a .md/.markdown file path, a directory path (all .md/.markdown inside, non-recursive), or an http(s) URL (HTML auto-converted to markdown).",
+  description: "Upload sources to a knowledge. --files accepts a JSON array; each entry can be a supported file path, a directory path (supported files inside, non-recursive), or an http(s) URL (HTML auto-converted to markdown).",
   flags: [
     { name: "name", type: "string", required: true, desc: "Knowledge base name" },
     {
       name: "files",
       type: "json",
       required: true,
-      desc: `JSON array of strings. Each entry can be a .md/.markdown file path, a directory path, or an http(s) URL. Example: '["./a.md","./docs","https://example.com/page"]'`
+      desc: `JSON array of strings. Each entry can be a supported file path, a directory path, or an http(s) URL. Example: '["./a.md","./docs","https://example.com/page"]'`
     }
   ],
   risk: "write",
@@ -202,7 +259,7 @@ var add = {
     const name = ctx.str("name");
     const files = await collectFiles(ctx);
     const form = buildForm(name, files);
-    const result = await httpUpload(API_PATH2, form, {}, ctx.host());
+    const result = await kbUpload(ctx, API_PATH2, form);
     return {
       uploaded: files.map((f) => ({ filename: f.filename, origin: f.origin, source: f.source })),
       result
@@ -243,7 +300,7 @@ var compile = {
     url: `${ctx.host().replace(/\/$/, "")}${API_PATH3}`,
     body: buildBody2(ctx)
   }),
-  execute: async (ctx) => ctx.api("POST", API_PATH3, {}, buildBody2(ctx))
+  execute: async (ctx) => kbApi(ctx, "POST", API_PATH3, {}, buildBody2(ctx))
 };
 
 // src/commands/te-kb/remove.ts
@@ -264,12 +321,17 @@ var remove = {
     url: `${ctx.host().replace(/\/$/, "")}${API_PATH4}`,
     body: buildBody3(ctx)
   }),
-  execute: async (ctx) => httpDelete(API_PATH4, {}, buildBody3(ctx), ctx.host())
+  execute: async (ctx) => kbApi(ctx, "DELETE", API_PATH4, {}, buildBody3(ctx))
 };
 
 // src/commands/te-kb/create.ts
 var API_PATH5 = "/agent/api/external/knowledge-bases/create";
 var VALID_SCOPES = /* @__PURE__ */ new Set(["personal", "company"]);
+function getScope(ctx) {
+  const scope = ctx.str("scope") || "company";
+  validateScope(scope);
+  return scope;
+}
 function validateScope(scope) {
   if (!VALID_SCOPES.has(scope)) {
     throw new Error(`Invalid --scope: ${scope}. Must be one of: personal | company`);
@@ -292,7 +354,7 @@ function normalizeTags(raw) {
 }
 function buildBody4(ctx) {
   const body = {
-    scope: ctx.str("scope"),
+    scope: getScope(ctx),
     name: ctx.str("name")
   };
   const description = ctx.str("description");
@@ -310,7 +372,7 @@ var create = {
   command: "+new",
   description: "Create a new knowledge.",
   flags: [
-    { name: "scope", type: "string", required: true, desc: "Knowledge base scope: personal | company" },
+    { name: "scope", type: "string", required: false, default: "company", desc: "Knowledge base scope: personal | company (default: company)" },
     { name: "name", type: "string", required: true, desc: "Knowledge base name (\u226430 chars, unique per scope)" },
     { name: "description", type: "string", required: false, desc: "Optional description (\u2264200 chars)" },
     { name: "tags", type: "json", required: false, desc: `Optional JSON array of tags (max 2, each \u226415 chars). Example: '["t1","t2"]'` },
@@ -319,7 +381,7 @@ var create = {
   ],
   risk: "write",
   validate: (ctx) => {
-    validateScope(ctx.str("scope"));
+    getScope(ctx);
     normalizeTags(ctx.json("tags"));
   },
   dryRun: (ctx) => ({
@@ -327,7 +389,7 @@ var create = {
     url: `${ctx.host().replace(/\/$/, "")}${API_PATH5}`,
     body: buildBody4(ctx)
   }),
-  execute: async (ctx) => ctx.api("POST", API_PATH5, {}, buildBody4(ctx))
+  execute: async (ctx) => kbApi(ctx, "POST", API_PATH5, {}, buildBody4(ctx))
 };
 
 // src/commands/te-kb/rm-source.ts
@@ -352,7 +414,7 @@ var rmSource = {
     url: `${ctx.host().replace(/\/$/, "")}${API_PATH6}`,
     body: buildBody5(ctx)
   }),
-  execute: async (ctx) => httpDelete(API_PATH6, {}, buildBody5(ctx), ctx.host())
+  execute: async (ctx) => kbApi(ctx, "DELETE", API_PATH6, {}, buildBody5(ctx))
 };
 
 // src/commands/te-kb/schema.ts
@@ -381,7 +443,128 @@ var schema = {
     url: `${ctx.host().replace(/\/$/, "")}${API_PATH7}`,
     body: buildBody6(ctx)
   }),
-  execute: async (ctx) => ctx.api("POST", API_PATH7, {}, buildBody6(ctx))
+  execute: async (ctx) => kbApi(ctx, "POST", API_PATH7, {}, buildBody6(ctx))
+};
+
+// src/commands/te-kb/kb-index.ts
+var API_PATH8 = "/agent/api/external/knowledge-bases/index";
+function buildBody7(ctx) {
+  const body = {};
+  const sources = ctx.json("sources");
+  if (sources) body.sources = sources;
+  const locale = ctx.str("locale");
+  if (locale) body.locale = locale;
+  return body;
+}
+var kbIndex = {
+  service: "kb",
+  command: "+index",
+  description: "List accessible knowledge bases and their index.md navigation maps via POST /agent/api/external/knowledge-bases/index. Call this first to see the layout, then use +grep to locate and +read to open pages.",
+  flags: [
+    {
+      name: "sources",
+      type: "json",
+      required: false,
+      desc: 'Optional JSON array of knowledge base refs to scope the index, e.g. [{"scope":"company","name":"engineering-handbook"}]. Omit to list all accessible knowledge bases.'
+    },
+    { name: "locale", type: "string", required: false, desc: "Optional locale: zh | en | ja | ko" }
+  ],
+  risk: "read",
+  dryRun: (ctx) => ({
+    method: "POST",
+    url: `${ctx.host().replace(/\/$/, "")}${API_PATH8}`,
+    body: buildBody7(ctx)
+  }),
+  execute: async (ctx) => kbApi(ctx, "POST", API_PATH8, {}, buildBody7(ctx))
+};
+
+// src/commands/te-kb/kb-grep.ts
+var API_PATH9 = "/agent/api/external/knowledge-bases/grep";
+function buildBody8(ctx) {
+  const body = {
+    query: ctx.str("query")
+  };
+  const sources = ctx.json("sources");
+  if (sources) body.sources = sources;
+  const topK = ctx.num("top-k");
+  if (topK) body.topK = topK;
+  const locale = ctx.str("locale");
+  if (locale) body.locale = locale;
+  return body;
+}
+var kbGrep = {
+  service: "kb",
+  command: "+grep",
+  description: "Keyword-search knowledge base pages via POST /agent/api/external/knowledge-bases/grep. Returns matched lines with path, line number, breadcrumb and context snippet. Use +read to open a page returned here.",
+  flags: [
+    {
+      name: "query",
+      type: "string",
+      required: true,
+      alias: "q",
+      desc: "Keywords to search across the knowledge bases"
+    },
+    {
+      name: "sources",
+      type: "json",
+      required: false,
+      desc: 'Optional JSON array of knowledge base refs to scope the search, e.g. [{"scope":"company","name":"engineering-handbook"}]. Omit to search all accessible knowledge bases.'
+    },
+    { name: "top-k", type: "number", required: false, desc: "Max number of hits to return (1-50, default 10)" },
+    { name: "locale", type: "string", required: false, desc: "Optional locale: zh | en | ja | ko" }
+  ],
+  risk: "read",
+  dryRun: (ctx) => ({
+    method: "POST",
+    url: `${ctx.host().replace(/\/$/, "")}${API_PATH9}`,
+    body: buildBody8(ctx)
+  }),
+  execute: async (ctx) => kbApi(ctx, "POST", API_PATH9, {}, buildBody8(ctx))
+};
+
+// src/commands/te-kb/kb-read.ts
+var API_PATH10 = "/agent/api/external/knowledge-bases/read";
+function buildBody9(ctx) {
+  const body = {
+    source: ctx.json("source"),
+    path: ctx.str("path")
+  };
+  const offset = ctx.num("offset");
+  if (offset) body.offset = offset;
+  const limit = ctx.num("limit");
+  if (limit) body.limit = limit;
+  const locale = ctx.str("locale");
+  if (locale) body.locale = locale;
+  return body;
+}
+var kbRead = {
+  service: "kb",
+  command: "+read",
+  description: "Read a full knowledge base page (or a line window) via POST /agent/api/external/knowledge-bases/read. Use after +index / +grep locate a candidate page.",
+  flags: [
+    {
+      name: "source",
+      type: "json",
+      required: true,
+      desc: 'Knowledge base ref pointing to exactly one base, e.g. {"scope":"company","name":"engineering-handbook"}'
+    },
+    {
+      name: "path",
+      type: "string",
+      required: true,
+      desc: "Page path relative to the knowledge base root, e.g. wiki/concepts/data-model.md or index.md"
+    },
+    { name: "offset", type: "number", required: false, desc: "Start line (1-based). Omit to read from the beginning." },
+    { name: "limit", type: "number", required: false, desc: "Max number of lines to return (1-2000)." },
+    { name: "locale", type: "string", required: false, desc: "Optional locale: zh | en | ja | ko" }
+  ],
+  risk: "read",
+  dryRun: (ctx) => ({
+    method: "POST",
+    url: `${ctx.host().replace(/\/$/, "")}${API_PATH10}`,
+    body: buildBody9(ctx)
+  }),
+  execute: async (ctx) => kbApi(ctx, "POST", API_PATH10, {}, buildBody9(ctx))
 };
 
 // src/commands/te-kb/index.ts
@@ -392,7 +575,10 @@ var commands = [
   remove,
   create,
   rmSource,
-  schema
+  schema,
+  kbIndex,
+  kbGrep,
+  kbRead
 ];
 var te_kb_default = commands;
 export {
@@ -400,6 +586,9 @@ export {
   compile,
   create,
   te_kb_default as default,
+  kbGrep,
+  kbIndex,
+  kbRead,
   query,
   remove,
   rmSource,

package/dist/{te-meta-25SLZFXJ.js → te-meta-NQ4U3VJW.js}

@@ -2,9 +2,10 @@ import {
   callMcpTool,
   parseMcpResult,
   resolveMcpUrl
-} from "./chunk-MPFTXJFG.js";
-import "./chunk-24ZKQWG2.js";
-import "./chunk-TMMUBSKW.js";
+} from "./chunk-2NGWYMLB.js";
+import "./chunk-SRRUNQIM.js";
+import "./chunk-BI5ZVD6T.js";
+import "./chunk-MNHE6SSI.js";
 
 // src/commands/te-meta/shared.ts
 function createMcpCommand(config) {
@@ -533,8 +534,28 @@ var listEntities = createMcpCommand({
   })
 });
 
+// src/commands/te-meta/entity/create-entity.ts
+var createEntity = createMcpCommand({
+  command: "+create_entity",
+  description: "Create a new entity that links an event or user property as an analysis dimension.",
+  flags: [
+    { name: "project_id", type: "number", required: true, desc: "Project ID", alias: "p" },
+    { name: "entity_name", type: "string", required: true, desc: "Entity name" },
+    { name: "column_name", type: "string", required: true, desc: "Property column name to associate with the entity" },
+    { name: "table_type", type: "number", required: true, desc: "Table type: 0 = event property, 1 = user property" }
+  ],
+  risk: "write",
+  mcpService: "te_analysis_extend",
+  buildArgs: (ctx) => ({
+    projectId: ctx.num("project_id"),
+    entityName: ctx.str("entity_name"),
+    columnName: ctx.str("column_name"),
+    tableType: ctx.num("table_type")
+  })
+});
+
 // src/commands/te-meta/entity/index.ts
-var commands3 = [listEntities];
+var commands3 = [listEntities, createEntity];
 var entity_default = commands3;
 
 // src/commands/te-meta/index.ts