@thinkingai/ae-cli 1.0.22 → 1.0.27

package/dist/te-agent-FLTDYV3L.js

@@ -0,0 +1,522 @@
+import {
+  deleteFromMainApp,
+  getFromMainApp,
+  patchToMainApp,
+  postToMainApp,
+  uploadToMainApp
+} from "./chunk-TG64AQV4.js";
+
+// src/commands/te-agent/models.ts
+var BASE_PATH = "/api/sandbox/agent/models";
+var listModels = {
+  service: "agent",
+  command: "+list-models",
+  description: "List models visible to current user (personal / company / system)",
+  flags: [
+    { name: "scope", type: "string", required: false, desc: "Filter by scope: personal | company | system" }
+  ],
+  risk: "read",
+  validate: (ctx) => {
+    const scope = ctx.str("scope");
+    if (scope && !["personal", "company", "system"].includes(scope)) {
+      throw new Error("--scope \u5FC5\u987B\u662F personal\u3001company \u6216 system");
+    }
+  },
+  dryRun: (ctx) => {
+    const scope = ctx.str("scope");
+    const qs = scope ? `?scope=${encodeURIComponent(scope)}` : "";
+    return { method: "GET", url: `${BASE_PATH}${qs}` };
+  },
+  execute: async (ctx) => {
+    const scope = ctx.str("scope");
+    const qs = scope ? `?scope=${encodeURIComponent(scope)}` : "";
+    return getFromMainApp(`${BASE_PATH}${qs}`);
+  }
+};
+var addModel = {
+  service: "agent",
+  command: "+add-model",
+  description: "Add a custom model (personal scope)",
+  flags: [
+    { name: "model-id", type: "string", required: true, desc: "Model identifier (e.g. gpt-4o)" },
+    { name: "name", type: "string", required: true, desc: "Display name" },
+    { name: "base-url", type: "string", required: true, desc: "API base URL" },
+    { name: "api-key", type: "string", required: false, desc: "API key" },
+    { name: "provider", type: "string", required: false, desc: "Provider name" },
+    { name: "description", type: "string", required: false, desc: "Description" },
+    { name: "context-length", type: "number", required: false, desc: "Context window length" }
+  ],
+  risk: "write",
+  validate: (ctx) => {
+    const baseUrl = ctx.str("baseUrl");
+    try {
+      new URL(baseUrl);
+    } catch {
+      throw new Error("--base-url \u5FC5\u987B\u662F\u6709\u6548 URL");
+    }
+  },
+  dryRun: (ctx) => ({
+    method: "POST",
+    url: BASE_PATH,
+    body: {
+      modelId: ctx.str("modelId"),
+      displayName: ctx.str("name"),
+      baseUrl: ctx.str("baseUrl"),
+      apiKey: ctx.str("apiKey") || void 0,
+      provider: ctx.str("provider") || void 0,
+      description: ctx.str("description") || void 0,
+      contextLength: ctx.num("contextLength") || void 0
+    }
+  }),
+  execute: async (ctx) => {
+    return postToMainApp(BASE_PATH, {
+      modelId: ctx.str("modelId"),
+      displayName: ctx.str("name"),
+      baseUrl: ctx.str("baseUrl"),
+      apiKey: ctx.str("apiKey") || void 0,
+      provider: ctx.str("provider") || void 0,
+      description: ctx.str("description") || void 0,
+      contextLength: ctx.num("contextLength") || void 0
+    });
+  }
+};
+var delModel = {
+  service: "agent",
+  command: "+del-model",
+  description: "Delete a personal model",
+  flags: [
+    { name: "id", type: "string", required: true, desc: "Model record ID (CUID)" }
+  ],
+  risk: "write",
+  dryRun: (ctx) => ({
+    method: "DELETE",
+    url: `${BASE_PATH}?id=${encodeURIComponent(ctx.str("id"))}`
+  }),
+  execute: async (ctx) => {
+    return deleteFromMainApp(`${BASE_PATH}?id=${encodeURIComponent(ctx.str("id"))}`);
+  }
+};
+var toggleModel = {
+  service: "agent",
+  command: "+toggle-model",
+  description: "Enable or disable a model",
+  flags: [
+    { name: "id", type: "string", required: true, desc: "Model record ID (CUID)" },
+    { name: "enabled", type: "boolean", required: true, desc: "true to enable, false to disable" }
+  ],
+  risk: "write",
+  dryRun: (ctx) => ({
+    method: "PATCH",
+    url: BASE_PATH,
+    body: { id: ctx.str("id"), enabled: ctx.bool("enabled") }
+  }),
+  execute: async (ctx) => {
+    return patchToMainApp(BASE_PATH, {
+      id: ctx.str("id"),
+      enabled: ctx.bool("enabled")
+    });
+  }
+};
+
+// src/commands/te-agent/mcps.ts
+var BASE_PATH2 = "/api/sandbox/agent/mcp-servers";
+var listMcps = {
+  service: "agent",
+  command: "+list-mcps",
+  description: "List MCP servers visible to current user",
+  flags: [
+    { name: "scope", type: "string", required: false, desc: "Filter by scope: personal | company | system" }
+  ],
+  risk: "read",
+  validate: (ctx) => {
+    const scope = ctx.str("scope");
+    if (scope && !["personal", "company", "system"].includes(scope)) {
+      throw new Error("--scope \u5FC5\u987B\u662F personal\u3001company \u6216 system");
+    }
+  },
+  dryRun: (ctx) => {
+    const scope = ctx.str("scope");
+    const qs = scope ? `?scope=${encodeURIComponent(scope)}` : "";
+    return { method: "GET", url: `${BASE_PATH2}${qs}` };
+  },
+  execute: async (ctx) => {
+    const scope = ctx.str("scope");
+    const qs = scope ? `?scope=${encodeURIComponent(scope)}` : "";
+    return getFromMainApp(`${BASE_PATH2}${qs}`);
+  }
+};
+var addMcp = {
+  service: "agent",
+  command: "+add-mcp",
+  description: "Add an MCP server (personal scope)",
+  flags: [
+    { name: "name", type: "string", required: true, desc: "Server name (2-64 chars, letter start, [a-zA-Z0-9_-])" },
+    { name: "url", type: "string", required: true, desc: "MCP server URL" },
+    { name: "display-name", type: "string", required: false, desc: "Display name (max 100)" },
+    { name: "description", type: "string", required: false, desc: "Description (max 255)" },
+    { name: "transport", type: "string", required: false, default: "http", desc: "Transport: sse | http" },
+    { name: "headers", type: "json", required: false, desc: "HTTP headers as JSON object" }
+  ],
+  risk: "write",
+  validate: (ctx) => {
+    const name = ctx.str("name");
+    if (!/^[a-zA-Z][a-zA-Z0-9_-]*$/.test(name)) {
+      throw new Error("--name \u5FC5\u987B\u4EE5\u5B57\u6BCD\u5F00\u5934\uFF0C\u4EC5\u5305\u542B\u5B57\u6BCD\u3001\u6570\u5B57\u3001_\u3001-");
+    }
+    if (name.length < 2 || name.length > 64) {
+      throw new Error("--name \u957F\u5EA6\u5FC5\u987B\u5728 2-64 \u4E4B\u95F4");
+    }
+    try {
+      new URL(ctx.str("url"));
+    } catch {
+      throw new Error("--url \u5FC5\u987B\u662F\u6709\u6548 URL");
+    }
+    if (!/^https?:\/\//i.test(ctx.str("url"))) {
+      throw new Error("--url \u4EC5\u652F\u6301 http:// \u6216 https:// \u534F\u8BAE");
+    }
+    const transport = ctx.str("transport");
+    if (transport && !["sse", "http"].includes(transport)) {
+      throw new Error("--transport \u5FC5\u987B\u662F sse \u6216 http");
+    }
+  },
+  dryRun: (ctx) => ({
+    method: "POST",
+    url: BASE_PATH2,
+    body: {
+      name: ctx.str("name"),
+      url: ctx.str("url"),
+      displayName: ctx.str("displayName") || void 0,
+      description: ctx.str("description") || void 0,
+      transport: ctx.str("transport") || "http",
+      headers: ctx.json("headers") || void 0
+    }
+  }),
+  execute: async (ctx) => {
+    return postToMainApp(BASE_PATH2, {
+      name: ctx.str("name"),
+      url: ctx.str("url"),
+      displayName: ctx.str("displayName") || void 0,
+      description: ctx.str("description") || void 0,
+      transport: ctx.str("transport") || "http",
+      headers: ctx.json("headers") || void 0
+    });
+  }
+};
+var delMcp = {
+  service: "agent",
+  command: "+del-mcp",
+  description: "Delete a personal MCP server",
+  flags: [
+    { name: "id", type: "string", required: true, desc: "MCP server record ID (CUID)" }
+  ],
+  risk: "write",
+  dryRun: (ctx) => ({
+    method: "DELETE",
+    url: `${BASE_PATH2}?id=${encodeURIComponent(ctx.str("id"))}`
+  }),
+  execute: async (ctx) => {
+    return deleteFromMainApp(`${BASE_PATH2}?id=${encodeURIComponent(ctx.str("id"))}`);
+  }
+};
+var toggleMcp = {
+  service: "agent",
+  command: "+toggle-mcp",
+  description: "Enable or disable an MCP server",
+  flags: [
+    { name: "id", type: "string", required: true, desc: "MCP server record ID (CUID)" },
+    { name: "enabled", type: "boolean", required: true, desc: "true to enable, false to disable" }
+  ],
+  risk: "write",
+  dryRun: (ctx) => ({
+    method: "PATCH",
+    url: BASE_PATH2,
+    body: { id: ctx.str("id"), enabled: ctx.bool("enabled") }
+  }),
+  execute: async (ctx) => {
+    return patchToMainApp(BASE_PATH2, {
+      id: ctx.str("id"),
+      enabled: ctx.bool("enabled")
+    });
+  }
+};
+
+// src/commands/te-agent/skills.ts
+var BASE_PATH3 = "/api/sandbox/agent/skills";
+async function resolveInstructions(raw) {
+  if (raw === "@-") {
+    const chunks = [];
+    for await (const chunk of process.stdin) {
+      chunks.push(chunk);
+    }
+    const text = Buffer.concat(chunks).toString("utf8").trim();
+    if (!text) throw new Error("stdin \u4E3A\u7A7A\uFF0C\u65E0\u6CD5\u8BFB\u53D6 instructions");
+    return text;
+  }
+  return raw;
+}
+var listSkills = {
+  service: "agent",
+  command: "+list-skills",
+  description: "List Skills visible to current user",
+  flags: [
+    { name: "scope", type: "string", required: false, desc: "Filter by scope: personal | company | system" }
+  ],
+  risk: "read",
+  validate: (ctx) => {
+    const scope = ctx.str("scope");
+    if (scope && !["personal", "company", "system"].includes(scope)) {
+      throw new Error("--scope \u5FC5\u987B\u662F personal\u3001company \u6216 system");
+    }
+  },
+  dryRun: (ctx) => {
+    const scope = ctx.str("scope");
+    const qs = scope ? `?scope=${encodeURIComponent(scope)}` : "";
+    return { method: "GET", url: `${BASE_PATH3}${qs}` };
+  },
+  execute: async (ctx) => {
+    const scope = ctx.str("scope");
+    const qs = scope ? `?scope=${encodeURIComponent(scope)}` : "";
+    return getFromMainApp(`${BASE_PATH3}${qs}`);
+  }
+};
+var addSkill = {
+  service: "agent",
+  command: "+add-skill",
+  description: "Create a custom Skill (personal scope)",
+  flags: [
+    { name: "name", type: "string", required: true, desc: "Skill name (1-80 chars)" },
+    { name: "description", type: "string", required: true, desc: "Skill description" },
+    { name: "instructions", type: "string", required: true, desc: "Skill instructions (use @- to read from stdin)" },
+    { name: "display-name", type: "string", required: false, desc: "Display name (max 100)" }
+  ],
+  risk: "write",
+  validate: (ctx) => {
+    const name = ctx.str("name");
+    if (name.length < 1 || name.length > 80) {
+      throw new Error("--name \u957F\u5EA6\u5FC5\u987B\u5728 1-80 \u4E4B\u95F4");
+    }
+  },
+  dryRun: (ctx) => ({
+    method: "POST",
+    url: BASE_PATH3,
+    body: {
+      name: ctx.str("name"),
+      description: ctx.str("description"),
+      instructions: ctx.str("instructions") === "@-" ? "(from stdin)" : ctx.str("instructions"),
+      displayName: ctx.str("displayName") || void 0
+    }
+  }),
+  execute: async (ctx) => {
+    const instructions = await resolveInstructions(ctx.str("instructions"));
+    return postToMainApp(BASE_PATH3, {
+      name: ctx.str("name"),
+      description: ctx.str("description"),
+      instructions,
+      displayName: ctx.str("displayName") || void 0
+    });
+  }
+};
+var delSkill = {
+  service: "agent",
+  command: "+del-skill",
+  description: "Delete a personal Skill (physical delete)",
+  flags: [
+    { name: "id", type: "string", required: true, desc: "Skill record ID (CUID)" }
+  ],
+  risk: "write",
+  dryRun: (ctx) => ({
+    method: "DELETE",
+    url: `${BASE_PATH3}?id=${encodeURIComponent(ctx.str("id"))}`
+  }),
+  execute: async (ctx) => {
+    return deleteFromMainApp(`${BASE_PATH3}?id=${encodeURIComponent(ctx.str("id"))}`);
+  }
+};
+var toggleSkill = {
+  service: "agent",
+  command: "+toggle-skill",
+  description: "Enable or disable a Skill",
+  flags: [
+    { name: "id", type: "string", required: true, desc: "Skill record ID (CUID)" },
+    { name: "enabled", type: "boolean", required: true, desc: "true to enable, false to disable" }
+  ],
+  risk: "write",
+  dryRun: (ctx) => ({
+    method: "PATCH",
+    url: BASE_PATH3,
+    body: { id: ctx.str("id"), enabled: ctx.bool("enabled") }
+  }),
+  execute: async (ctx) => {
+    return patchToMainApp(BASE_PATH3, {
+      id: ctx.str("id"),
+      enabled: ctx.bool("enabled")
+    });
+  }
+};
+
+// src/commands/te-agent/attachments.ts
+import { readFileSync, existsSync } from "fs";
+import { resolve, basename, extname } from "path";
+var LIST_PATH = "/api/sandbox/agent/attachments";
+var UPLOAD_PATH = "/api/sandbox/agent/attachments/upload";
+function guessMimeType(filePath) {
+  const ext = extname(filePath).toLowerCase();
+  const map = {
+    ".png": "image/png",
+    ".jpg": "image/jpeg",
+    ".jpeg": "image/jpeg",
+    ".gif": "image/gif",
+    ".webp": "image/webp",
+    ".svg": "image/svg+xml",
+    ".pdf": "application/pdf",
+    ".md": "text/markdown",
+    ".txt": "text/plain",
+    ".csv": "text/csv",
+    ".json": "application/json",
+    ".html": "text/html",
+    ".xml": "application/xml",
+    ".doc": "application/msword",
+    ".docx": "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+    ".xls": "application/vnd.ms-excel",
+    ".xlsx": "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+    ".pptx": "application/vnd.openxmlformats-officedocument.presentationml.presentation",
+    ".zip": "application/zip"
+  };
+  return map[ext] || "application/octet-stream";
+}
+function resolveFilePaths(ctx) {
+  const file = ctx.str("file");
+  const files = ctx.str("files");
+  if (file && files) {
+    throw new Error("--file \u548C --files \u4E0D\u80FD\u540C\u65F6\u4F7F\u7528");
+  }
+  if (!file && !files) {
+    throw new Error("\u5FC5\u987B\u6307\u5B9A --file \u6216 --files");
+  }
+  if (file) return [file];
+  try {
+    const parsed = JSON.parse(files);
+    if (!Array.isArray(parsed)) throw new Error("--files \u5FC5\u987B\u662F JSON \u6570\u7EC4");
+    return parsed.map((p) => {
+      if (typeof p !== "string") throw new Error("--files \u6570\u7EC4\u5143\u7D20\u5FC5\u987B\u662F\u5B57\u7B26\u4E32");
+      return p;
+    });
+  } catch (err) {
+    if (err instanceof SyntaxError) throw new Error("--files \u5FC5\u987B\u662F\u6709\u6548 JSON \u6570\u7EC4");
+    throw err;
+  }
+}
+async function uploadFiles(filePaths) {
+  const formData = new FormData();
+  for (const filePath of filePaths) {
+    const resolved = resolve(filePath);
+    if (!existsSync(resolved)) {
+      throw new Error(`\u6587\u4EF6\u4E0D\u5B58\u5728\uFF1A${resolved}`);
+    }
+    const buffer = readFileSync(resolved);
+    const mimeType = guessMimeType(resolved);
+    const blob = new Blob([buffer], { type: mimeType });
+    formData.append("file", blob, basename(resolved));
+  }
+  return uploadToMainApp(UPLOAD_PATH, formData);
+}
+var listAttachments = {
+  service: "agent",
+  command: "+list-attachments",
+  description: "List user attachments (paginated)",
+  flags: [
+    { name: "q", type: "string", required: false, desc: "Search keyword" },
+    { name: "type", type: "string", required: false, desc: "Filter by type: image | document" },
+    { name: "page", type: "number", required: false, desc: "Page number (default 1)" },
+    { name: "page-size", type: "number", required: false, desc: "Page size (default 20, max 100)" }
+  ],
+  risk: "read",
+  validate: (ctx) => {
+    const type = ctx.str("type");
+    if (type && !["image", "document"].includes(type)) {
+      throw new Error("--type \u5FC5\u987B\u662F image \u6216 document");
+    }
+  },
+  dryRun: (ctx) => {
+    const params = new URLSearchParams();
+    if (ctx.str("q")) params.set("q", ctx.str("q"));
+    if (ctx.str("type")) params.set("type", ctx.str("type"));
+    if (ctx.num("page")) params.set("page", String(ctx.num("page")));
+    if (ctx.num("pageSize")) params.set("pageSize", String(ctx.num("pageSize")));
+    const qs = params.toString();
+    return { method: "GET", url: `${LIST_PATH}${qs ? "?" + qs : ""}` };
+  },
+  execute: async (ctx) => {
+    const params = new URLSearchParams();
+    if (ctx.str("q")) params.set("q", ctx.str("q"));
+    if (ctx.str("type")) params.set("type", ctx.str("type"));
+    if (ctx.num("page")) params.set("page", String(ctx.num("page")));
+    if (ctx.num("pageSize")) params.set("pageSize", String(ctx.num("pageSize")));
+    const qs = params.toString();
+    return getFromMainApp(`${LIST_PATH}${qs ? "?" + qs : ""}`);
+  }
+};
+var addAttachment = {
+  service: "agent",
+  command: "+add-attachment",
+  description: "Upload sandbox file(s) to attachment library",
+  flags: [
+    { name: "file", type: "string", required: false, desc: "Single file path" },
+    { name: "files", type: "string", required: false, desc: `JSON array of file paths, e.g. '["./a.png","./b.pdf"]'` }
+  ],
+  risk: "write",
+  validate: (ctx) => {
+    resolveFilePaths(ctx);
+  },
+  dryRun: (ctx) => {
+    const paths = resolveFilePaths(ctx);
+    return { method: "POST", url: UPLOAD_PATH, body: { files: paths } };
+  },
+  execute: async (ctx) => {
+    const paths = resolveFilePaths(ctx);
+    return uploadFiles(paths);
+  }
+};
+var delAttachment = {
+  service: "agent",
+  command: "+del-attachment",
+  description: "Soft-delete an attachment",
+  flags: [
+    { name: "id", type: "string", required: true, desc: "Attachment ID" }
+  ],
+  risk: "write",
+  dryRun: (ctx) => ({
+    method: "DELETE",
+    url: `${LIST_PATH}?id=${encodeURIComponent(ctx.str("id"))}`
+  }),
+  execute: async (ctx) => {
+    return deleteFromMainApp(`${LIST_PATH}?id=${encodeURIComponent(ctx.str("id"))}`);
+  }
+};
+
+// src/commands/te-agent/index.ts
+var commands = [
+  // 模型管理
+  listModels,
+  addModel,
+  delModel,
+  toggleModel,
+  // MCP 服务管理
+  listMcps,
+  addMcp,
+  delMcp,
+  toggleMcp,
+  // Skill 管理
+  listSkills,
+  addSkill,
+  delSkill,
+  toggleSkill,
+  // 附件库管理
+  listAttachments,
+  addAttachment,
+  delAttachment
+];
+var te_agent_default = commands;
+export {
+  te_agent_default as default
+};