茂道 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)
|