Fix in Completions API prompt (#29)

Author: dmitripikus

synthetic-multi-round-qa/multi-round-qa.py

@@ -173,7 +173,7 @@ async def _async_launch_request(self, messages: List[Dict[str, str]],  max_token
             else:
                 # Use completions API
                 # Convert messages to a prompt string
-                prompt = "\n".join([f"{msg['role'].upper()}: {msg['content']}" for msg in messages])
+                prompt = "\n".join([f"{msg['role'].upper()}: {msg['content']}" for msg in messages]) + "\nASSISTANT:"
 
                 response = await self.client.completions.create(
                     model=self.model,
@@ -213,7 +213,7 @@ async def _async_launch_request(self, messages: List[Dict[str, str]],  max_token
                             stream=False,
                         )
                     else:
-                        prompt = "\n".join([f"{msg['role'].upper()}: {msg['content']}" for msg in messages])
+                        prompt = "\n".join([f"{msg['role'].upper()}: {msg['content']}" for msg in messages]) + "\nASSISTANT:"
                         final_response = await self.client.completions.create(
                             model=self.model,
                             prompt=prompt,