apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-24 14:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update make tools: code -> remove_comments(code).

Browse source
This commit is contained in:
刘棒棒 2023-12-21 16:58:02 +08:00
parent 94b352cf23
commit 6d36511249
1 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

17
metagpt/roles/ml_engineer.py
View file

@ -20,6 +20,7 @@ from metagpt.roles.kaggle_manager import DownloadData, SubmitResult
from metagpt.schema import Message, Plan
from metagpt.utils.save_code import save_code_file
from metagpt.utils.recovery_util import save_history, load_history
from metagpt.utils.common import remove_comments
class MLEngineer(Role):
@ -189,7 +190,7 @@ class MLEngineer(Role):
result, success = await self.execute_code.run(code)
print(result)
# make tools for successful code and long code.
if success and self.make_udfs and len(code.split('\n')) > 4:
if success and self.make_udfs and len(remove_comments(code).split('\n')) > 4:
logger.info('Execute code successfully. Now start to make tools ...')
await self.make_tools(code=code)
self.working_memory.add(
@ -326,12 +327,12 @@ if __name__ == "__main__":
role.use_udfs = False
await role.run(requirement)
# use udfs
role.reset()
role.make_udfs = False
role.use_udfs = True
role.use_code_steps = False
role.use_tools = False
await role.run(requirement)
# role.reset()
# role.make_udfs = False
# role.use_udfs = True
# role.use_code_steps = False
# role.use_tools = False
# await role.run(requirement)
# requirement = "Perform data analysis on the provided data. Train a model to predict the target variable Survived. Include data preprocessing, feature engineering, and modeling in your pipeline. The metric is accuracy."
@ -381,4 +382,4 @@ if __name__ == "__main__":
logger.exception(f"An error occurred: {e}, save trajectory here: {save_path}")
fire.Fire(main)
fire.Fire(run_udfs)