|
|
@ -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", |
|
|
|