beam-boot / scripts /hf_permissions.py
StarAtNyte1's picture
Publish Beam Boot / Super Beam source
324ecd2 verified
Raw History Blame Contribute Delete
987 Bytes
import json
import os
import urllib.request
import modal
app = modal.App('beamboot-hf-permissions')
@app.function(image=modal.Image.debian_slim(python_version='3.12'), secrets=[modal.Secret.from_name('steerability-submission-write')])
def inspect():
results=[]
for name,value in os.environ.items():
upper=name.upper()
if value and ('HF_' in upper or 'HUGGING' in upper) and ('TOKEN' in upper or 'KEY' in upper):
try:
req=urllib.request.Request('https://huggingface.co/api/whoami-v2',headers={'Authorization':'Bearer '+value})
with urllib.request.urlopen(req,timeout=15) as response: data=json.load(response)
results.append({'variable':name,'name':data.get('name'),'role':data.get('auth',{}).get('accessToken',{}).get('role')})
except Exception as e: results.append({'variable':name,'error':type(e).__name__})
return results
@app.local_entrypoint()
def main(): print(inspect.remote())