File size: 4,219 Bytes
dfb73fe |
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 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 |
#!/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()
|