wangjia
f18db021c3
调度： - model/scoring.ts: score = 自身分(P0=3/P1=2/P2=1) + 已完成依赖分(链条惯性) + 等待解锁的 blocked 任务分(解锁加权)，编排器与 nextExecutable 同一打分 - createTask 校验 deps 存在且同项目(依赖图天然无环) - daemon 重启中断自愈: executing 任务标 failed run 后重新入队(reconcileInterrupted) 执行管线： - executor/cc.ts: 公共 headless CC 执行器(转录/超时/模型回退重试) - executor/reviewer.ts: 执行后自动复审(只读 CC 审 diff)，固定模板 summary (做了什么/怎么做/测试/CodeReview/安全Review/结论) + VERDICT 解析 - executor/models.ts: 按复杂度选模型(easy→sonnet/medium→opus/hard→fable5)， env 可覆盖、project.model 最优先、不可用自动回退链 - runner: 测试/构建命令白名单(npm/go/shellcheck/make/pytest)，prompt 要求实跑测试 - TaskResult 加 summary/verdict; RunKind 加 reviewer - 容器收口: 已拆解 Hard 子任务全 done → 容器自动 done(afterDone 逐级向上) 看板： - 五徽章组(待审批/待执行/执行中/被阻塞/总量，hover 展开，均不含已完成) - 归档区: 深度1整树完成沉底，时间倒序分页(10/20/50/100 chip 选择) - 归档详情对话框: 全属性/执行历史与时长/审批记录/状态流转时间线(含相关人或事) - Agent 面板显示调度模式 + 各复杂度实际模型 - 结果闸展示复审 summary + 建议通过/拒绝徽章 - 筛选修复(组选与单选分离、已拆解移出进行中)、同步按钮收进配置面板、 保存配置自动收起、预览全宽、被依赖阻塞→被阻塞 - API: GET /api/tasks/:id/events(任务级事件时间线)、/api/agents 带 scheduling/models 测试: 49/49(新增 scoring/复审/模型/容器收口/deps 校验/中断恢复) Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
99 lines
4.2 KiB
TypeScript
99 lines
4.2 KiB
TypeScript
import { test } from 'node:test';
|
||
import assert from 'node:assert/strict';
|
||
import { pickModel, isModelError, pickFallbackModel, MODEL_FALLBACK_CHAIN } from '../src/executor/models.js';
|
||
import type { Project, Task } from '../src/model/types.js';
|
||
import type { Complexity } from '../src/model/complexity.js';
|
||
|
||
function fakeTask(complexity: Complexity): Task {
|
||
return { complexity } as Task;
|
||
}
|
||
function fakeProject(model: string | null = null): Project {
|
||
return { model } as Project;
|
||
}
|
||
|
||
/** 临时设置 env，跑完恢复（避免污染其他用例） */
|
||
function withEnv(vars: Record<string, string>, fn: () => void): void {
|
||
const saved = new Map<string, string | undefined>();
|
||
for (const [k, v] of Object.entries(vars)) {
|
||
saved.set(k, process.env[k]);
|
||
process.env[k] = v;
|
||
}
|
||
try {
|
||
fn();
|
||
} finally {
|
||
for (const [k, v] of saved) {
|
||
if (v === undefined) delete process.env[k];
|
||
else process.env[k] = v;
|
||
}
|
||
}
|
||
}
|
||
|
||
const ENV_KEYS = [
|
||
'MAESTRO_MODEL_EASY', 'MAESTRO_MODEL_MEDIUM', 'MAESTRO_MODEL_HARD',
|
||
'MAESTRO_MODEL_REVIEW_EASY', 'MAESTRO_MODEL_REVIEW_MEDIUM', 'MAESTRO_MODEL_REVIEW_HARD',
|
||
];
|
||
|
||
test('pickModel：复杂度映射默认值（executor / reviewer）', () => {
|
||
// 防止外部 env 干扰：先清空再断言默认
|
||
withEnv(Object.fromEntries(ENV_KEYS.map((k) => [k, ''])), () => {
|
||
const p = fakeProject();
|
||
assert.equal(pickModel(fakeTask('easy'), p, 'executor'), 'claude-sonnet-4-6');
|
||
assert.equal(pickModel(fakeTask('medium'), p, 'executor'), 'claude-opus-4-8');
|
||
assert.equal(pickModel(fakeTask('hard'), p, 'executor'), 'claude-fable-5');
|
||
assert.equal(pickModel(fakeTask('easy'), p, 'reviewer'), 'claude-sonnet-4-6');
|
||
assert.equal(pickModel(fakeTask('medium'), p, 'reviewer'), 'claude-opus-4-8');
|
||
assert.equal(pickModel(fakeTask('hard'), p, 'reviewer'), 'claude-opus-4-8');
|
||
});
|
||
});
|
||
|
||
test('pickModel：project.model 最优先（两个角色、所有复杂度都用它）', () => {
|
||
withEnv({ MAESTRO_MODEL_HARD: 'claude-haiku-4-5' }, () => {
|
||
const p = fakeProject('claude-opus-4-6');
|
||
for (const c of ['easy', 'medium', 'hard'] as const) {
|
||
assert.equal(pickModel(fakeTask(c), p, 'executor'), 'claude-opus-4-6');
|
||
assert.equal(pickModel(fakeTask(c), p, 'reviewer'), 'claude-opus-4-6');
|
||
}
|
||
});
|
||
});
|
||
|
||
test('pickModel：env 覆盖默认值，且 executor/reviewer 各用各的 env', () => {
|
||
withEnv({
|
||
MAESTRO_MODEL_EASY: 'claude-haiku-4-5',
|
||
MAESTRO_MODEL_REVIEW_HARD: 'claude-fable-5',
|
||
}, () => {
|
||
const p = fakeProject();
|
||
assert.equal(pickModel(fakeTask('easy'), p, 'executor'), 'claude-haiku-4-5');
|
||
assert.equal(pickModel(fakeTask('easy'), p, 'reviewer'), 'claude-sonnet-4-6'); // reviewer 不受 executor env 影响
|
||
assert.equal(pickModel(fakeTask('hard'), p, 'reviewer'), 'claude-fable-5');
|
||
assert.equal(pickModel(fakeTask('hard'), p, 'executor'), 'claude-fable-5'); // executor hard 默认即 fable
|
||
});
|
||
});
|
||
|
||
test('pickModel：env 为空白字符串视为未设置', () => {
|
||
withEnv({ MAESTRO_MODEL_MEDIUM: ' ' }, () => {
|
||
assert.equal(pickModel(fakeTask('medium'), fakeProject(), 'executor'), 'claude-opus-4-8');
|
||
});
|
||
});
|
||
|
||
test('isModelError：模型不可用类错误才触发回退', () => {
|
||
// 应触发
|
||
assert.ok(isModelError('model claude-fable-5 not_found'));
|
||
assert.ok(isModelError('Invalid model: claude-x'));
|
||
assert.ok(isModelError('permission denied for model claude-fable-5'));
|
||
assert.ok(isModelError('404 model does not exist'));
|
||
assert.ok(isModelError('The requested model is unavailable'));
|
||
assert.ok(isModelError('unknown model "claude-fable-5"'));
|
||
// 不应触发
|
||
assert.ok(!isModelError('执行超时'));
|
||
assert.ok(!isModelError('network connection refused'));
|
||
assert.ok(!isModelError('not found: /tmp/some/file')); // 没提到 model
|
||
assert.ok(!isModelError('model produced empty output')); // 提到 model 但不是可用性错误
|
||
});
|
||
|
||
test('pickFallbackModel：回退链取第一个 ≠ 失败模型的', () => {
|
||
assert.deepEqual([...MODEL_FALLBACK_CHAIN], ['claude-opus-4-8', 'claude-sonnet-4-6']);
|
||
assert.equal(pickFallbackModel('claude-fable-5'), 'claude-opus-4-8');
|
||
assert.equal(pickFallbackModel('claude-opus-4-8'), 'claude-sonnet-4-6');
|
||
assert.equal(pickFallbackModel('claude-sonnet-4-6'), 'claude-opus-4-8');
|
||
});
|