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

modify add action to set action

Browse source
This commit is contained in:
geekan 2024-01-10 17:54:13 +08:00
parent 07d34bda7a
commit 91e6564586
19 changed files with 34 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/metagpt/serialize_deserialize/test_serdeser_base.py
View file

@ -67,7 +67,7 @@ class RoleA(Role):
def __init__(self, **kwargs):
super(RoleA, self).__init__(**kwargs)
self.add_actions([ActionPass])
self.set_actions([ActionPass])
self._watch([UserRequirement])
@ -79,7 +79,7 @@ class RoleB(Role):
def __init__(self, **kwargs):
super(RoleB, self).__init__(**kwargs)
self.add_actions([ActionOK, ActionRaise])
self.set_actions([ActionOK, ActionRaise])
self._watch([ActionPass])
self.rc.react_mode = RoleReactMode.BY_ORDER
@ -92,7 +92,7 @@ class RoleC(Role):
def __init__(self, **kwargs):
super(RoleC, self).__init__(**kwargs)
self.add_actions([ActionOK, ActionRaise])
self.set_actions([ActionOK, ActionRaise])
self._watch([UserRequirement])
self.rc.react_mode = RoleReactMode.BY_ORDER
self.rc.memory.ignore_id = True

8
tests/metagpt/test_role.py
View file

@ -33,7 +33,7 @@ class MockAction(Action):
class MockRole(Role):
def __init__(self, name="", profile="", goal="", constraints="", desc=""):
super().__init__(name=name, profile=profile, goal=goal, constraints=constraints, desc=desc)
self.add_actions([MockAction()])
self.set_actions([MockAction()])
def test_basic():
@ -111,7 +111,7 @@ async def test_send_to():
def test_init_action():
role = Role()
role.add_actions([MockAction, MockAction])
role.set_actions([MockAction, MockAction])
assert len(role.actions) == 2
@ -127,7 +127,7 @@ async def test_recover():
role.publish_message(None)
role.llm = mock_llm
role.add_actions([MockAction, MockAction])
role.set_actions([MockAction, MockAction])
role.recovered = True
role.latest_observed_msg = Message(content="recover_test")
role.rc.state = 0
@ -144,7 +144,7 @@ async def test_think_act():
mock_llm.aask.side_effect = ["ok"]
role = Role()
role.add_actions([MockAction])
role.set_actions([MockAction])
await role.think()
role.rc.memory.add(Message("run"))
assert len(role.get_memories()) == 1