135 lines
3.4 KiB
JavaScript
135 lines
3.4 KiB
JavaScript
#!/usr/bin/env node
|
||
/**
|
||
* Skill 流程 + 总管 LLM 演示 CLI
|
||
*
|
||
* npm run dev # 真实 LLM（需 OPENAI_API_KEY）
|
||
* npm run demo # Mock LLM
|
||
*/
|
||
import * as readline from "node:readline/promises";
|
||
import { stdin as input, stdout as output } from "node:process";
|
||
import { loadLlmConfigOptional } from "../config/env.js";
|
||
import {
|
||
createMockMainAgentResponse,
|
||
MockLlmProvider,
|
||
} from "../llm/client.js";
|
||
import { createDefaultMainAgentLlm } from "../runtime/llm-factory.js";
|
||
import { formatSession, PhaseRuntime } from "../runtime/phase-runtime.js";
|
||
|
||
const useMock = process.argv.includes("--mock");
|
||
|
||
function printSession(runtime: PhaseRuntime): void {
|
||
console.log("\n--- 会话 ---");
|
||
console.log(formatSession(runtime.getSession()));
|
||
const index = runtime.getBlackboard().listTagIndex();
|
||
if (index.length > 0) {
|
||
console.log("blackboard tags:");
|
||
for (const entry of index) {
|
||
console.log(` - ${entry.tag} [${entry.source}]`);
|
||
}
|
||
}
|
||
console.log("----------------\n");
|
||
}
|
||
|
||
function printHelp(): void {
|
||
console.log(`命令:
|
||
<文本> 选 skill / 提交输入 / 驳回时发修改意见
|
||
/approve 确认总管建议的 worker
|
||
/accept 接受当前产物
|
||
/reject-art 拒绝当前产物
|
||
/status 查看状态
|
||
/quit 退出`);
|
||
}
|
||
|
||
async function main(): Promise<void> {
|
||
const llm = useMock
|
||
? new MockLlmProvider([
|
||
createMockMainAgentResponse({
|
||
action: "run_worker",
|
||
reason: "建议生成大纲",
|
||
workerId: "outline-worker",
|
||
requiresApproval: true,
|
||
}),
|
||
createMockMainAgentResponse({
|
||
action: "finish",
|
||
reason: "完成",
|
||
requiresApproval: false,
|
||
}),
|
||
])
|
||
: loadLlmConfigOptional()
|
||
? createDefaultMainAgentLlm()
|
||
: null;
|
||
|
||
if (!llm) {
|
||
console.error("请设置 OPENAI_API_KEY 或使用 --mock");
|
||
process.exit(1);
|
||
}
|
||
|
||
if (useMock || !loadLlmConfigOptional()) {
|
||
console.log("使用 Mock LLM\n");
|
||
}
|
||
|
||
const runtime = new PhaseRuntime({
|
||
autoStubWorker: true,
|
||
llm,
|
||
onMessage: (msg) => console.log(msg),
|
||
});
|
||
|
||
await runtime.start();
|
||
printSession(runtime);
|
||
printHelp();
|
||
|
||
const rl = readline.createInterface({ input, output });
|
||
|
||
try {
|
||
while (true) {
|
||
const line = (await rl.question("> ")).trim();
|
||
if (!line) continue;
|
||
|
||
if (line === "/quit") break;
|
||
if (line === "/help") {
|
||
printHelp();
|
||
continue;
|
||
}
|
||
if (line === "/status") {
|
||
printSession(runtime);
|
||
continue;
|
||
}
|
||
if (line === "/approve") {
|
||
await runtime.approve();
|
||
printSession(runtime);
|
||
continue;
|
||
}
|
||
if (line === "/accept") {
|
||
await runtime.acceptArtifact();
|
||
printSession(runtime);
|
||
continue;
|
||
}
|
||
if (line === "/reject-art") {
|
||
await runtime.rejectArtifact("用户拒绝产物");
|
||
printSession(runtime);
|
||
continue;
|
||
}
|
||
|
||
const reason = runtime.getSession().waitingReason;
|
||
if (reason?.kind === "approve_step") {
|
||
await runtime.rejectStep(line);
|
||
} else {
|
||
await runtime.submitInput(line);
|
||
}
|
||
printSession(runtime);
|
||
|
||
if (runtime.getSession().phase === "done") {
|
||
console.log("流程已完成。");
|
||
break;
|
||
}
|
||
}
|
||
} finally {
|
||
rl.close();
|
||
}
|
||
}
|
||
|
||
main().catch((err) => {
|
||
console.error(err instanceof Error ? err.message : err);
|
||
process.exit(1);
|
||
});
|