assert-ai.test.ts

// node --test ask-claude.test.ts
import { test } from "node:test";
import { query, type Options } from "@anthropic-ai/claude-code";

async function assertAI(prompt: string, options: Options = {}): Promise<void> {
  const MARKER = "**ASSERT_OK**";
  const abortController = new AbortController();
  const finalPrompt = `Assert user query. Return ${MARKER} if it is true. \n\n${prompt}`;
  for await (const message of query({
    prompt: finalPrompt,
    options: {
      maxTurns: 7,
      allowedTools: ["Read(*)"],
      cwd: process.cwd(),
      abortController,
      ...options,
    },
  })) {
    if (message.type === "assistant") {
      const c = message.message.content;
      const res: string = Array.isArray(c) ? c.map((c) => c.text).join("") : c;
      if (res.includes(MARKER)) {
        abortController.abort();
        return;
      }
    }
  }
  throw new Error("No response");
}

test("assertAI", async () => {
  await assertAI(`Exists README.md`);
});