add pydantic v2 support and change role's private fields into public

examples/agent_creator.py

@@ -17,7 +17,7 @@ MULTI_ACTION_AGENT_CODE_EXAMPLE = EXAMPLE_CODE_FILE.read_text()
 
 
 class CreateAgent(Action):
-    PROMPT_TEMPLATE = """
+    PROMPT_TEMPLATE: str = """
     ### BACKGROUND
     You are using an agent framework called metagpt to write agents capable of different actions,
     the usage of metagpt can be illustrated by the following example:
@@ -64,9 +64,9 @@ class AgentCreator(Role):
         self._init_actions([CreateAgent])
 
     async def _act(self) -> Message:
-        logger.info(f"{self._setting}: to do {self._rc.todo}({self._rc.todo.name})")
-        todo = self._rc.todo
-        msg = self._rc.memory.get()[-1]
+        logger.info(f"{self._setting}: to do {self.rc.todo}({self.rc.todo.name})")
+        todo = self.rc.todo
+        msg = self.rc.memory.get()[-1]
 
         instruction = msg.content
         code_text = await CreateAgent().run(example=self.agent_template, instruction=instruction)

examples/build_customized_agent.py

@@ -16,7 +16,7 @@ from metagpt.schema import Message
 
 
 class SimpleWriteCode(Action):
-    PROMPT_TEMPLATE = """
+    PROMPT_TEMPLATE: str = """
     Write a python function that can {instruction} and provide two runnnable test cases.
     Return ```python your_code_here ``` with NO other texts,
     your code:
@@ -60,8 +60,8 @@ class SimpleCoder(Role):
         self._init_actions([SimpleWriteCode])
 
     async def _act(self) -> Message:
-        logger.info(f"{self._setting}: to do {self._rc.todo}({self._rc.todo.name})")
-        todo = self._rc.todo  # todo will be SimpleWriteCode()
+        logger.info(f"{self._setting}: to do {self.rc.todo}({self.rc.todo.name})")
+        todo = self.rc.todo  # todo will be SimpleWriteCode()
 
         msg = self.get_memories(k=1)[0]  # find the most recent messages
         code_text = await todo.run(msg.content)
@@ -80,16 +80,16 @@ class RunnableCoder(Role):
         self._set_react_mode(react_mode=RoleReactMode.BY_ORDER.value)
 
     async def _act(self) -> Message:
-        logger.info(f"{self._setting}: to do {self._rc.todo}({self._rc.todo.name})")
+        logger.info(f"{self._setting}: to do {self.rc.todo}({self.rc.todo.name})")
         # By choosing the Action by order under the hood
         # todo will be first SimpleWriteCode() then SimpleRunCode()
-        todo = self._rc.todo
+        todo = self.rc.todo
 
         msg = self.get_memories(k=1)[0]  # find the most k recent messages
         result = await todo.run(msg.content)
 
         msg = Message(content=result, role=self.profile, cause_by=type(todo))
-        self._rc.memory.add(msg)
+        self.rc.memory.add(msg)
         return msg
 
 

examples/build_customized_multi_agents.py

@@ -22,7 +22,7 @@ def parse_code(rsp):
 
 
 class SimpleWriteCode(Action):
-    PROMPT_TEMPLATE = """
+    PROMPT_TEMPLATE: str = """
     Write a python function that can {instruction}.
     Return ```python your_code_here ``` with NO other texts,
     your code:
@@ -50,7 +50,7 @@ class SimpleCoder(Role):
 
 
 class SimpleWriteTest(Action):
-    PROMPT_TEMPLATE = """
+    PROMPT_TEMPLATE: str = """
     Context: {context}
     Write {k} unit tests using pytest for the given function, assuming you have imported it.
     Return ```python your_code_here ``` with NO other texts,
@@ -80,8 +80,8 @@ class SimpleTester(Role):
         self._watch([SimpleWriteCode, SimpleWriteReview])  # feel free to try this too
 
     async def _act(self) -> Message:
-        logger.info(f"{self._setting}: to do {self._rc.todo}({self._rc.todo.name})")
-        todo = self._rc.todo
+        logger.info(f"{self._setting}: to do {self.rc.todo}({self.rc.todo.name})")
+        todo = self.rc.todo
 
         # context = self.get_memories(k=1)[0].content # use the most recent memory as context
         context = self.get_memories()  # use all memories as context
@@ -93,7 +93,7 @@ class SimpleTester(Role):
 
 
 class SimpleWriteReview(Action):
-    PROMPT_TEMPLATE = """
+    PROMPT_TEMPLATE: str = """
     Context: {context}
     Review the test cases and provide one critical comments:
     """

examples/debate.py

@@ -59,12 +59,12 @@ class Debator(Role):
     async def _observe(self) -> int:
         await super()._observe()
         # accept messages sent (from opponent) to self, disregard own messages from the last round
-        self._rc.news = [msg for msg in self._rc.news if msg.send_to == {self.name}]
-        return len(self._rc.news)
+        self.rc.news = [msg for msg in self.rc.news if msg.send_to == {self.name}]
+        return len(self.rc.news)
 
     async def _act(self) -> Message:
-        logger.info(f"{self._setting}: to do {self._rc.todo}({self._rc.todo.name})")
-        todo = self._rc.todo  # An instance of SpeakAloud
+        logger.info(f"{self._setting}: to do {self.rc.todo}({self.rc.todo.name})")
+        todo = self.rc.todo  # An instance of SpeakAloud
 
         memories = self.get_memories()
         context = "\n".join(f"{msg.sent_from}: {msg.content}" for msg in memories)
@@ -79,7 +79,7 @@ class Debator(Role):
             sent_from=self.name,
             send_to=self.opponent_name,
         )
-        self._rc.memory.add(msg)
+        self.rc.memory.add(msg)
 
         return msg