106 lines
3.6 KiB
JavaScript
106 lines
3.6 KiB
JavaScript
/**
|
||
* 回合执行器按工具执行模式分流（A3.4）。
|
||
*
|
||
* 两种模式必须都覆盖，否则 Pi 那条路只有真模型测试才碰得到：
|
||
* platform —— 平台执行并回填（假引擎走这条）
|
||
* runtime —— 引擎自己执行（Pi + 扩展桥），平台只观察事件、不得重复执行
|
||
*/
|
||
import assert from 'node:assert/strict';
|
||
import { test } from 'node:test';
|
||
|
||
import { runTurn } from '@aps/server/app/turn-runner';
|
||
|
||
function stubSession(mode, events) {
|
||
return {
|
||
id: 'stub',
|
||
toolExecution: mode,
|
||
async *send() {
|
||
for (const event of events) yield event;
|
||
},
|
||
resolveToolResult() {},
|
||
abort() {},
|
||
async dispose() {},
|
||
};
|
||
}
|
||
|
||
const toolCallEvents = [
|
||
{ type: 'text', delta: '我来调用。' },
|
||
{ type: 'tool_call', callId: 'c-1', capabilityId: 'test.echo', args: { q: 1 } },
|
||
{ type: 'tool_result', callId: 'c-1', receipt: { executionId: 'exec-1', status: 'succeeded' } },
|
||
{ type: 'turn_end', reason: 'stop' },
|
||
];
|
||
|
||
function collect() {
|
||
const events = [];
|
||
return { events, publish: (event) => events.push(event) };
|
||
}
|
||
|
||
test('runtime 模式：平台不执行工具，只观察 tool.call 与 tool.result', async () => {
|
||
const { events, publish } = collect();
|
||
let invocations = 0;
|
||
const result = await runTurn({
|
||
runtimeSession: stubSession('runtime', toolCallEvents),
|
||
sessionId: 's-1',
|
||
runId: 'r-1',
|
||
assistantMessageId: 'm-1',
|
||
text: '调用试试',
|
||
publish,
|
||
invoke: async () => {
|
||
invocations += 1;
|
||
return { executionId: 'platform-should-not-run', status: 'succeeded' };
|
||
},
|
||
signal: new AbortController().signal,
|
||
});
|
||
|
||
assert.equal(invocations, 0, '引擎模式下平台不得再执行一次（否则等于重复调用）');
|
||
const types = events.map((event) => event.type);
|
||
assert.deepEqual(types, ['run.state', 'message.delta', 'tool.call', 'tool.result', 'message.completed', 'run.state']);
|
||
assert.equal(result.toolCalls, 1);
|
||
assert.equal(result.finishReason, 'stop');
|
||
});
|
||
|
||
test('platform 模式：平台执行并把回执回填给运行时', async () => {
|
||
const { events, publish } = collect();
|
||
const resolved = [];
|
||
const session = stubSession('platform', toolCallEvents);
|
||
session.resolveToolResult = (callId, receipt) => resolved.push({ callId, receipt });
|
||
|
||
let invocations = 0;
|
||
await runTurn({
|
||
runtimeSession: session,
|
||
sessionId: 's-1',
|
||
runId: 'r-1',
|
||
assistantMessageId: 'm-1',
|
||
text: '调用试试',
|
||
publish,
|
||
invoke: async () => {
|
||
invocations += 1;
|
||
return { executionId: 'exec-platform', status: 'succeeded' };
|
||
},
|
||
signal: new AbortController().signal,
|
||
});
|
||
|
||
assert.equal(invocations, 1, '平台模式必须执行一次');
|
||
assert.deepEqual(resolved, [{ callId: 'c-1', receipt: { executionId: 'exec-platform', status: 'succeeded' } }]);
|
||
const types = events.map((event) => event.type);
|
||
assert.ok(types.includes('tool.result'), types.join(','));
|
||
});
|
||
|
||
test('错误回合的结束原因不会被 turn_end 覆盖回 stop', async () => {
|
||
const { events, publish } = collect();
|
||
const result = await runTurn({
|
||
runtimeSession: stubSession('runtime', [
|
||
{ type: 'error', kind: 'forbidden', message: '401 无效密钥' },
|
||
{ type: 'turn_end', reason: 'stop' },
|
||
]),
|
||
sessionId: 's-1',
|
||
runId: 'r-1',
|
||
assistantMessageId: 'm-1',
|
||
text: '触发错误',
|
||
publish,
|
||
invoke: async () => ({ executionId: 'x', status: 'succeeded' }),
|
||
signal: new AbortController().signal,
|
||
});
|
||
assert.equal(result.finishReason, 'error');
|
||
assert.equal(events.find((event) => event.type === 'message.completed').data.finishReason, 'error');
|
||
}); |