File size: 3,876 Bytes
461adca | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | """
Test script for configuration system.
"""
import os
from config import get_settings, validate_configuration
def test_configuration():
"""Test configuration loading and validation."""
print("=" * 60)
print("Testing Configuration System")
print("=" * 60)
# Test loading default configuration (without API key validation for testing)
print("\n1. Loading default configuration...")
try:
settings = get_settings(validate_api_keys=False)
print(f"✅ Configuration loaded successfully!")
print(f" Environment: {settings.app.environment}")
print(f" Debug mode: {settings.app.debug}")
print(f" Local directory: {settings.aws.local_dir}")
except Exception as e:
print(f"❌ Failed to load configuration: {str(e)}")
return False
# Test validation
print("\n2. Validating configuration...")
is_valid = validate_configuration(settings, print_report=True)
if not is_valid:
print("\n❌ Configuration validation failed!")
return False
# Test environment-specific loading
print("\n3. Testing environment-specific configuration...")
# Test development environment
print("\n a) Development environment:")
try:
dev_settings = get_settings(environment='development', reload=True, validate_api_keys=False)
print(f" ✅ Loaded development config")
print(f" Debug mode: {dev_settings.app.debug}")
print(f" Environment: {dev_settings.app.environment}")
except Exception as e:
print(f" ❌ Failed: {str(e)}")
# Test production environment
print("\n b) Production environment:")
try:
prod_settings = get_settings(environment='production', reload=True, validate_api_keys=False)
print(f" ✅ Loaded production config")
print(f" Debug mode: {prod_settings.app.debug}")
print(f" Environment: {prod_settings.app.environment}")
except Exception as e:
print(f" ❌ Failed: {str(e)}")
# Test configuration values
print("\n4. Testing configuration values...")
settings = get_settings(reload=True, validate_api_keys=False)
print(f" LlamaIndex settings:")
print(f" Context window: {settings.llama_index.context_window}")
print(f" Chunk size: {settings.llama_index.chunk_size}")
print(f" Similarity top k: {settings.llama_index.similarity_top_k}")
print(f"\n Session settings:")
print(f" Timeout: {settings.session.timeout_minutes} minutes")
print(f" Max sessions: {settings.session.max_sessions}")
print(f" Storage type: {settings.session.storage_type}")
print(f"\n Model providers: {', '.join(settings.models.providers)}")
# Test model configurations
print(f"\n Generation models:")
for provider in settings.models.providers:
models = getattr(settings.models.generation, provider, [])
if models:
default_model = next((m for m in models if m.default), models[0])
print(f" {provider}: {default_model.display_name}")
print(f"\n Analysis models:")
for provider in settings.models.providers:
models = getattr(settings.models.analysis, provider, [])
if models:
default_model = next((m for m in models if m.default), models[0])
print(f" {provider}: {default_model.display_name}")
print("\n" + "=" * 60)
print("✅ All configuration tests passed!")
print("=" * 60)
return True
if __name__ == "__main__":
# Set test environment variables if not present
if not os.getenv('OPENAI_API_KEY'):
print("⚠️ Warning: OPENAI_API_KEY not set. Some validations may fail.")
success = test_configuration()
exit(0 if success else 1)
|