LearnIT/test/rag_services_test.dart

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...');
    });
  });
}