Environment Setup

#!/usr/bin/env python3
"""Verify AI development environment setup."""

import sys

def check_python():
    print(f"Python version: {sys.version}")
    assert sys.version_info >= (3, 10), "Python 3.10+ required"
    print("  Python: OK")

def check_packages():
    packages = [
        ("openai", "openai"),
        ("anthropic", "anthropic"),
        ("transformers", "transformers"),
        ("sentence_transformers", "sentence-transformers"),
        ("tiktoken", "tiktoken"),
        ("torch", "torch"),
        ("dotenv", "python-dotenv"),
    ]

    for import_name, pip_name in packages:
        try:
            __import__(import_name)
            print(f"  {pip_name}: OK")
        except ImportError:
            print(f"  {pip_name}: MISSING - run: uv pip install {pip_name}")

def check_gpu():
    try:
        import torch
        if torch.cuda.is_available():
            print(f"  CUDA: Available ({torch.cuda.get_device_name(0)})")
        elif hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():
            print("  MPS (Apple Silicon): Available")
        else:
            print("  GPU: Not available (CPU only)")
    except ImportError:
        print("  GPU: PyTorch not installed")

def check_api_keys():
    import os
    from dotenv import load_dotenv
    load_dotenv()

    keys = [
        ("OPENAI_API_KEY", "OpenAI"),
        ("ANTHROPIC_API_KEY", "Anthropic"),
        ("HF_TOKEN", "HuggingFace"),
    ]

    for key, name in keys:
        if os.getenv(key):
            print(f"  {name}: Set")
        else:
            print(f"  {name}: Not set (optional)")

def test_openai():
    import os
    from dotenv import load_dotenv
    load_dotenv()

    if not os.getenv("OPENAI_API_KEY"):
        print("  OpenAI API: Skipped (no key)")
        return

    from openai import OpenAI
    client = OpenAI()
    response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[{"role": "user", "content": "Say 'test successful' in 2 words"}],
        max_tokens=10
    )
    print(f"  OpenAI API: OK - {response.choices[0].message.content}")

if __name__ == "__main__":
    print("\n=== Environment Verification ===\n")

    print("1. Python Version")
    check_python()

    print("\n2. Required Packages")
    check_packages()

    print("\n3. GPU Availability")
    check_gpu()

    print("\n4. API Keys (.env)")
    check_api_keys()

    print("\n5. API Connection Test")
    try:
        test_openai()
    except Exception as e:
        print(f"  OpenAI API: Error - {e}")

    print("\n=== Verification Complete ===\n")