Files
bigwo/test2/server/tests/run_real_test.js

69 lines
2.4 KiB
JavaScript
Raw Permalink Normal View History

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);
}
})();
}
}