agent-resource-management 1.3.0 → 1.4.1

package/dist/main.js

@@ -230,12 +230,78 @@ class ApiClient {
     }
     return res.arrayBuffer();
   }
+  async createAgent(data) {
+    const res = await this.request("/agents", {
+      method: "POST",
+      body: JSON.stringify(data)
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+  async updateAgent(id, data) {
+    const res = await this.request(`/agents/${id}`, {
+      method: "PUT",
+      body: JSON.stringify(data)
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+  async deleteAgent(id) {
+    const res = await this.request(`/agents/${id}`, {
+      method: "DELETE"
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+  async bindSkillToAgent(agentId, skillId, config) {
+    const res = await this.request(`/agents/${agentId}/skills`, {
+      method: "POST",
+      body: JSON.stringify({ skillId, config })
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+  async unbindSkillFromAgent(agentId, skillId) {
+    const res = await this.request(`/agents/${agentId}/skills?skillId=${skillId}`, {
+      method: "DELETE"
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+  async bindKnowledgeToAgent(agentId, knowledgeId, retrievalConfig) {
+    const res = await this.request(`/agents/${agentId}/knowledges`, {
+      method: "POST",
+      body: JSON.stringify({ knowledgeId, retrievalConfig })
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+  async unbindKnowledgeFromAgent(agentId, knowledgeId) {
+    const res = await this.request(`/agents/${agentId}/knowledges?knowledgeId=${knowledgeId}`, {
+      method: "DELETE"
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+  async getAgentByName(name) {
+    const result = await this.listAgents(name, 1, 1);
+    return result.agents.find((a) => a.name === name) || null;
+  }
 }
 
 // src/lib/storage.ts
 import { writeFileSync, readFileSync, existsSync, mkdirSync } from "fs";
 import { join } from "path";
-var CONFIG_DIR = join(process.env.HOME || "/root", ".adk");
+var CONFIG_DIR = join(process.env.HOME || "/root", ".arm");
 var CONFIG_FILE = join(CONFIG_DIR, "config.json");
 function ensureConfigDir() {
   if (!existsSync(CONFIG_DIR)) {
@@ -821,6 +887,229 @@ import { writeFileSync as writeFileSync3, existsSync as existsSync4 } from "fs";
 import { join as join4 } from "path";
 import { execSync as execSync3 } from "child_process";
 import { mkdtempSync as mkdtempSync3, rmSync as rmSync3 } from "fs";
+function outputJson(result) {
+  console.log(JSON.stringify(result, null, 2));
+}
+function getJsonFlag() {
+  return process.argv.includes("--json") || process.argv.includes("-j");
+}
+async function createAgent(name, options = {}) {
+  const config = loadConfig();
+  if (!config?.token) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "NOT_LOGGED_IN", message: "未登录，请先运行 arm login" } });
+      process.exit(1);
+    }
+    error("未登录，请先运行 arm login");
+    process.exit(1);
+  }
+  const client = new ApiClient(config.serverUrl, config.token);
+  try {
+    const skills = options.skills?.map((skillId, index) => ({
+      skillId,
+      config: options.skillConfigs?.[index] ? JSON.parse(options.skillConfigs[index]) : undefined
+    }));
+    const knowledges = options.knowledges?.map((knowledgeId, index) => ({
+      knowledgeId,
+      retrievalConfig: options.knowledgeConfigs?.[index] ? JSON.parse(options.knowledgeConfigs[index]) : undefined
+    }));
+    const result = await client.createAgent({
+      name,
+      description: options.description,
+      prompt: options.prompt,
+      avatar: options.avatar,
+      skills,
+      knowledges
+    });
+    if (getJsonFlag()) {
+      outputJson({ success: true, data: result });
+      return;
+    }
+    success(`Agent "${name}" 创建成功 (ID: ${result.id})`);
+  } catch (err) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "CREATE_FAILED", message: err instanceof Error ? err.message : "未知错误" } });
+      process.exit(1);
+    }
+    error(`创建失败: ${err instanceof Error ? err.message : "未知错误"}`);
+    process.exit(1);
+  }
+}
+async function updateAgent(id, options = {}) {
+  const config = loadConfig();
+  if (!config?.token) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "NOT_LOGGED_IN", message: "未登录，请先运行 arm login" } });
+      process.exit(1);
+    }
+    error("未登录，请先运行 arm login");
+    process.exit(1);
+  }
+  const client = new ApiClient(config.serverUrl, config.token);
+  try {
+    const updateData = {};
+    if (options.name !== undefined)
+      updateData.name = options.name;
+    if (options.description !== undefined)
+      updateData.description = options.description;
+    if (options.prompt !== undefined)
+      updateData.prompt = options.prompt;
+    if (options.avatar !== undefined)
+      updateData.avatar = options.avatar;
+    if (options.status !== undefined)
+      updateData.status = options.status;
+    const result = await client.updateAgent(id, updateData);
+    if (getJsonFlag()) {
+      outputJson({ success: true, data: result });
+      return;
+    }
+    success(`Agent "${id}" 更新成功`);
+  } catch (err) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "UPDATE_FAILED", message: err instanceof Error ? err.message : "未知错误" } });
+      process.exit(1);
+    }
+    error(`更新失败: ${err instanceof Error ? err.message : "未知错误"}`);
+    process.exit(1);
+  }
+}
+async function deleteAgent(id) {
+  const config = loadConfig();
+  if (!config?.token) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "NOT_LOGGED_IN", message: "未登录，请先运行 arm login" } });
+      process.exit(1);
+    }
+    error("未登录，请先运行 arm login");
+    process.exit(1);
+  }
+  const client = new ApiClient(config.serverUrl, config.token);
+  try {
+    await client.deleteAgent(id);
+    if (getJsonFlag()) {
+      outputJson({ success: true, data: { id } });
+      return;
+    }
+    success(`Agent "${id}" 删除成功`);
+  } catch (err) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "DELETE_FAILED", message: err instanceof Error ? err.message : "未知错误" } });
+      process.exit(1);
+    }
+    error(`删除失败: ${err instanceof Error ? err.message : "未知错误"}`);
+    process.exit(1);
+  }
+}
+async function bindSkill(id, skillId, config) {
+  const configStore = loadConfig();
+  if (!configStore?.token) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "NOT_LOGGED_IN", message: "未登录，请先运行 arm login" } });
+      process.exit(1);
+    }
+    error("未登录，请先运行 arm login");
+    process.exit(1);
+  }
+  const client = new ApiClient(configStore.serverUrl, configStore.token);
+  try {
+    const parsedConfig = config ? JSON.parse(config) : undefined;
+    await client.bindSkillToAgent(id, skillId, parsedConfig);
+    if (getJsonFlag()) {
+      outputJson({ success: true, data: { agentId: id, skillId, config: parsedConfig } });
+      return;
+    }
+    success(`Skill "${skillId}" 已绑定到 Agent "${id}"`);
+  } catch (err) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "BIND_FAILED", message: err instanceof Error ? err.message : "未知错误" } });
+      process.exit(1);
+    }
+    error(`绑定失败: ${err instanceof Error ? err.message : "未知错误"}`);
+    process.exit(1);
+  }
+}
+async function unbindSkill(id, skillId) {
+  const config = loadConfig();
+  if (!config?.token) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "NOT_LOGGED_IN", message: "未登录，请先运行 arm login" } });
+      process.exit(1);
+    }
+    error("未登录，请先运行 arm login");
+    process.exit(1);
+  }
+  const client = new ApiClient(config.serverUrl, config.token);
+  try {
+    await client.unbindSkillFromAgent(id, skillId);
+    if (getJsonFlag()) {
+      outputJson({ success: true, data: { agentId: id, skillId } });
+      return;
+    }
+    success(`Skill "${skillId}" 已从 Agent "${id}" 解绑`);
+  } catch (err) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "UNBIND_FAILED", message: err instanceof Error ? err.message : "未知错误" } });
+      process.exit(1);
+    }
+    error(`解绑失败: ${err instanceof Error ? err.message : "未知错误"}`);
+    process.exit(1);
+  }
+}
+async function bindKnowledge(id, knowledgeId, retrievalConfig) {
+  const configStore = loadConfig();
+  if (!configStore?.token) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "NOT_LOGGED_IN", message: "未登录，请先运行 arm login" } });
+      process.exit(1);
+    }
+    error("未登录，请先运行 arm login");
+    process.exit(1);
+  }
+  const client = new ApiClient(configStore.serverUrl, configStore.token);
+  try {
+    const parsedConfig = retrievalConfig ? JSON.parse(retrievalConfig) : undefined;
+    await client.bindKnowledgeToAgent(id, knowledgeId, parsedConfig);
+    if (getJsonFlag()) {
+      outputJson({ success: true, data: { agentId: id, knowledgeId, retrievalConfig: parsedConfig } });
+      return;
+    }
+    success(`Knowledge "${knowledgeId}" 已绑定到 Agent "${id}"`);
+  } catch (err) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "BIND_FAILED", message: err instanceof Error ? err.message : "未知错误" } });
+      process.exit(1);
+    }
+    error(`绑定失败: ${err instanceof Error ? err.message : "未知错误"}`);
+    process.exit(1);
+  }
+}
+async function unbindKnowledge(id, knowledgeId) {
+  const config = loadConfig();
+  if (!config?.token) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "NOT_LOGGED_IN", message: "未登录，请先运行 arm login" } });
+      process.exit(1);
+    }
+    error("未登录，请先运行 arm login");
+    process.exit(1);
+  }
+  const client = new ApiClient(config.serverUrl, config.token);
+  try {
+    await client.unbindKnowledgeFromAgent(id, knowledgeId);
+    if (getJsonFlag()) {
+      outputJson({ success: true, data: { agentId: id, knowledgeId } });
+      return;
+    }
+    success(`Knowledge "${knowledgeId}" 已从 Agent "${id}" 解绑`);
+  } catch (err) {
+    if (getJsonFlag()) {
+      outputJson({ success: false, error: { code: "UNBIND_FAILED", message: err instanceof Error ? err.message : "未知错误" } });
+      process.exit(1);
+    }
+    error(`解绑失败: ${err instanceof Error ? err.message : "未知错误"}`);
+    process.exit(1);
+  }
+}
 async function listAgents() {
   const config = loadConfig();
   if (!config?.token) {
@@ -1283,6 +1572,145 @@ async function main() {
           }
           await downloadAgent(args[2], args[3]);
           break;
+        case "create":
+          if (!args[2]) {
+            console.error(`用法: arm agent create <name> [--description="..."] [--prompt="..."] [--avatar="..."] [--skill=id] [--knowledge=id] [--skill-config='{...}'] [--knowledge-config='{...}'] [--json]`);
+            process.exit(1);
+          }
+          {
+            const name = args[2];
+            const options = {};
+            const skills = [];
+            const knowledges = [];
+            const skillConfigs = [];
+            const knowledgeConfigs = [];
+            for (let i = 3;i < args.length; i++) {
+              const arg = args[i];
+              if (arg.startsWith("--description=")) {
+                options.description = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--prompt=")) {
+                options.prompt = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--avatar=")) {
+                options.avatar = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--skill=")) {
+                skills.push(arg.split("=").slice(1).join("="));
+              } else if (arg.startsWith("--knowledge=")) {
+                knowledges.push(arg.split("=").slice(1).join("="));
+              } else if (arg.startsWith("--skill-config=")) {
+                skillConfigs.push(arg.split("=").slice(1).join("="));
+              } else if (arg.startsWith("--knowledge-config=")) {
+                knowledgeConfigs.push(arg.split("=").slice(1).join("="));
+              }
+            }
+            await createAgent(name, {
+              description: options.description,
+              prompt: options.prompt,
+              avatar: options.avatar,
+              skills,
+              knowledges,
+              skillConfigs,
+              knowledgeConfigs
+            });
+          }
+          break;
+        case "update":
+          if (!args[2]) {
+            console.error('用法: arm agent update <id> [--name="..."] [--description="..."] [--prompt="..."] [--avatar="..."] [--status=active|draft] [--json]');
+            process.exit(1);
+          }
+          {
+            const id = args[2];
+            const options = {};
+            for (let i = 3;i < args.length; i++) {
+              const arg = args[i];
+              if (arg.startsWith("--name=")) {
+                options.name = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--description=")) {
+                options.description = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--prompt=")) {
+                options.prompt = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--avatar=")) {
+                options.avatar = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--status=")) {
+                options.status = arg.split("=").slice(1).join("=");
+              }
+            }
+            await updateAgent(id, {
+              name: options.name,
+              description: options.description,
+              prompt: options.prompt,
+              avatar: options.avatar,
+              status: options.status
+            });
+          }
+          break;
+        case "delete":
+          if (!args[2]) {
+            console.error("用法: arm agent delete <id> [--json]");
+            process.exit(1);
+          }
+          await deleteAgent(args[2]);
+          break;
+        case "bind":
+          if (!args[2]) {
+            console.error("用法: arm agent bind <id> --skill=<skillId> [--skill-config='{...}'] 或 arm agent bind <id> --knowledge=<knowledgeId> [--knowledge-config='{...}'] [--json]");
+            process.exit(1);
+          }
+          {
+            const id = args[2];
+            let skillId;
+            let knowledgeId;
+            let skillConfig;
+            let knowledgeConfig;
+            for (let i = 3;i < args.length; i++) {
+              const arg = args[i];
+              if (arg.startsWith("--skill=")) {
+                skillId = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--knowledge=")) {
+                knowledgeId = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--skill-config=")) {
+                skillConfig = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--knowledge-config=")) {
+                knowledgeConfig = arg.split("=").slice(1).join("=");
+              }
+            }
+            if (skillId) {
+              await bindSkill(id, skillId, skillConfig);
+            } else if (knowledgeId) {
+              await bindKnowledge(id, knowledgeId, knowledgeConfig);
+            } else {
+              console.error("用法: arm agent bind <id> --skill=<skillId> 或 --knowledge=<knowledgeId>");
+              process.exit(1);
+            }
+          }
+          break;
+        case "unbind":
+          if (!args[2]) {
+            console.error("用法: arm agent unbind <id> --skill=<skillId> 或 --knowledge=<knowledgeId> [--json]");
+            process.exit(1);
+          }
+          {
+            const id = args[2];
+            let skillId;
+            let knowledgeId;
+            for (let i = 3;i < args.length; i++) {
+              const arg = args[i];
+              if (arg.startsWith("--skill=")) {
+                skillId = arg.split("=").slice(1).join("=");
+              } else if (arg.startsWith("--knowledge=")) {
+                knowledgeId = arg.split("=").slice(1).join("=");
+              }
+            }
+            if (skillId) {
+              await unbindSkill(id, skillId);
+            } else if (knowledgeId) {
+              await unbindKnowledge(id, knowledgeId);
+            } else {
+              console.error("用法: arm agent unbind <id> --skill=<skillId> 或 --knowledge=<knowledgeId>");
+              process.exit(1);
+            }
+          }
+          break;
         default:
           console.log(`
 可用命令:
@@ -1290,6 +1718,14 @@ async function main() {
   arm agent search <keyword>          搜索 Agent
   arm agent info <name>               查看 Agent 详情
   arm agent download <name> [dir]     下载 Agent
+  arm agent create <name>             创建 Agent (--description, --prompt, --avatar, --skill, --knowledge)
+  arm agent update <id>               更新 Agent (--name, --description, --prompt, --avatar, --status)
+  arm agent delete <id>                删除 Agent
+  arm agent bind <id> --skill=<id>    绑定 Skill 到 Agent
+  arm agent unbind <id> --skill=<id>  解绑 Skill
+  arm agent bind <id> --knowledge=<id> 绑定 Knowledge 到 Agent
+  arm agent unbind <id> --knowledge=<id> 解绑 Knowledge
+  所有命令支持 --json 参数获取机器可读输出
 `);
       }
       break;
@@ -1319,8 +1755,16 @@ Agent Resource Management (arm)
   arm agent search <keyword>          搜索 Agent
   arm agent info <name>               查看 Agent 详情
   arm agent download <name> [dir]     下载 Agent
+  arm agent create <name>             创建 Agent
+  arm agent update <id>               更新 Agent
+  arm agent delete <id>                删除 Agent
+  arm agent bind <id> --skill=<id>    绑定 Skill
+  arm agent unbind <id> --skill=<id>  解绑 Skill
+  arm agent bind <id> --knowledge=<id> 绑定 Knowledge
+  arm agent unbind <id> --knowledge=<id> 解绑 Knowledge
   arm server                          显示当前服务端
   arm server set <url>                设置服务端
+  使用 arm <entity> -h 查看详细帮助
 `);
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-resource-management",
-  "version": "1.3.0",
+  "version": "1.4.1",
   "type": "module",
   "bin": {
     "arm": "./dist/main.js"