Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
| 56 |
-
prompt = format_prompt(DESC_PROMPT,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
logger.info("prompt:" + prompt)
|
| 58 |
result = self.llm_caller(prompt)
|
| 59 |
logger.info("spy interact result: {}".format(result))
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 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 |
-
|
| 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 |
-
|
| 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
|