agency-orchestrator 0.1.0

package/dist/agents/loader.js

@@ -0,0 +1,93 @@
+/**
+ * 加载 agency-agents 的 .md 文件，提取角色定义
+ *
+ * 文件格式:
+ * ---
+ * name: 角色名
+ * description: 描述
+ * emoji: 🔧
+ * ---
+ * # 角色标题
+ * ...system prompt 内容...
+ */
+import { readFileSync, existsSync, readdirSync } from 'node:fs';
+import { join, resolve } from 'node:path';
+/**
+ * 加载指定角色的定义
+ * @param agentsDir agency-agents 的 agents 目录路径
+ * @param rolePath 角色路径，如 "engineering/engineering-sre"
+ */
+export function loadAgent(agentsDir, rolePath) {
+    const fullPath = resolve(agentsDir, `${rolePath}.md`);
+    if (!existsSync(fullPath)) {
+        throw new Error(`角色文件不存在: ${fullPath}\n请确认 agents_dir 和 role 路径正确`);
+    }
+    const content = readFileSync(fullPath, 'utf-8');
+    return parseAgentFile(content, rolePath);
+}
+/**
+ * 解析 agent .md 文件内容
+ */
+function parseAgentFile(content, rolePath) {
+    const frontmatterMatch = content.match(/^---\s*\n([\s\S]*?)\n---\s*\n([\s\S]*)$/);
+    if (!frontmatterMatch) {
+        // 没有 frontmatter，整个文件当 system prompt
+        return {
+            name: rolePath,
+            description: '',
+            systemPrompt: content.trim(),
+        };
+    }
+    const frontmatterRaw = frontmatterMatch[1];
+    const body = frontmatterMatch[2];
+    // 简单解析 frontmatter（不用 js-yaml 避免循环依赖，frontmatter 结构简单）
+    const meta = {};
+    for (const line of frontmatterRaw.split('\n')) {
+        const colonIdx = line.indexOf(':');
+        if (colonIdx > 0) {
+            const key = line.slice(0, colonIdx).trim();
+            const value = line.slice(colonIdx + 1).trim().replace(/^["']|["']$/g, '');
+            meta[key] = value;
+        }
+    }
+    return {
+        name: meta.name || rolePath,
+        description: meta.description || '',
+        emoji: meta.emoji,
+        tools: meta.tools,
+        systemPrompt: body.trim(),
+    };
+}
+/**
+ * 列出所有可用角色
+ */
+export function listAgents(agentsDir) {
+    const dir = resolve(agentsDir);
+    if (!existsSync(dir)) {
+        throw new Error(`agents 目录不存在: ${dir}`);
+    }
+    const agents = [];
+    // 遍历子目录
+    for (const dept of readdirSync(dir, { withFileTypes: true })) {
+        if (!dept.isDirectory())
+            continue;
+        // 跳过非 agent 目录
+        if (dept.name.startsWith('.') || dept.name === 'node_modules' || dept.name === 'scripts' ||
+            dept.name === 'integrations' || dept.name === 'examples')
+            continue;
+        const deptDir = join(dir, dept.name);
+        for (const file of readdirSync(deptDir, { withFileTypes: true })) {
+            if (!file.isFile() || !file.name.endsWith('.md'))
+                continue;
+            const rolePath = `${dept.name}/${file.name.replace('.md', '')}`;
+            try {
+                const agent = loadAgent(agentsDir, rolePath);
+                agents.push(agent);
+            }
+            catch {
+                // 跳过无法解析的文件
+            }
+        }
+    }
+    return agents;
+}

package/dist/cli.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/cli.js

@@ -0,0 +1,276 @@
+#!/usr/bin/env node
+/**
+ * agency-orchestrator CLI
+ *
+ * 用法:
+ *   ao run workflow.yaml --input key=value --input file=@path.md
+ *   ao validate workflow.yaml
+ *   ao plan workflow.yaml
+ *   ao roles --agents-dir ./agents
+ */
+import { readFileSync, existsSync } from 'node:fs';
+import { resolve } from 'node:path';
+import { execSync } from 'node:child_process';
+import { parseWorkflow, validateWorkflow } from './core/parser.js';
+import { buildDAG, formatDAG } from './core/dag.js';
+import { listAgents } from './agents/loader.js';
+import { run } from './index.js';
+const args = process.argv.slice(2);
+const command = args[0];
+async function main() {
+    if (!command || command === '--help' || command === '-h') {
+        printHelp();
+        return;
+    }
+    switch (command) {
+        case 'run':
+            await handleRun();
+            break;
+        case 'validate':
+            handleValidate();
+            break;
+        case 'plan':
+            handlePlan();
+            break;
+        case 'roles':
+            handleRoles();
+            break;
+        case 'init':
+            await handleInit();
+            break;
+        case '--version':
+        case '-v':
+            console.log(getVersion());
+            break;
+        default: {
+            // 容错：用户可能漏了空格，如 "planworkflows/x.yaml"
+            const knownCmds = ['run', 'validate', 'plan', 'roles'];
+            const match = knownCmds.find(c => command.startsWith(c) && command.length > c.length);
+            if (match) {
+                console.error(`看起来少了个空格？试试:\n  ao ${match} ${command.slice(match.length)}\n`);
+            }
+            else {
+                console.error(`未知命令: ${command}\n`);
+                printHelp();
+            }
+            process.exit(1);
+        }
+    }
+}
+async function handleRun() {
+    const filePath = args[1];
+    if (!filePath) {
+        console.error('用法: ao run <workflow.yaml> [--input key=value ...]');
+        process.exit(1);
+    }
+    const inputs = parseInputArgs();
+    const outputDir = getArgValue('--output') || '.ao-output';
+    const quiet = args.includes('--quiet') || args.includes('-q');
+    try {
+        const result = await run(resolve(filePath), inputs, { outputDir, quiet });
+        process.exit(result.success ? 0 : 1);
+    }
+    catch (err) {
+        console.error(`\n错误: ${err instanceof Error ? err.message : err}`);
+        process.exit(1);
+    }
+}
+function handleValidate() {
+    const filePath = args[1];
+    if (!filePath) {
+        console.error('用法: ao validate <workflow.yaml>');
+        process.exit(1);
+    }
+    try {
+        const workflow = parseWorkflow(resolve(filePath));
+        const errors = validateWorkflow(workflow);
+        if (errors.length === 0) {
+            console.log(`  ${workflow.name} — 校验通过`);
+            console.log(`  ${workflow.steps.length} 个步骤, ${(workflow.inputs || []).length} 个输入`);
+        }
+        else {
+            console.error(`  ${workflow.name} — 校验失败:\n`);
+            for (const err of errors) {
+                console.error(`  - ${err}`);
+            }
+            process.exit(1);
+        }
+    }
+    catch (err) {
+        console.error(`错误: ${err instanceof Error ? err.message : err}`);
+        process.exit(1);
+    }
+}
+function handlePlan() {
+    const filePath = args[1];
+    if (!filePath) {
+        console.error('用法: ao plan <workflow.yaml>');
+        process.exit(1);
+    }
+    try {
+        const workflow = parseWorkflow(resolve(filePath));
+        const errors = validateWorkflow(workflow);
+        if (errors.length > 0) {
+            console.error(`校验失败:\n${errors.map(e => `  - ${e}`).join('\n')}`);
+            process.exit(1);
+        }
+        const dag = buildDAG(workflow);
+        console.log(`\n  ${workflow.name}\n`);
+        console.log(formatDAG(dag));
+    }
+    catch (err) {
+        console.error(`错误: ${err instanceof Error ? err.message : err}`);
+        process.exit(1);
+    }
+}
+async function handleInit() {
+    const targetDir = resolve('agency-agents-zh');
+    if (existsSync(targetDir)) {
+        console.log(`  agency-agents-zh 已存在，跳过下载`);
+        // 尝试更新
+        try {
+            execSync('git pull', { cwd: targetDir, stdio: 'pipe' });
+            console.log('  已更新到最新版本');
+        }
+        catch {
+            console.log('  (更新失败，使用现有版本)');
+        }
+    }
+    else {
+        console.log('  正在下载 agency-agents-zh (186 个 AI 角色定义)...\n');
+        try {
+            execSync('git clone --depth 1 https://github.com/jnMetaCode/agency-agents-zh.git', { stdio: 'inherit' });
+            console.log('\n  下载完成!');
+        }
+        catch {
+            console.error('\n  下载失败，请手动克隆:');
+            console.error('  git clone https://github.com/jnMetaCode/agency-agents-zh.git');
+            process.exit(1);
+        }
+    }
+    // 显示角色数量
+    const agents = listAgents(targetDir);
+    console.log(`  共 ${agents.length} 个角色可用\n`);
+    console.log('  接下来你可以:');
+    console.log('    ao roles                              查看所有角色');
+    console.log('    ao plan workflows/product-review.yaml  查看执行计划');
+    console.log('    ao run workflows/story-creation.yaml   运行工作流');
+}
+function handleRoles() {
+    const agentsDir = getArgValue('--agents-dir') || resolveAgentsDir();
+    try {
+        const agents = listAgents(resolve(agentsDir));
+        console.log(`\n  共 ${agents.length} 个角色 (${agentsDir}):\n`);
+        for (const agent of agents) {
+            const emoji = agent.emoji || ' ';
+            console.log(`  ${emoji} ${agent.name} — ${agent.description || '(无描述)'}`);
+        }
+    }
+    catch (err) {
+        console.error(`错误: ${err instanceof Error ? err.message : err}`);
+        process.exit(1);
+    }
+}
+/** 解析 --input key=value 和 --input key=@file 参数 */
+function parseInputArgs() {
+    const inputs = {};
+    for (let i = 2; i < args.length; i++) {
+        if (args[i] === '--input' || args[i] === '-i') {
+            const pair = args[++i];
+            if (!pair) {
+                console.error('--input 需要 key=value 参数');
+                process.exit(1);
+            }
+            const eqIdx = pair.indexOf('=');
+            if (eqIdx < 1) {
+                console.error(`无效的 input 格式: ${pair} (应为 key=value)`);
+                process.exit(1);
+            }
+            const key = pair.slice(0, eqIdx);
+            let value = pair.slice(eqIdx + 1);
+            // @file 语法：从文件读取值
+            if (value.startsWith('@')) {
+                const filePath = resolve(value.slice(1));
+                try {
+                    value = readFileSync(filePath, 'utf-8');
+                }
+                catch {
+                    console.error(`无法读取文件: ${filePath}`);
+                    process.exit(1);
+                }
+            }
+            inputs[key] = value;
+        }
+    }
+    return inputs;
+}
+/**
+ * 自动查找 agents 目录，按优先级：
+ * 1. ./agency-agents-zh (ao init 下载的)
+ * 2. ../agency-agents-zh (同级目录)
+ * 3. ./agents (自定义)
+ */
+function resolveAgentsDir() {
+    const candidates = [
+        './agency-agents-zh',
+        '../agency-agents-zh',
+        './agents',
+    ];
+    for (const dir of candidates) {
+        const full = resolve(dir);
+        if (existsSync(full))
+            return dir;
+    }
+    return './agency-agents-zh';
+}
+function getArgValue(flag) {
+    const idx = args.indexOf(flag);
+    if (idx >= 0 && idx + 1 < args.length) {
+        return args[idx + 1];
+    }
+    return undefined;
+}
+function getVersion() {
+    try {
+        const pkg = JSON.parse(readFileSync(new URL('../package.json', import.meta.url), 'utf-8'));
+        return pkg.version;
+    }
+    catch {
+        return '0.1.0';
+    }
+}
+function printHelp() {
+    console.log(`
+  agency-orchestrator — Multi-Agent Workflow Engine
+  基于 agency-agents-zh 的多智能体编排引擎
+
+  Quick Start:
+    ao init                           下载 186 个 AI 角色定义
+    ao roles                          查看所有可用角色
+    ao plan <workflow.yaml>           查看执行计划 (DAG)
+    ao run <workflow.yaml> [options]   执行工作流
+
+  Commands:
+    init                              下载/更新 agency-agents-zh
+    run <workflow.yaml>               执行工作流
+    validate <workflow.yaml>          校验工作流定义
+    plan <workflow.yaml>              查看执行计划
+    roles [--agents-dir path]         列出可用角色
+
+  Options:
+    --input, -i key=value    传入输入变量
+    --input, -i key=@file    从文件读取变量值
+    --output dir             输出目录 (默认 .ao-output/)
+    --quiet, -q              静默模式
+    --version, -v            版本号
+
+  Examples:
+    ao init
+    ao run workflows/story-creation.yaml -i premise='一个时间旅行的故事' -i style='悬疑'
+    ao run workflows/product-review.yaml -i prd_content=@prd.md
+    ao plan workflows/content-pipeline.yaml
+
+  Agents: https://github.com/jnMetaCode/agency-agents-zh
+  `);
+}
+main();

package/dist/connectors/claude.d.ts

@@ -0,0 +1,6 @@
+import type { LLMConnector, LLMResult, LLMConfig } from '../types.js';
+export declare class ClaudeConnector implements LLMConnector {
+    private client;
+    constructor(apiKey?: string);
+    chat(systemPrompt: string, userMessage: string, config: LLMConfig): Promise<LLMResult>;
+}

package/dist/connectors/claude.js

@@ -0,0 +1,38 @@
+/**
+ * Claude API Connector
+ */
+import Anthropic from '@anthropic-ai/sdk';
+export class ClaudeConnector {
+    client;
+    constructor(apiKey) {
+        this.client = new Anthropic({
+            apiKey: apiKey || process.env.ANTHROPIC_API_KEY,
+        });
+        if (!this.client.apiKey) {
+            throw new Error('缺少 ANTHROPIC_API_KEY\n' +
+                '请设置环境变量: export ANTHROPIC_API_KEY=your-key\n' +
+                '或在 workflow YAML 中配置');
+        }
+    }
+    async chat(systemPrompt, userMessage, config) {
+        const response = await this.client.messages.create({
+            model: config.model,
+            max_tokens: config.max_tokens || 4096,
+            system: systemPrompt,
+            messages: [
+                { role: 'user', content: userMessage },
+            ],
+        });
+        const content = response.content
+            .filter(block => block.type === 'text')
+            .map(block => block.type === 'text' ? block.text : '')
+            .join('\n');
+        return {
+            content,
+            usage: {
+                input_tokens: response.usage.input_tokens,
+                output_tokens: response.usage.output_tokens,
+            },
+        };
+    }
+}

package/dist/connectors/interface.d.ts

@@ -0,0 +1,5 @@
+/**
+ * LLM Connector 接口
+ * 所有 provider 实现这个接口
+ */
+export type { LLMConnector, LLMResult, LLMConfig } from '../types.js';

package/dist/connectors/interface.js

@@ -0,0 +1 @@
+export {};

package/dist/connectors/ollama.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Ollama Connector — 本地模型，不需要 API key
+ */
+import type { LLMConnector, LLMResult, LLMConfig } from '../types.js';
+export declare class OllamaConnector implements LLMConnector {
+    private baseUrl;
+    constructor(baseUrl?: string);
+    chat(systemPrompt: string, userMessage: string, config: LLMConfig): Promise<LLMResult>;
+}

package/dist/connectors/ollama.js

@@ -0,0 +1,36 @@
+export class OllamaConnector {
+    baseUrl;
+    constructor(baseUrl) {
+        const raw = baseUrl || process.env.OLLAMA_BASE_URL || 'http://localhost:11434';
+        this.baseUrl = raw.replace(/\/+$/, '');
+    }
+    async chat(systemPrompt, userMessage, config) {
+        const response = await fetch(`${this.baseUrl}/api/chat`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({
+                model: config.model,
+                messages: [
+                    { role: 'system', content: systemPrompt },
+                    { role: 'user', content: userMessage },
+                ],
+                stream: false,
+                options: {
+                    num_predict: config.max_tokens || 2048,
+                },
+            }),
+        });
+        if (!response.ok) {
+            const text = await response.text();
+            throw new Error(`Ollama error ${response.status}: ${text}`);
+        }
+        const data = await response.json();
+        return {
+            content: data.message.content,
+            usage: {
+                input_tokens: data.prompt_eval_count || 0,
+                output_tokens: data.eval_count || 0,
+            },
+        };
+    }
+}

package/dist/connectors/openai-compatible.d.ts

@@ -0,0 +1,14 @@
+/**
+ * OpenAI Compatible Connector
+ * 支持 DeepSeek、智谱、通义、Moonshot 等兼容 OpenAI 格式的 API
+ */
+import type { LLMConnector, LLMResult, LLMConfig } from '../types.js';
+export declare class OpenAICompatibleConnector implements LLMConnector {
+    private apiKey;
+    private baseUrl;
+    constructor(options?: {
+        apiKey?: string;
+        baseUrl?: string;
+    });
+    chat(systemPrompt: string, userMessage: string, config: LLMConfig): Promise<LLMResult>;
+}

package/dist/connectors/openai-compatible.js

@@ -0,0 +1,43 @@
+export class OpenAICompatibleConnector {
+    apiKey;
+    baseUrl;
+    constructor(options = {}) {
+        this.apiKey = options.apiKey || process.env.OPENAI_API_KEY || '';
+        this.baseUrl = options.baseUrl || process.env.OPENAI_BASE_URL || 'https://api.openai.com/v1';
+        // 去掉末尾的 /
+        this.baseUrl = this.baseUrl.replace(/\/+$/, '');
+        if (!this.apiKey) {
+            throw new Error('缺少 API Key，请通过参数或环境变量传入');
+        }
+    }
+    async chat(systemPrompt, userMessage, config) {
+        const response = await fetch(`${this.baseUrl}/chat/completions`, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'Authorization': `Bearer ${this.apiKey}`,
+            },
+            body: JSON.stringify({
+                model: config.model,
+                max_tokens: config.max_tokens || 4096,
+                messages: [
+                    { role: 'system', content: systemPrompt },
+                    { role: 'user', content: userMessage },
+                ],
+            }),
+        });
+        if (!response.ok) {
+            const text = await response.text();
+            throw new Error(`API error ${response.status}: ${text}`);
+        }
+        const data = await response.json();
+        const content = data.choices?.[0]?.message?.content || '';
+        return {
+            content,
+            usage: {
+                input_tokens: data.usage?.prompt_tokens || 0,
+                output_tokens: data.usage?.completion_tokens || 0,
+            },
+        };
+    }
+}

package/dist/core/dag.d.ts

@@ -0,0 +1,17 @@
+/**
+ * DAG 构建和拓扑排序
+ */
+import type { WorkflowDefinition, DAGNode } from '../types.js';
+export interface DAG {
+    nodes: Map<string, DAGNode>;
+    /** 拓扑排序后的执行层级，每层内的节点可并行 */
+    levels: string[][];
+}
+/**
+ * 从 WorkflowDefinition 构建 DAG
+ */
+export declare function buildDAG(workflow: WorkflowDefinition): DAG;
+/**
+ * 格式化 DAG 为可读文本（用于 `ao plan` 命令）
+ */
+export declare function formatDAG(dag: DAG): string;

package/dist/core/dag.js

@@ -0,0 +1,90 @@
+/**
+ * 从 WorkflowDefinition 构建 DAG
+ */
+export function buildDAG(workflow) {
+    const nodes = new Map();
+    // 创建所有节点
+    for (const step of workflow.steps) {
+        nodes.set(step.id, {
+            step,
+            dependencies: step.depends_on || [],
+            dependents: [],
+            status: 'pending',
+        });
+    }
+    // 构建反向依赖（谁依赖我）
+    for (const [id, node] of nodes) {
+        for (const dep of node.dependencies) {
+            const depNode = nodes.get(dep);
+            if (!depNode) {
+                throw new Error(`step "${id}" 依赖不存在的 step: "${dep}"`);
+            }
+            depNode.dependents.push(id);
+        }
+    }
+    // 拓扑排序 — 按层分组（同层可并行）
+    const levels = topologicalLevels(nodes);
+    return { nodes, levels };
+}
+/**
+ * 拓扑排序，返回执行层级
+ * 每个层级内的节点互不依赖，可并行执行
+ *
+ * 例如:
+ *   A → B → D
+ *   A → C → D
+ * 结果: [[A], [B, C], [D]]
+ */
+function topologicalLevels(nodes) {
+    const inDegree = new Map();
+    for (const [id, node] of nodes) {
+        inDegree.set(id, node.dependencies.length);
+    }
+    const levels = [];
+    const remaining = new Set(nodes.keys());
+    while (remaining.size > 0) {
+        // 找出当前入度为 0 的节点
+        const currentLevel = [];
+        for (const id of remaining) {
+            if (inDegree.get(id) === 0) {
+                currentLevel.push(id);
+            }
+        }
+        if (currentLevel.length === 0) {
+            throw new Error('工作流存在循环依赖，无法拓扑排序');
+        }
+        // 移除本层节点，更新入度
+        for (const id of currentLevel) {
+            remaining.delete(id);
+            const node = nodes.get(id);
+            for (const dep of node.dependents) {
+                inDegree.set(dep, inDegree.get(dep) - 1);
+            }
+        }
+        levels.push(currentLevel);
+    }
+    return levels;
+}
+/**
+ * 格式化 DAG 为可读文本（用于 `ao plan` 命令）
+ */
+export function formatDAG(dag) {
+    const lines = ['执行计划:\n'];
+    for (let i = 0; i < dag.levels.length; i++) {
+        const level = dag.levels[i];
+        const parallel = level.length > 1;
+        for (let j = 0; j < level.length; j++) {
+            const node = dag.nodes.get(level[j]);
+            const step = node.step;
+            const prefix = parallel ? (j === 0 ? '┌' : j === level.length - 1 ? '└' : '├') : '→';
+            const tag = parallel ? ' (并行)' : '';
+            lines.push(`  第${i + 1}层 ${prefix} [${step.id}] ${step.role || step.type}${tag}`);
+            if (node.dependencies.length > 0) {
+                lines.push(`         依赖: ${node.dependencies.join(', ')}`);
+            }
+        }
+        if (i < dag.levels.length - 1)
+            lines.push('  │');
+    }
+    return lines.join('\n');
+}

package/dist/core/executor.d.ts

@@ -0,0 +1,20 @@
+/**
+ * DAG 执行引擎 — 核心调度器
+ */
+import type { DAGNode, LLMConnector, LLMConfig, WorkflowResult } from '../types.js';
+import type { DAG } from './dag.js';
+export interface ExecutorOptions {
+    connector: LLMConnector;
+    agentsDir: string;
+    llmConfig: LLMConfig;
+    concurrency: number;
+    inputs: Map<string, string>;
+    /** 每步完成的回调 */
+    onStepComplete?: (node: DAGNode) => void;
+    onStepStart?: (node: DAGNode) => void;
+    /** 一批并行步骤开始前的回调 */
+    onBatchStart?: (nodes: DAGNode[]) => void;
+    /** 一批并行步骤全部完成后的回调（按顺序） */
+    onBatchComplete?: (nodes: DAGNode[]) => void;
+}
+export declare function executeDAG(dag: DAG, options: ExecutorOptions): Promise<WorkflowResult>;