IA e pequenas coisas a funcionar
This commit is contained in:
253
test/rag_services_test.dart
Normal file
253
test/rag_services_test.dart
Normal file
@@ -0,0 +1,253 @@
|
||||
import 'package:flutter_test/flutter_test.dart';
|
||||
import '../lib/core/services/content_service.dart';
|
||||
import '../lib/core/services/vector_service.dart';
|
||||
import '../lib/core/services/rag_service.dart';
|
||||
import '../lib/core/services/rag_ai_service.dart';
|
||||
import '../lib/core/models/content_chunk.dart';
|
||||
|
||||
void main() {
|
||||
group('AI Tutor RAG Services Tests', () {
|
||||
setUpAll(() async {
|
||||
// Setup test data
|
||||
print('🧪 Setting up test environment...');
|
||||
});
|
||||
|
||||
test('VectorService - Generate Embedding', () async {
|
||||
print('🔍 Testing VectorService embedding generation...');
|
||||
|
||||
const testText =
|
||||
'A fotossíntese é o processo pelo qual as plantas convertem luz solar em energia química.';
|
||||
|
||||
final embedding = VectorService.generateEmbedding(testText);
|
||||
|
||||
// Verify embedding properties
|
||||
expect(embedding.length, equals(384)); // Standard embedding size
|
||||
expect(
|
||||
embedding.every((x) => x >= -1.0 && x <= 1.0),
|
||||
isTrue,
|
||||
); // Normalized
|
||||
|
||||
print(
|
||||
'✅ Embedding generated successfully: ${embedding.length} dimensions',
|
||||
);
|
||||
});
|
||||
|
||||
test('VectorService - Cosine Similarity', () {
|
||||
print('🔍 Testing cosine similarity calculation...');
|
||||
|
||||
const text1 = 'A fotossíntese é importante para as plantas.';
|
||||
const text2 = 'As plantas usam fotossíntese para produzir energia.';
|
||||
const text3 = 'Os animais precisam de comida para sobreviver.';
|
||||
|
||||
final embedding1 = VectorService.generateEmbedding(text1);
|
||||
final embedding2 = VectorService.generateEmbedding(text2);
|
||||
final embedding3 = VectorService.generateEmbedding(text3);
|
||||
|
||||
final similarity12 = VectorService.cosineSimilarity(
|
||||
embedding1,
|
||||
embedding2,
|
||||
);
|
||||
final similarity13 = VectorService.cosineSimilarity(
|
||||
embedding1,
|
||||
embedding3,
|
||||
);
|
||||
|
||||
// Similar texts should have higher similarity
|
||||
expect(similarity12, greaterThan(similarity13));
|
||||
expect(similarity12, greaterThan(0.5)); // Should be reasonably similar
|
||||
expect(similarity13, lessThan(0.3)); // Should be less similar
|
||||
|
||||
print('✅ Cosine similarity working correctly');
|
||||
print(' Similar (fotossíntese): ${similarity12.toStringAsFixed(3)}');
|
||||
print(
|
||||
' Different (plantas vs animais): ${similarity13.toStringAsFixed(3)}',
|
||||
);
|
||||
});
|
||||
|
||||
test('RAGService - Mode Instructions', () {
|
||||
print('🔍 Testing RAG service mode instructions...');
|
||||
|
||||
// Test different modes
|
||||
final explanationMode = RAGService._getModeInstructions(
|
||||
TutorMode.explanation,
|
||||
);
|
||||
final tutorMode = RAGService._getModeInstructions(TutorMode.tutor);
|
||||
final explorationMode = RAGService._getModeInstructions(
|
||||
TutorMode.exploration,
|
||||
);
|
||||
|
||||
expect(explanationMode, contains('explicações detalhadas'));
|
||||
expect(tutorMode, contains('método socrático'));
|
||||
expect(explorationMode, contains('exploração'));
|
||||
|
||||
print('✅ Mode instructions generated correctly');
|
||||
print(' Explanation: "${explanationMode.substring(0, 50)}..."');
|
||||
print(' Tutor: "${tutorMode.substring(0, 50)}..."');
|
||||
print(' Exploration: "${explorationMode.substring(0, 50)}..."');
|
||||
});
|
||||
|
||||
test('RAGService - Context Building', () {
|
||||
print('🔍 Testing context window building...');
|
||||
|
||||
// Create mock content chunks
|
||||
final chunks = [
|
||||
ContentChunk(
|
||||
id: 'test1',
|
||||
contentId: 'content1',
|
||||
text:
|
||||
'A fotossíntese é o processo biológico que converte luz solar em energia química.',
|
||||
subject: 'Biologia',
|
||||
concept: 'Fotossíntese',
|
||||
unit: 'Processos Biológicos',
|
||||
difficulty: 0.6,
|
||||
grade: 10,
|
||||
embedding: List.filled(384, 0.1),
|
||||
sourceDocument: 'Biologia Manual.pdf',
|
||||
metadata: {},
|
||||
createdAt: DateTime.now(),
|
||||
),
|
||||
ContentChunk(
|
||||
id: 'test2',
|
||||
contentId: 'content1',
|
||||
text:
|
||||
'Durante a fotossíntese, as plantas absorvem CO2 e liberam oxigénio.',
|
||||
subject: 'Biologia',
|
||||
concept: 'Fotossíntese',
|
||||
unit: 'Processos Biológicos',
|
||||
difficulty: 0.7,
|
||||
grade: 10,
|
||||
embedding: List.filled(384, 0.2),
|
||||
sourceDocument: 'Biologia Manual.pdf',
|
||||
metadata: {},
|
||||
createdAt: DateTime.now(),
|
||||
),
|
||||
];
|
||||
|
||||
final context = RAGService._buildContextWindow(
|
||||
chunks,
|
||||
'O que é fotossíntese?',
|
||||
TutorMode.explanation,
|
||||
);
|
||||
|
||||
expect(context, contains('CONTEÚDO EDUCACIONAL RELEVANTE'));
|
||||
expect(context, contains('Fotossíntese'));
|
||||
expect(context, contains('Biologia'));
|
||||
expect(context, contains('INSTRUÇÕES DE TUTORIA'));
|
||||
|
||||
print('✅ Context window built successfully');
|
||||
print(' Context length: ${context.length} characters');
|
||||
print(' Contains ${chunks.length} source chunks');
|
||||
});
|
||||
|
||||
test('RAGAIService - Service Availability', () async {
|
||||
print('🔍 Testing Ollama service availability...');
|
||||
|
||||
try {
|
||||
final isAvailable = await RAGAIService.isServiceAvailable();
|
||||
|
||||
if (isAvailable) {
|
||||
print('✅ Ollama service is available');
|
||||
|
||||
// Test model info
|
||||
final modelInfo = await RAGAIService.getModelInfo();
|
||||
if (modelInfo != null) {
|
||||
print(' Model: ${modelInfo['name']}');
|
||||
print(' Size: ${modelInfo['size']}');
|
||||
print(' Modified: ${modelInfo['modified']}');
|
||||
}
|
||||
|
||||
// Test simple query
|
||||
final testResponse = await RAGAIService.testService();
|
||||
print(' Test response: "$testResponse"');
|
||||
} else {
|
||||
print('⚠️ Ollama service is not available');
|
||||
print(' This is expected if the service is not running');
|
||||
}
|
||||
} catch (e) {
|
||||
print('❌ Error testing service availability: $e');
|
||||
}
|
||||
});
|
||||
|
||||
test('RAG Pipeline - End-to-End Simulation', () async {
|
||||
print('🔍 Testing complete RAG pipeline simulation...');
|
||||
|
||||
try {
|
||||
const userQuery = 'O que é fotossíntese?';
|
||||
const mode = TutorMode.explanation;
|
||||
|
||||
// Step 1: Generate query embedding
|
||||
final queryEmbedding = VectorService.generateEmbedding(userQuery);
|
||||
print(' ✅ Query embedding generated');
|
||||
|
||||
// Step 2: Simulate content retrieval (mock data)
|
||||
final mockChunks = [
|
||||
ContentChunk(
|
||||
id: 'chunk1',
|
||||
contentId: 'content1',
|
||||
text:
|
||||
'A fotossíntese é o processo pelo qual as plantas, algas e algumas bactérias convertem luz solar em energia química.',
|
||||
subject: 'Biologia',
|
||||
concept: 'Fotossíntese',
|
||||
unit: 'Processos Biológicos',
|
||||
difficulty: 0.6,
|
||||
grade: 10,
|
||||
embedding: VectorService.generateEmbedding(
|
||||
'fotossíntese plantas energia',
|
||||
),
|
||||
sourceDocument: 'Biologia Manual.pdf',
|
||||
metadata: {'page': 45},
|
||||
createdAt: DateTime.now(),
|
||||
),
|
||||
];
|
||||
|
||||
// Step 3: Build context
|
||||
final context = RAGService._buildContextWindow(
|
||||
mockChunks,
|
||||
userQuery,
|
||||
mode,
|
||||
);
|
||||
print(' ✅ Context built (${context.length} chars)');
|
||||
|
||||
// Step 4: Test AI service if available
|
||||
try {
|
||||
final ragResponse = await RAGAIService.generateRAGResponse(
|
||||
userQuery: userQuery,
|
||||
context: context,
|
||||
mode: mode,
|
||||
sources: mockChunks,
|
||||
);
|
||||
|
||||
print(' ✅ RAG response generated');
|
||||
print(' Answer: "${ragResponse.answer.substring(0, 100)}..."');
|
||||
print(' Confidence: ${ragResponse.confidence.toStringAsFixed(2)}');
|
||||
print(' Sources: ${ragResponse.sources.length}');
|
||||
print(
|
||||
' Related concepts: ${ragResponse.relatedConcepts.join(', ')}',
|
||||
);
|
||||
} catch (e) {
|
||||
print(' ⚠️ AI service not available, using mock response');
|
||||
|
||||
// Create mock response
|
||||
final mockResponse = RAGService._createRAGResponse(
|
||||
query: userQuery,
|
||||
context: context,
|
||||
mode: mode,
|
||||
sources: mockChunks,
|
||||
);
|
||||
|
||||
print(' ✅ Mock RAG response created');
|
||||
print(' Answer: "${mockResponse.answer.substring(0, 100)}..."');
|
||||
print(' Confidence: ${mockResponse.confidence.toStringAsFixed(2)}');
|
||||
}
|
||||
|
||||
print('✅ RAG pipeline simulation completed successfully');
|
||||
} catch (e) {
|
||||
print('❌ Error in RAG pipeline: $e');
|
||||
}
|
||||
});
|
||||
|
||||
tearDownAll(() async {
|
||||
print('🧹 Cleaning up test environment...');
|
||||
});
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user