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

Merge branch 'tool_manage_new' into 'code_intepreter'

Browse source 
rm make tools in ci for now

See merge request agents/data_agents_opt!61
This commit is contained in:
林义章 2024-01-25 06:34:24 +00:00
commit 5f624844cc
3 changed files with 6 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

21
tests/metagpt/roles/run_code_interpreter.py
View file

@ -9,9 +9,7 @@ from metagpt.schema import Plan
from metagpt.utils.recovery_util import load_history, save_history
async def run_code_interpreter(
role_class, requirement, auto_run, use_tools, use_code_steps, make_udfs, use_udfs, save_dir, tools
):
async def run_code_interpreter(role_class, requirement, auto_run, use_tools, use_code_steps, save_dir, tools):
"""
The main function to run the MLEngineer with optional history loading.
@ -25,16 +23,13 @@ async def run_code_interpreter(
"""
if role_class == "ci":
role = CodeInterpreter(
goal=requirement, auto_run=auto_run, use_tools=use_tools, make_udfs=make_udfs, tools=tools
)
role = CodeInterpreter(goal=requirement, auto_run=auto_run, use_tools=use_tools, tools=tools)
else:
role = MLEngineer(
goal=requirement,
auto_run=auto_run,
use_tools=use_tools,
use_code_steps=use_code_steps,
make_udfs=make_udfs,
tools=tools,
)
@ -50,10 +45,10 @@ async def run_code_interpreter(
try:
await role.run(requirement)
except Exception as e:
save_path = save_history(role, save_dir)
logger.exception(f"An error occurred: {e}, save trajectory here: {save_path}")
save_history(role, save_dir)
if __name__ == "__main__":
# requirement = "Run data analysis on sklearn Iris dataset, include a plot"
@ -73,8 +68,6 @@ if __name__ == "__main__":
role_class = "mle"
auto_run = True
use_tools = True
make_udfs = False
use_udfs = False
tools = []
# tools = ["FillMissingValue", "CatCross", "non_existing_test"]
@ -84,13 +77,9 @@ if __name__ == "__main__":
auto_run: bool = auto_run,
use_tools: bool = use_tools,
use_code_steps: bool = False,
make_udfs: bool = make_udfs,
use_udfs: bool = use_udfs,
save_dir: str = save_dir,
tools=tools,
):
await run_code_interpreter(
role_class, requirement, auto_run, use_tools, use_code_steps, make_udfs, use_udfs, save_dir, tools
)
await run_code_interpreter(role_class, requirement, auto_run, use_tools, use_code_steps, save_dir, tools)
fire.Fire(main)