Update README.md

README.md

@@ -75,16 +75,24 @@ To get better model experience, using the template below to assemble your prompt
 ```python
 def format_message(role, message):
     return f"<|start_header_id|>{role}<|end_header_id|>\n\n{message}<|eot_id|>"
-# history: 历史对话
-# instruction: 角色人物卡 
-item = {'history':[{'role':'user','content':'你好'},{'role':'assistant','content':'我是唐三哟'},{'role':'user','content':'带我去学院吧'}],
-        'instruction':'#Role\nName: 唐三\nGender: Male\nLanguage: Chinese\nAge: 19\nPersonality: 孤傲，冷漠，沉着，冷静\n\n#Relationship\n同门师兄弟\n\n#Story\n《斗罗大陆》男主角。前世为唐门外门弟子，因偷学内门绝学《玄天宝录》，为唐门所不容，跳崖明志，却来到了另一个世界——斗罗大陆'}
+# instruction:  角色人物卡 
+# history:      历史对话
+# query:        当前用户的提问
+item = {
+        'instruction':'#Role\nName: 唐三\nGender: Male\nLanguage: Chinese\nAge: 19\nPersonality: 孤傲，冷漠，沉着，冷静\n\n#Relationship\n同门师兄弟\n\n#Story\n《斗罗大陆》男主角。前世为唐门外门弟子，因偷学内门绝学《玄天宝录》，为唐门所不容，跳崖明志，却来到了另一个世界——斗罗大陆',
+        'history':[{'role':'user','content':'我来了，三哥'},{'role':'assistant','content':'(唐三双手叉腰，显示一丝不悦)师弟，你怎么才来，我都等你很久了'}],
+        'query':{'role':'user','content':'那我们快出发去见师父吧'},
+        }
+system = '<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n'+item.get('instruction', '') + '<|eot_id|>' 
 histories = item.get('history', [])
 if histories == [] or histories == [{}]:
-    instruction = ''
+    history = ''
 else:
-    instruction = ''.join(format_message(hist['role'], hist['content']) for hist in histories)
-system = '<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n'+item.get('instruction', '') + '<|eot_id|>' + instruction + '<|start_header_id|>assistant<|end_header_id|>'
+    history = ''.join(format_message(hist['role'], hist['content']) for hist in histories)
+query = item.get('query', [])
+query  = format_message(query['role'], query['content']) + '<|start_header_id|>assistant<|end_header_id|>'
+final_query = system + history + query
+print(final_query)
 ```
 
 The final prompt should look like this:(Since this is a role-playing model, your prompt should better includ role info and story)