diff --git a/week2/community-contributions/day1_three_party_chat.ipynb b/week2/community-contributions/day1_three_party_chat.ipynb index 5979e94..e51b170 100644 --- a/week2/community-contributions/day1_three_party_chat.ipynb +++ b/week2/community-contributions/day1_three_party_chat.ipynb @@ -92,7 +92,7 @@ " if isinstance(self.api, OpenAI):\n", " completion = self.api.chat.completions.create(\n", " model=self.name,\n", - " messages=messages,\n", + " messages=[{\"role\": \"system\", \"content\": system}] + messages,\n", " max_tokens=300\n", " )\n", " completion = completion.choices[0].message.content\n", @@ -112,7 +112,6 @@ " return self.parse_answer(completion)\n", "\n", " def parse_answer(self, answer):\n", - " # \n", " # Remove prefix 'Name:' from answer if present.\n", " regex = r\"(?P\\w+): (?P.*)\"\n", " match = re.match(regex, answer, re.DOTALL)\n", @@ -146,8 +145,6 @@ " self.role = system_prompt\n", " self.initial_msg = initial_message\n", " self.messages = [] # keeps conversation history\n", - " if isinstance(self.model.api, OpenAI) and self.role:\n", - " self.messages = [{\"role\": \"system\", \"content\": self.role}]\n", " self.last_msg = \"\"\n", "\n", " def speak(self):\n",