import fs from 'node:fs';

import promptfoo from 'promptfoo';

const prompts = [
  // Prompts can be raw text...
  'Write a haiku about: {{body}}',

  // Or message arrays...
  [
    {
      role: 'system',
      content: 'You are speaking to a time traveler. Begin your conversation.',
    },
    {
      role: 'user',
      content: '{{body}}',
    },
  ],

  // Or files
  'file://prompt.txt',

  // Or functions
  (vars) => {
    const genres = ['fantasy', 'sci-fi', 'mystery', 'horror'];
    const characters = {
      fantasy: 'wizard',
      'sci-fi': 'alien',
      mystery: 'detective',
      horror: 'ghost',
    };
    const genre = genres[Math.floor(Math.random() * genres.length)];
    const character = characters[genre];
    return [
      {
        role: 'system',
        content: `You have encountered a ${character} in a ${genre} setting. Engage with the character using the knowledge typical for that genre.`,
      },
      {
        role: 'user',
        content: '{{body}}',
      },
    ];
  },
];

const customFunction = (prompt, context) => {
  // Call an LLM API in this function, or any other logic.
  console.log(`Prompt: ${prompt}, vars: ${JSON.stringify(context.vars)}`);
  return {
    output: '<LLM output>',
  };
};
customFunction.label = 'My custom function';

const providers = [
  // Call the OpenAI API GPT 4o Mini
  'openai:gpt-4.1-mini',

  // Or use a custom function.
  customFunction,
];

const tests = [
  {
    vars: {
      body: 'Hello world',
    },
  },
  {
    vars: {
      body: "I'm hungry",
    },
    assert: [
      {
        type: 'javascript',
        value: (output) => {
          // Logic for checking LLM output goes here...
          const pass = output.includes('foo bar');
          return {
            pass,
            score: pass ? 1.0 : 0.0,
            reason: pass ? 'Output contained substring' : 'Output did not contain substring',
          };
        },
      },
    ],
  },
  {
    vars: {
      body: 'Tell me a joke',
    },
    options: {
      // Transform functions can be passed directly when using the node package.
      transform: (output) => output.toLowerCase(),
    },
    assert: [
      {
        type: 'contains',
        value: 'joke',
        // Per-assertion transforms also accept functions.
        transform: (output) => output.trim(),
      },
    ],
  },
];

(async () => {
  const evalRecord = await promptfoo.evaluate({
    prompts,
    providers,
    tests,

    // Persist results locally so they are visible in the promptfoo viewer
    writeLatestResults: true,

    // Uncomment to enable sharing (requires cloud account or self-hosted server)
    // sharing: true,
  });
  const results = await evalRecord.toEvaluateSummary();

  console.log('RESULTS:');
  const resultsString = JSON.stringify(results, null, 2);
  console.log(resultsString);

  // Display shareable URL if available
  if (results.shareableUrl) {
    console.log('\nView results online:', results.shareableUrl);
  }

  fs.writeFileSync('output.json', resultsString);
  console.log('Wrote output.json');
})();