Function scriptedAgent

test-hook-integration.mjs:15–39

({ script, hooks, autoApprove = true })

// Build an Agent whose model output is scripted. `script` is an array of
// generator functions, one per model round; each yields stream chunks.
function scriptedAgent({ script, hooks, autoApprove = true }) {
	let calls = 0
	const sentMessages = []
	AxonClient.prototype.createMessage = async function* (_systemPrompt, messages) {
		sentMessages.push(messages)
		const step = script[Math.min(calls, script.length - 1)]
		calls++
		yield* step()
	}
	const events = []
	const agent = new Agent({
		cwd: process.cwd(),
		token: "x",
		modelId: "axon-code-2-5-pro",
		autoApproveEdits: autoApprove,
		autoApproveSafeCommands: autoApprove,
		hooks,
		callbacks: {
			onEvent: (e) => events.push(e),
			requestApproval: async () => "yes",
			requestFollowup: async () => "answer",
		},
	})
	return { agent, events, sentMessages, calls: () => calls }
}

const sayThenEnd = (t) => function* () { yield { type: "text", text: t } }
const toolThen = (name, args) =>