Class TestRAGIntegration

tests/unit/test_all_embedders.py:295–321

Test RAG class integration with different embedders.

class TestRAGIntegration:
    """Test RAG class integration with different embedders."""
    
    def test_rag_initialization(self):
        """Test RAG initialization with different embedder configurations."""
        from api.rag import RAG
        
        # Test with default configuration
        try:
            rag = RAG(provider="google", model="gemini-1.5-flash")
            assert rag is not None, "RAG should be initialized"
            assert hasattr(rag, 'embedder'), "RAG should have embedder"
            assert hasattr(rag, 'is_ollama_embedder'), "RAG should have is_ollama_embedder attribute"
        except Exception as e:
            logger.warning(f"RAG initialization failed (might be expected if keys missing): {e}")

    def test_rag_embedder_type_detection(self):
        """Test that RAG correctly detects embedder type."""
        from api.rag import RAG
        
        try:
            rag = RAG()
            # Should have the embedder type detection logic
            assert hasattr(rag, 'is_ollama_embedder'), "RAG should detect embedder type"
            assert isinstance(rag.is_ollama_embedder, bool), "is_ollama_embedder should be boolean"
        except Exception as e:
            logger.warning(f"RAG initialization failed: {e}")


class TestEnvironmentVariableHandling: