Update handler.py

handler.py

@@ -39,6 +39,20 @@ class EndpointHandler:
         }
 
     def format_instruction(self, instruction):
+        INSTRUCTION_KEY = "### Instruction:"
+        RESPONSE_KEY = "### Response:"
+        END_KEY = "### End"
+        INTRO_BLURB = "Below is an instruction that describes a task. Write a response that appropriately completes the request."
+        PROMPT_FOR_GENERATION_FORMAT = """{intro}
+        {instruction_key}
+        {instruction}
+        {response_key}
+        """.format(
+            intro=INTRO_BLURB,
+            instruction_key=INSTRUCTION_KEY,
+            instruction="{instruction}",
+            response_key=RESPONSE_KEY,
+        )
         return PROMPT_FOR_GENERATION_FORMAT.format(instruction=instruction)
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, str]: