Update app.py

app.py

@@ -47,6 +47,7 @@ class SpyAgent(BasicAgent):
     def interact(self, req=AgentReq) -> AgentResp:
         logger.info("spy interact: {}".format(req))
         if req.status == STATUS_ROUND:
+            # 填充 DESC_PROMPT 模板
             prompt = format_prompt(DESC_PROMPT,
                                    {"name": self.memory.load_variable("name"),
                                     "word": self.memory.load_variable("word"),
@@ -55,12 +56,15 @@ class SpyAgent(BasicAgent):
             logger.info("prompt:" + prompt)
             result = self.llm_caller(prompt)
             logger.info("spy interact result: {}".format(result))
+            # 将描述发送到公共聊天区域
+            self.memory.append_history(f"{self.memory.load_variable('name')}: {result}")
             return AgentResp(success=True, result=result, errMsg=None)
 
         elif req.status == STATUS_VOTE:
             self.memory.append_history('主持人: 到了投票的时候了。每个人，请指向你认为可能是卧底的人。')
             choices = [name for name in req.message.split(",") if name != self.memory.load_variable("name")]  # 排除自己
             self.memory.set_variable("choices", choices)
+            # 填充 VOTE_PROMPT 模板
             prompt = format_prompt(VOTE_PROMPT, {"name": self.memory.load_variable("name"),
                                                  "choices": choices,
                                                  "history": "\n".join(self.memory.load_history())
@@ -68,6 +72,8 @@ class SpyAgent(BasicAgent):
             logger.info("prompt:" + prompt)
             result = self.llm_caller(prompt)
             logger.info("spy interact result: {}".format(result))
+            # 将投票结果发送到公共聊天区域
+            self.memory.append_history(f"{self.memory.load_variable('name')}: 我投票给 {result}。")
             return AgentResp(success=True, result=result, errMsg=None)
         else:
             raise NotImplementedError