Upload 3 files

resource_group_automation.py

@@ -0,0 +1,152 @@
+import requests
+from environs import Env
+
+# Initialize environment
+env = Env()
+env.read_env() # Looks for .env files in current directory
+
+# Set your token (in a real environment, get this from env vars)
+HF_TOKEN = env("HF_TOKEN")
+ORG_NAME = env("ORG_NAME")
+BASE_URL = env("BASE_URL")
+
+# HTTP Headers for API requests
+headers = {
+    "Authorization": f"Bearer {HF_TOKEN}",
+    "Content-Type": "application/json"
+}
+
+# First thing's first: create the repo (or use existing one)
+def create_repository(org_name, repo_name):
+    """Create repository or use existing one"""
+    # First check if it already exists
+    repo_url = f"{BASE_URL}/models/{org_name}/{repo_name}"
+    check_response = requests.get(repo_url, headers=headers)
+    
+    if check_response.status_code == 200:
+        print(f"Good news! Repository {org_name}/{repo_name} already exists!")
+        return True
+        
+    # Try to create it if it doesn't exist
+    create_url = f"{BASE_URL}/repos/create"
+    print(f"Houston, stand by: Creating repo with URL: {create_url}")
+    data = {
+        "type": "model",
+        "name": repo_name,
+        "organization": org_name,
+        "private": True
+    }
+
+    response = requests.post(create_url, headers=headers, json=data)
+    if response.status_code == 200:
+        print(f"Woohoo!!! Created repository: {org_name}/{repo_name}")
+        return True
+    else:
+        # Check if the error is because the repo already exists
+        error_data = response.json()
+        if "error" in error_data and "already created" in error_data["error"]:
+            print(f"Repository {org_name}/{repo_name} already exists! Moving on.")
+            return True
+        else:
+            print(f"Son of a Lambda Cold Start! Failed to create repository: {response.text}")
+            return False
+
+def create_resource_group_for_repo(repo_name, creator_username):
+    """
+    The magical function that creates a resource group for a new repository.
+    """
+    # Step 0: Make the repo first or we're going nowhere fast
+    if not create_repository(ORG_NAME, repo_name):
+        print("Hey, yo! Slow your roll! Can't make a resource group for a repo that doesn't exist!")
+        return False
+
+    # Step 1: Create our glorious resource group
+    rg_name = f"rg-{repo_name}"  # Simpler name, less chance for API to throw a fit
+    rg_url = f"{BASE_URL}/organizations/{ORG_NAME}/resource-groups"
+    print(f"Conjuring a resource group with URL: {rg_url}")
+    
+    # Let's be direct - just create it with minimal data
+    data = {
+        "name": rg_name,
+        "description": f"Automagically created for {repo_name} by the Wizard of Automation"
+    }
+    
+    # Throw the request into the void and pray for a 200 or 201
+    response = requests.post(rg_url, headers=headers, json=data)
+
+    # Print full response for debugging (you know it will happen)
+    print(f"Resource group creation response: {response.status_code} - {response.text}")
+    
+    success = False
+    if response.status_code == 201 or response.status_code == 200:
+        try:
+            response_data = response.json()
+            if "id" in response_data:
+                success = True
+                rg_id = response_data.get("id")
+                print(f"Resource group successfully summoned! ID: {rg_id}")
+            else:
+                print("Response has unexpected format - missing 'id' field")
+        except Exception as e:
+            print(f"Error parsing response: {e}")
+    
+    if not success:
+        print(f"Son of a Nutcracker! Resource group creation failed: {response.text}")
+        return False
+    
+    # We're in business! Grab that ID
+    rg_id = response.json().get("id")
+    print(f"I stole, I mean grabbed that Resource Group ID because I need it: {rg_id}")
+    
+    # Step 2: Connect the repo to our shiny new resource group
+    repo_url = f"{BASE_URL}/models/{ORG_NAME}/{repo_name}/resource-group"
+    repo_data = {
+        "resourceGroupId": rg_id
+    }
+    
+    repo_response = requests.post(repo_url, headers=headers, json=repo_data)
+    
+    if repo_response.status_code != 200:
+        print(f"Son of an API! Failed to connect repo: {repo_response.text}")
+        return False
+    
+    print(f"Repository {repo_name} has joined the resource group party!")
+    
+    # Step 3: Make the creator an admin - they brought this repo into the world after all
+    member_url = f"{BASE_URL}/organizations/{ORG_NAME}/resource-groups/{rg_id}/users"
+    member_data = {
+    "users": [
+        {
+            "user": creator_username,
+            "role": "admin"
+        }
+    ]
+}
+    
+    member_response = requests.post(member_url, headers=headers, json=member_data)
+    
+    if member_response.status_code != 200:
+        print(f"Son of a broken websocket! Creator couldn't be added: {member_response.text}")
+        # We'll continue anyway - the group and repo connection are working
+    else:
+        print(f"Creator {creator_username} is now the admin overlord of this resource group!")
+    
+    # Victory dance
+    return True
+
+# Demo time: This simulates what would happen when a new repo is created
+
+def main():
+    repo_name = "testing-magic"
+    creator = "joshhayles"
+    
+    print(f"A WIZARD created a new Repo: {ORG_NAME}/{repo_name}")
+    result = create_resource_group_for_repo(repo_name, creator)
+    
+    if result:
+        print("Ohh, snap! Successfully set up resource group for the new repository!")
+    else:
+        print("Damn. Failed to set up resource group")
+
+if __name__ == "__main__":
+    main()

son-of-a-nutcracker.md

@@ -0,0 +1,74 @@
+# Automation Nutcracker: HF-ROCKS Edition
+
+Featuring:
+- A Wizard, 
+- a token,
+- a nutcracker
+- and Oscar the Grouch
+  
+How I walked into a terminal and got jack-hammered slapped with errors. And please don't ask me what jack-hammered slapped means. Nobody knows. 
+
+## Symptoms
+- Terminal vomit featuring `Son of a Nutcracker`
+- API responses colder than a Lambda start
+- The lingering suspicion that the machine knows you're trying to be cool :sunglasses:
+
+---
+
+## Cause of Death
+- **Ego-Driven Org Name Change**: Trying to flex by renaming the org to `HF-ROCKS` instead of just leaving `eh-quizz` alone.
+- **Token Permissions in Witness Protection**: The API didn't lie; my token was missing org-level permissions. But it still pissed me off.
+- **Blind Faith in the UI**: Thinking that what I saw in the Hugging Face dashboard actually **mattered** in the API response.
+
+---
+
+## Autopsy Findings
+Here's how it all unraveled:
+
+### The First Heartbreak
+```
+A WIZARD created a new Repo: HF ROCKS/testing-magic
+Son of a Nutcracker. Failed to create resource group: {"error":"Sorry, we can't find the page you are looking for."}
+Damn. Failed to set up resource group
+```
+
+Conclusion: The API didn't give a flying horse monkey crap about my cool new org name.
+
+And once again, please don't ask me what flying horse monkey crap means. Nobody knows.
+
+---
+
+### Forensics Report
+After scraping away the layers with `test.py`, I finally hit paydirt:
+```json
+"orgs": [{"type":"org","id":"67a287f99b8fb9f109323d45","name":"eh-quizz","fullname":"HF-ROCKS",...}]
+```
+**Mismatched `name` field detected.**
+
+My whole plan went straight into Oscar the Grouch's trashcan because the API was locked on the original name, not the flashy one I gave it in the UI.
+
+---
+
+### Resurrection Steps
+1. Update token permissions to include `Org` scope (the hidden boss level nobody talks about).
+2. Accept the fact that API responses don't care about your branding ambitions.
+3. Run the script again...
+
+---
+
+## Golden Sunset 
+```
+Resource group successfully summoned! ID: 67c9ed9e658c9a336cb87170
+Repository testing-magic has joined the resource group party!
+Creator joshhayles is now the admin overlord of this resource group!
+Ohh, snap! Successfully set up resource group for the new repository!
+```
+
+Lessons learned?
+- The API is always right.
+- You're not cool enough to rename your org name mid-script.
+- Automation without humility will absolutely clown you.
+
+---
+
+**Autopsy Complete** 

test.py

@@ -0,0 +1,191 @@
+# I need to print some stuff out so I know what I'm dealing with here
+# Go ahead, run the file. I DARE ya!
+
+import requests
+import json
+from environs import Env
+
+# Initialize environment
+env = Env()
+env.read_env()
+
+HF_TOKEN = env("HF_TOKEN")
+ORG_NAME = env("ORG_NAME") 
+BASE_URL = "https://huggingface.co/api"
+
+headers = {
+    "Authorization": f"Bearer {HF_TOKEN}",
+    "Content-Type": "application/json"
+}
+
+def print_section(title):
+    """Print a simple section header"""
+    print("\n" + "=" * 50)
+    print(title)
+    print("=" * 50)
+
+def format_response(label, response):
+    """Format and print JSON response with clear structure"""
+    print(f"\n{label}: {response.status_code}")
+    
+    if response.status_code == 200:
+        try:
+            # Parse and return the JSON data
+            data = response.json()
+            # Format and print it with indentation
+            formatted_json = json.dumps(data, indent=2)
+            print(formatted_json)
+            return data
+        except json.JSONDecodeError:
+            print("Error: Could not parse JSON response")
+            print(response.text)
+            return None
+    else:
+        print(f"Error: {response.text}")
+        return None
+
+# First, check authentication and user info
+print_section("AUTHENTICATION CHECK")
+org_url = f"{BASE_URL}/whoami-v2"
+response = requests.get(org_url, headers=headers)
+user_data = format_response("Whoami response", response)
+
+# List existing resource groups
+print_section("RESOURCE GROUPS")
+list_rg_url = f"{BASE_URL}/organizations/{ORG_NAME}/resource-groups"
+response = requests.get(list_rg_url, headers=headers)
+resource_groups = format_response("List resource groups response", response)
+
+# Create a test resource group only if it doesn't already exist
+print_section("RESOURCE GROUP CREATION TEST")
+new_group_name = "test-resource-group"
+new_group_description = "A test resource group"
+
+# Check if a group with this name already exists
+group_exists = False
+if resource_groups:
+    for group in resource_groups:
+        if group.get("name") == new_group_name:
+            group_exists = True
+            print(f"\nResource group '{new_group_name}' already exists with ID: {group.get('id')}")
+            print("Skipping creation to avoid duplicates.")
+            break
+
+# Only create the group if it doesn't exist
+if not group_exists:
+    create_rg_url = f"{BASE_URL}/organizations/{ORG_NAME}/resource-groups"
+    data = {
+        "name": new_group_name,
+        "description": new_group_description
+    }
+    response = requests.post(create_rg_url, headers=headers, json=data)
+    format_response("Create resource group response", response)
+
+"""
+Expected response:
+
+==================================================
+AUTHENTICATION CHECK
+==================================================
+
+Whoami response: 200
+{
+  "type": "user",
+  "id": "67c8834889772d508b6fa33c",
+  "name": "joshhayles",
+  "fullname": "Josh Hayles",
+  "isPro": false,
+  "avatarUrl": "https://cdn-avatars.huggingface.co/v1/production/uploads/67c8834889772d508b6fa33c/oZqi8zNQsTCSunm5NFnhE.png",
+  "orgs": [
+    {
+      "type": "org",
+      "id": "67a287f99b8fb9f109323d45",
+      "name": "eh-quizz",
+      "fullname": "eh-quizz",
+      "email": "[email protected]",
+      "canPay": false,
+      "periodEnd": 1743465599,
+      "avatarUrl": "https://cdn-avatars.huggingface.co/v1/production/uploads/67c8834889772d508b6fa33c/dU5WKVWLSq0jaYjG-zJBg.png",
+      "roleInOrg": "admin",
+      "isEnterprise": true
+    }
+  ],
+  "auth": {
+    "type": "access_token",
+    "accessToken": {
+      "displayName": "wuzzup-token",
+      "role": "fineGrained",
+      "createdAt": "2025-03-05T23:19:18.771Z",
+      "fineGrained": {
+        "canReadGatedRepos": true,
+        "global": [
+          "inference.serverless.write",
+          "discussion.write",
+          "post.write"
+        ],
+        "scoped": [
+          {
+            "entity": {
+              "_id": "67c9e5b9b98e0e0b6605992f",
+              "type": "model",
+              "name": "eh-quizz/testing-magic"
+            },
+            "permissions": [
+              "repo.content.read",
+              "repo.write"
+            ]
+          },
+          {
+            "entity": {
+              "_id": "67a287f99b8fb9f109323d45",
+              "type": "org",
+              "name": "eh-quizz"
+            },
+            "permissions": [
+              "repo.content.read",
+              "discussion.write",
+              "repo.write",
+              "org.read",
+              "org.write",
+              "resourceGroup.write"
+            ]
+          },
+          {
+            "entity": {
+              "_id": "67c8834889772d508b6fa33c",
+              "type": "user",
+              "name": "joshhayles"
+            },
+            "permissions": [
+              "repo.content.read",
+              "inference.endpoints.infer.write",
+              "user.webhooks.read",
+              "discussion.write"
+            ]
+          }
+        ]
+      }
+    }
+  }
+}
+
+==================================================
+RESOURCE GROUPS
+==================================================
+
+List resource groups response: 200
+[]
+
+==================================================
+RESOURCE GROUP CREATION TEST
+==================================================
+
+Create resource group response: 200
+{
+  "id": "67ca1e89dd6c6628fcb2ca6d",
+  "name": "test-resource-group",
+  "description": "A test resource group",
+  "users": [],
+  "repos": []
+}
+"""