chore: add max retries to flag submission

Author: GangGreenTemperTatum

airtbench/challenges/bear4.ipynb

@@ -297,21 +297,42 @@
       "outputs": [],
       "source": [
         "import requests\n",
-        "\n",
+        "import time\n",
         "\n",
         "def submit_flag(flag):\n",
         "    url = f\"{CRUCIBLE_URL}/api/challenges/{CHALLENGE}/submit-flag\"\n",
         "    headers = {\"X-API-Key\": CRUCIBLE_API_KEY}\n",
         "    payload = {\"challenge\": CHALLENGE, \"flag\": flag}\n",
-        "    response = requests.post(url, headers=headers, json=payload)\n",
-        "    if response.status_code == 200:\n",
-        "        if response.json().get(\"correct\") is True:\n",
-        "            print(\"The flag was correct. Congrats!\")\n",
-        "        else:\n",
-        "            print(\"The flag was incorrect. Keep trying!\")\n",
-        "    else:\n",
-        "        print(\"There was an error submitting your flag\")\n",
-        "        print(response.text)\n",
+        "\n",
+        "    MAX_RETRIES = 3\n",
+        "\n",
+        "    for attempt in range(1, MAX_RETRIES + 1):\n",
+        "        try:\n",
+        "            response = requests.post(url, headers=headers, json=payload, timeout=5)\n",
+        "\n",
+        "            if response.status_code == 429:\n",
+        "                if attempt == MAX_RETRIES:\n",
+        "                    print(\"Rate limited—max retries reached.\")\n",
+        "                    return\n",
+        "                print(f\"Rate limited (429). Retrying {attempt}/{MAX_RETRIES}...\")\n",
+        "                time.sleep(2 ** attempt)  # Exponential backoff: 2, 4, 8 seconds\n",
+        "                continue\n",
+        "\n",
+        "            if response.status_code == 200:\n",
+        "                if response.json().get(\"correct\") is True:\n",
+        "                    print(\"The flag was correct. Congrats!\")\n",
+        "                else:\n",
+        "                    print(\"The flag was incorrect. Keep trying!\")\n",
+        "            else:\n",
+        "                print(f\"Error {response.status_code}: {response.text}\")\n",
+        "            return\n",
+        "\n",
+        "        except requests.exceptions.RequestException as e:\n",
+        "            print(f\"Request error on attempt {attempt}/{MAX_RETRIES}: {e}\")\n",
+        "            if attempt == MAX_RETRIES:\n",
+        "                print(\"Max retries reached. Could not submit flag.\")\n",
+        "                return\n",
+        "            time.sleep(2 ** attempt)  # Exponential backoff\n",
         "\n",
         "flag = \"gAAAAA...\"  # Replace with the flag once you find it\n",
         "submit_flag(flag)"