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

fix: role timeout not passing in

Browse source
This commit is contained in:
Yizhou Chi 2024-10-16 09:50:24 +08:00
parent 541f8a1b10
commit 7794b99005
2 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
expo/MCTS.py
View file

@ -20,7 +20,11 @@ from metagpt.utils.common import read_json_file
def initialize_di_root_node(state, reflection: bool = True):
role = ResearchAssistant(
node_id="0", start_task_id=state["start_task_id"], use_reflection=reflection, role_dir=state["node_dir"]
node_id="0",
start_task_id=state["start_task_id"],
use_reflection=reflection,
role_dir=state["node_dir"],
role_timeout=state["role_timeout"],
)
return role, Node(parent=None, state=state, action=None, value=0)

7
expo/run_experiment.py
View file

@ -9,7 +9,7 @@ from expo.experimenter.experimenter import Experimenter
from expo.experimenter.mcts import MCTSExperimenter
def get_args():
def get_args(cmd=True):
parser = argparse.ArgumentParser()
parser.add_argument("--name", type=str, default="")
parser.add_argument(
@ -22,7 +22,10 @@ def get_args():
get_di_args(parser)
get_mcts_args(parser)
get_aug_exp_args(parser)
return parser.parse_args()
if cmd:
return parser.parse_args()
else:
return parser.parse_args("")
def get_mcts_args(parser):