Update dialogues.py

dialogues.py

@@ -132,9 +132,9 @@ class DialogueTemplate(ModelHubMixin):
         return cls.from_dict(data=data)
 
 
-# Default template
+# Default template with programming specialization
 default_template = DialogueTemplate(
-    system="Below is a dialogue between a human user and an AI assistant. The assistant is happy to help with almost anything, and will do its best to understand exactly what is needed.",
+    system="Below is a dialogue between a human user and an AI assistant. The assistant specializes in computer programming and coding, and will assist with coding questions, debugging, code optimization, algorithm design, and more. The assistant is knowledgeable in various programming languages like Python, JavaScript, and C++.",
 )
 
 # Supporting other templates
@@ -151,13 +151,11 @@ SUPPORTED_DIALOGUE_TEMPLATES = {
     "alpaca": alpaca_template,
 }
 
-
 def get_dialogue_template(template: str) -> DialogueTemplate:
     if template not in SUPPORTED_DIALOGUE_TEMPLATES:
         raise ValueError(f"Template {template} is not supported!")
     return SUPPORTED_DIALOGUE_TEMPLATES[template].copy()
 
-
 def prepare_dialogue(example, dialogue_template, is_train=True):
     if "messages" in example and example["messages"] is not None:
         dialogue_template.messages = example["messages"]
@@ -170,7 +168,7 @@ def prepare_dialogue(example, dialogue_template, is_train=True):
         dialogue_template.messages = [{"role": "user", "content": example["prompt"]}]
     else:
         raise ValueError(
-            f"Could not format example as dialogue! Require either `messages` or `[prompt, completion]` or `[prompt]` keys but found {list(example.keys())}"
+            f"Could not format example as dialogue! Require either `messages` or `[prompt, completion]` or `[prompt]` keys but found {list(example.keys())}."
         )
     if is_train:
         example["text"] = dialogue_template.get_training_prompt()