brad/LLM_Engineering_OLD
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor: Add system prompt to messages in the Model class, not in Participant

Browse Source
This commit is contained in:
Sandra Neuhäußer
2025-04-04 14:37:49 +02:00
parent 4fdc04a601
commit ac585aaad1
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
week2/community-contributions/day1_three_party_chat.ipynb
View File

@@ -92,7 +92,7 @@
" if isinstance(self.api, OpenAI):\n", " if isinstance(self.api, OpenAI):\n",
" completion = self.api.chat.completions.create(\n", " completion = self.api.chat.completions.create(\n",
" model=self.name,\n", " model=self.name,\n",
" messages=messages,\n", " messages=[{\"role\": \"system\", \"content\": system}] + messages,\n",
" max_tokens=300\n", " max_tokens=300\n",
" )\n", " )\n",
" completion = completion.choices[0].message.content\n", " completion = completion.choices[0].message.content\n",
@@ -112,7 +112,6 @@
" return self.parse_answer(completion)\n", " return self.parse_answer(completion)\n",
"\n", "\n",
" def parse_answer(self, answer):\n", " def parse_answer(self, answer):\n",
" # \n",
" # Remove prefix 'Name:' from answer if present.\n", " # Remove prefix 'Name:' from answer if present.\n",
" regex = r\"(?P<name>\\w+): (?P<content>.*)\"\n", " regex = r\"(?P<name>\\w+): (?P<content>.*)\"\n",
" match = re.match(regex, answer, re.DOTALL)\n", " match = re.match(regex, answer, re.DOTALL)\n",
@@ -146,8 +145,6 @@
" self.role = system_prompt\n", " self.role = system_prompt\n",
" self.initial_msg = initial_message\n", " self.initial_msg = initial_message\n",
" self.messages = [] # keeps conversation history\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", " self.last_msg = \"\"\n",
"\n", "\n",
" def speak(self):\n", " def speak(self):\n",