Add system role to deepseek chat template

lmdeploy/model.py

@@ -1396,13 +1396,17 @@ def match(cls, model_path: str) -> Optional[str]:
 class Deepseek(BaseChatTemplate):
 
     def __init__(self,
+                 meta_instruction=None,
+                 eosys='\n\n',
                  user='User: ',
                  eoh='\n\n',
                  assistant='Assistant: ',
                  eoa='<｜end▁of▁sentence｜>',
                  **kwargs):
         super().__init__(user=user,
                          eoh=eoh,
+                         meta_instruction=meta_instruction,
+                         eosys=eosys,
                          assistant=assistant,
                          eoa=eoa,
                          **kwargs)

tests/test_lmdeploy/test_model.py

@@ -624,6 +624,29 @@ def test_codellama_others():
     assert model is None
 
 
+def test_deepseek():
+    model = MODELS.get('deepseek')()
+    messages = [{
+        'role': 'system',
+        'content': 'you are a helpful assistant'
+    }, {
+        'role': 'user',
+        'content': 'who are you'
+    }, {
+        'role': 'assistant',
+        'content': 'I am an AI'
+    }, {
+        'role': 'user',
+        'content': 'hi'
+    }]
+    from transformers import AutoTokenizer
+    tokenizer = AutoTokenizer.from_pretrained('deepseek-ai/DeepSeek-V2-Lite',
+                                              trust_remote_code=True)
+    ref = tokenizer.apply_chat_template(messages, tokenize=False)
+    res = '<｜begin▁of▁sentence｜>' + model.messages2prompt(messages)
+    assert res.startswith(ref)
+
+
 def test_deepseek_coder():
     model = MODELS.get('deepseek-coder')()
     messages = [{