Simplify

Author: oeken

tutorials/needle.ipynb

@@ -64,7 +64,7 @@
    "source": [
     "system_messages = [{\"role\": \"system\", \"content\": r.content} for r in results] # results from Needle\n",
     "user_message = {\n",
-    "    \"role\": \"system\",\n",
+    "    \"role\": \"user\",\n",
     "    \"content\": f\"\"\"\n",
     "        Only answer the question based on the provided results data. \n",
     "        If there is no data in the provided data for the question, do not try to generate an answer. \n",
@@ -74,7 +74,7 @@
     "\n",
     "openai_client = OpenAI()\n",
     "answer = openai_client.chat.completions.create(\n",
-    "    model=\"gpt-3.5-turbo\",\n",
+    "    model=\"gpt-4o-mini\",\n",
     "    messages=[\n",
     "        *system_messages,\n",
     "        user_message,\n",