setup_check.py

#!/usr/bin/env python3
"""
Setup verification script

Run this after installing dependencies to verify your setup is correct.
"""

import sys
import os


def check_python_version():
    """Check Python version is 3.10+"""
    version = sys.version_info
    if version.major < 3 or (version.major == 3 and version.minor < 10):
        print("❌ Python 3.10+ required")
        print(f"   Current version: {version.major}.{version.minor}.{version.micro}")
        return False
    print(f"✅ Python {version.major}.{version.minor}.{version.micro}")
    return True


def check_dependencies():
    """Check if required packages are installed"""
    try:
        import llama_cpp
        print("✅ llama-cpp-python installed")
        return True
    except ImportError:
        print("❌ llama-cpp-python not found")
        print("   Install with: pip install llama-cpp-python")
        return False


def check_model_directory():
    """Check if models directory exists"""
    if os.path.isdir("models"):
        print("✅ models/ directory exists")
        
        # Check for GGUF files
        files = [f for f in os.listdir("models") if f.endswith(".gguf")]
        if files:
            print(f"✅ Found {len(files)} GGUF model(s):")
            for f in files:
                size_mb = os.path.getsize(f"models/{f}") / (1024**2)
                print(f"   - {f} ({size_mb:.1f} MB)")
        else:
            print("⚠️  No GGUF models found in models/")
            print("   Download a model and place it in models/")
        return True
    else:
        print("❌ models/ directory not found")
        return False


def check_structure():
    """Check repository structure"""
    required_dirs = ["shared", "agent", "lessons"]
    all_exist = True
    
    for dir_name in required_dirs:
        if os.path.isdir(dir_name):
            print(f"✅ {dir_name}/ directory exists")
        else:
            print(f"❌ {dir_name}/ directory not found")
            all_exist = False
    
    return all_exist


def main():
    """Run all checks"""
    print("="*50)
    print("AI Agents from Scratch - Setup Verification")
    print("="*50)
    print()
    
    checks = [
        ("Python Version", check_python_version),
        ("Dependencies", check_dependencies),
        ("Models Directory", check_model_directory),
        ("Repository Structure", check_structure),
    ]
    
    results = []
    
    for name, check_func in checks:
        print(f"\n{name}:")
        results.append(check_func())
    
    print("\n" + "="*50)
    if all(results):
        print("✅ All checks passed! You're ready to start learning.")
        print("\nNext steps:")
        print("1. Read lessons/01_basic_llm_chat.md")
        print("2. Run: python complete_example.py")
    else:
        print("⚠️  Some checks failed. Please fix the issues above.")
    print("="*50)


if __name__ == "__main__":
    main()