File size: 4,149 Bytes
53e769b |
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 |
#!/usr/bin/env python3
"""
Simple Space Authentication Verification (GitHub Secrets)
This script can be added to your Hugging Face Space to verify
that authentication is working correctly using GitHub secrets.
Usage:
Add this 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 verify_space_authentication():
"""Verify authentication is working in the Space using GitHub secrets."""
print("🔐 Verifying 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")
print(" - The Space will automatically have access to this secret")
return False
print(f"✅ HF_TOKEN found: {token[:8]}...{token[-4:]}")
print(f" - Source: GitHub secrets")
try:
# Test authentication
login(token=token)
user_info = whoami()
username = user_info["name"]
print(f"✅ Authentication successful!")
print(f" - Username: {username}")
# Test API access
api = HfApi()
print(f"✅ API access working")
# Test repository creation
print(f"\n🧪 Testing Repository Creation")
repo_name = "test-openllm-verification"
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 verification tests passed!")
print(f" - Authentication: ✅ Working")
print(f" - Repository Creation: ✅ Working")
print(f" - GitHub Secrets Integration: ✅ Working")
print(f" - Ready for training and model uploads!")
return True
except Exception as e:
print(f"❌ Authentication 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 verification function."""
print("🚀 OpenLLM - Space Authentication Verification (GitHub Secrets)")
print("=" * 65)
success = verify_space_authentication()
if success:
print(f"\n✅ Verification 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❌ Verification failed")
print(f" - Please fix authentication issues before training")
sys.exit(1)
if __name__ == "__main__":
main()
|