/** @file The model, called for real through the Workers AI binding. */
import { env } from "cloudflare:workers";
import { expect, test } from "vitest";

import { makeWorkersAiModel, MODEL } from "../src/model.ts";

test("the model counts the rs in strawberry", async () => {
	const model = makeWorkersAiModel(env.AI, MODEL);
	const reply = await model.complete([
		{
			role: "user",
			content: 'How many times does the letter r appear in "strawberry"? Answer with just the number.',
		},
	]);
	expect(reply.trim()).toMatch(/^3\b/);
});