Switch from urllib to requests to improve reliability (#867)

* Switch from urllib to requests to improve reliability

* Keep ruff linter-specific

* update

* update

* update

ch07/05_dataset-generation/llama3-ollama.ipynb

@@ -194,8 +194,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import urllib.request\n",
     "import json\n",
+    "import requests\n",
     "\n",
     "def query_model(prompt, model=\"llama3\", url=\"http://localhost:11434/api/chat\", role=\"user\"):\n",
     "    # Create the data payload as a dictionary\n",
@@ -209,25 +209,21 @@
     "        ]\n",
     "    }\n",
     "\n",
-    "    # Convert the dictionary to a JSON formatted string and encode it to bytes\n",
-    "    payload = json.dumps(data).encode(\"utf-8\")\n",
-    "\n",
-    "    # Create a request object, setting the method to POST and adding necessary headers\n",
-    "    request = urllib.request.Request(url, data=payload, method=\"POST\")\n",
-    "    request.add_header(\"Content-Type\", \"application/json\")\n",
-    "\n",
-    "    # Send the request and capture the response\n",
-    "    response_data = \"\"\n",
-    "    with urllib.request.urlopen(request) as response:\n",
-    "        # Read and decode the response\n",
-    "        while True:\n",
-    "            line = response.readline().decode(\"utf-8\")\n",
+    "    # Send the POST request\n",
+    "    with requests.post(url, json=data, stream=True, timeout=30) as r:\n",
+    "        r.raise_for_status()\n",
+    "        response_data = \"\"\n",
+    "        for line in r.iter_lines(decode_unicode=True):\n",
     "            if not line:\n",
-    "                break\n",
+    "                continue\n",
     "            response_json = json.loads(line)\n",
-    "            response_data += response_json[\"message\"][\"content\"]\n",
+    "            if \"message\" in response_json:\n",
+    "                response_data += response_json[\"message\"][\"content\"]\n",
     "\n",
-    "    return response_data"
+    "    return response_data\n",
+    "\n",
+    "result = query_model(\"What do Llamas eat?\")\n",
+    "print(result)"
    ]
   },
   {
@@ -498,7 +494,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.4"
+   "version": "3.10.16"
   }
  },
  "nbformat": 4,