const path = require('path');

require('dotenv').config({ path: path.join(__dirname, '../.env') });

console.log('Checking environment variables...');
console.log('VOLC_ARK_API_KEY:', process.env.VOLC_ARK_API_KEY ? 'Set' : 'Not set');
console.log('VOLC_ARK_ENDPOINT_ID:', process.env.VOLC_ARK_ENDPOINT_ID);
console.log('VOLC_ARK_KNOWLEDGE_BASE_IDS:', process.env.VOLC_ARK_KNOWLEDGE_BASE_IDS);
console.log('VOLC_ARK_KNOWLEDGE_ENDPOINT_ID:', process.env.VOLC_ARK_KNOWLEDGE_ENDPOINT_ID);

const hasRequiredConfig = process.env.VOLC_ARK_API_KEY && 
                          process.env.VOLC_ARK_ENDPOINT_ID && 
                          process.env.VOLC_ARK_KNOWLEDGE_BASE_IDS &&
                          process.env.VOLC_ARK_KNOWLEDGE_ENDPOINT_ID;

if (hasRequiredConfig) {
  console.log('\nAll required configs found! Running real test...\n');
  
  const VikingTester = require('./viking_retrieval_performance_with_mock');
  
  (async () => {
    try {
      const tester = new VikingTester({ mockMode: false });
      await tester.runFullSuite();
    } catch (err) {
      console.error('Test failed:', err);
    }
  })();
} else {
  console.log('\nMissing required environment variables. Checking parent directory...');
  
  require('dotenv').config({ path: path.join(__dirname, '../../.env') });
  
  console.log('VOLC_ARK_API_KEY (parent):', process.env.VOLC_ARK_API_KEY ? 'Set' : 'Not set');
  
  const hasRequiredConfigParent = process.env.VOLC_ARK_API_KEY && 
                                   process.env.VOLC_ARK_ENDPOINT_ID && 
                                   process.env.VOLC_ARK_KNOWLEDGE_BASE_IDS &&
                                   process.env.VOLC_ARK_KNOWLEDGE_ENDPOINT_ID;
  
  if (hasRequiredConfigParent) {
    console.log('\nConfigs found in parent directory! Running real test...\n');
    
    const VikingTester = require('./viking_retrieval_performance_with_mock');
    
    (async () => {
      try {
        const tester = new VikingTester({ mockMode: false });
        await tester.runFullSuite();
      } catch (err) {
        console.error('Test failed:', err);
      }
    })();
  } else {
    console.log('\nNo configs found. Running mock test instead...\n');
    
    const VikingTester = require('./viking_retrieval_performance_with_mock');
    
    (async () => {
      try {
        const tester = new VikingTester({ mockMode: true });
        await tester.runFullSuite();
      } catch (err) {
        console.error('Test failed:', err);
      }
    })();
  }
}