accept goal during run; move more logic from role to planner

This commit is contained in:
yzlin 2024-01-30 16:40:13 +08:00
parent 7aa89a3204
commit 606f1b8f9c
3 changed files with 50 additions and 45 deletions

View file

@ -23,10 +23,9 @@ async def run_code_interpreter(role_class, requirement, auto_run, use_tools, use
"""
if role_class == "ci":
role = CodeInterpreter(goal=requirement, auto_run=auto_run, use_tools=use_tools, tools=tools)
role = CodeInterpreter(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,