ComputerAssignments/CA10_knowledge_graphs/tests/test_api_keys.py

#!/usr/bin/env python3
"""
Test script for API keys
"""

import os
from dotenv import load_dotenv

# Load environment variables
load_dotenv()


def test_api_keys():
    """Test if API keys are properly configured"""

    print("🔑 Testing API Keys Configuration")
    print("=" * 40)

    # Test OpenAI
    openai_key = os.getenv("OPENAI_API_KEY")
    if openai_key and openai_key != "your_openai_api_key_here":
        print("✅ OpenAI API Key: Configured")
        print(f"   Key: {openai_key[:10]}...")
    else:
        print("❌ OpenAI API Key: Not configured")

    # Test Anthropic
    anthropic_key = os.getenv("ANTHROPIC_API_KEY")
    if anthropic_key and anthropic_key != "your_anthropic_api_key_here":
        print("✅ Anthropic API Key: Configured")
        print(f"   Key: {anthropic_key[:10]}...")
    else:
        print("❌ Anthropic API Key: Not configured")

    print("\n💡 To configure API keys:")
    print("   1. Edit the .env file")
    print("   2. Replace 'your_openai_api_key_here' with your actual key")
    print("   3. Replace 'your_anthropic_api_key_here' with your actual key")


if __name__ == "__main__":
    test_api_keys()