alphaagent-ace/verify_installation.py at main · SourceShift/alphaagent-ace

215 lines (172 loc) · 5.64 KB
#!/usr/bin/env python3
Verification script to check ACE RAG Gemini installation.
- All modules can be imported
- Configuration is valid
- Dependencies are installed
- Basic functionality works
from pathlib import Path
def check_imports():
    """Check that all modules can be imported."""
    print("Checking imports...")
        import ace_rag
        from ace_rag import Config
        from ace_rag.rag_engine import RAGEngine
        from ace_rag.gemini_client import GeminiClient
        from ace_rag.vector_store import VectorStore
        from ace_rag.document_processor import DocumentProcessor
        from ace_rag.playbook import Playbook
        from ace_rag.ace_generator import ACEGenerator
        from ace_rag.ace_reflector import ACEReflector
        from ace_rag.ace_curator import ACECurator
        print("✓ All modules imported successfully")
        return True
    except ImportError as e:
        print(f"✗ Import error: {e}")
        return False
def check_dependencies():
    """Check that required dependencies are installed."""
    print("\nChecking dependencies...")
    required_packages = [
        "google.generativeai",
        "pydantic",
        "dotenv",
        "faiss",
        "numpy",
    missing = []
    for package in required_packages:
        try:
            __import__(package)
            print(f"  ✓ {package}")
        except ImportError:
            print(f"  ✗ {package} (missing)")
            missing.append(package)
    if missing:
        print(f"\n✗ Missing packages: {', '.join(missing)}")
        print("Install with: pip install -r requirements.txt")
        return False
    print("✓ All dependencies installed")
    return True
def check_configuration():
    """Check configuration setup."""
    print("\nChecking configuration...")
        from ace_rag import Config
        from ace_rag.exceptions import ConfigurationException
        # Check .env file exists
        env_file = Path(".env")
        if not env_file.exists():
            print("  ⚠ .env file not found")
            print("  Copy .env.example to .env and add your API key")
            return False
        # Try to load config
        try:
            config = Config.from_env()
            print("  ✓ Configuration loaded")
            # Check API key is not default
            if config.gemini.api_key == "your-api-key-here":
                print("  ✗ API key not configured in .env")
                return False
            print("  ✓ API key configured")
            return True
        except ConfigurationException as e:
            print(f"  ✗ Configuration error: {e}")
            return False
    except Exception as e:
        print(f"  ✗ Unexpected error: {e}")
        return False
def check_data_directories():
    """Check that data directories are created."""
    print("\nChecking data directories...")
    directories = ["data", "data/vector_store", "data/playbook"]
    for dir_path in directories:
        path = Path(dir_path)
        if path.exists():
            print(f"  ✓ {dir_path}")
        else:
            print(f"  ⚠ {dir_path} (will be created on first use)")
    return True
def check_models():
    """Check that data models work correctly."""
    print("\nChecking data models...")
        from ace_rag.models import (
            Document,
            Chunk,
            QueryTrajectory,
            FusionMethod,
        # Test Document
        doc = Document(
            id="test",
            content="Test content",
            metadata={}
        assert doc.id == "test"
        print("  ✓ Document model")
        # Test Chunk
        chunk = Chunk(
            id="chunk1",
            document_id="test",
            content="Chunk content",
            position=0
        assert chunk.document_id == "test"
        print("  ✓ Chunk model")
        # Test QueryTrajectory
        trajectory = QueryTrajectory(
            id="traj1",
            original_query="test",
            expanded_query="expanded test",
            temperature=0.5,
            fusion_method=FusionMethod.MEAN
        assert trajectory.temperature == 0.5
        print("  ✓ QueryTrajectory model")
        print("✓ All models working correctly")
        return True
    except Exception as e:
        print(f"✗ Model error: {e}")
        return False
def run_verification():
    """Run all verification checks."""
    print("=" * 60)
    print("ACE RAG Gemini Installation Verification")
    print("=" * 60)
    checks = [
        ("Imports", check_imports),
        ("Dependencies", check_dependencies),
        ("Configuration", check_configuration),
        ("Data Directories", check_data_directories),
        ("Models", check_models),
    results = {}
    for name, check_func in checks:
        try:
            results[name] = check_func()
        except Exception as e:
            print(f"\n✗ {name} check failed with error: {e}")
            results[name] = False
    print("\n" + "=" * 60)
    print("Verification Summary")
    print("=" * 60)
    for name, result in results.items():
        status = "✓ PASS" if result else "✗ FAIL"
        print(f"{status}: {name}")
    all_passed = all(results.values())
    if all_passed:
        print("\n✓ All checks passed! Installation is ready.")
        print("\nNext steps:")
        print("1. Run: python examples/basic_usage.py")
        print("2. Or: python examples/ace_evolution_demo.py")
        return 0
        print("\n✗ Some checks failed. Please fix the issues above.")
        return 1
if __name__ == "__main__":
    sys.exit(run_verification())
Provide feedback

Saved searches

Use saved searches to filter your results more quickly

FilesExpand file tree

verify_installation.py

Latest commit

History

verify_installation.py

File metadata and controls
