update examples

examples/agent_creator.py

@@ -55,16 +55,13 @@ class CreateAgent(Action):
 
 
 class AgentCreator(Role):
-    def __init__(
-        self,
-        name: str = "Matrix",
-        profile: str = "AgentCreator",
-        agent_template: str = MULTI_ACTION_AGENT_CODE_EXAMPLE,
-        **kwargs,
-    ):
-        super().__init__(name, profile, **kwargs)
+    name: str = "Matrix"
+    profile: str = "AgentCreator"
+    agent_template: str = MULTI_ACTION_AGENT_CODE_EXAMPLE
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
         self._init_actions([CreateAgent])
-        self.agent_template = agent_template
 
     async def _act(self) -> Message:
         logger.info(f"{self._setting}: ready to {self._rc.todo}")
@@ -86,10 +83,6 @@ if __name__ == "__main__":
 
         creator = AgentCreator(agent_template=agent_template)
 
-        # msg = """Write an agent called SimpleTester that will take any code snippet (str)
-        #     and return a testing code (str) for testing
-        #     the given code snippet. Use pytest as the testing framework."""
-
         msg = """
         Write an agent called SimpleTester that will take any code snippet (str) and do the following:
         1. write a testing code (str) for testing the given code snippet, save the testing code as a .py file in the current working directory;

examples/build_customized_agent.py

@@ -10,9 +10,8 @@ import subprocess
 import fire
 
 from metagpt.actions import Action
-from metagpt.llm import LLM
 from metagpt.logs import logger
-from metagpt.roles import Role
+from metagpt.roles.role import Role, RoleReactMode
 from metagpt.schema import Message
 
 
@@ -23,8 +22,7 @@ class SimpleWriteCode(Action):
     your code:
     """
 
-    def __init__(self, name: str = "SimpleWriteCode", context=None, llm: LLM = None):
-        super().__init__(name, context, llm)
+    name: str = "SimpleWriteCode"
 
     async def run(self, instruction: str):
         prompt = self.PROMPT_TEMPLATE.format(instruction=instruction)
@@ -44,8 +42,7 @@ class SimpleWriteCode(Action):
 
 
 class SimpleRunCode(Action):
-    def __init__(self, name: str = "SimpleRunCode", context=None, llm: LLM = None):
-        super().__init__(name, context, llm)
+    name: str = "SimpleRunCode"
 
     async def run(self, code_text: str):
         result = subprocess.run(["python3", "-c", code_text], capture_output=True, text=True)
@@ -55,13 +52,11 @@ class SimpleRunCode(Action):
 
 
 class SimpleCoder(Role):
-    def __init__(
-        self,
-        name: str = "Alice",
-        profile: str = "SimpleCoder",
-        **kwargs,
-    ):
-        super().__init__(name, profile, **kwargs)
+    name: str = "Alice"
+    profile: str = "SimpleCoder"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
         self._init_actions([SimpleWriteCode])
 
     async def _act(self) -> Message:
@@ -76,15 +71,13 @@ class SimpleCoder(Role):
 
 
 class RunnableCoder(Role):
-    def __init__(
-        self,
-        name: str = "Alice",
-        profile: str = "RunnableCoder",
-        **kwargs,
-    ):
-        super().__init__(name, profile, **kwargs)
+    name: str = "Alice"
+    profile: str = "RunnableCoder"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
         self._init_actions([SimpleWriteCode, SimpleRunCode])
-        self._set_react_mode(react_mode="by_order")
+        self._set_react_mode(react_mode=RoleReactMode.BY_ORDER.value)
 
     async def _act(self) -> Message:
         logger.info(f"{self._setting}: ready to {self._rc.todo}")

examples/build_customized_multi_agents.py

@@ -8,7 +8,6 @@ import re
 import fire
 
 from metagpt.actions import Action, UserRequirement
-from metagpt.llm import LLM
 from metagpt.logs import logger
 from metagpt.roles import Role
 from metagpt.schema import Message
@@ -28,9 +27,7 @@ class SimpleWriteCode(Action):
     Return ```python your_code_here ``` with NO other texts,
     your code:
     """
-
-    def __init__(self, name: str = "SimpleWriteCode", context=None, llm: LLM = None):
-        super().__init__(name, context, llm)
+    name: str = "SimpleWriteCode"
 
     async def run(self, instruction: str):
         prompt = self.PROMPT_TEMPLATE.format(instruction=instruction)
@@ -43,13 +40,11 @@ class SimpleWriteCode(Action):
 
 
 class SimpleCoder(Role):
-    def __init__(
-        self,
-        name: str = "Alice",
-        profile: str = "SimpleCoder",
-        **kwargs,
-    ):
-        super().__init__(name, profile, **kwargs)
+    name: str = "Alice"
+    profile: str = "SimpleCoder"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
         self._watch([UserRequirement])
         self._init_actions([SimpleWriteCode])
 
@@ -62,8 +57,7 @@ class SimpleWriteTest(Action):
     your code:
     """
 
-    def __init__(self, name: str = "SimpleWriteTest", context=None, llm: LLM = None):
-        super().__init__(name, context, llm)
+    name: str = "SimpleWriteTest"
 
     async def run(self, context: str, k: int = 3):
         prompt = self.PROMPT_TEMPLATE.format(context=context, k=k)
@@ -76,13 +70,11 @@ class SimpleWriteTest(Action):
 
 
 class SimpleTester(Role):
-    def __init__(
-        self,
-        name: str = "Bob",
-        profile: str = "SimpleTester",
-        **kwargs,
-    ):
-        super().__init__(name, profile, **kwargs)
+    name: str = "Bob"
+    profile: str = "SimpleTester"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
         self._init_actions([SimpleWriteTest])
         # self._watch([SimpleWriteCode])
         self._watch([SimpleWriteCode, SimpleWriteReview])  # feel free to try this too
@@ -106,8 +98,7 @@ class SimpleWriteReview(Action):
     Review the test cases and provide one critical comments:
     """
 
-    def __init__(self, name: str = "SimpleWriteReview", context=None, llm: LLM = None):
-        super().__init__(name, context, llm)
+    name: str = "SimpleWriteReview"
 
     async def run(self, context: str):
         prompt = self.PROMPT_TEMPLATE.format(context=context)
@@ -118,13 +109,11 @@ class SimpleWriteReview(Action):
 
 
 class SimpleReviewer(Role):
-    def __init__(
-        self,
-        name: str = "Charlie",
-        profile: str = "SimpleReviewer",
-        **kwargs,
-    ):
-        super().__init__(name, profile, **kwargs)
+    name: str = "Charlie"
+    profile: str = "SimpleReviewer"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
         self._init_actions([SimpleWriteReview])
         self._watch([SimpleWriteTest])
 
@@ -147,7 +136,7 @@ async def main(
     )
 
     team.invest(investment=investment)
-    team.start_project(idea)
+    team.run_project(idea)
     await team.run(n_round=n_round)