ALwrity Version 0.5.0 (Fastapi + React )

2025-08-06 12:48:02 +05:30
parent f28a919caa
commit 32f97fa6b3
476 changed files with 115544 additions and 28747 deletions
--- a/docs/test_gemini_fix.py
+++ b/docs/test_gemini_fix.py
@@ -0,0 +1,119 @@
+#!/usr/bin/env python3
+"""
+Test script to verify the Gemini provider fixes.
+"""
+
+import os
+import sys
+from pathlib import Path
+
+# Add the backend directory to the path
+sys.path.append(str(Path(__file__).parent / 'backend'))
+
+from llm_providers.gemini_provider import gemini_text_response, gemini_pro_text_gen, test_gemini_api_key
+
+def test_gemini_text_response():
+    """Test the basic text response function."""
+    try:
+        print("🧪 Testing Gemini text response...")
+        
+        # Test with a simple prompt
+        prompt = "Hello, how are you today?"
+        response = gemini_text_response(prompt, temperature=0.1, max_tokens=50)
+        
+        if response and len(response) > 0:
+            print("✅ Gemini text response: PASSED")
+            print(f"   - Response: {response[:100]}...")
+            return True
+        else:
+            print("❌ Gemini text response: FAILED (Empty response)")
+            return False
+            
+    except Exception as e:
+        print(f"❌ Gemini text response: FAILED (Error: {e})")
+        return False
+
+def test_gemini_pro_text_gen():
+    """Test the legacy text generation function."""
+    try:
+        print("🧪 Testing Gemini Pro text generation...")
+        
+        # Test with a simple prompt
+        prompt = "What is the capital of France?"
+        response = gemini_pro_text_gen(prompt, temperature=0.1, max_tokens=50)
+        
+        if response and len(response) > 0 and not response.startswith("Error"):
+            print("✅ Gemini Pro text generation: PASSED")
+            print(f"   - Response: {response[:100]}...")
+            return True
+        else:
+            print(f"❌ Gemini Pro text generation: FAILED (Response: {response})")
+            return False
+            
+    except Exception as e:
+        print(f"❌ Gemini Pro text generation: FAILED (Error: {e})")
+        return False
+
+async def test_gemini_api_key_validation():
+    """Test the API key validation function."""
+    try:
+        print("🧪 Testing Gemini API key validation...")
+        
+        # Get API key from environment
+        api_key = os.getenv('GEMINI_API_KEY')
+        if not api_key:
+            print("❌ Gemini API key validation: FAILED (No API key found)")
+            return False
+        
+        # Test the API key
+        is_valid, message = await test_gemini_api_key(api_key)
+        
+        if is_valid:
+            print("✅ Gemini API key validation: PASSED")
+            print(f"   - Message: {message}")
+            return True
+        else:
+            print(f"❌ Gemini API key validation: FAILED (Message: {message})")
+            return False
+            
+    except Exception as e:
+        print(f"❌ Gemini API key validation: FAILED (Error: {e})")
+        return False
+
+async def main():
+    """Run all Gemini tests."""
+    print("🧪 Testing Gemini Provider Fixes")
+    print("=" * 50)
+    
+    tests = [
+        test_gemini_text_response,
+        test_gemini_pro_text_gen,
+        test_gemini_api_key_validation
+    ]
+    
+    passed = 0
+    total = len(tests)
+    
+    for test in tests:
+        if test == test_gemini_api_key_validation:
+            result = await test()
+        else:
+            result = test()
+        
+        if result:
+            passed += 1
+        print()
+    
+    print("=" * 50)
+    print(f"📊 Test Results: {passed}/{total} tests passed")
+    
+    if passed == total:
+        print("🎉 All Gemini tests passed! The fixes are working correctly.")
+        return 0
+    else:
+        print("⚠️ Some Gemini tests failed. Please check the API key and configuration.")
+        return 1
+
+if __name__ == "__main__":
+    import asyncio
+    sys.exit(asyncio.run(main()))