import { query } from "@anthropic-ai/claude-agent-sdk";

for await (const message of query({
  prompt:
    "Get a cited adversarial evaluation plan for SupportBot v2.3. Do not generate adversarial inputs.",
  options: {
    mcpServers: {
      "k2-adversarial-context": {
        command: "python",
        args: ["scripts/k2_adversarial_mcp_server.py"],
        env: {
          K2_ADV_BUNDLE_DIR:
            "docs/customer-demos/demo-adversarial-context/k2-assets"
        }
      }
    },
    allowedTools: [
      "mcp__k2-adversarial-context__get_evaluation_plan"
    ]
  }
})) {
  if (message.type === "system" && message.subtype === "init") {
    console.log(message.mcp_servers);
  }
  if (message.type === "result" && message.subtype === "success") {
    console.log(message.result);
  }
}