zishan111 commited on
Commit
79700a3
·
verified ·
1 Parent(s): 2bafd42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -14
app.py CHANGED
@@ -52,34 +52,38 @@ class SpyAgent(BasicAgent):
52
  def interact(self, req=AgentReq) -> AgentResp:
53
  logger.info("spy interact: {}".format(req))
54
  if req.status == STATUS_ROUND:
55
- # 填充 DESC_PROMPT 模板
56
- prompt = format_prompt(DESC_PROMPT,{"name": self.memory.load_variable("name"),"word": self.memory.load_variable("word"),)
 
 
 
 
57
  logger.info("prompt:" + prompt)
58
  result = self.llm_caller(prompt)
59
  logger.info("spy interact result: {}".format(result))
60
- # 将描述发送到公共聊天区域
61
- self.memory.append_history(
62
- f"{self.memory.load_variable('name')}: {result}")
63
- # 如果描述准确
64
- if result == self.memory.load_variable("word"):
65
- self.memory.append_history(f"(能准确描述你的词汇的发言内容)")
66
- # 发言完毕
67
- self.memory.append_history(f"{self.memory.load_variable('name')}发言完毕。")
68
- return AgentResp(success=True, result=result, errMsg=None)
69
 
70
  elif req.status == STATUS_VOTE:
71
  self.memory.append_history('主持人: 到了投票的时候了。每个人,请指向你认为可能是卧底的人。')
72
  choices = [name for name in req.message.split(",") if name != self.memory.load_variable("name")] # 排除自己
73
  self.memory.set_variable("choices", choices)
74
- # 填充 VOTE_PROMPT 模板
75
  prompt = format_prompt(VOTE_PROMPT, {"name": self.memory.load_variable("name"),
76
  "choices": choices,
77
  "history": "\n".join(self.memory.load_history())
 
78
  logger.info("prompt:" + prompt)
79
  result = self.llm_caller(prompt)
80
  logger.info("spy interact result: {}".format(result))
81
- # 将投票结果发送到公共聊天区域
82
- self.memory.append_history(f"{self.memory.load_variable('name')}: 我投票给 {result}。")
83
  return AgentResp(success=True, result=result, errMsg=None)
84
  else:
85
  raise NotImplementedError
 
52
  def interact(self, req=AgentReq) -> AgentResp:
53
  logger.info("spy interact: {}".format(req))
54
  if req.status == STATUS_ROUND:
55
+
56
+ prompt = format_prompt(DESC_PROMPT,
57
+ {"name": self.memory.load_variable("name"),
58
+ "word": self.memory.load_variable("word"),
59
+ "history": "\n".join(self.memory.load_history())
60
+ })
61
  logger.info("prompt:" + prompt)
62
  result = self.llm_caller(prompt)
63
  logger.info("spy interact result: {}".format(result))
64
+
65
+
66
+
67
+
68
+
69
+
70
+
71
+ return AgentResp(success=True, result=result, errMsg=None)
 
72
 
73
  elif req.status == STATUS_VOTE:
74
  self.memory.append_history('主持人: 到了投票的时候了。每个人,请指向你认为可能是卧底的人。')
75
  choices = [name for name in req.message.split(",") if name != self.memory.load_variable("name")] # 排除自己
76
  self.memory.set_variable("choices", choices)
77
+
78
  prompt = format_prompt(VOTE_PROMPT, {"name": self.memory.load_variable("name"),
79
  "choices": choices,
80
  "history": "\n".join(self.memory.load_history())
81
+ })
82
  logger.info("prompt:" + prompt)
83
  result = self.llm_caller(prompt)
84
  logger.info("spy interact result: {}".format(result))
85
+
86
+
87
  return AgentResp(success=True, result=result, errMsg=None)
88
  else:
89
  raise NotImplementedError