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 player profile with const

Browse source
This commit is contained in:
better629 2024-04-10 20:46:59 +08:00
parent a466bc9243
commit fe0e62806d
3 changed files with 16 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

7
examples/werewolf_game/evals/eval.py
View file

@ -17,6 +17,7 @@ from tqdm import tqdm
from utils import Utils
from metagpt.const import DEFAULT_WORKSPACE_ROOT, METAGPT_ROOT
from metagpt.environment.werewolf.const import RoleType
class Vote:
@ -92,20 +93,20 @@ class Vote:
# find all werewolves
werewolves = []
for match in pattern.finditer(text):
if match.group(2) == "Werewolf":
if match.group(2) == RoleType.WEREWOLF.value:
werewolves.append(match.group(1))
# find all non_werewolves
non_werewolves = []
for match in pattern.finditer(text):
if match.group(2) != "Werewolf":
if match.group(2) != RoleType.WEREWOLF.value:
non_werewolves.append(match.group(1))
num_non_werewolves = len(non_werewolves)
# count players other than werewolves made the correct votes
correct_votes = 0
for match in pattern.finditer(text):
if match.group(2) != "Werewolf" and match.group(3) in werewolves:
if match.group(2) != RoleType.WEREWOLF.value and match.group(3) in werewolves:
correct_votes += 1
# cal the rateability of non_werewolves