Spaces:
Running
Running
茂道
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)
|