opc-agent 3.0.1 → 4.0.1

package/src/cli/setup.ts

@@ -0,0 +1,314 @@
+import * as readline from 'readline';
+import * as fs from 'fs';
+import * as path from 'path';
+import * as http from 'http';
+import * as https from 'https';
+import * as os from 'os';
+
+// ── Colors ──────────────────────────────────────────────────────────────────
+const c = {
+  green: (s: string) => `\x1b[32m${s}\x1b[0m`,
+  red: (s: string) => `\x1b[31m${s}\x1b[0m`,
+  yellow: (s: string) => `\x1b[33m${s}\x1b[0m`,
+  blue: (s: string) => `\x1b[34m${s}\x1b[0m`,
+  cyan: (s: string) => `\x1b[36m${s}\x1b[0m`,
+  bold: (s: string) => `\x1b[1m${s}\x1b[0m`,
+  dim: (s: string) => `\x1b[2m${s}\x1b[0m`,
+  magenta: (s: string) => `\x1b[35m${s}\x1b[0m`,
+};
+
+// ── Types ───────────────────────────────────────────────────────────────────
+interface SetupConfig {
+  provider: 'ollama' | 'openai' | 'deepseek' | 'qwen' | 'anthropic';
+  model?: string;
+  embeddingModel?: string;
+  apiKey?: string;
+  baseUrl?: string;
+}
+
+interface AgentTemplate {
+  id: string;
+  icon: string;
+  name: string;
+  description: string;
+  defaultAgentName: string;
+}
+
+const TEMPLATES: AgentTemplate[] = [
+  { id: 'customer-service', icon: '🎧', name: '客服助手', description: '回答客户问题，自动学习产品知识', defaultAgentName: '我的客服助手' },
+  { id: 'content-writer', icon: '✍️', name: '写作助手', description: '帮你写文章、邮件、报告', defaultAgentName: '我的写作助手' },
+  { id: 'data-analyst', icon: '📊', name: '数据分析师', description: '分析数据，生成洞察', defaultAgentName: '我的数据分析师' },
+  { id: 'translator', icon: '🌐', name: '翻译助手', description: '多语言翻译，越用越准', defaultAgentName: '我的翻译助手' },
+  { id: 'general', icon: '🤖', name: '通用助手', description: '什么都能聊', defaultAgentName: '我的AI助手' },
+];
+
+const OPC_HOME = path.join(os.homedir(), '.opc');
+const CONFIG_PATH = path.join(OPC_HOME, 'config.json');
+
+// ── Readline helpers ────────────────────────────────────────────────────────
+export function createRL(input?: NodeJS.ReadableStream, output?: NodeJS.WritableStream): readline.Interface {
+  return readline.createInterface({
+    input: input ?? process.stdin,
+    output: output ?? process.stdout,
+  });
+}
+
+function ask(rl: readline.Interface, question: string): Promise<string> {
+  return new Promise((resolve) => rl.question(question, (a) => resolve(a.trim())));
+}
+
+// ── HTTP helpers ────────────────────────────────────────────────────────────
+function httpGet(url: string, timeout = 5000): Promise<{ status: number; body: string }> {
+  return new Promise((resolve, reject) => {
+    const mod = url.startsWith('https') ? https : http;
+    const req = mod.get(url, { timeout }, (res) => {
+      let body = '';
+      res.on('data', (chunk: Buffer) => { body += chunk; });
+      res.on('end', () => resolve({ status: res.statusCode ?? 0, body }));
+    });
+    req.on('error', reject);
+    req.on('timeout', () => { req.destroy(); reject(new Error('timeout')); });
+  });
+}
+
+// ── Step 1: Welcome ─────────────────────────────────────────────────────────
+function printWelcome(): void {
+  console.log('');
+  console.log(c.bold('  🎉 欢迎使用 OPC Agent！'));
+  console.log(c.dim('     让我们用 3 分钟配置你的第一个 AI Agent。'));
+  console.log('');
+}
+
+// ── Step 2: Model detection ─────────────────────────────────────────────────
+async function detectOllama(): Promise<{ running: boolean; models: string[] }> {
+  try {
+    const { body } = await httpGet('http://localhost:11434/api/tags');
+    const data = JSON.parse(body);
+    const models = (data.models || []).map((m: any) => m.name || m.model);
+    return { running: true, models };
+  } catch {
+    return { running: false, models: [] };
+  }
+}
+
+async function stepModel(rl: readline.Interface): Promise<SetupConfig> {
+  console.log(c.bold('📡 Step 1/4: 配置 AI 模型'));
+  console.log(c.dim('  正在检测本地 Ollama...'));
+  console.log('');
+
+  const ollama = await detectOllama();
+
+  if (ollama.running) {
+    console.log(c.green('  ✔ Ollama 运行中！已安装模型：'));
+    if (ollama.models.length === 0) {
+      console.log(c.dim('    （无模型）'));
+    } else {
+      for (const m of ollama.models) {
+        console.log(`    • ${m}`);
+      }
+    }
+    console.log('');
+
+    const hasChat = ollama.models.some((m) => m.startsWith('qwen2.5'));
+    const hasEmbed = ollama.models.some((m) => m.startsWith('nomic-embed-text'));
+
+    if (!hasChat) {
+      const dl = await ask(rl, `  推荐模型 ${c.cyan('qwen2.5:7b')} 未安装，是否下载？ [Y/n] `);
+      if (dl.toLowerCase() !== 'n') {
+        console.log(c.dim('  → 请在终端运行: ollama pull qwen2.5:7b'));
+      }
+    }
+
+    if (!hasEmbed) {
+      const dl = await ask(rl, `  推荐 Embedding 模型 ${c.cyan('nomic-embed-text')} 未安装，是否下载？ [Y/n] `);
+      if (dl.toLowerCase() !== 'n') {
+        console.log(c.dim('  → 请在终端运行: ollama pull nomic-embed-text'));
+      }
+    }
+
+    console.log('');
+    return {
+      provider: 'ollama',
+      model: hasChat ? ollama.models.find((m) => m.startsWith('qwen2.5'))! : 'qwen2.5:7b',
+      embeddingModel: hasEmbed ? 'nomic-embed-text' : 'nomic-embed-text',
+      baseUrl: 'http://localhost:11434',
+    };
+  }
+
+  // Ollama not running
+  console.log(c.yellow('  ⚠ 未检测到 Ollama'));
+  console.log('');
+  console.log('  请选择：');
+  console.log(`    ${c.cyan('A')} ) 安装 Ollama（推荐，免费本地运行）`);
+  console.log(`    ${c.cyan('B')} ) 使用云端 API`);
+  console.log('');
+
+  const choice = await ask(rl, '  你的选择 [A/B]: ');
+
+  if (choice.toUpperCase() !== 'B') {
+    console.log('');
+    console.log(c.bold('  📦 安装 Ollama：'));
+    console.log('');
+    if (process.platform === 'win32') {
+      console.log('    1. 访问 https://ollama.com/download');
+      console.log('    2. 下载 Windows 版并安装');
+      console.log('    3. 安装后运行: ollama pull qwen2.5:7b');
+    } else if (process.platform === 'darwin') {
+      console.log('    brew install ollama');
+      console.log('    ollama serve &');
+      console.log('    ollama pull qwen2.5:7b');
+    } else {
+      console.log('    curl -fsSL https://ollama.com/install.sh | sh');
+      console.log('    ollama serve &');
+      console.log('    ollama pull qwen2.5:7b');
+    }
+    console.log('');
+    console.log(c.dim('  安装完成后重新运行 opc setup'));
+    return { provider: 'ollama', model: 'qwen2.5:7b', embeddingModel: 'nomic-embed-text', baseUrl: 'http://localhost:11434' };
+  }
+
+  // Cloud API
+  console.log('');
+  console.log('  选择云端 Provider：');
+  console.log(`    ${c.cyan('1')} ) OpenAI`);
+  console.log(`    ${c.cyan('2')} ) DeepSeek`);
+  console.log(`    ${c.cyan('3')} ) 通义千问 (Qwen)`);
+  console.log(`    ${c.cyan('4')} ) Anthropic`);
+  console.log('');
+
+  const providerChoice = await ask(rl, '  选择 [1-4]: ');
+  const providers: Record<string, { provider: SetupConfig['provider']; baseUrl: string; model: string; testUrl: string }> = {
+    '1': { provider: 'openai', baseUrl: 'https://api.openai.com/v1', model: 'gpt-4o-mini', testUrl: 'https://api.openai.com/v1/models' },
+    '2': { provider: 'deepseek', baseUrl: 'https://api.deepseek.com/v1', model: 'deepseek-chat', testUrl: 'https://api.deepseek.com/v1/models' },
+    '3': { provider: 'qwen', baseUrl: 'https://dashscope.aliyuncs.com/compatible-mode/v1', model: 'qwen-plus', testUrl: 'https://dashscope.aliyuncs.com/compatible-mode/v1/models' },
+    '4': { provider: 'anthropic', baseUrl: 'https://api.anthropic.com/v1', model: 'claude-3-5-sonnet-20241022', testUrl: 'https://api.anthropic.com/v1/models' },
+  };
+
+  const p = providers[providerChoice] ?? providers['1'];
+  console.log('');
+  const apiKey = await ask(rl, `  请输入 ${c.bold(p.provider)} API Key: `);
+
+  if (apiKey) {
+    console.log(c.dim('  正在测试连接...'));
+    try {
+      await httpGet(p.testUrl);
+      console.log(c.green('  ✔ 连接成功！'));
+    } catch {
+      console.log(c.yellow('  ⚠ 无法验证连接，但配置已保存，稍后可测试'));
+    }
+  }
+
+  console.log('');
+  return { provider: p.provider, model: p.model, apiKey, baseUrl: p.baseUrl };
+}
+
+// ── Step 3: Choose template ─────────────────────────────────────────────────
+async function stepTemplate(rl: readline.Interface): Promise<AgentTemplate> {
+  console.log(c.bold('📋 Step 2/4: 选择 Agent 模板'));
+  console.log('');
+
+  for (let i = 0; i < TEMPLATES.length; i++) {
+    const t = TEMPLATES[i];
+    console.log(`    ${c.cyan(String(i + 1))} ) ${t.icon} ${c.bold(t.name)} — ${c.dim(t.description)}`);
+  }
+  console.log(`    ${c.cyan('6')} ) 📋 更多模板...（打开 Studio 浏览）`);
+  console.log('');
+
+  const choice = await ask(rl, '  选择模板 [1-6]: ');
+  const idx = parseInt(choice, 10) - 1;
+
+  if (choice === '6') {
+    console.log(c.dim('  → 运行 opc studio 浏览更多模板'));
+    return TEMPLATES[4]; // default to general
+  }
+
+  const template = TEMPLATES[idx] ?? TEMPLATES[4];
+  console.log(c.green(`  ✔ 已选择: ${template.icon} ${template.name}`));
+  console.log('');
+  return template;
+}
+
+// ── Step 4: Create agent ────────────────────────────────────────────────────
+async function stepCreateAgent(rl: readline.Interface, template: AgentTemplate, config: SetupConfig): Promise<{ name: string; id: string; dir: string }> {
+  console.log(c.bold('🤖 Step 3/4: 创建 Agent'));
+  console.log('');
+
+  const name = (await ask(rl, `  Agent 名称 [${template.defaultAgentName}]: `)) || template.defaultAgentName;
+  const description = await ask(rl, `  简短描述（可选）: `);
+
+  // Generate a slug id
+  const id = name
+    .replace(/[^\w\u4e00-\u9fff]/g, '-')
+    .replace(/-+/g, '-')
+    .replace(/^-|-$/g, '')
+    .toLowerCase() || `agent-${Date.now()}`;
+
+  const agentDir = path.join(OPC_HOME, 'agents', id);
+  fs.mkdirSync(agentDir, { recursive: true });
+
+  // Write agent config
+  const agentConfig = {
+    id,
+    name,
+    description: description || template.description,
+    template: template.id,
+    provider: config.provider,
+    model: config.model,
+    embeddingModel: config.embeddingModel,
+    apiKey: config.apiKey,
+    baseUrl: config.baseUrl,
+    createdAt: new Date().toISOString(),
+  };
+
+  fs.writeFileSync(path.join(agentDir, 'config.json'), JSON.stringify(agentConfig, null, 2));
+
+  // Initialize brain directory
+  const brainDir = path.join(agentDir, 'brain');
+  fs.mkdirSync(brainDir, { recursive: true });
+  fs.writeFileSync(path.join(brainDir, 'README.md'), `# ${name}\n\n将知识文档放在这里，Agent 会自动学习。\n`);
+
+  // Save global config
+  fs.mkdirSync(OPC_HOME, { recursive: true });
+  const globalConfig = fs.existsSync(CONFIG_PATH) ? JSON.parse(fs.readFileSync(CONFIG_PATH, 'utf-8')) : {};
+  globalConfig.defaultAgent = id;
+  globalConfig.provider = config.provider;
+  globalConfig.model = config.model;
+  globalConfig.baseUrl = config.baseUrl;
+  if (config.apiKey) globalConfig.apiKey = config.apiKey;
+  if (config.embeddingModel) globalConfig.embeddingModel = config.embeddingModel;
+  fs.writeFileSync(CONFIG_PATH, JSON.stringify(globalConfig, null, 2));
+
+  console.log('');
+  console.log(`  ${c.green('✔')} Agent 已创建: ${c.bold(name)}`);
+  console.log(`  ${c.dim('    目录: ' + agentDir)}`);
+  console.log('');
+
+  return { name, id, dir: agentDir };
+}
+
+// ── Step 5: Completion ──────────────────────────────────────────────────────
+function printCompletion(agentName: string): void {
+  console.log(c.bold(`  ✅ 你的 AI Agent「${agentName}」已创建！`));
+  console.log('');
+  console.log('  启动方式：');
+  console.log(`    ${c.cyan('opc studio')}    — 打开网页管理界面`);
+  console.log(`    ${c.cyan('opc chat')}      — 终端直接对话`);
+  console.log(`    ${c.cyan('opc start')}     — 后台运行`);
+  console.log('');
+  console.log(c.dim('  你的 Agent 会自动学习和进化，越用越聪明！🧬'));
+  console.log('');
+}
+
+// ── Main entry ──────────────────────────────────────────────────────────────
+export async function runSetup(input?: NodeJS.ReadableStream, output?: NodeJS.WritableStream): Promise<void> {
+  const rl = createRL(input, output);
+  try {
+    printWelcome();
+    const config = await stepModel(rl);
+    const template = await stepTemplate(rl);
+    const agent = await stepCreateAgent(rl, template, config);
+    printCompletion(agent.name);
+  } finally {
+    rl.close();
+  }
+}

package/src/cli.ts

@@ -36,6 +36,9 @@ import type { CronJob } from './core/scheduler';
 import type { Span } from './traces';
 import { spawn } from 'child_process';
 import { searchRoles, getPopularRoles, getRole, getCategories } from 'agent-workstation';
+import { fetchTemplates as hubFetchTemplates, fetchTemplate as hubFetchTemplate, fetchBrainSeeds, isHubAvailable } from './hub/client';
+import type { HubTemplate } from './hub/client';
+import { downloadAndLearnBrainSeeds } from './hub/brain-seed';
 
 const program = new Command();
 
@@ -358,9 +361,34 @@ export class EchoSkill extends BaseSkill {
     }
 
     const name = opts.yes ? (nameArg ?? 'my-agent') : (nameArg ?? await promptUser('Project name', 'my-agent'));
-    const template = opts.yes
-      ? (opts.template ?? 'customer-service')
-      : (opts.template ?? await select('Select a template:', Object.entries(TEMPLATES).map(([value, { label }]) => ({ value, label }))));
+
+    // Try fetching templates from Hub API, fall back to bundled
+    let hubTemplates: HubTemplate[] = [];
+    let useHub = false;
+    try {
+      const hubAvailable = await isHubAvailable();
+      if (hubAvailable) {
+        hubTemplates = await hubFetchTemplates();
+        if (hubTemplates.length > 0) useHub = true;
+      }
+    } catch {
+      // Hub unreachable — fall back to bundled templates
+    }
+
+    let template: string;
+    let selectedHubTemplate: HubTemplate | undefined;
+    if (opts.yes) {
+      template = opts.template ?? 'customer-service';
+    } else if (opts.template) {
+      template = opts.template;
+      if (useHub) selectedHubTemplate = hubTemplates.find(t => t.id === template);
+    } else if (useHub) {
+      console.log(`  ${icon.info} ${color.dim('Using templates from Workstation Hub')}`);
+      template = await select('Select a template:', hubTemplates.map(t => ({ value: t.id, label: `${t.name} - ${t.description}` })));
+      selectedHubTemplate = hubTemplates.find(t => t.id === template);
+    } else {
+      template = await select('Select a template:', Object.entries(TEMPLATES).map(([value, { label }]) => ({ value, label })));
+    }
 
     const dir = path.resolve(name);
     if (fs.existsSync(dir)) {
@@ -703,6 +731,22 @@ on startup to understand the project context.
     console.log(`   ${icon.file} Dockerfile`);
     console.log(`   ${icon.file} README.md`);
     console.log(`\n   Template: ${color.cyan(template)}`);
+
+    // Download brain-seed files from Hub if available
+    if (selectedHubTemplate) {
+      try {
+        const seeds = await fetchBrainSeeds(selectedHubTemplate.id);
+        if (seeds.length > 0) {
+          const result = await downloadAndLearnBrainSeeds(dir, seeds);
+          console.log(`\n   📚 Imported ${color.bold(String(result.savedFiles.length))} knowledge files into brain-seed/`);
+          if (result.learnedCount > 0) {
+            console.log(`   🧠 Auto-learned ${color.bold(String(result.learnedCount))} files into local DeepBrain`);
+          }
+        }
+      } catch {
+        // Brain-seed download failed — non-fatal, project still usable
+      }
+    }
     console.log(`\n${color.bold('Next steps:')}`);
     console.log(`   1. cd ${name}`);
     console.log(`   2. npm install`);
@@ -901,6 +945,46 @@ program
     console.log(`\n   ${color.dim('Press Ctrl+C to stop.')}\n`);
   });
 
+// ── Serve command (OpenAI-compatible API) ────────────────────
+
+program
+  .command('serve')
+  .description('Start OpenAI-compatible API server')
+  .option('-f, --file <file>', 'OAD file', 'oad.yaml')
+  .option('-p, --port <port>', 'Port', '8080')
+  .option('-H, --host <host>', 'Host', '0.0.0.0')
+  .option('-k, --api-key <key>', 'API key for auth')
+  .action(async (opts: { file: string; port: string; host: string; apiKey?: string }) => {
+    loadDotEnv();
+    const { APIServer } = require('./core/api-server');
+    const runtime = new AgentRuntime();
+    await runtime.loadConfig(opts.file);
+    await runtime.initialize();
+    await runtime.start();
+    const agent = runtime.getAgent();
+
+    const server = new APIServer({
+      port: parseInt(opts.port) || 8080,
+      host: opts.host,
+      apiKey: opts.apiKey ?? process.env.OPC_API_KEY,
+      agent,
+    });
+    await server.start();
+
+    const name = agent?.name ?? 'unknown';
+    console.log(`\n${icon.rocket} OpenAI-compatible API server running`);
+    console.log(`   Agent: ${color.bold(name)}`);
+    console.log(`   URL:   ${color.cyan(`http://${opts.host}:${opts.port}`)}`);
+    console.log(`   Auth:  ${opts.apiKey ? color.green('enabled') : color.yellow('disabled')}`);
+    console.log(`\n   Endpoints:`);
+    console.log(`     POST /v1/chat/completions`);
+    console.log(`     GET  /v1/models`);
+    console.log(`     POST /v1/embeddings`);
+    console.log(`     GET  /health`);
+    console.log(`     GET  /v1/agent/status`);
+    console.log(`\n   ${color.dim('Press Ctrl+C to stop.')}\n`);
+  });
+
 // ── Info command ─────────────────────────────────────────────
 
 program
@@ -2070,13 +2154,71 @@ program
   .command('studio')
   .description('Start OPC Studio web UI')
   .option('--port <port>', 'Port to listen on', '4000')
+  .option('--no-open', 'Do not open browser automatically')
   .action(async (opts: any) => {
     const { StudioServer } = require('./studio/server');
-    const server = new StudioServer({
-      port: parseInt(opts.port, 10),
-      agentDir: process.cwd(),
+    const net = require('net');
+    const port = parseInt(opts.port, 10);
+
+    const checkPort = (p: number): Promise<boolean> => new Promise((resolve) => {
+      const sock = new net.Socket();
+      sock.setTimeout(400);
+      sock.once('connect', () => { sock.destroy(); resolve(true); });
+      sock.once('error', () => { sock.destroy(); resolve(false); });
+      sock.once('timeout', () => { sock.destroy(); resolve(false); });
+      sock.connect(p, 'localhost');
     });
+
+    const server = new StudioServer({ port, agentDir: process.cwd() });
     await server.start();
+
+    // Try to start sub-module UI servers with graceful fallback
+    const subModules = [
+      { name: 'DeepBrain', icon: '🧠', pkg: 'deepbrain', port: 4001, serveMethod: 'serveUI' },
+      { name: 'AgentKits', icon: '📊', pkg: 'agent-kits', port: 4002, serveMethod: 'serveUI' },
+      { name: 'Workstation', icon: '👤', pkg: 'agent-workstation', port: 4003, serveMethod: 'serveUI' },
+    ];
+
+    const moduleStatuses: string[] = [];
+    for (const mod of subModules) {
+      try {
+        const already = await checkPort(mod.port);
+        if (already) {
+          moduleStatuses.push(`  ${icon.success} ${mod.icon} ${mod.name} already running on :${mod.port}`);
+          continue;
+        }
+        const modExports = require(mod.pkg);
+        if (typeof modExports[mod.serveMethod] === 'function') {
+          modExports[mod.serveMethod]({ port: mod.port });
+          await new Promise(r => setTimeout(r, 600));
+          const started = await checkPort(mod.port);
+          moduleStatuses.push(started
+            ? `  ${icon.success} ${mod.icon} ${mod.name} started on :${mod.port}`
+            : `  ${icon.warn} ${mod.icon} ${mod.name} failed to start`);
+        } else {
+          moduleStatuses.push(`  ${color.dim('○')} ${mod.icon} ${mod.name} no serve method`);
+        }
+      } catch {
+        moduleStatuses.push(`  ${color.dim('○')} ${mod.icon} ${mod.name} not installed`);
+      }
+    }
+
+    if (moduleStatuses.length > 0) {
+      console.log('\nModules:');
+      moduleStatuses.forEach(s => console.log(s));
+    }
+
+    const url = `http://localhost:${port}`;
+    console.log(`\n${icon.success} OPC Studio ready → ${color.cyan(url)}`);
+
+    if (opts.open !== false) {
+      try {
+        const { exec } = require('child_process');
+        const openCmd = process.platform === 'darwin' ? 'open' : process.platform === 'win32' ? 'start ""' : 'xdg-open';
+        exec(`${openCmd} ${url}`);
+      } catch {}
+    }
+
     console.log(color.dim('Press Ctrl+C to stop'));
   });