agent-resource-management 1.3.0 → 1.4.1

package/src/cmd/agent.ts

@@ -6,6 +6,293 @@ import { join } from 'path';
 import { execSync } from 'child_process';
 import { mkdtempSync, rmSync } from 'fs';
 
+interface JsonResult<T = unknown> {
+  success: boolean;
+  data?: T;
+  error?: {
+    code: string;
+    message: string;
+  };
+}
+
+function outputJson<T>(result: JsonResult<T>): void {
+  console.log(JSON.stringify(result, null, 2));
+}
+
+function getJsonFlag(): boolean {
+  return process.argv.includes('--json') || process.argv.includes('-j');
+}
+
+function getAgentIdentifier(identifier: string): { id: string; name: string } {
+  if (identifier.includes('-')) {
+    return { id: identifier, name: identifier };
+  }
+  return { id: identifier, name: identifier };
+}
+
+export async function createAgent(
+  name: string,
+  options: {
+    description?: string;
+    prompt?: string;
+    avatar?: string;
+    skills?: string[];
+    knowledges?: string[];
+    skillConfigs?: string[];
+    knowledgeConfigs?: string[];
+  } = {}
+): Promise<void> {
+  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);
+  }
+}
+
+export async function updateAgent(
+  id: string,
+  options: {
+    name?: string;
+    description?: string;
+    prompt?: string;
+    avatar?: string;
+    status?: 'active' | 'draft';
+  } = {}
+): Promise<void> {
+  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: Partial<{
+      name: string;
+      description: string;
+      prompt: string;
+      avatar: string;
+      status: 'active' | 'draft';
+    }> = {};
+
+    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);
+  }
+}
+
+export async function deleteAgent(id: string): Promise<void> {
+  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);
+  }
+}
+
+export async function bindSkill(id: string, skillId: string, config?: string): Promise<void> {
+  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);
+  }
+}
+
+export async function unbindSkill(id: string, skillId: string): Promise<void> {
+  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);
+  }
+}
+
+export async function bindKnowledge(id: string, knowledgeId: string, retrievalConfig?: string): Promise<void> {
+  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);
+  }
+}
+
+export async function unbindKnowledge(id: string, knowledgeId: string): Promise<void> {
+  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);
+  }
+}
+
 export async function listAgents(): Promise<void> {
   const config = loadConfig();
   if (!config?.token) {

package/src/lib/client.ts

@@ -271,4 +271,91 @@ export class ApiClient {
 
     return res.arrayBuffer();
   }
+
+  async createAgent(data: {
+    name: string;
+    description?: string;
+    prompt?: string;
+    avatar?: string;
+    skills?: Array<{ skillId: string; config?: Record<string, unknown> }>;
+    knowledges?: Array<{ knowledgeId: string; retrievalConfig?: { topK?: number; similarityThreshold?: number } }>;
+  }): Promise<Agent> {
+    const res = await this.request<Agent>('/agents', {
+      method: 'POST',
+      body: JSON.stringify(data),
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async updateAgent(id: string, data: Partial<{
+    name: string;
+    description: string;
+    prompt: string;
+    avatar: string;
+    status: 'active' | 'draft';
+  }>): Promise<Agent> {
+    const res = await this.request<Agent>(`/agents/${id}`, {
+      method: 'PUT',
+      body: JSON.stringify(data),
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async deleteAgent(id: string): Promise<void> {
+    const res = await this.request<null>(`/agents/${id}`, {
+      method: 'DELETE',
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+
+  async bindSkillToAgent(agentId: string, skillId: string, config?: Record<string, unknown>): Promise<void> {
+    const res = await this.request<null>(`/agents/${agentId}/skills`, {
+      method: 'POST',
+      body: JSON.stringify({ skillId, config }),
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+
+  async unbindSkillFromAgent(agentId: string, skillId: string): Promise<void> {
+    const res = await this.request<null>(`/agents/${agentId}/skills?skillId=${skillId}`, {
+      method: 'DELETE',
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+
+  async bindKnowledgeToAgent(agentId: string, knowledgeId: string, retrievalConfig?: { topK?: number; similarityThreshold?: number }): Promise<void> {
+    const res = await this.request<null>(`/agents/${agentId}/knowledges`, {
+      method: 'POST',
+      body: JSON.stringify({ knowledgeId, retrievalConfig }),
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+
+  async unbindKnowledgeFromAgent(agentId: string, knowledgeId: string): Promise<void> {
+    const res = await this.request<null>(`/agents/${agentId}/knowledges?knowledgeId=${knowledgeId}`, {
+      method: 'DELETE',
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+
+  async getAgentByName(name: string): Promise<Agent | null> {
+    const result = await this.listAgents(name, 1, 1);
+    return result.agents.find(a => a.name === name) || null;
+  }
 }

package/src/lib/storage.ts

@@ -1,7 +1,7 @@
 import { writeFileSync, readFileSync, existsSync, mkdirSync } from 'fs';
 import { join } from 'path';
 
-const CONFIG_DIR = join(process.env.HOME || '/root', '.adk');
+const CONFIG_DIR = join(process.env.HOME || '/root', '.arm');
 const CONFIG_FILE = join(CONFIG_DIR, 'config.json');
 
 interface Config {

package/src/main.ts

@@ -1,6 +1,6 @@
 import { login, logout, getCurrentUser } from './cmd/auth';
 import { listSkills, searchSkills, infoSkill, downloadSkill, uploadSkill, mySkills, deleteSkill, validateSkill } from './cmd/skill';
-import { listAgents, searchAgents, infoAgent, downloadAgent } from './cmd/agent';
+import { listAgents, searchAgents, infoAgent, downloadAgent, createAgent, updateAgent, deleteAgent, bindSkill, unbindSkill, bindKnowledge, unbindKnowledge } from './cmd/agent';
 import { listKnowledge, searchKnowledge, infoKnowledge, downloadKnowledge, uploadKnowledge, myKnowledge, deleteKnowledge } from './cmd/knowledge';
 import { showServer, setServer } from './cmd/server';
 
@@ -186,6 +186,153 @@ 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: Record<string, string | undefined> = {};
+            const skills: string[] = [];
+            const knowledges: string[] = [];
+            const skillConfigs: string[] = [];
+            const knowledgeConfigs: string[] = [];
+
+            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: Record<string, string | undefined> = {};
+
+            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 as 'active' | 'draft' | undefined,
+            });
+          }
+          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: string | undefined;
+            let knowledgeId: string | undefined;
+            let skillConfig: string | undefined;
+            let knowledgeConfig: string | undefined;
+
+            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: string | undefined;
+            let knowledgeId: string | undefined;
+
+            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(`
 可用命令:
@@ -193,6 +340,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;
@@ -223,8 +378,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 查看详细帮助
 `);
   }
 }