apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-27 14:25:20 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update log_parse_json and custom werewolf num in prompt

Browse source
This commit is contained in:
better629 2024-04-10 17:42:56 +08:00
parent c43a1cada7
commit ade46df232
5 changed files with 19 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

19
metagpt/ext/werewolf/actions/common_actions.py
View file

@ -11,6 +11,14 @@ from metagpt.logs import logger
from metagpt.utils.common import parse_json_code_block
def log_and_parse_json(name: str, rsp: str) -> dict:
rsp = rsp.replace("\n", " ")
logger.debug(f"{name} result: {rsp}")
json_blocks = parse_json_code_block(rsp)
rsp_json = json.loads(json_blocks[0])
return rsp_json
class Speak(Action):
"""Action: Any speak action in a game"""
@ -66,8 +74,7 @@ class Speak(Action):
)
rsp = await self._aask(prompt)
rsp = rsp.replace("\n", " ")
rsp_json = json.loads(rsp)
rsp_json = log_and_parse_json(self.name, rsp)
return rsp_json["RESPONSE"]
@ -183,8 +190,7 @@ class NighttimeWhispers(Action):
)
rsp = await self._aask(prompt)
rsp = rsp.replace("\n", " ")
rsp_json = json.loads(rsp)
rsp_json = log_and_parse_json(self.name, rsp)
return f"{self.name} " + rsp_json["RESPONSE"]
@ -229,9 +235,6 @@ class Reflect(Action):
)
rsp = await self._aask(prompt)
rsp = rsp.replace("\n", " ")
logger.debug(f"{self.name} result: {rsp}")
json_blocks = parse_json_code_block(rsp)
rsp_json = json.loads(json_blocks[0])
rsp_json = log_and_parse_json(self.name, rsp)
return json.dumps(rsp_json["REFLECTION"])

3
metagpt/ext/werewolf/actions/moderator_actions.py
View file

@ -11,7 +11,8 @@ class InstructSpeak(Action):
)
content = instruction_info["content"]
if "{living_players}" in content and "{werewolf_players}" in content:
content = content.format(living_players=living_players, werewolf_players=werewolf_players)
content = content.format(living_players=living_players, werewolf_players=werewolf_players,
werewolf_num=len(werewolf_players))
if "{living_players}" in content:
content = content.format(living_players=living_players)
if "{werewolf_players}" in content: