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
|