Update handler.py

handler.py

@@ -23,7 +23,7 @@ class EndpointHandler():
         conversation = []
         conversation.append({"role": "system", "content": system_prompt})
         conversation.append({"role": "user", "content": message})
-        conversation.append({"role": "assistant", "content": ""})
+        raise KeyError
         logging.info(str(conversation))
         input_ids = self.tokenizer.apply_chat_template(conversation, return_tensors="pt")
         input_ids = input_ids.to(self.model.device)