38 lines
957 B
JavaScript
38 lines
957 B
JavaScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
const {runToolLoopRounds} = await import("../dist/ai/tool-loop-runner.js");
|
|
|
|
test("tool loop runner stops when handler requests it", async () => {
|
|
const rounds = [];
|
|
|
|
await runToolLoopRounds({
|
|
maxRounds: 5,
|
|
async onRound(round) {
|
|
rounds.push(round);
|
|
return {shouldContinue: round < 1};
|
|
},
|
|
});
|
|
|
|
assert.deepEqual(rounds, [0, 1]);
|
|
});
|
|
|
|
test("tool loop runner calls max rounds hook when handler never stops", async () => {
|
|
const rounds = [];
|
|
let maxRoundsReached = -1;
|
|
|
|
await runToolLoopRounds({
|
|
maxRounds: 3,
|
|
async onRound(round) {
|
|
rounds.push(round);
|
|
return {shouldContinue: true};
|
|
},
|
|
onMaxRoundsReached(round) {
|
|
maxRoundsReached = round;
|
|
},
|
|
});
|
|
|
|
assert.deepEqual(rounds, [0, 1, 2]);
|
|
assert.equal(maxRoundsReached, 2);
|
|
});
|