Fix refactoring

Author: ZainRizvi

tools/analytics/org/analyze_runner_usage.py

@@ -79,6 +79,12 @@
 GITHUB_TOKEN = os.getenv("GITHUB_TOKEN")
 ORG_NAME = None  # Will be set by argparse
 
+# GitHub API headers
+HEADERS = {
+    "Authorization": f"Bearer {GITHUB_TOKEN}",
+    "Accept": "application/vnd.github+json",
+}
+
 # List of repositories to exclude in the format 'org/repo'
 EXCLUDED_REPOS = [
     "pytorch/pytorch",
@@ -150,7 +156,7 @@ def get_repos(org: str) -> List[str]:
     while True:
         url = f"{BASE_URL}/orgs/{org}/repos?per_page=100&page={page}"
         logging.debug(f"[get_repos] Requesting URL: {url}")
-        data = make_cached_request(url)
+        data = make_cached_request(url, HEADERS)
         if data is None:
             logging.error(f"[get_repos] Failed to fetch page {page} for org: {org}")
             break
@@ -186,7 +192,7 @@ def get_workflow_runs(org: str, repo: str) -> List[Dict]:
     while True:
         url = f"{BASE_URL}/repos/{org}/{repo}/actions/runs?per_page=100&page={page}&created=>={WORKFLOW_RUN_LOOKBACK}"
         logging.debug(f"[get_workflow_runs] Requesting URL: {url}")
-        response_data = make_cached_request(url)
+        response_data = make_cached_request(url, HEADERS)
         if response_data is None:
             logging.error(
                 f"[get_workflow_runs] Failed to fetch page {page} for repo: {repo}"
@@ -271,7 +277,7 @@ def get_jobs_for_run(
     )
     url = f"{BASE_URL}/repos/{org}/{repo}/actions/runs/{run_id}/jobs"
     logging.debug(f"[get_jobs_for_run] Requesting URL: {url}")
-    response_data = make_cached_request(url)
+    response_data = make_cached_request(url, HEADERS)
     if response_data is None:
         logging.error(
             f"[get_jobs_for_run] Failed to fetch jobs for run {run_id} in repo: {repo}"

tools/analytics/org/cache_manager.py

@@ -130,21 +130,15 @@ def clear_cache():
         logging.info(f"[clear_cache] Cache directory does not exist: {CACHE_DIR}")
 
 
-HEADERS = {
-    "Authorization": f"Bearer {GITHUB_TOKEN}",
-    "Accept": "application/vnd.github+json",
-}
-
-
 def make_cached_request(
-    url: str, headers: Optional[Dict[str, str]] = None
+    url: str, headers: Dict[str, str]
 ) -> Optional[Dict]:
     """
     Make an HTTP request with caching. Returns the JSON response if successful.
 
     Args:
         url: The URL to request
-        headers: Optional headers for the request
+        headers: Headers for the request (required)
 
     Returns:
         JSON response data if successful, None if failed
@@ -158,7 +152,7 @@ def make_cached_request(
     # Make actual HTTP request
     logging.info(f"[make_cached_request] Making HTTP request to: {url}")
     try:
-        response = requests.get(url, headers=headers or HEADERS)
+        response = requests.get(url, headers=headers)
         response.raise_for_status()
         data = response.json()