Merge pull request #886 from garylin2099/ci_dev

add ml example

examples/ci/machine_learning.py

@@ -0,0 +1,13 @@
+import asyncio
+
+from metagpt.roles.ci.code_interpreter import CodeInterpreter
+
+
+async def main(requirement: str):
+    role = CodeInterpreter(auto_run=True, use_tools=False)
+    await role.run(requirement)
+
+
+if __name__ == "__main__":
+    requirement = "Run data analysis on sklearn Wine recognition dataset, include a plot, and train a model to predict wine class (20% as validation), and show validation accuracy."
+    asyncio.run(main(requirement))

examples/ci/ml_engineer_with_tools.py

@@ -3,8 +3,8 @@ import asyncio
 from metagpt.roles.ci.ml_engineer import MLEngineer
 
 
-async def main(requirement: str, auto_run: bool = True, use_tools: bool = True):
-    role = MLEngineer(goal=requirement, auto_run=auto_run, use_tools=use_tools)
+async def main(requirement: str):
+    role = MLEngineer(auto_run=True, use_tools=True)
     await role.run(requirement)