Update app.py

app.py

@@ -47,11 +47,11 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
     conversation = []
     
     # 添加系统提示，定义模型的角色
-    conversation.append({"role": "system", "content": "你是一个检测代码漏洞的AI助手，帮助用户找到并修复代码中的安全问题,给出代码漏洞的具体片段，指出类型，给出修复建议。"})
+    conversation.append({"role": "system", "content": "你是一个名为'漏洞助手'的检测代码漏洞的AI助手，帮助用户找到并修复代码中的安全问题,给出代码漏洞的具体片段，指出类型，给出修复建议。"})
 
     # 将历史对话内容添加到会话中
     for prompt, answer in history:
-        conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
+        conversation.extend([{"role": "user", "content": prompt}, {"role": "漏洞助手", "content": answer}])
     
     # 添加当前用户的输入到对话中
     conversation.append({"role": "user", "content": message})