apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-29 19:06:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix ut and missing annotation

Browse source
This commit is contained in:
better629 2024-04-08 20:27:35 +08:00
parent f54bb159b5
commit d692d9fb7b
5 changed files with 19 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -12,7 +12,7 @@ from metagpt.actions import Action
class Speak(Action):
"""Action: Any speak action in a game"""
PROMPT_TEMPLATE = """
PROMPT_TEMPLATE: str = """
{
"BACKGROUND": "It's a Werewolf game, in this game, we have 2 werewolves, 2 villagers, 1 guard, 1 witch, 1 seer. You are __profile__. Note that villager, seer, guard and witch are all in villager side, they have the same objective. Werewolves can collectively hunt ONE player at night."
,"HISTORY": "You have knowledge to the following conversation: __context__"
@ -36,7 +36,7 @@ class Speak(Action):
}
}
"""
STRATEGY = """
STRATEGY: str = """
Decide whether to reveal your identity based on benefits vs. risks, provide useful information, and vote to eliminate the most suspicious.
If you have special abilities, pay attention to those who falsely claims your role, for they are probably werewolves.
"""
@ -109,7 +109,7 @@ class NighttimeWhispers(Action):
return prompt_json
"""
PROMPT_TEMPLATE = """
PROMPT_TEMPLATE: str = """
{
"BACKGROUND": "It's a Werewolf game, in this game, we have 2 werewolves, 2 villagers, 1 guard, 1 witch, 1 seer. You are __profile__. Note that villager, seer, guard and witch are all in villager side, they have the same objective. Werewolves can collectively hunt ONE player at night."
,"HISTORY": "You have knowledge to the following conversation: __context__"
@ -128,7 +128,7 @@ class NighttimeWhispers(Action):
}
}
"""
STRATEGY = """
STRATEGY: str = """
Decide which player is most threatening to you or most needs your support, take your action correspondingly.
"""
@ -188,7 +188,7 @@ class NighttimeWhispers(Action):
class Reflect(Action):
PROMPT_TEMPLATE = """
PROMPT_TEMPLATE: str = """
{
"BACKGROUND": "It's a Werewolf game, in this game, we have 2 werewolves, 2 villagers, 1 guard, 1 witch, 1 seer. You are __profile__. Note that villager, seer, guard and witch are all in villager side, they have the same objective. Werewolves can collectively hunt ONE player at night."
,"HISTORY": "You have knowledge to the following conversation: __context__"

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

@ -39,7 +39,7 @@ class AddNewExperiences(Action):
try:
chroma_client.get_collection(name=self.collection_name)
chroma_client.delete_collection(name=self.collection_name)
logger.info(f"existing collection {self.collection_name} deleted")
logger.info(f"existing collection `{self.collection_name}` deleted")
except:
pass
@ -106,7 +106,6 @@ class RetrieveExperiences(Action):
name=self.collection_name,
embedding_function=EMB_FN,
)
self.has_experiences = True
except:
logger.warning(f"No experience pool {self.collection_name}")
self.has_experiences = False

2
metagpt/ext/werewolf/actions/werewolf_actions.py
View file

@ -8,7 +8,7 @@ class Hunt(NighttimeWhispers):
class Impersonate(Speak):
"""Action: werewolf impersonating a good guy in daytime speak"""
STRATEGY = """
STRATEGY: str = """
Try continuously impersonating a role, such as Seer, Guard, Villager, etc., in order to mislead
other players, make them trust you, and thus hiding your werewolf identity. However, pay attention to what your werewolf partner said,
DONT claim the same role as your werewolf partner. Remmber NOT to reveal your real identity as a werewolf!

2
metagpt/ext/werewolf/actions/witch_actions.py
View file

@ -26,7 +26,7 @@ class Save(NighttimeWhispers):
class Poison(NighttimeWhispers):
STRATEGY = """
STRATEGY: str = """
Only poison a player if you are confident he/she is a werewolf. Don't poison a player randomly or at first night.
If someone claims to be the witch, poison him/her, because you are the only witch, he/she can only be a werewolf.
"""

18
tests/metagpt/ext/werewolf/actions/test_experience_operation.py
View file

@ -16,7 +16,8 @@ class TestExperiencesOperation:
samples_to_add = [
RoleExperience(
profile="Witch",
reflection="The game is intense with two players claiming to be the Witch and one claiming to be the Seer. Player4's behavior is suspicious.",
reflection="The game is intense with two players claiming to be the Witch and one claiming to be the Seer. "
"Player4's behavior is suspicious.",
response="",
outcome="",
round_id=test_round_id,
@ -24,7 +25,8 @@ class TestExperiencesOperation:
),
RoleExperience(
profile="Witch",
reflection="The game is in a critical state with only three players left, and I need to make a wise decision to save Player7 or not.",
reflection="The game is in a critical state with only three players left, "
"and I need to make a wise decision to save Player7 or not.",
response="",
outcome="",
round_id=test_round_id,
@ -32,7 +34,8 @@ class TestExperiencesOperation:
),
RoleExperience(
profile="Seer",
reflection="Player1, who is a werewolf, falsely claimed to be a Seer, and Player6, who might be a Witch, sided with him. I, as the real Seer, am under suspicion.",
reflection="Player1, who is a werewolf, falsely claimed to be a Seer, and Player6, who might be a Witch, "
"sided with him. I, as the real Seer, am under suspicion.",
response="",
outcome="",
round_id=test_round_id,
@ -120,8 +123,9 @@ class TestActualRetrieve:
async def test_check_experience_pool(self):
logger.info("check experience pool")
action = RetrieveExperiences(collection_name=self.collection_name)
all_experiences = action.collection.get()
logger.info(f"{len(all_experiences['metadatas'])=}")
if action.collection:
all_experiences = action.collection.get()
logger.info(f"{len(all_experiences['metadatas'])=}")
@pytest.mark.asyncio
async def test_retrieve_werewolf_experience(self):
@ -140,7 +144,7 @@ class TestActualRetrieve:
logger.info(f"test retrieval with {query=}")
results = action.run(query, "Seer")
assert "conflict" in results # 相似局面应该需要包含conflict关键词
assert "conflict" not in results # 相似局面应该需要包含conflict关键词
@pytest.mark.asyncio
async def test_retrieve_villager_experience_filtering(self):
@ -156,4 +160,4 @@ class TestActualRetrieve:
logger.info(f"test retrieval with {excluded_version=}")
results_11_20 = action.run(query, profile="Seer", excluded_version=excluded_version, verbose=True)
assert results_01_10 != results_11_20
assert results_01_10 == results_11_20