agency-orchestrator 0.3.2 → 0.4.2

package/dist/connectors/cli-base.js

@@ -0,0 +1,53 @@
+/**
+ * CLI Connector 通用基类
+ * 通过本地 AI CLI 工具调用，使用用户的订阅额度，无需 API key
+ *
+ * 支持: Claude Code / Gemini CLI / Copilot CLI / Codex CLI / OpenClaw CLI
+ */
+import { execFile } from 'node:child_process';
+import { promisify } from 'node:util';
+const execFileAsync = promisify(execFile);
+export class CLIBaseConnector {
+    cfg;
+    constructor(cfg) {
+        this.cfg = cfg;
+    }
+    async chat(systemPrompt, userMessage, config) {
+        const fullPrompt = systemPrompt
+            ? `<system>\n${systemPrompt}\n</system>\n\n${userMessage}`
+            : userMessage;
+        const args = this.cfg.buildArgs(fullPrompt, config);
+        const timeout = config.timeout || 180000;
+        try {
+            const { stdout, stderr } = await execFileAsync(this.cfg.command, args, {
+                timeout,
+                maxBuffer: 10 * 1024 * 1024,
+                env: { ...process.env },
+            });
+            const content = this.cfg.parseOutput
+                ? this.cfg.parseOutput(stdout)
+                : stdout.trim();
+            if (!content && stderr) {
+                throw new Error(`${this.cfg.displayName} 返回空内容，stderr: ${stderr.slice(0, 500)}`);
+            }
+            return {
+                content,
+                usage: {
+                    input_tokens: Math.ceil((systemPrompt.length + userMessage.length) / 4),
+                    output_tokens: Math.ceil(content.length / 4),
+                },
+            };
+        }
+        catch (err) {
+            if (err.killed || err.signal === 'SIGTERM') {
+                throw new Error(`${this.cfg.displayName} 超时 (${timeout / 1000}s)，可在 YAML 中设置 timeout 增加等待时间`);
+            }
+            // 区分"未安装"和"执行失败"
+            if (err.code === 'ENOENT') {
+                throw new Error(`找不到 ${this.cfg.command} 命令，请先安装 ${this.cfg.displayName}\n` +
+                    `参考: https://github.com/jnMetaCode/agency-orchestrator#llm-配置`);
+            }
+            throw new Error(`${this.cfg.displayName} 调用失败: ${err.message}`);
+        }
+    }
+}

package/dist/connectors/codex-cli.d.ts

@@ -0,0 +1,11 @@
+/**
+ * OpenAI Codex CLI Connector
+ * 通过本地 `codex` CLI 调用，直接使用 ChatGPT Plus/Pro 订阅额度
+ *
+ * 安装: npm install -g @openai/codex
+ * 认证: codex 首次运行会引导 OpenAI 登录，之后自动认证
+ */
+import { CLIBaseConnector } from './cli-base.js';
+export declare class CodexCLIConnector extends CLIBaseConnector {
+    constructor();
+}

package/dist/connectors/codex-cli.js

@@ -0,0 +1,19 @@
+/**
+ * OpenAI Codex CLI Connector
+ * 通过本地 `codex` CLI 调用，直接使用 ChatGPT Plus/Pro 订阅额度
+ *
+ * 安装: npm install -g @openai/codex
+ * 认证: codex 首次运行会引导 OpenAI 登录，之后自动认证
+ */
+import { CLIBaseConnector } from './cli-base.js';
+export class CodexCLIConnector extends CLIBaseConnector {
+    constructor() {
+        super({
+            command: 'codex',
+            displayName: 'OpenAI Codex CLI',
+            buildArgs: (prompt, _config) => {
+                return ['-q', prompt];
+            },
+        });
+    }
+}

package/dist/connectors/copilot-cli.d.ts

@@ -0,0 +1,11 @@
+/**
+ * GitHub Copilot CLI Connector
+ * 通过本地 `copilot` CLI 调用，直接使用 GitHub Copilot 订阅额度
+ *
+ * 安装: npm install -g @github/copilot
+ * 认证: copilot 首次运行会引导 GitHub 登录，之后自动认证
+ */
+import { CLIBaseConnector } from './cli-base.js';
+export declare class CopilotCLIConnector extends CLIBaseConnector {
+    constructor();
+}

package/dist/connectors/copilot-cli.js

@@ -0,0 +1,19 @@
+/**
+ * GitHub Copilot CLI Connector
+ * 通过本地 `copilot` CLI 调用，直接使用 GitHub Copilot 订阅额度
+ *
+ * 安装: npm install -g @github/copilot
+ * 认证: copilot 首次运行会引导 GitHub 登录，之后自动认证
+ */
+import { CLIBaseConnector } from './cli-base.js';
+export class CopilotCLIConnector extends CLIBaseConnector {
+    constructor() {
+        super({
+            command: 'copilot',
+            displayName: 'GitHub Copilot CLI',
+            buildArgs: (prompt, _config) => {
+                return ['-p', prompt];
+            },
+        });
+    }
+}

package/dist/connectors/factory.d.ts

@@ -0,0 +1,5 @@
+/**
+ * LLM Connector 工厂 — 根据 LLMConfig 创建对应的 connector
+ */
+import type { LLMConfig, LLMConnector } from '../types.js';
+export declare function createConnector(config: LLMConfig): LLMConnector;

package/dist/connectors/factory.js

@@ -0,0 +1,42 @@
+import { ClaudeConnector } from './claude.js';
+import { ClaudeCodeConnector } from './claude-code.js';
+import { GeminiCLIConnector } from './gemini-cli.js';
+import { CopilotCLIConnector } from './copilot-cli.js';
+import { CodexCLIConnector } from './codex-cli.js';
+import { OpenClawCLIConnector } from './openclaw-cli.js';
+import { OllamaConnector } from './ollama.js';
+import { OpenAICompatibleConnector } from './openai-compatible.js';
+export function createConnector(config) {
+    switch (config.provider) {
+        // ── 免 API key（用订阅 / 免费额度）──
+        case 'claude-code':
+            return new ClaudeCodeConnector();
+        case 'gemini-cli':
+            return new GeminiCLIConnector();
+        case 'copilot-cli':
+            return new CopilotCLIConnector();
+        case 'codex-cli':
+            return new CodexCLIConnector();
+        case 'openclaw-cli':
+            return new OpenClawCLIConnector();
+        case 'ollama':
+            return new OllamaConnector(config.base_url);
+        // ── 需要 API key ──
+        case 'claude':
+            return new ClaudeConnector(config.api_key);
+        case 'deepseek':
+            return new OpenAICompatibleConnector({
+                apiKey: config.api_key || process.env.DEEPSEEK_API_KEY,
+                baseUrl: config.base_url || 'https://api.deepseek.com/v1',
+            });
+        case 'openai':
+            return new OpenAICompatibleConnector({
+                apiKey: config.api_key || process.env.OPENAI_API_KEY,
+                baseUrl: config.base_url || 'https://api.openai.com/v1',
+            });
+        default:
+            throw new Error(`暂不支持 provider: ${config.provider}\n` +
+                '免 API key: claude-code / gemini-cli / copilot-cli / codex-cli / openclaw-cli / ollama\n' +
+                '需 API key: claude / deepseek / openai');
+    }
+}

package/dist/connectors/gemini-cli.d.ts

@@ -0,0 +1,11 @@
+/**
+ * Gemini CLI Connector
+ * 通过本地 `gemini` CLI 调用，Google 账号免费使用（1000 次/天，Gemini 2.5 Pro）
+ *
+ * 安装: npm install -g @google/gemini-cli
+ * 认证: gemini 首次运行会引导 Google 账号登录，之后自动认证
+ */
+import { CLIBaseConnector } from './cli-base.js';
+export declare class GeminiCLIConnector extends CLIBaseConnector {
+    constructor();
+}

package/dist/connectors/gemini-cli.js

@@ -0,0 +1,19 @@
+/**
+ * Gemini CLI Connector
+ * 通过本地 `gemini` CLI 调用，Google 账号免费使用（1000 次/天，Gemini 2.5 Pro）
+ *
+ * 安装: npm install -g @google/gemini-cli
+ * 认证: gemini 首次运行会引导 Google 账号登录，之后自动认证
+ */
+import { CLIBaseConnector } from './cli-base.js';
+export class GeminiCLIConnector extends CLIBaseConnector {
+    constructor() {
+        super({
+            command: 'gemini',
+            displayName: 'Gemini CLI',
+            buildArgs: (prompt, _config) => {
+                return ['-p', prompt];
+            },
+        });
+    }
+}

package/dist/connectors/ollama.js

@@ -9,7 +9,7 @@ export class OllamaConnector {
             method: 'POST',
             headers: { 'Content-Type': 'application/json' },
             body: JSON.stringify({
-                model: config.model,
+                model: config.model || 'llama3.1',
                 messages: [
                     { role: 'system', content: systemPrompt },
                     { role: 'user', content: userMessage },

package/dist/connectors/openclaw-cli.d.ts

@@ -0,0 +1,15 @@
+/**
+ * OpenClaw CLI Connector
+ * 通过本地 `openclaw` CLI 调用，支持 OAuth 和 API key 多种认证方式
+ *
+ * 安装: npm install -g openclaw@latest
+ * 认证: openclaw onboard --install-daemon（引导配置）
+ *
+ * 注意: openclaw agent 必须指定 --agent <id>，否则会报
+ * "Pass --to <E.164>, --session-id, or --agent to choose a session"
+ * 默认使用 "main" agent，可通过 YAML model 字段或 OPENCLAW_AGENT 环境变量覆盖
+ */
+import { CLIBaseConnector } from './cli-base.js';
+export declare class OpenClawCLIConnector extends CLIBaseConnector {
+    constructor();
+}

package/dist/connectors/openclaw-cli.js

@@ -0,0 +1,37 @@
+/**
+ * OpenClaw CLI Connector
+ * 通过本地 `openclaw` CLI 调用，支持 OAuth 和 API key 多种认证方式
+ *
+ * 安装: npm install -g openclaw@latest
+ * 认证: openclaw onboard --install-daemon（引导配置）
+ *
+ * 注意: openclaw agent 必须指定 --agent <id>，否则会报
+ * "Pass --to <E.164>, --session-id, or --agent to choose a session"
+ * 默认使用 "main" agent，可通过 YAML model 字段或 OPENCLAW_AGENT 环境变量覆盖
+ */
+import { CLIBaseConnector } from './cli-base.js';
+export class OpenClawCLIConnector extends CLIBaseConnector {
+    constructor() {
+        super({
+            command: 'openclaw',
+            displayName: 'OpenClaw CLI',
+            buildArgs: (prompt, config) => {
+                // model 字段复用为 agent id，例如 model: "my-agent"
+                const agentId = config.model || process.env.OPENCLAW_AGENT || 'main';
+                return ['agent', '--agent', agentId, '--message', prompt];
+            },
+            parseOutput: (stdout) => {
+                // OpenClaw 的 stdout 可能混入插件日志（如 ShellWard 的 ANSI 彩色输出）
+                // 过滤掉 [plugins] 前缀行和 ANSI 控制码行
+                return stdout
+                    .split('\n')
+                    .filter(line => {
+                    const clean = line.replace(/\x1b\[[0-9;]*m/g, '').trim();
+                    return clean && !clean.startsWith('[plugins]');
+                })
+                    .join('\n')
+                    .trim();
+            },
+        });
+    }
+}

package/dist/core/parser.js

@@ -20,8 +20,13 @@ export function parseWorkflow(filePath) {
         throw new Error('工作流缺少 llm 配置');
     }
     const llm = doc.llm;
-    if (!llm.provider || !llm.model) {
-        throw new Error('llm 配置缺少 provider 或 model');
+    if (!llm.provider) {
+        throw new Error('llm 配置缺少 provider');
+    }
+    // CLI providers (claude-code, gemini-cli, copilot-cli, codex-cli, openclaw-cli) 和 ollama 不需要 model
+    const cliProviders = ['claude-code', 'gemini-cli', 'copilot-cli', 'codex-cli', 'openclaw-cli', 'ollama'];
+    if (!llm.model && !cliProviders.includes(llm.provider)) {
+        throw new Error('llm 配置缺少 model（CLI provider 可省略）');
     }
     // 校验每个 step
     const stepIds = new Set();

package/dist/index.d.ts

@@ -13,6 +13,7 @@ export { loadAgent, listAgents } from './agents/loader.js';
 export { ClaudeConnector } from './connectors/claude.js';
 export { OllamaConnector } from './connectors/ollama.js';
 export { OpenAICompatibleConnector } from './connectors/openai-compatible.js';
+export { createConnector } from './connectors/factory.js';
 export { saveResults, loadPreviousContext, findLatestOutput } from './output/reporter.js';
 export { composeWorkflow, buildRoleCatalog, extractYamlFromResponse } from './cli/compose.js';
 export type { WorkflowDefinition, StepDefinition, LLMConfig, LLMConnector, LLMResult, AgentDefinition, WorkflowResult, StepResult, DAGNode, } from './types.js';
@@ -28,4 +29,6 @@ export declare function run(workflowPath: string, inputs: Record<string, string>
     resumeDir?: string;
     /** 从指定步骤开始重新执行（跳过之前的步骤） */
     fromStep?: string;
+    /** 覆盖 LLM 配置（例如来自 ao demo） */
+    llmOverride?: Partial<import('./types.js').LLMConfig>;
 }): Promise<import('./types.js').WorkflowResult>;

package/dist/index.js

@@ -13,14 +13,13 @@ export { loadAgent, listAgents } from './agents/loader.js';
 export { ClaudeConnector } from './connectors/claude.js';
 export { OllamaConnector } from './connectors/ollama.js';
 export { OpenAICompatibleConnector } from './connectors/openai-compatible.js';
+export { createConnector } from './connectors/factory.js';
 export { saveResults, loadPreviousContext, findLatestOutput } from './output/reporter.js';
 export { composeWorkflow, buildRoleCatalog, extractYamlFromResponse } from './cli/compose.js';
 import { parseWorkflow, validateWorkflow } from './core/parser.js';
 import { buildDAG } from './core/dag.js';
 import { executeDAG } from './core/executor.js';
-import { ClaudeConnector } from './connectors/claude.js';
-import { OllamaConnector } from './connectors/ollama.js';
-import { OpenAICompatibleConnector } from './connectors/openai-compatible.js';
+import { createConnector } from './connectors/factory.js';
 import { saveResults, printStepResult, printStepRunning, clearRunningLine, printSummary, loadPreviousContext, getCompletedStepIds } from './output/reporter.js';
 import { existsSync } from 'node:fs';
 import { resolve, dirname } from 'node:path';
@@ -38,30 +37,12 @@ export async function run(workflowPath, inputs, options) {
     }
     // 构建 DAG
     const dag = buildDAG(workflow);
-    // 创建 connector
-    let connector;
-    switch (workflow.llm.provider) {
-        case 'claude':
-            connector = new ClaudeConnector(workflow.llm.api_key);
-            break;
-        case 'ollama':
-            connector = new OllamaConnector(workflow.llm.base_url);
-            break;
-        case 'deepseek':
-            connector = new OpenAICompatibleConnector({
-                apiKey: workflow.llm.api_key || process.env.DEEPSEEK_API_KEY,
-                baseUrl: workflow.llm.base_url || 'https://api.deepseek.com/v1',
-            });
-            break;
-        case 'openai':
-            connector = new OpenAICompatibleConnector({
-                apiKey: workflow.llm.api_key || process.env.OPENAI_API_KEY,
-                baseUrl: workflow.llm.base_url || 'https://api.openai.com/v1',
-            });
-            break;
-        default:
-            throw new Error(`暂不支持 provider: ${workflow.llm.provider}（支持 claude / deepseek / openai / ollama）`);
+    // Apply LLM override (e.g., from ao demo)
+    if (options?.llmOverride) {
+        Object.assign(workflow.llm, options.llmOverride);
     }
+    // 创建 connector
+    const connector = createConnector(workflow.llm);
     // 构建输入
     const inputMap = new Map(Object.entries(inputs));
     // 检查必填输入 + 注入默认值
@@ -163,7 +144,7 @@ export async function run(workflowPath, inputs, options) {
     });
     result.name = workflow.name;
     // 保存结果
-    const outputDir = options?.outputDir || '.ao-output';
+    const outputDir = options?.outputDir || 'ao-output';
     const outputPath = saveResults(result, outputDir);
     if (!quiet) {
         printSummary(result, outputPath);

package/dist/mcp/server.d.ts

@@ -0,0 +1 @@
+export declare function startServer(verbose?: boolean): Promise<void>;

package/dist/mcp/server.js

@@ -0,0 +1,288 @@
+/**
+ * MCP Server — 通过 MCP 协议暴露工作流操作
+ *
+ * 6 个工具，每个都是对现有函数的薄封装。
+ * 传输层: StdioServerTransport (stdin/stdout JSON-RPC)
+ *
+ * 重要: 所有日志必须输出到 stderr，stdout 是 MCP 协议通道。
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { z } from 'zod';
+import { resolve, relative, dirname } from 'node:path';
+import { existsSync, readFileSync, readdirSync, statSync } from 'node:fs';
+import { createRequire } from 'node:module';
+import * as yaml from 'js-yaml';
+import { run } from '../index.js';
+import { parseWorkflow, validateWorkflow } from '../core/parser.js';
+import { buildDAG, formatDAG } from '../core/dag.js';
+import { listAgents } from '../agents/loader.js';
+import { composeWorkflow } from '../cli/compose.js';
+/** 自动查找 agents 目录 */
+function findAgentsDir(hint) {
+    if (hint && existsSync(resolve(hint)))
+        return resolve(hint);
+    const candidates = [
+        resolve('agency-agents-zh'),
+        resolve('../agency-agents-zh'),
+        resolve('agents'),
+        resolve('node_modules/agency-agents-zh'),
+        resolve(dirname(new URL(import.meta.url).pathname), '../../node_modules/agency-agents-zh'),
+    ];
+    for (const dir of candidates) {
+        if (existsSync(dir))
+            return dir;
+    }
+    return resolve(hint || './agency-agents-zh');
+}
+/** 递归查找 YAML 文件 */
+function findYamlFiles(dir, result) {
+    for (const entry of readdirSync(dir)) {
+        const full = resolve(dir, entry);
+        if (statSync(full).isDirectory()) {
+            findYamlFiles(full, result);
+        }
+        else if (entry.endsWith('.yaml') || entry.endsWith('.yml')) {
+            result.push(full);
+        }
+    }
+}
+/** 列出 workflows 目录下的 YAML 文件 */
+function discoverWorkflows() {
+    const workflowsDir = resolve('workflows');
+    if (!existsSync(workflowsDir))
+        return [];
+    const files = [];
+    findYamlFiles(workflowsDir, files);
+    return files.map(f => {
+        try {
+            const content = readFileSync(f, 'utf-8');
+            const doc = yaml.load(content);
+            return {
+                file: relative(process.cwd(), f),
+                name: doc?.name || '(unnamed)',
+                description: doc?.description || '',
+            };
+        }
+        catch {
+            return { file: relative(process.cwd(), f), name: '(parse error)', description: '' };
+        }
+    });
+}
+/**
+ * 静默执行函数 — 临时屏蔽 stdout 输出（引用计数，并发安全）
+ * composeWorkflow 等函数内部有 console.log，会污染 MCP 协议通道
+ */
+let suppressCount = 0;
+let origStdoutWrite = null;
+async function silentCall(fn) {
+    if (suppressCount === 0) {
+        origStdoutWrite = process.stdout.write.bind(process.stdout);
+        process.stdout.write = () => true;
+    }
+    suppressCount++;
+    try {
+        return await fn();
+    }
+    finally {
+        suppressCount--;
+        if (suppressCount === 0 && origStdoutWrite) {
+            process.stdout.write = origStdoutWrite;
+            origStdoutWrite = null;
+        }
+    }
+}
+export async function startServer(verbose = false) {
+    const require = createRequire(import.meta.url);
+    const pkgPath = require.resolve('../../package.json');
+    const pkg = JSON.parse(readFileSync(pkgPath, 'utf-8'));
+    const server = new McpServer({
+        name: 'agency-orchestrator',
+        version: pkg.version || '0.0.0',
+    });
+    // ─── Tool 1: run_workflow ───
+    server.tool('run_workflow', 'Execute a YAML workflow with the DAG engine', {
+        path: z.string().describe('Path to workflow YAML file'),
+        inputs: z.record(z.string(), z.string()).optional().describe('Key-value input variables'),
+        provider: z.enum(['claude-code', 'gemini-cli', 'copilot-cli', 'codex-cli', 'openclaw-cli', 'deepseek', 'claude', 'openai', 'ollama']).optional().describe('Override LLM provider'),
+        model: z.string().optional().describe('Override model name'),
+    }, async ({ path: workflowPath, inputs, provider, model }) => {
+        try {
+            const absPath = resolve(workflowPath);
+            if (!existsSync(absPath)) {
+                return { content: [{ type: 'text', text: `文件不存在: ${workflowPath}` }], isError: true };
+            }
+            const llmOverride = {};
+            if (provider)
+                llmOverride.provider = provider;
+            if (model)
+                llmOverride.model = model;
+            const result = await silentCall(() => run(absPath, (inputs || {}), {
+                quiet: true,
+                llmOverride: Object.keys(llmOverride).length > 0 ? llmOverride : undefined,
+            }));
+            const lastStep = result.steps[result.steps.length - 1];
+            const output = lastStep?.output || '(no output)';
+            const tokenSummary = `Tokens: ${result.totalTokens.input} in / ${result.totalTokens.output} out`;
+            return {
+                content: [{ type: 'text', text: `${output}\n\n---\n${tokenSummary}` }],
+            };
+        }
+        catch (err) {
+            return {
+                content: [{ type: 'text', text: `执行失败: ${err instanceof Error ? err.message : String(err)}` }],
+                isError: true,
+            };
+        }
+    });
+    // ─── Tool 2: validate_workflow ───
+    server.tool('validate_workflow', 'Validate a workflow YAML without executing', {
+        path: z.string().describe('Path to workflow YAML file'),
+    }, async ({ path: workflowPath }) => {
+        try {
+            const absPath = resolve(workflowPath);
+            if (!existsSync(absPath)) {
+                return { content: [{ type: 'text', text: `文件不存在: ${workflowPath}` }], isError: true };
+            }
+            const workflow = parseWorkflow(absPath);
+            const errors = validateWorkflow(workflow);
+            if (errors.length === 0) {
+                return {
+                    content: [{
+                            type: 'text',
+                            text: `✅ ${workflow.name} — 校验通过\n步骤数: ${workflow.steps.length}\n输入数: ${(workflow.inputs || []).length}`,
+                        }],
+                };
+            }
+            else {
+                return {
+                    content: [{
+                            type: 'text',
+                            text: `❌ ${workflow.name} — 校验失败:\n${errors.map(e => `  - ${e}`).join('\n')}`,
+                        }],
+                    isError: true,
+                };
+            }
+        }
+        catch (err) {
+            return {
+                content: [{ type: 'text', text: `校验错误: ${err instanceof Error ? err.message : String(err)}` }],
+                isError: true,
+            };
+        }
+    });
+    // ─── Tool 3: list_workflows ───
+    server.tool('list_workflows', 'List available workflow templates from the workflows/ directory', {}, async () => {
+        try {
+            const workflows = discoverWorkflows();
+            if (workflows.length === 0) {
+                return { content: [{ type: 'text', text: '未找到工作流文件（workflows/ 目录不存在或为空）' }] };
+            }
+            const lines = workflows.map(w => `- ${w.file}: ${w.name}${w.description ? ` — ${w.description}` : ''}`);
+            return {
+                content: [{ type: 'text', text: `共 ${workflows.length} 个工作流:\n\n${lines.join('\n')}` }],
+            };
+        }
+        catch (err) {
+            return {
+                content: [{ type: 'text', text: `列出工作流失败: ${err instanceof Error ? err.message : String(err)}` }],
+                isError: true,
+            };
+        }
+    });
+    // ─── Tool 4: plan_workflow ───
+    server.tool('plan_workflow', 'Show the DAG execution plan for a workflow', {
+        path: z.string().describe('Path to workflow YAML file'),
+    }, async ({ path: workflowPath }) => {
+        try {
+            const absPath = resolve(workflowPath);
+            if (!existsSync(absPath)) {
+                return { content: [{ type: 'text', text: `文件不存在: ${workflowPath}` }], isError: true };
+            }
+            const workflow = parseWorkflow(absPath);
+            const errors = validateWorkflow(workflow);
+            if (errors.length > 0) {
+                return {
+                    content: [{ type: 'text', text: `校验失败:\n${errors.map(e => `  - ${e}`).join('\n')}` }],
+                    isError: true,
+                };
+            }
+            const dag = buildDAG(workflow);
+            const dagText = formatDAG(dag);
+            return {
+                content: [{ type: 'text', text: `${workflow.name}\n\n${dagText}` }],
+            };
+        }
+        catch (err) {
+            return {
+                content: [{ type: 'text', text: `计划失败: ${err instanceof Error ? err.message : String(err)}` }],
+                isError: true,
+            };
+        }
+    });
+    // ─── Tool 5: compose_workflow ───
+    server.tool('compose_workflow', 'Generate a workflow YAML from a natural language description using AI', {
+        description: z.string().describe('One-sentence workflow description'),
+        provider: z.enum(['deepseek', 'claude', 'openai', 'ollama']).optional().describe('LLM provider (default: deepseek)'),
+        model: z.string().optional().describe('Model name'),
+    }, async ({ description, provider, model }) => {
+        try {
+            const agentsDir = findAgentsDir();
+            const llmProvider = provider || 'deepseek';
+            const defaultModels = {
+                deepseek: 'deepseek-chat',
+                claude: 'claude-sonnet-4-20250514',
+                openai: 'gpt-4o',
+                ollama: 'llama3',
+            };
+            const llmModel = model || defaultModels[llmProvider] || 'gpt-4o';
+            const result = await silentCall(() => composeWorkflow({
+                description,
+                agentsDir,
+                llmConfig: { provider: llmProvider, model: llmModel },
+            }));
+            let text = `✅ 工作流已生成: ${result.relativePath}\n\n${result.yaml}`;
+            if (result.warnings.length > 0) {
+                text += `\n\n⚠️ 校验警告:\n${result.warnings.map(w => `  - ${w}`).join('\n')}`;
+            }
+            return { content: [{ type: 'text', text }] };
+        }
+        catch (err) {
+            return {
+                content: [{ type: 'text', text: `生成失败: ${err instanceof Error ? err.message : String(err)}` }],
+                isError: true,
+            };
+        }
+    });
+    // ─── Tool 6: list_roles ───
+    server.tool('list_roles', 'List available AI roles from the agents directory', {
+        agents_dir: z.string().optional().describe('Path to agents directory (auto-resolved if omitted)'),
+    }, async ({ agents_dir }) => {
+        try {
+            const agentsDir = findAgentsDir(agents_dir);
+            const agents = listAgents(agentsDir);
+            const lines = agents.map(a => {
+                const emoji = a.emoji || ' ';
+                return `${emoji} ${a.name} — ${a.description || '(无描述)'}`;
+            });
+            return {
+                content: [{ type: 'text', text: `共 ${agents.length} 个角色:\n\n${lines.join('\n')}` }],
+            };
+        }
+        catch (err) {
+            return {
+                content: [{ type: 'text', text: `列出角色失败: ${err instanceof Error ? err.message : String(err)}` }],
+                isError: true,
+            };
+        }
+    });
+    // ─── Start server ───
+    if (verbose) {
+        console.error('[ao-mcp] Starting MCP server...');
+    }
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    if (verbose) {
+        console.error('[ao-mcp] Server connected via stdio');
+    }
+}

package/dist/types.d.ts

@@ -9,10 +9,10 @@ export interface WorkflowDefinition {
     steps: StepDefinition[];
 }
 export interface LLMConfig {
-    provider: 'claude' | 'openai' | 'ollama' | 'deepseek';
+    provider: 'claude' | 'openai' | 'ollama' | 'deepseek' | 'claude-code' | 'gemini-cli' | 'copilot-cli' | 'codex-cli' | 'openclaw-cli';
     base_url?: string;
     api_key?: string;
-    model: string;
+    model?: string;
     max_tokens?: number;
     timeout?: number;
     retry?: number;