Add authentication test script

space_auth_test.py

@@ -0,0 +1,127 @@
+#!/usr/bin/env python3
+"""
+Space Authentication Test for OpenLLM Training
+
+This script verifies that Hugging Face authentication is working correctly
+in your Space using GitHub secrets. Add this to your Space to test before training.
+
+Usage:
+    Add this file to your Space and run it to verify authentication.
+"""
+
+import os
+import sys
+
+try:
+    from huggingface_hub import HfApi, login, whoami, create_repo, delete_repo
+    HF_AVAILABLE = True
+except ImportError:
+    HF_AVAILABLE = False
+    print("❌ huggingface_hub not installed")
+    sys.exit(1)
+
+
+def test_space_authentication():
+    """Test authentication in Hugging Face Space using GitHub secrets."""
+    print("🔐 Testing Space Authentication (GitHub Secrets)")
+    print("=" * 55)
+    
+    # Check if we're in a Space
+    space_vars = ["SPACE_ID", "SPACE_HOST", "SPACE_REPO_ID"]
+    is_space = any(os.getenv(var) for var in space_vars)
+    
+    if is_space:
+        print("✅ Running in Hugging Face Space environment")
+        for var in space_vars:
+            value = os.getenv(var)
+            if value:
+                print(f"   - {var}: {value}")
+    else:
+        print("ℹ️ Running in local environment")
+    
+    # Check HF_TOKEN from GitHub secrets
+    token = os.getenv("HF_TOKEN")
+    if not token:
+        print("❌ HF_TOKEN not found in environment")
+        print("   - Please set HF_TOKEN in your GitHub repository secrets")
+        print("   - Go to GitHub repository → Settings → Secrets and variables → Actions")
+        print("   - Add HF_TOKEN with your Hugging Face token")
+        return False
+    
+    print(f"✅ HF_TOKEN found: {token[:8]}...{token[-4:]}")
+    print(f"   - Source: GitHub secrets")
+    
+    try:
+        # Test authentication
+        print("\n🔄 Testing authentication...")
+        login(token=token)
+        
+        user_info = whoami()
+        username = user_info["name"]
+        
+        print(f"✅ Authentication successful!")
+        print(f"   - Username: {username}")
+        
+        # Test API access
+        print("\n🔄 Testing API access...")
+        api = HfApi()
+        print(f"✅ API access working")
+        
+        # Test repository creation
+        print("\n🧪 Testing Repository Creation")
+        repo_name = "test-openllm-auth"
+        repo_id = f"{username}/{repo_name}"
+        
+        print(f"🔄 Creating test repository: {repo_id}")
+        create_repo(
+            repo_id=repo_id,
+            repo_type="model",
+            exist_ok=True,
+            private=True
+        )
+        print(f"✅ Repository created successfully")
+        
+        # Clean up
+        print(f"🔄 Cleaning up test repository...")
+        delete_repo(repo_id=repo_id, repo_type="model")
+        print(f"✅ Repository deleted")
+        
+        print(f"\n🎉 All authentication tests passed!")
+        print(f"   - Authentication: ✅ Working")
+        print(f"   - API Access: ✅ Working")
+        print(f"   - Repository Creation: ✅ Working")
+        print(f"   - GitHub Secrets Integration: ✅ Working")
+        print(f"   - Ready for OpenLLM training and model uploads!")
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ Authentication test failed: {e}")
+        print(f"\n🔧 Troubleshooting:")
+        print(f"1. Check if HF_TOKEN is set correctly in GitHub repository secrets")
+        print(f"2. Verify token has 'Write' permissions")
+        print(f"3. Check Space logs for detailed error messages")
+        print(f"4. Ensure your Space is connected to the GitHub repository")
+        return False
+
+
+def main():
+    """Main test function."""
+    print("🚀 OpenLLM - Space Authentication Test")
+    print("=" * 45)
+    
+    success = test_space_authentication()
+    
+    if success:
+        print(f"\n✅ Authentication test completed successfully!")
+        print(f"   - Your Space is ready for OpenLLM training")
+        print(f"   - Model uploads will work correctly")
+        print(f"   - GitHub secrets integration is working")
+    else:
+        print(f"\n❌ Authentication test failed")
+        print(f"   - Please fix authentication issues before training")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()