Update cefr_utils.py

cefr_utils.py

@@ -7,6 +7,9 @@ description = {
     "A1": "Can interact in a simple way but communication is totally dependent on repetition at a slower rate, rephrasing and repair. Can ask and answer simple questions, initiate and respond to simple statements in areas of immediate need or on very familiar topics."
 }
 
+def format_context(context):
+    return os.linesep.join([("A" if (i%2==0) else "B") + ": " + turn["text"] for i, turn in enumerate(context)])
+    
 def get_messages(instruction, item, apply_chat_template, system_msg, next_speaker="A"):
     item['messages'] = [{"role": "system", "content": f"Only output {next_speaker}'s response."}] if system_msg else []
     item['messages'] += [{"role": "user", "content": f"{instruction}\nDialog:\n{format_context(item['context'])}\n"}]