apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-01 20:03:28 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

make timeout as argument

Browse source
This commit is contained in:
Yizhou Chi 2024-10-10 18:54:40 +08:00
parent 2fc8f20de6
commit eb460d3e19
7 changed files with 19 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
expo/experimenter/aug.py
View file

@ -34,7 +34,9 @@ class AugExperimenter(Experimenter):
results = []
for i in range(self.args.num_experiments):
di = ResearchAssistant(node_id=str(i), use_reflection=self.args.reflection)
di = ResearchAssistant(
node_id=str(i), use_reflection=self.args.reflection, role_timeout=self.args.role_timeout
)
di.role_dir = f"{di.role_dir}_{self.args.task}"
requirement = user_requirement + EXPS_PROMPT.format(experience=exps[i])
print(requirement)

5
expo/experimenter/experimenter.py
View file

@ -27,6 +27,7 @@ class Experimenter:
low_is_better=self.args.low_is_better,
name=self.args.name,
special_instruction=self.args.special_instruction,
args=self.args,
)
async def run_di(self, di, user_requirement, run_idx):
@ -82,7 +83,9 @@ class Experimenter:
results = []
for i in range(self.args.num_experiments):
di = ResearchAssistant(node_id="0", use_reflection=self.args.reflection)
di = ResearchAssistant(
node_id="0", use_reflection=self.args.reflection, role_timeout=self.args.role_timeout
)
score_dict = await self.run_di(di, user_requirement, run_idx=i)
results.append(
{"idx": i, "score_dict": score_dict, "user_requirement": user_requirement, "args": vars(self.args)}