zishan111 commited on
Commit
0f334b1
·
verified ·
1 Parent(s): 4668386

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -53,19 +53,19 @@ class SpyAgent(BasicAgent):
53
  logger.info("spy interact: {}".format(req))
54
  if req.status == STATUS_ROUND:
55
  # 填充 DESC_PROMPT 模板
56
- prompt = format_prompt(DESC_PROMPT,
57
- {"name": self.memory.load_variable("name"),
58
- "word": self.memory.load_variable("word"),)
59
  logger.info("prompt:" + prompt)
60
  result = self.llm_caller(prompt)
61
  logger.info("spy interact result: {}".format(result))
62
  # 将描述发送到公共聊天区域
63
- self.memory.append_history(f"{self.memory.load_variable('name')}: {result}")
 
64
  # 如果描述准确
65
  if result == self.memory.load_variable("word"):
66
  self.memory.append_history(f"(能准确描述你的词汇的发言内容)")
67
  # 发言完毕
68
- self.memory.append_history(f"{self.memory.load_variable('name')}发言完毕。")
 
69
  return AgentResp(success=True, result=result, errMsg=None)
70
 
71
  elif req.status == STATUS_VOTE:
 
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(
68
+ f"{self.memory.load_variable('name')}发言完毕。")
69
  return AgentResp(success=True, result=result, errMsg=None)
70
 
71
  elif req.status == STATUS_VOTE: