ALwrity version 0.5.5

docs/alwrity_test_scripts/test_env_check.py

@@ -0,0 +1,142 @@
+#!/usr/bin/env python3
+"""
+Test script to check environment variables and API key loading.
+"""
+
+import os
+import sys
+from pathlib import Path
+
+# Add the backend directory to the Python path
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from dotenv import load_dotenv
+
+def test_environment_loading():
+    """Test environment variable loading."""
+    print("🔍 Testing environment variable loading...")
+    
+    # Check current working directory
+    print(f"Current working directory: {os.getcwd()}")
+    
+    # Check if .env file exists in various locations
+    possible_env_paths = [
+        Path('.env'),  # Current directory
+        Path('../.env'),  # Parent directory
+        Path('../../.env'),  # Grandparent directory
+        Path('../../../.env'),  # Great-grandparent directory
+        Path('backend/.env'),  # Backend directory
+    ]
+    
+    print("\n📁 Checking for .env files:")
+    for env_path in possible_env_paths:
+        if env_path.exists():
+            print(f"✅ Found .env file: {env_path.absolute()}")
+        else:
+            print(f"❌ No .env file: {env_path.absolute()}")
+    
+    # Try to load .env from different locations
+    print("\n🔄 Attempting to load .env files:")
+    for env_path in possible_env_paths:
+        if env_path.exists():
+            print(f"Loading .env from: {env_path.absolute()}")
+            load_dotenv(env_path)
+            break
+    else:
+        print("⚠️ No .env file found, trying to load from current directory")
+        load_dotenv()
+    
+    # Check environment variables
+    print("\n🔑 Checking environment variables:")
+    env_vars_to_check = [
+        'GEMINI_API_KEY',
+        'GOOGLE_API_KEY',
+        'OPENAI_API_KEY',
+        'DATABASE_URL',
+        'SECRET_KEY'
+    ]
+    
+    for var in env_vars_to_check:
+        value = os.getenv(var)
+        if value:
+            # Show first few characters for security
+            masked_value = value[:8] + "..." if len(value) > 8 else "***"
+            print(f"✅ {var}: {masked_value}")
+        else:
+            print(f"❌ {var}: Not set")
+    
+    # Test specific Gemini API key loading
+    print("\n🤖 Testing Gemini API key loading:")
+    gemini_key = os.getenv('GEMINI_API_KEY')
+    if gemini_key:
+        print(f"✅ GEMINI_API_KEY found: {gemini_key[:8]}...")
+        
+        # Test if the key looks valid
+        if len(gemini_key) > 20:
+            print("✅ API key length looks valid")
+        else:
+            print("⚠️ API key seems too short")
+    else:
+        print("❌ GEMINI_API_KEY not found")
+        
+        # Check alternative names
+        alternative_keys = ['GOOGLE_API_KEY', 'GEMINI_KEY', 'GOOGLE_AI_API_KEY']
+        for alt_key in alternative_keys:
+            alt_value = os.getenv(alt_key)
+            if alt_value:
+                print(f"⚠️ Found alternative key {alt_key}: {alt_value[:8]}...")
+    
+    return gemini_key is not None
+
+def test_gemini_provider_import():
+    """Test importing the Gemini provider."""
+    print("\n🧪 Testing Gemini provider import...")
+    
+    try:
+        from services.llm_providers.gemini_provider import gemini_structured_json_response
+        print("✅ Successfully imported gemini_structured_json_response")
+        return True
+    except Exception as e:
+        print(f"❌ Failed to import Gemini provider: {e}")
+        return False
+
+def test_ai_service_manager_import():
+    """Test importing the AI service manager."""
+    print("\n🧪 Testing AI service manager import...")
+    
+    try:
+        from services.ai_service_manager import AIServiceManager
+        print("✅ Successfully imported AIServiceManager")
+        
+        # Try to create an instance
+        ai_manager = AIServiceManager()
+        print("✅ Successfully created AIServiceManager instance")
+        return True
+    except Exception as e:
+        print(f"❌ Failed to import/create AI service manager: {e}")
+        return False
+
+if __name__ == "__main__":
+    print("🚀 Starting environment and API key validation tests")
+    print("=" * 60)
+    
+    # Test environment loading
+    env_ok = test_environment_loading()
+    
+    # Test imports
+    gemini_import_ok = test_gemini_provider_import()
+    ai_manager_ok = test_ai_service_manager_import()
+    
+    print("\n" + "=" * 60)
+    print("📊 Test Results Summary:")
+    print(f"Environment loading: {'✅ PASS' if env_ok else '❌ FAIL'}")
+    print(f"Gemini provider import: {'✅ PASS' if gemini_import_ok else '❌ FAIL'}")
+    print(f"AI service manager: {'✅ PASS' if ai_manager_ok else '❌ FAIL'}")
+    
+    if not env_ok:
+        print("\n💡 To fix environment issues:")
+        print("1. Create a .env file in the backend directory")
+        print("2. Add your GEMINI_API_KEY to the .env file")
+        print("3. Example: GEMINI_API_KEY=your_actual_api_key_here")
+    
+    print("\n" + "=" * 60)