69 lines
2.4 KiB
JavaScript
69 lines
2.4 KiB
JavaScript
|
|
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);
|
||
|
|
}
|
||
|
|
})();
|
||
|
|
}
|
||
|
|
}
|