diff --git a/week2/day3.ipynb b/week2/day3.ipynb index e9d0348..2dd936b 100644 --- a/week2/day3.ipynb +++ b/week2/day3.ipynb @@ -224,14 +224,16 @@ "metadata": {}, "outputs": [], "source": [ + "# Fixed a bug in this function brilliantly identified by student Gabor M.!\n", + "# I've also improved the structure of this function\n", + "\n", "def chat(message, history):\n", - " messages = [{\"role\": \"system\", \"content\": system_message}] + history + [{\"role\": \"user\", \"content\": message}]\n", "\n", + " relevant_system_message = system_message\n", " if 'belt' in message:\n", - " messages.append({\"role\": \"system\", \"content\": \"For added context, the store does not sell belts, \\\n", - "but be sure to point out other items on sale\"})\n", + " relevant_system_message += \" The store does not sell belts; if you are asked for belts, be sure to point out other items on sale.\"\n", " \n", - " messages.append({\"role\": \"user\", \"content\": message})\n", + " messages = [{\"role\": \"system\", \"content\": relevant_system_message}] + history + [{\"role\": \"user\", \"content\": message}]\n", "\n", " stream = openai.chat.completions.create(model=MODEL, messages=messages, stream=True)\n", "\n",