@starlink-awaken/agentmesh 1.0.2 → 1.2.0

package/dist/src/cli.js

@@ -0,0 +1,410 @@
+#!/usr/bin/env bun
+/**
+ * Agent Mesh Gateway CLI
+ * 统一命令行接口 — start, setup, health, models, quota, config, doctor, help
+ */
+import { existsSync, readFileSync } from 'node:fs';
+import { resolve, dirname, join } from 'node:path';
+import { initLogger } from './core/logger.js';
+const PROJECT_ROOT = resolve(dirname(import.meta.dir), '..');
+const VERSION = '1.1.0';
+const BANNER = `
+   █████╗  ██████╗ ███████╗███╗   ██╗████████╗
+  ██╔══██╗██╔════╝ ██╔════╝████╗  ██║╚══██╔══╝
+  ███████║██║  ███╗█████╗  ██╔██╗ ██║   ██║
+  ██╔══██║██║   ██║██╔══╝  ██║╚██╗██║   ██║
+  ██║  ██║╚██████╔╝███████╗██║ ╚████║   ██║
+  ╚═╝  ╚═╝ ╚═════╝ ╚══════╝╚═╝  ╚═══╝   ╚═╝
+  𝙼 𝙴 𝚂 𝙷    𝙶 𝙰 𝚃 𝙴 𝚆 𝙰 𝚈   v${VERSION}
+`;
+const BASE_URL = () => {
+    const host = Bun.env.AGENT_GATEWAY_HOST || '127.0.0.1';
+    const port = Bun.env.AGENT_GATEWAY_PORT || '3000';
+    return `http://${host}:${port}`;
+};
+async function apiRequest(path, opts) {
+    const resp = await fetch(`${BASE_URL()}${path}`, {
+        ...opts,
+        headers: { 'Content-Type': 'application/json', ...opts?.headers },
+        signal: AbortSignal.timeout(15_000),
+    });
+    if (!resp.ok) {
+        const err = (await resp.json().catch(() => ({ message: resp.statusText })));
+        throw new Error(`[${resp.status}] ${err.message || 'Unknown error'}`);
+    }
+    return resp.json();
+}
+// ===========================================================================
+// Help
+// ===========================================================================
+function showHelp(topic) {
+    if (topic === 'start' || topic === 'run') {
+        console.log(`
+  agentmesh start — 启动网关服务
+
+  用法:
+    agentmesh start [--port PORT] [--host HOST]
+
+  选项:
+    --port  监听端口 (默认: 3000)
+    --host  监听地址 (默认: 0.0.0.0)
+
+  示例:
+    agentmesh start
+    agentmesh start --port 9400`);
+        return;
+    }
+    if (topic === 'setup') {
+        console.log(`  agentmesh setup — 交互式初始化向导
+
+  引导配置:
+    - API Key (DeepSeek, OpenAI, OpenRouter)
+    - 服务端口和日志级别
+    - 生成 .env 配置文件`);
+        return;
+    }
+    if (topic === 'models' || topic === 'model') {
+        console.log(`  agentmesh models — 列出可用模型`);
+        return;
+    }
+    if (topic === 'quota') {
+        console.log(`  agentmesh quota — 查看配额状态 (需 codexbar)
+
+  显示所有 Provider 的实时配额/余额信息。
+  首次调用可能较慢 (codexbar 数据采集)。`);
+        return;
+    }
+    if (topic === 'config') {
+        console.log(`  agentmesh config — 配置管理
+
+  用法:
+    agentmesh config show    显示当前配置
+    agentmesh config path    显示配置文件路径
+    agentmesh config edit    在编辑器中打开配置`);
+        return;
+    }
+    console.log(`
+${BANNER}
+  🤖 Agent Mesh Gateway CLI  v${VERSION}
+
+  用法:
+    agentmesh <command> [options]
+
+  管理命令:
+    start                 启动网关服务
+    setup                 交互式初始化向导 (配置API Key等)
+    doctor                系统诊断检查
+
+  接入命令:
+    connect [tool]        一键接入 AI 工具 (--dry-run 预览)
+    disconnect [tool]     恢复工具配置到接入前状态
+
+  查询命令:
+    health, status        健康检查
+    models                列出可用模型
+    quota                 查看配额状态
+    agents                列出已注册 Agent
+    tasks                 列出任务列表
+
+  配置命令:
+    config show           显示当前配置
+    config path           显示配置文件路径
+    config edit           在编辑器中打开配置
+
+  获取帮助:
+    help                  显示此帮助
+    help <command>        显示特定命令帮助
+
+  示例:
+    agentmesh start
+    agentmesh setup
+    agentmesh connect --dry-run    # 预览接入变更
+    agentmesh connect all           # 接入所有工具
+    agentmesh disconnect all        # 恢复所有工具
+
+  文档: https://github.com/starlink-awaken/agentmesh
+`);
+}
+// ===========================================================================
+// Commands
+// ===========================================================================
+async function cmdStart(args) {
+    // 解析参数
+    const portArg = args.indexOf('--port');
+    const hostArg = args.indexOf('--host');
+    const port = portArg >= 0 ? args[portArg + 1] || '3000' : '3000';
+    const host = hostArg >= 0 ? args[hostArg + 1] || '0.0.0.0' : '0.0.0.0';
+    // 设置环境变量传递给服务器
+    Bun.env.AGENT_GATEWAY_PORT = port;
+    Bun.env.AGENT_GATEWAY_HOST = host;
+    console.log(`${BANNER}
+  🚀 Starting Agent Mesh Gateway...
+  ═══════════════════════════════════════
+  HTTP:       http://${host}:${port}
+  Health:     http://${host}:${port}/health
+  Models:     http://${host}:${port}/v1/models
+  Quota:      http://${host}:${port}/model-gateway/quota
+  Docs:       http://${host}:${port}/docs
+  ═══════════════════════════════════════
+`);
+    // 动态导入并启动服务器
+    await import('./index.js');
+    // 服务器会在导入时启动
+}
+async function cmdHealth() {
+    try {
+        const data = await apiRequest('/health');
+        console.log(`\n  ✅ Gateway Running`);
+        console.log(`  Status:     ${data.status}`);
+        console.log(`  Agents:     ${data.agents?.length || 0}`);
+        console.log(`  Timestamp:  ${new Date(data.timestamp).toISOString()}\n`);
+    }
+    catch (err) {
+        console.error(`\n  ❌ Gateway not reachable at ${BASE_URL()}`);
+        console.error(`  Error: ${err.message}`);
+        console.log(`\n  Start it with: agentmesh start\n`);
+    }
+}
+async function cmdModels() {
+    try {
+        const data = await apiRequest('/v1/models');
+        console.log('\n  📋 Available Models:\n');
+        const byProvider = {};
+        for (const m of data.data || []) {
+            (byProvider[m.owned_by] ??= []).push(m.id);
+        }
+        for (const [provider, models] of Object.entries(byProvider)) {
+            console.log(`  ${provider}:`);
+            models.forEach(m => console.log(`    - ${m}`));
+            console.log('');
+        }
+    }
+    catch (err) {
+        console.error(`\n  ❌ ${err.message}\n`);
+    }
+}
+async function cmdQuota() {
+    try {
+        console.log('\n  ⏳ Fetching quota data (may take ~15s)...');
+        const data = await apiRequest('/model-gateway/quota', { signal: AbortSignal.timeout(60_000) });
+        console.log('\n  📊 Provider Quota Status:\n');
+        if (!data || Object.keys(data).length === 0) {
+            console.log('  No quota data available. Is codexbar installed?\n');
+            return;
+        }
+        for (const [name, info] of Object.entries(data)) {
+            const icon = info.available ? '🟢' : '🔴';
+            console.log(`  ${icon} ${name.padEnd(15)} ${info.summary}`);
+        }
+        console.log('');
+    }
+    catch (err) {
+        console.error(`\n  ❌ ${err.message}\n`);
+    }
+}
+async function cmdAgents() {
+    try {
+        const data = await apiRequest('/agents');
+        console.log('\n  📋 Registered Agents:\n');
+        data.forEach(a => console.log(`  ${a.status === 'online' ? '🟢' : '⚫'} ${a.id.padEnd(18)} ${a.name.padEnd(22)} [${a.status}]\n     ${(a.capabilities || []).join(', ')}\n`));
+    }
+    catch (err) {
+        console.error(`\n  ❌ ${err.message}\n`);
+    }
+}
+async function cmdTasks() {
+    try {
+        const tasks = await apiRequest('/tasks');
+        console.log('\n  📋 Tasks:\n');
+        if (!tasks.length) {
+            console.log('  (none)\n');
+            return;
+        }
+        tasks.forEach(t => console.log(`  ${t.id?.slice(0, 8)}...  ${t.status?.padEnd(10)} ${new Date(t.created_at).toLocaleString()}`));
+        console.log('');
+    }
+    catch (err) {
+        console.error(`\n  ❌ ${err.message}\n`);
+    }
+}
+async function cmdConfig(args) {
+    const configPath = join(PROJECT_ROOT, 'config', 'gateway.yaml');
+    const sub = args[0];
+    if (sub === 'path') {
+        console.log(`\n  Config: ${configPath}\n`);
+    }
+    else if (sub === 'edit') {
+        const editor = Bun.env.EDITOR || Bun.env.VISUAL || 'vim';
+        console.log(`  Opening ${configPath} with ${editor}...`);
+        Bun.spawnSync([editor, configPath], { stdio: ['inherit', 'inherit', 'inherit'] });
+    }
+    else {
+        // show
+        try {
+            const content = readFileSync(configPath, 'utf-8');
+            console.log(`\n  ── ${configPath} ──\n`);
+            console.log(content.split('\n').map(l => `  ${l}`).join('\n'));
+        }
+        catch {
+            console.error(`\n  ❌ Config not found: ${configPath}\n`);
+        }
+    }
+}
+async function cmdDoctor() {
+    console.log('\n  🔍 Agent Mesh Gateway Diagnostics\n');
+    const checks = [];
+    // bun
+    try {
+        const proc = Bun.spawnSync({ cmd: ['bun', '--version'], stdout: 'pipe' });
+        const ver = new TextDecoder().decode(proc.stdout).trim();
+        checks.push(['Bun Runtime', true, ver]);
+    }
+    catch {
+        checks.push(['Bun Runtime', false, 'Not found']);
+    }
+    // codexbar
+    try {
+        const proc = Bun.spawnSync({ cmd: ['codexbar', '--version'], stdout: 'pipe', stderr: 'pipe' });
+        checks.push(['codexbar', proc.exitCode === 0, proc.exitCode === 0 ? 'OK' : 'Warning']);
+    }
+    catch {
+        checks.push(['codexbar', false, 'Install for quota tracking']);
+    }
+    // API keys
+    for (const [name, env] of [['DeepSeek', 'DEEPSEEK_API_KEY'], ['OpenAI', 'OPENAI_API_KEY'], ['OpenRouter', 'OPENROUTER_API_KEY']]) {
+        checks.push([`${name} API Key`, !!Bun.env[env], Bun.env[env] ? 'Configured (env)' : 'Not set']);
+    }
+    // .env file
+    const envPath = join(PROJECT_ROOT, '.env');
+    checks.push(['.env file', existsSync(envPath), existsSync(envPath) ? 'Exists' : 'Run: agentmesh setup']);
+    // docker
+    try {
+        const proc = Bun.spawnSync({ cmd: ['docker', '--version'], stdout: 'pipe' });
+        checks.push(['Docker', true, new TextDecoder().decode(proc.stdout).trim()]);
+    }
+    catch {
+        checks.push(['Docker', false, 'Not required']);
+    }
+    // config
+    const configPath = join(PROJECT_ROOT, 'config', 'gateway.yaml');
+    checks.push(['Config file', existsSync(configPath), existsSync(configPath) ? 'OK' : 'Missing!']);
+    // Print
+    for (const [name, ok, detail] of checks) {
+        console.log(`  ${ok ? '✅' : '⚠️'} ${name.padEnd(20)} ${detail}`);
+    }
+    // Gateway status
+    try {
+        await apiRequest('/health');
+        console.log(`  ✅ Gateway              Running at ${BASE_URL()}`);
+    }
+    catch {
+        console.log('  ⚠️ Gateway              Not running (start: agentmesh start)');
+    }
+    console.log('');
+}
+// ===========================================================================
+// Main
+// ===========================================================================
+async function main() {
+    const args = Bun.argv.slice(2);
+    initLogger({ level: Bun.env.LOG_LEVEL || 'info', dir: join(PROJECT_ROOT, 'logs') });
+    if (args.length === 0 || args[0] === 'help' || args[0] === '--help' || args[0] === '-h') {
+        showHelp(args[1]);
+        return;
+    }
+    const cmd = args[0];
+    const rest = args.slice(1);
+    try {
+        switch (cmd) {
+            case 'start':
+            case 'run':
+                await cmdStart(rest);
+                break;
+            case 'setup':
+            case 'init':
+                const { runSetup } = await import('./cli/setup.js');
+                await runSetup();
+                break;
+            case 'health':
+            case 'status':
+                await cmdHealth();
+                break;
+            case 'models':
+            case 'model':
+                await cmdModels();
+                break;
+            case 'quota':
+                await cmdQuota();
+                break;
+            case 'agents':
+            case 'ls':
+                await cmdAgents();
+                break;
+            case 'tasks':
+                await cmdTasks();
+                break;
+            case 'config':
+                await cmdConfig(rest);
+                break;
+            case 'connect': {
+                // 子命令: connect list
+                if (rest[0] === 'list') {
+                    const { listDetectedTools } = await import('./cli/connect.js');
+                    const tools = listDetectedTools();
+                    console.log('\n  🔍 检测到的 AI 工具:\n');
+                    for (const t of tools) {
+                        console.log(`  ${t.installed ? '🟢' : '⚫'} ${t.name.padEnd(18)} ${t.description}`);
+                        if (t.configPath)
+                            console.log(`     Config: ${t.configPath}`);
+                    }
+                    console.log(`\n  执行 \`agentmesh connect\` 进入交互式选择`);
+                    console.log(`  执行 \`agentmesh connect all --dry-run\` 预览接入变更\n`);
+                    break;
+                }
+                // 无参数或 -i → 交互式选择
+                const hasTargets = rest.some(a => !a.startsWith('--'));
+                const isInteractive = rest.includes('-i') || rest.includes('--interactive') || !hasTargets;
+                if (isInteractive && !rest.includes('--dry-run')) {
+                    const { interactiveConnect } = await import('./cli/connect.js');
+                    await interactiveConnect();
+                    break;
+                }
+                const { connectTools } = await import('./cli/connect.js');
+                const dryRun = rest.includes('--dry-run');
+                const targets = rest.filter(a => !a.startsWith('--') && a !== '-i' && a !== '--interactive');
+                const results = await connectTools(targets.length ? targets : ['all'], { dryRun });
+                console.log('\n  接入结果:\n');
+                for (const r of results) {
+                    const icon = r.status === 'ok' ? '✅' : r.status === 'skipped' ? '⏭️' : '❌';
+                    console.log(`  ${icon} ${r.tool.padEnd(16)} ${r.detail}`);
+                }
+                if (dryRun)
+                    console.log('\n  ⚠️  DRY-RUN 模式，未实际修改文件');
+                else
+                    console.log(`\n  💡 使用 \`agentmesh disconnect\` 恢复配置`);
+                break;
+            }
+            case 'disconnect': {
+                const { disconnectTools } = await import('./cli/connect.js');
+                const targets = rest.filter(a => !a.startsWith('--'));
+                const results = await disconnectTools(targets.length ? targets : ['all']);
+                console.log('\n  断开结果:\n');
+                for (const r of results) {
+                    const icon = r.status === 'ok' ? '✅' : '❌';
+                    console.log(`  ${icon} ${r.tool.padEnd(16)} ${r.detail}`);
+                }
+                break;
+            }
+            case 'doctor':
+            case 'check':
+                await cmdDoctor();
+                break;
+            default:
+                console.error(`\n  ❌ Unknown command: ${cmd}\n  Run 'agentmesh help' for available commands.\n`);
+        }
+    }
+    catch (err) {
+        console.error(`\n  ❌ Error: ${err.message}\n`);
+    }
+}
+main();

package/dist/src/core/agent-registry.d.ts

@@ -0,0 +1,48 @@
+import type { AgentAdapter } from '../adapters/base.js';
+import type { Agent } from '../types/index.js';
+export declare class AgentRegistry {
+    private adapters;
+    private initialized;
+    constructor();
+    /**
+     * 初始化注册表
+     */
+    initialize(): void;
+    /**
+     * 从配置注册所有 Agent
+     */
+    private registerAllFromConfig;
+    /**
+     * 注册自定义适配器
+     */
+    register(adapter: AgentAdapter): void;
+    /**
+     * 获取适配器
+     */
+    get(agentId: string): AgentAdapter | undefined;
+    /**
+     * 获取所有适配器
+     */
+    getAll(): AgentAdapter[];
+    /**
+     * 获取所有 Agent 信息
+     */
+    getAgents(): Agent[];
+    /**
+     * 根据能力查找 Agent
+     */
+    findByCapability(capability: string): Agent[];
+    /**
+     * 检查适配器是否存在
+     */
+    has(agentId: string): boolean;
+    /**
+     * 检查适配器是否健康
+     */
+    checkHealth(agentId: string): Promise<boolean>;
+    /**
+     * 获取可用 Agent 列表（健康检查）
+     */
+    getAvailableAgents(): Promise<Agent[]>;
+}
+export declare const agentRegistry: AgentRegistry;