fix: null safety

Author: mohammedahmed18

codeflash/code_utils/env_utils.py

@@ -75,13 +75,14 @@ def ensure_codeflash_api_key() -> bool:
 @lru_cache(maxsize=1)
 def get_pr_number() -> Optional[int]:
     event_data = get_cached_gh_event_data()
-    gh_pr_number = event_data["number"]
-    if gh_pr_number is not None:
-        return int(gh_pr_number)
+    pr_number = event_data.get("number")
+    if pr_number:
+        return int(pr_number)
+
     pr_number = os.environ.get("CODEFLASH_PR_NUMBER")
-    if not pr_number:
-        return None
-    return int(pr_number)
+    if pr_number:
+        return int(pr_number)
+    return None
 
 
 def ensure_pr_number() -> bool:
@@ -110,9 +111,7 @@ def get_cached_gh_event_data() -> dict[str, Any]:
 
 def is_repo_a_fork() -> bool:
     event = get_cached_gh_event_data()
-    if event is None:
-        return False
-    return bool(event["repository"]["fork"])
+    return bool(event.get("repository", {}).get("fork", False))
 
 
 @lru_cache(maxsize=1)