新修改。

examples/agent_creator.py

@@ -5,18 +5,18 @@ Author: garylin2099
 '''
 import re
 
-from metagpt.const import PROJECT_ROOT, WORKSPACE_ROOT
 from metagpt.actions import Action
+from metagpt.const import PROJECT_ROOT, WORKSPACE_ROOT
+from metagpt.logs import logger
 from metagpt.roles import Role
 from metagpt.schema import Message
-from metagpt.logs import logger
 
 with open(PROJECT_ROOT / "examples/build_customized_agent.py", "r") as f:
     # use official example script to guide AgentCreator
     MULTI_ACTION_AGENT_CODE_EXAMPLE = f.read()
 
-class CreateAgent(Action):
 
+class CreateAgent(Action):
     PROMPT_TEMPLATE = """
     ### BACKGROUND
     You are using an agent framework called metagpt to write agents capable of different actions,
@@ -34,7 +34,6 @@ class CreateAgent(Action):
     """
 
     async def run(self, example: str, instruction: str):
-
         prompt = self.PROMPT_TEMPLATE.format(example=example, instruction=instruction)
         # logger.info(prompt)
 
@@ -53,13 +52,14 @@ class CreateAgent(Action):
             f.write(code_text)
         return code_text
 
+
 class AgentCreator(Role):
     def __init__(
-        self,
-        name: str = "Matrix",
-        profile: str = "AgentCreator",
-        agent_template: str = MULTI_ACTION_AGENT_CODE_EXAMPLE,
-        **kwargs,
+            self,
+            name: str = "Matrix",
+            profile: str = "AgentCreator",
+            agent_template: str = MULTI_ACTION_AGENT_CODE_EXAMPLE,
+            **kwargs,
     ):
         super().__init__(name, profile, **kwargs)
         self._init_actions([CreateAgent])
@@ -76,11 +76,12 @@ class AgentCreator(Role):
 
         return msg
 
+
 if __name__ == "__main__":
     import asyncio
 
-    async def main():
 
+    async def main():
         agent_template = MULTI_ACTION_AGENT_CODE_EXAMPLE
 
         creator = AgentCreator(agent_template=agent_template)
@@ -97,4 +98,5 @@ if __name__ == "__main__":
         """
         await creator.run(msg)
 
+
     asyncio.run(main())

examples/build_customized_agent.py

@@ -3,19 +3,19 @@ Filename: MetaGPT/examples/build_customized_agent.py
 Created Date: Tuesday, September 19th 2023, 6:52:25 pm
 Author: garylin2099
 '''
+import asyncio
 import re
 import subprocess
-import asyncio
 
 import fire
 
 from metagpt.actions import Action
+from metagpt.logs import logger
 from metagpt.roles import Role
 from metagpt.schema import Message
-from metagpt.logs import logger
+
 
 class SimpleWriteCode(Action):
-
     PROMPT_TEMPLATE = """
     Write a python function that can {instruction} and provide two runnnable test cases.
     Return ```python your_code_here ``` with NO other texts,
@@ -35,7 +35,6 @@ class SimpleWriteCode(Action):
         super().__init__(name, context, llm)
 
     async def run(self, instruction: str):
-
         prompt = self.PROMPT_TEMPLATE.format(instruction=instruction)
 
         rsp = await self._aask(prompt)
@@ -51,6 +50,7 @@ class SimpleWriteCode(Action):
         code_text = match.group(1) if match else rsp
         return code_text
 
+
 class SimpleRunCode(Action):
     def __init__(self, name="SimpleRunCode", context=None, llm=None):
         super().__init__(name, context, llm)
@@ -61,12 +61,13 @@ class SimpleRunCode(Action):
         logger.info(f"{code_result=}")
         return code_result
 
+
 class SimpleCoder(Role):
     def __init__(
-        self,
-        name: str = "Alice",
-        profile: str = "SimpleCoder",
-        **kwargs,
+            self,
+            name: str = "Alice",
+            profile: str = "SimpleCoder",
+            **kwargs,
     ):
         super().__init__(name, profile, **kwargs)
         self._init_actions([SimpleWriteCode])
@@ -75,7 +76,7 @@ class SimpleCoder(Role):
         logger.info(f"{self._setting}: ready to {self._rc.todo}")
         todo = self._rc.todo
 
-        msg = self._rc.memory.get()[-1] # retrieve the latest memory
+        msg = self._rc.memory.get()[-1]  # retrieve the latest memory
         instruction = msg.content
 
         code_text = await SimpleWriteCode().run(instruction)
@@ -83,12 +84,13 @@ class SimpleCoder(Role):
 
         return msg
 
+
 class RunnableCoder(Role):
     def __init__(
-        self,
-        name: str = "Alice",
-        profile: str = "RunnableCoder",
-        **kwargs,
+            self,
+            name: str = "Alice",
+            profile: str = "RunnableCoder",
+            **kwargs,
     ):
         super().__init__(name, profile, **kwargs)
         self._init_actions([SimpleWriteCode, SimpleRunCode])
@@ -128,6 +130,7 @@ class RunnableCoder(Role):
             await self._act()
         return Message(content="All job done", role=self.profile)
 
+
 def main(msg="write a function that calculates the sum of a list"):
     # role = SimpleCoder()
     role = RunnableCoder()
@@ -135,5 +138,6 @@ def main(msg="write a function that calculates the sum of a list"):
     result = asyncio.run(role.run(msg))
     logger.info(result)
 
+
 if __name__ == '__main__':
     fire.Fire(main)

examples/debate.py

@@ -5,13 +5,15 @@ Author: garylin2099
 '''
 import asyncio
 import platform
+
 import fire
 
-from metagpt.software_company import SoftwareCompany
 from metagpt.actions import Action, BossRequirement
+from metagpt.logs import logger
 from metagpt.roles import Role
 from metagpt.schema import Message
-from metagpt.logs import logger
+from metagpt.software_company import SoftwareCompany
+
 
 class ShoutOut(Action):
     """Action: Shout out loudly in a debate (quarrel)"""
@@ -31,7 +33,6 @@ class ShoutOut(Action):
         super().__init__(name, context, llm)
 
     async def run(self, context: str, name: str, opponent_name: str):
-
         prompt = self.PROMPT_TEMPLATE.format(context=context, name=name, opponent_name=opponent_name)
         # logger.info(prompt)
 
@@ -39,12 +40,13 @@ class ShoutOut(Action):
 
         return rsp
 
+
 class Trump(Role):
     def __init__(
-        self,
-        name: str = "Trump",
-        profile: str = "Republican",
-        **kwargs,
+            self,
+            name: str = "Trump",
+            profile: str = "Republican",
+            **kwargs,
     ):
         super().__init__(name, profile, **kwargs)
         self._init_actions([ShoutOut])
@@ -55,7 +57,7 @@ class Trump(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]  
+        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:
@@ -79,12 +81,13 @@ class Trump(Role):
 
         return msg
 
+
 class Biden(Role):
     def __init__(
-        self,
-        name: str = "Biden",
-        profile: str = "Democrat",
-        **kwargs,
+            self,
+            name: str = "Biden",
+            profile: str = "Democrat",
+            **kwargs,
     ):
         super().__init__(name, profile, **kwargs)
         self._init_actions([ShoutOut])
@@ -120,6 +123,7 @@ class Biden(Role):
 
         return msg
 
+
 async def startup(idea: str, investment: float = 3.0, n_round: int = 5,
                   code_review: bool = False, run_tests: bool = False):
     """We reuse the startup paradigm for roles to interact with each other.

examples/search_with_specific_engine.py

@@ -6,11 +6,12 @@ from metagpt.tools import SearchEngineType
 
 async def main():
     # Serper API
-    #await Searcher(engine = SearchEngineType.SERPER_GOOGLE).run(["What are some good sun protection products?","What are some of the best beaches?"])
+    # await Searcher(engine = SearchEngineType.SERPER_GOOGLE).run(["What are some good sun protection products?","What are some of the best beaches?"])
     # SerpAPI
-    #await Searcher(engine=SearchEngineType.SERPAPI_GOOGLE).run("What are the best ski brands for skiers?")
+    # await Searcher(engine=SearchEngineType.SERPAPI_GOOGLE).run("What are the best ski brands for skiers?")
     # Google API
     await Searcher(engine=SearchEngineType.DIRECT_GOOGLE).run("What are the most interesting human facts?")
 
+
 if __name__ == '__main__':
     asyncio.run(main())

examples/sk_agent.py

@@ -9,7 +9,6 @@ import asyncio
 
 from semantic_kernel.core_skills import FileIOSkill, MathSkill, TextSkill, TimeSkill
 from semantic_kernel.planning import SequentialPlanner
-
 # from semantic_kernel.planning import SequentialPlanner
 from semantic_kernel.planning.action_planner.action_planner import ActionPlanner
 

examples/use_off_the_shelf_agent.py

@@ -5,8 +5,9 @@ Author: garylin2099
 '''
 import asyncio
 
-from metagpt.roles.product_manager import ProductManager
 from metagpt.logs import logger
+from metagpt.roles.product_manager import ProductManager
+
 
 async def main():
     msg = "Write a PRD for a snake game"
@@ -14,5 +15,6 @@ async def main():
     result = await role.run(msg)
     logger.info(result.content[:100])
 
+
 if __name__ == '__main__':
     asyncio.run(main())

examples/write_tutorial.py

@@ -18,4 +18,3 @@ async def main():
 
 if __name__ == '__main__':
     asyncio.run(main())
-