agent-resource-management 1.3.0

package/src/lib/client.ts

@@ -0,0 +1,274 @@
+import type { User, Skill, SkillListResponse, ApiResponse, LoginResponse, Agent, AgentListResponse } from '@pkg/types/skill';
+
+export class ApiClient {
+  private token: string | null = null;
+  private serverUrl: string;
+
+  constructor(serverUrl: string, token?: string) {
+    this.serverUrl = serverUrl.replace(/\/$/, '');
+    this.token = token || null;
+  }
+
+  setToken(token: string) {
+    this.token = token;
+  }
+
+  clearToken() {
+    this.token = null;
+  }
+
+  private async request<T>(
+    path: string,
+    options: RequestInit = {}
+  ): Promise<ApiResponse<T>> {
+    const headers: Record<string, string> = {
+      'Content-Type': 'application/json',
+      ...(options.headers as Record<string, string>),
+    };
+
+    if (this.token) {
+      headers['Authorization'] = `Bearer ${this.token}`;
+    }
+
+    const res = await fetch(`${this.serverUrl}/api/v1${path}`, {
+      ...options,
+      headers,
+    });
+
+    return res.json();
+  }
+
+  async login(apiKey: string): Promise<LoginResponse> {
+    const res = await this.request<LoginResponse>('/auth/login', {
+      method: 'POST',
+      body: JSON.stringify({ apiKey }),
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async me(): Promise<User> {
+    const res = await this.request<User>('/auth/me');
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async listSkills(keyword?: string, page = 1, pageSize = 20): Promise<SkillListResponse> {
+    let path = `/skills?page=${page}&pageSize=${pageSize}`;
+    if (keyword) {
+      path += `&keyword=${encodeURIComponent(keyword)}`;
+    }
+    const res = await this.request<SkillListResponse>(path);
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async getSkill(name: string): Promise<Skill> {
+    const res = await this.request<Skill>(`/skills/${name}`);
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async getMySkills(): Promise<Skill[]> {
+    const res = await this.request<Skill[]>('/users/me/skills');
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async uploadSkill(filePath: string): Promise<Skill> {
+    const { readFileSync } = await import('fs');
+    const { basename } = await import('path');
+    const fileName = basename(filePath);
+    const fileBuffer = readFileSync(filePath);
+
+    const formData = new FormData();
+    const blob = new Blob([fileBuffer]);
+    formData.append('file', blob, fileName);
+
+    const headers: Record<string, string> = {};
+    if (this.token) {
+      headers['Authorization'] = `Bearer ${this.token}`;
+    }
+
+    const res = await fetch(`${this.serverUrl}/api/v1/skills`, {
+      method: 'POST',
+      headers,
+      body: formData,
+    });
+
+    const data = await res.json();
+    console.error('DEBUG server response:', JSON.stringify(data));
+    if (!data.ok) {
+      throw new Error(data.msg);
+    }
+    return data.data;
+  }
+
+  async deleteSkill(name: string): Promise<void> {
+    const res = await this.request<null>(`/skills/${name}`, {
+      method: 'DELETE',
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+
+  async downloadSkill(name: string): Promise<ArrayBuffer> {
+    const headers: Record<string, string> = {};
+    if (this.token) {
+      headers['Authorization'] = `Bearer ${this.token}`;
+    }
+
+    const res = await fetch(`${this.serverUrl}/api/v1/skills/${name}/download`, {
+      headers,
+    });
+
+    if (!res.ok) {
+      if (res.status === 404) {
+        throw new Error('Skill 不存在');
+      }
+      throw new Error('下载失败');
+    }
+
+    return res.arrayBuffer();
+  }
+
+  async listAgents(keyword?: string, page = 1, pageSize = 20): Promise<AgentListResponse> {
+    let path = `/agents?page=${page}&pageSize=${pageSize}`;
+    if (keyword) {
+      path += `&keyword=${encodeURIComponent(keyword)}`;
+    }
+    const res = await this.request<AgentListResponse>(path);
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async getAgent(id: string): Promise<Agent> {
+    const res = await this.request<Agent>(`/agents/${id}`);
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async downloadAgent(id: string): Promise<{ buffer: ArrayBuffer; version: string }> {
+    const headers: Record<string, string> = {};
+    if (this.token) {
+      headers['Authorization'] = `Bearer ${this.token}`;
+    }
+
+    const res = await fetch(`${this.serverUrl}/api/v1/agents/${id}/download`, {
+      headers,
+    });
+
+    if (!res.ok) {
+      if (res.status === 404) {
+        throw new Error('Agent 不存在');
+      }
+      throw new Error('下载失败');
+    }
+
+    const version = res.headers.get('X-Version') || 'unknown';
+    return {
+      buffer: await res.arrayBuffer(),
+      version,
+    };
+  }
+
+  async listKnowledge(keyword?: string, page = 1, pageSize = 20): Promise<{ knowledges: any[]; total: number }> {
+    let path = `/knowledges?page=${page}&pageSize=${pageSize}`;
+    if (keyword) {
+      path += `&keyword=${encodeURIComponent(keyword)}`;
+    }
+    const res = await this.request<{ knowledges: any[]; total: number }>(path);
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async getKnowledge(name: string): Promise<any> {
+    const res = await this.request<any>(`/knowledges/${name}`);
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async getMyKnowledge(): Promise<any[]> {
+    const res = await this.request<any[]>('/users/me/knowledges');
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+    return res.data;
+  }
+
+  async uploadKnowledge(filePath: string): Promise<any> {
+    const { readFileSync } = await import('fs');
+    const { basename } = await import('path');
+    const fileName = basename(filePath);
+    const fileBuffer = readFileSync(filePath);
+
+    const formData = new FormData();
+    const blob = new Blob([fileBuffer]);
+    formData.append('file', blob, fileName);
+
+    const headers: Record<string, string> = {};
+    if (this.token) {
+      headers['Authorization'] = `Bearer ${this.token}`;
+    }
+
+    const res = await fetch(`${this.serverUrl}/api/v1/knowledges`, {
+      method: 'POST',
+      headers,
+      body: formData,
+    });
+
+    const data = await res.json();
+    if (!data.ok) {
+      throw new Error(data.msg);
+    }
+    return data.data;
+  }
+
+  async deleteKnowledge(name: string): Promise<void> {
+    const res = await this.request<null>(`/knowledges/${name}`, {
+      method: 'DELETE',
+    });
+    if (!res.ok) {
+      throw new Error(res.msg);
+    }
+  }
+
+  async downloadKnowledge(name: string): Promise<ArrayBuffer> {
+    const headers: Record<string, string> = {};
+    if (this.token) {
+      headers['Authorization'] = `Bearer ${this.token}`;
+    }
+
+    const res = await fetch(`${this.serverUrl}/api/v1/knowledges/${name}/download`, {
+      headers,
+    });
+
+    if (!res.ok) {
+      if (res.status === 404) {
+        throw new Error('Knowledge 不存在');
+      }
+      throw new Error('下载失败');
+    }
+
+    return res.arrayBuffer();
+  }
+}

package/src/lib/formatter.ts

@@ -0,0 +1,196 @@
+export function formatSkill(skill: { name: string; description: string; downloadCount: number; license?: string }): string {
+  const lines = [
+    `\x1b[1m${skill.name}\x1b[0m`,
+    `  ${skill.description}`,
+    `  下载: ${skill.downloadCount}${skill.license ? ` | 许可: ${skill.license}` : ''}`,
+  ];
+  return lines.join('\n');
+}
+
+export function formatSkillDetail(skill: {
+  name: string;
+  description: string;
+  license?: string;
+  compatibility?: string;
+  allowedTools?: string[];
+  fileSize: number;
+  fileHash: string;
+  downloadCount: number;
+  publishedAt: string;
+  metadata?: Record<string, string>;
+}): string {
+  const lines = [
+    `\x1b[1m${skill.name}\x1b[0m`,
+    '',
+    `描述: ${skill.description}`,
+    `文件大小: ${skill.fileSize} bytes`,
+    `下载次数: ${skill.downloadCount}`,
+    `发布时间: ${skill.publishedAt}`,
+  ];
+
+  if (skill.license) {
+    lines.push(`许可: ${skill.license}`);
+  }
+  if (skill.compatibility) {
+    lines.push(`兼容性: ${skill.compatibility}`);
+  }
+  if (skill.allowedTools && skill.allowedTools.length > 0) {
+    lines.push(`允许工具: ${skill.allowedTools.join(', ')}`);
+  }
+  if (skill.metadata) {
+    lines.push(`元数据: ${JSON.stringify(skill.metadata)}`);
+  }
+
+  return lines.join('\n');
+}
+
+export function success(msg: string): void {
+  console.log(`\x1b[32m✓ ${msg}\x1b[0m`);
+}
+
+export function error(msg: string): void {
+  console.error(`\x1b[31m✗ ${msg}\x1b[0m`);
+}
+
+export function info(msg: string): void {
+  console.log(`\x1b[34mℹ ${msg}\x1b[0m`);
+}
+
+export function formatAgent(agent: {
+  name: string;
+  version: string;
+  description: string;
+  status: string;
+  skillsCount?: number;
+  knowledgesCount?: number;
+}): string {
+  const lines = [
+    `\x1b[1m${agent.name}\x1b[0m`,
+    `  ${agent.description}`,
+    `  版本: ${agent.version} | 状态: ${agent.status}${agent.skillsCount !== undefined ? ` | Skills: ${agent.skillsCount}` : ''}${agent.knowledgesCount !== undefined ? ` | Knowledges: ${agent.knowledgesCount}` : ''}`,
+  ];
+  return lines.join('\n');
+}
+
+export function formatAgentDetail(agent: {
+  name: string;
+  version: string;
+  description: string;
+  status: string;
+  prompt: string;
+  createdAt: string;
+  updatedAt: string;
+  skills?: Array<{
+    skill: { name: string };
+    config?: Record<string, unknown>;
+  }>;
+  knowledges?: Array<{
+    knowledgeId: string;
+    knowledgeName?: string;
+    retrievalConfig?: { topK?: number; similarityThreshold?: number };
+  }>;
+}): string {
+  const lines = [
+    `\x1b[1m${agent.name}\x1b[0m`,
+    '',
+    `版本: ${agent.version}`,
+    `描述: ${agent.description || '暂无描述'}`,
+    `状态: ${agent.status}`,
+    `创建时间: ${formatTime(agent.createdAt)}`,
+    `更新时间: ${formatTime(agent.updatedAt)}`,
+  ];
+
+  if (agent.skills && agent.skills.length > 0) {
+    lines.push('');
+    lines.push('Skills:');
+    for (const s of agent.skills) {
+      const hasConfig = s.config && Object.keys(s.config).length > 0;
+      lines.push(`  - ${s.skill.name}${hasConfig ? ` (config: ${JSON.stringify(s.config)})` : ''}`);
+    }
+  }
+
+  if (agent.knowledges && agent.knowledges.length > 0) {
+    lines.push('');
+    lines.push('Knowledges:');
+    for (const k of agent.knowledges) {
+      const name = k.knowledgeName || k.knowledgeId;
+      const topK = k.retrievalConfig?.topK;
+      lines.push(`  - ${name}${topK ? ` (topK: ${topK})` : ''}`);
+    }
+  }
+
+  if (agent.prompt) {
+    lines.push('');
+    lines.push('System Prompt:');
+    lines.push('---');
+    lines.push(agent.prompt);
+    lines.push('---');
+  }
+
+  return lines.join('\n');
+}
+
+function formatTime(isoString: string): string {
+  const date = new Date(isoString);
+  const year = date.getFullYear();
+  const month = String(date.getMonth() + 1).padStart(2, '0');
+  const day = String(date.getDate()).padStart(2, '0');
+  const hours = String(date.getHours()).padStart(2, '0');
+  const minutes = String(date.getMinutes()).padStart(2, '0');
+  const seconds = String(date.getSeconds()).padStart(2, '0');
+  return `${year}-${month}-${day} ${hours}:${minutes}:${seconds}`;
+}
+
+export function formatKnowledge(knowledge: {
+  name: string;
+  description: string;
+  type: string;
+  size: number;
+  downloadCount: number;
+}): string {
+  const sizeStr = knowledge.size > 1024 * 1024
+    ? `${(knowledge.size / (1024 * 1024)).toFixed(2)} MB`
+    : knowledge.size > 1024
+      ? `${(knowledge.size / 1024).toFixed(2)} KB`
+      : `${knowledge.size} B`;
+
+  const lines = [
+    `\x1b[1m${knowledge.name}\x1b[0m`,
+    `  ${knowledge.description}`,
+    `  类型: ${knowledge.type} | 大小: ${sizeStr} | 下载: ${knowledge.downloadCount}`,
+  ];
+  return lines.join('\n');
+}
+
+export function formatKnowledgeDetail(knowledge: {
+  name: string;
+  description: string;
+  type: string;
+  fileSize: number;
+  fileHash: string;
+  downloadCount: number;
+  publishedAt: string;
+  tags?: string[];
+}): string {
+  const sizeStr = knowledge.fileSize > 1024 * 1024
+    ? `${(knowledge.fileSize / (1024 * 1024)).toFixed(2)} MB`
+    : knowledge.fileSize > 1024
+      ? `${(knowledge.fileSize / 1024).toFixed(2)} KB`
+      : `${knowledge.fileSize} B`;
+
+  const lines = [
+    `\x1b[1m${knowledge.name}\x1b[0m`,
+    '',
+    `描述: ${knowledge.description}`,
+    `类型: ${knowledge.type}`,
+    `文件大小: ${sizeStr}`,
+    `下载次数: ${knowledge.downloadCount}`,
+    `发布时间: ${knowledge.publishedAt}`,
+  ];
+
+  if (knowledge.tags && knowledge.tags.length > 0) {
+    lines.push(`标签: ${knowledge.tags.join(', ')}`);
+  }
+
+  return lines.join('\n');
+}

package/src/lib/storage.ts

@@ -0,0 +1,63 @@
+import { writeFileSync, readFileSync, existsSync, mkdirSync } from 'fs';
+import { join } from 'path';
+
+const CONFIG_DIR = join(process.env.HOME || '/root', '.adk');
+const CONFIG_FILE = join(CONFIG_DIR, 'config.json');
+
+interface Config {
+  serverUrl: string;
+  token?: string;
+  user?: {
+    id: string;
+    name: string;
+    email: string;
+  };
+}
+
+function ensureConfigDir() {
+  if (!existsSync(CONFIG_DIR)) {
+    mkdirSync(CONFIG_DIR, { recursive: true });
+  }
+}
+
+export function saveConfig(config: Config): void {
+  ensureConfigDir();
+  writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
+}
+
+export function loadConfig(): Config | null {
+  try {
+    ensureConfigDir();
+    if (!existsSync(CONFIG_FILE)) {
+      return null;
+    }
+    const content = readFileSync(CONFIG_FILE, 'utf-8');
+    return JSON.parse(content);
+  } catch {
+    return null;
+  }
+}
+
+export function clearConfig(): void {
+  try {
+    if (existsSync(CONFIG_FILE)) {
+      const content = readFileSync(CONFIG_FILE, 'utf-8');
+      const config: Config = JSON.parse(content);
+      config.token = undefined;
+      config.user = undefined;
+      saveConfig(config);
+    }
+  } catch {
+  }
+}
+
+export function getServerUrl(): string {
+  const config = loadConfig();
+  return config?.serverUrl || 'http://localhost:3000';
+}
+
+export function setServerUrl(url: string): void {
+  const config = loadConfig() || {};
+  config.serverUrl = url;
+  saveConfig(config);
+}