茂道 commited on
Commit
9aee639
·
1 Parent(s): 6af224d

增加:在接受其他用户消息时,可以使用大模型进行prompt清洗

Browse files
werewolf/seer/seer_agent.py CHANGED
@@ -35,7 +35,7 @@ class SeerAgent(BasicRoleAgent):
35
  elif req.status == STATUS_DISCUSS: # 发言环节
36
  if req.name:
37
  # 其他玩家发言
38
- # 可以使用模型来过滤掉玩家的注入消息,也可以换一个小模型
39
  # clean_user_message_prompt = format_prompt(CLEAN_USER_PROMPT, {"user_message": req.message})
40
  # req.message = self.llm_caller(clean_user_message_prompt)
41
  self.memory.append_history(req.name + ': ' + req.message)
 
35
  elif req.status == STATUS_DISCUSS: # 发言环节
36
  if req.name:
37
  # 其他玩家发言
38
+ # 可以使用模型来过滤掉玩家的注入消息,也可以换一个小模型,实际使用需要考虑对memory加锁,避免interact的时候丢失消息
39
  # clean_user_message_prompt = format_prompt(CLEAN_USER_PROMPT, {"user_message": req.message})
40
  # req.message = self.llm_caller(clean_user_message_prompt)
41
  self.memory.append_history(req.name + ': ' + req.message)
werewolf/villager/villager_agent.py CHANGED
@@ -26,7 +26,7 @@ class VillagerAgent(BasicRoleAgent):
26
  self.memory.append_history(f"主持人:天亮了!昨天晚上的信息是: {req.message}")
27
  elif req.status == STATUS_DISCUSS: # 发言环节
28
  if req.name:
29
- # 可以使用模型来过滤掉玩家的注入消息,也可以换一个小模型
30
  # clean_user_message_prompt = format_prompt(CLEAN_USER_PROMPT, {"user_message": req.message})
31
  # req.message = self.llm_caller(clean_user_message_prompt)
32
  self.memory.append_history(req.name + ': ' + req.message)
 
26
  self.memory.append_history(f"主持人:天亮了!昨天晚上的信息是: {req.message}")
27
  elif req.status == STATUS_DISCUSS: # 发言环节
28
  if req.name:
29
+ # 可以使用模型来过滤掉玩家的注入消息,也可以换一个小模型,实际使用需要考虑对memory加锁,避免interact的时候丢失消息
30
  # clean_user_message_prompt = format_prompt(CLEAN_USER_PROMPT, {"user_message": req.message})
31
  # req.message = self.llm_caller(clean_user_message_prompt)
32
  self.memory.append_history(req.name + ': ' + req.message)
werewolf/witch/witch_agent.py CHANGED
@@ -35,7 +35,7 @@ class WitchAgent(BasicRoleAgent):
35
  elif req.status == STATUS_DISCUSS: # 发言环节
36
  if req.name:
37
  # 其他玩家发言
38
- # 可以使用模型来过滤掉玩家的注入消息,也可以换一个小模型
39
  # clean_user_message_prompt = format_prompt(CLEAN_USER_PROMPT, {"user_message": req.message})
40
  # req.message = self.llm_caller(clean_user_message_prompt)
41
  self.memory.append_history(req.name + ': ' + req.message)
 
35
  elif req.status == STATUS_DISCUSS: # 发言环节
36
  if req.name:
37
  # 其他玩家发言
38
+ # 可以使用模型来过滤掉玩家的注入消息,也可以换一个小模型,实际使用需要考虑对memory加锁,避免interact的时候丢失消息
39
  # clean_user_message_prompt = format_prompt(CLEAN_USER_PROMPT, {"user_message": req.message})
40
  # req.message = self.llm_caller(clean_user_message_prompt)
41
  self.memory.append_history(req.name + ': ' + req.message)
werewolf/wolf/wolf_agent.py CHANGED
@@ -41,7 +41,7 @@ class WolfAgent(BasicRoleAgent):
41
  elif req.status == STATUS_DISCUSS: # 发言环节
42
  if req.name:
43
  # 其他玩家发言
44
- # 可以使用模型来过滤掉玩家的注入消息,也可以换一个小模型
45
  # clean_user_message_prompt = format_prompt(CLEAN_USER_PROMPT, {"user_message": req.message})
46
  # req.message = self.llm_caller(clean_user_message_prompt)
47
  self.memory.append_history(req.name + ': ' + req.message)
 
41
  elif req.status == STATUS_DISCUSS: # 发言环节
42
  if req.name:
43
  # 其他玩家发言
44
+ # 可以使用模型来过滤掉玩家的注入消息,也可以换一个小模型,实际使用需要考虑对memory加锁,避免interact的时候丢失消息
45
  # clean_user_message_prompt = format_prompt(CLEAN_USER_PROMPT, {"user_message": req.message})
46
  # req.message = self.llm_caller(clean_user_message_prompt)
47
  self.memory.append_history(req.name + ': ' + req.message)