Files
cim_summary/backend/setup-test-data.js
Jon 57770fd99d feat: Implement hybrid LLM approach with enhanced prompts for CIM analysis
🎯 Major Features:
- Hybrid LLM configuration: Claude 3.7 Sonnet (primary) + GPT-4.5 (fallback)
- Task-specific model selection for optimal performance
- Enhanced prompts for all analysis types with proven results

🔧 Technical Improvements:
- Enhanced financial analysis with fiscal year mapping (100% success rate)
- Business model analysis with scalability assessment
- Market positioning analysis with TAM/SAM extraction
- Management team assessment with succession planning
- Creative content generation with GPT-4.5

📊 Performance & Cost Optimization:
- Claude 3.7 Sonnet: /5 per 1M tokens (82.2% MATH score)
- GPT-4.5: Premium creative content (5/50 per 1M tokens)
- ~80% cost savings using Claude for analytical tasks
- Automatic fallback system for reliability

 Proven Results:
- Successfully extracted 3-year financial data from STAX CIM
- Correctly mapped fiscal years (2023→FY-3, 2024→FY-2, 2025E→FY-1, LTM Mar-25→LTM)
- Identified revenue: 4M→1M→1M→6M (LTM)
- Identified EBITDA: 8.9M→3.9M→1M→7.2M (LTM)

🚀 Files Added/Modified:
- Enhanced LLM service with task-specific model selection
- Updated environment configuration for hybrid approach
- Enhanced prompt builders for all analysis types
- Comprehensive testing scripts and documentation
- Updated frontend components for improved UX

📚 References:
- Eden AI Model Comparison: Claude 3.7 Sonnet vs GPT-4.5
- Artificial Analysis Benchmarks for performance metrics
- Cost optimization based on model strengths and pricing
2025-07-28 16:46:06 -04:00

97 lines
3.0 KiB
JavaScript

#!/usr/bin/env node
/**
* Setup test data for agentic RAG database integration tests
* Creates test users and documents with proper UUIDs
*/
const { v4: uuidv4 } = require('uuid');
const db = require('./dist/config/database').default;
const bcrypt = require('bcrypt');
async function setupTestData() {
console.log('🔧 Setting up test data for agentic RAG database integration...\n');
try {
// Create test user
console.log('1. Creating test user...');
const testUserId = uuidv4();
const hashedPassword = await bcrypt.hash('testpassword123', 12);
await db.query(`
INSERT INTO users (id, email, password_hash, name, role, created_at, updated_at)
VALUES ($1, $2, $3, $4, $5, NOW(), NOW())
ON CONFLICT (email) DO NOTHING
`, [testUserId, 'test@agentic-rag.com', hashedPassword, 'Test User', 'admin']);
// Create test document
console.log('2. Creating test document...');
const testDocumentId = uuidv4();
await db.query(`
INSERT INTO documents (id, user_id, original_file_name, file_path, file_size, status, extracted_text, created_at, updated_at)
VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW())
`, [
testDocumentId,
testUserId,
'test-cim-document.pdf',
'/uploads/test-cim-document.pdf',
1024000,
'completed',
'This is a test CIM document for agentic RAG testing.'
]);
// Create test document for full flow
console.log('3. Creating test document for full flow...');
const testDocumentId2 = uuidv4();
await db.query(`
INSERT INTO documents (id, user_id, original_file_name, file_path, file_size, status, extracted_text, created_at, updated_at)
VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW())
`, [
testDocumentId2,
testUserId,
'test-cim-document-full.pdf',
'/uploads/test-cim-document-full.pdf',
2048000,
'completed',
'This is a comprehensive test CIM document for full agentic RAG flow testing.'
]);
console.log('✅ Test data setup completed successfully!');
console.log('\n📋 Test Data Summary:');
console.log(` Test User ID: ${testUserId}`);
console.log(` Test Document ID: ${testDocumentId}`);
console.log(` Test Document ID (Full Flow): ${testDocumentId2}`);
console.log(` Test User Email: test@agentic-rag.com`);
console.log(` Test User Password: testpassword123`);
// Export the IDs for use in tests
module.exports = {
testUserId,
testDocumentId,
testDocumentId2
};
return { testUserId, testDocumentId, testDocumentId2 };
} catch (error) {
console.error('❌ Failed to setup test data:', error);
throw error;
}
}
// Run setup if called directly
if (require.main === module) {
setupTestData()
.then(() => {
console.log('\n✨ Test data setup completed!');
process.exit(0);
})
.catch((error) => {
console.error('❌ Test data setup failed:', error);
process.exit(1);
});
}
module.exports = { setupTestData };