test_env_vars.py

#!/usr/bin/env python3
"""
Test script to verify environment variables are loaded correctly
"""
import os
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

def test_env_vars():
    """Test that environment variables are loaded correctly"""
    print("Testing environment variables...")
    
    # Test LiteLLM API Key
    api_key = os.getenv("LITELLM_API_KEY")
    if api_key:
        print(f"✓ LITELLM_API_KEY is set (value: {api_key[:5]}...)")
    else:
        print("✗ LITELLM_API_KEY is not set")
    
    # Test LiteLLM API URL
    api_url = os.getenv("LITELLM_API_URL")
    if api_url:
        print(f"✓ LITELLM_API_URL is set (value: {api_url})")
    else:
        print("✗ LITELLM_API_URL is not set")
    
    # Test LiteLLM API Base
    api_base = os.getenv("LITELLM_API_BASE")
    if api_base:
        print(f"✓ LITELLM_API_BASE is set (value: {api_base})")
    else:
        print("✗ LITELLM_API_BASE is not set")
    
    # Test AICHAT Config Path
    config_path = os.getenv("AICHAT_CONFIG_PATH")
    if config_path:
        print(f"✓ AICHAT_CONFIG_PATH is set (value: {config_path})")
    else:
        print("✗ AICHAT_CONFIG_PATH is not set")
    
    # Test Azure Subscription ID
    azure_sub_id = os.getenv("AZURE_SUBSCRIPTION_ID")
    if azure_sub_id:
        print(f"✓ AZURE_SUBSCRIPTION_ID is set (value: {azure_sub_id[:5]}...)")
    else:
        print("✗ AZURE_SUBSCRIPTION_ID is not set")

if __name__ == "__main__":
    test_env_vars()