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

drop retry

Browse source
This commit is contained in:
lidanyang 2024-01-09 17:45:46 +08:00
parent fd9f2416ff
commit 3eee6eff8c
1 changed files with 6 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

21
metagpt/actions/execute_code.py
View file

@ -55,7 +55,6 @@ class ExecutePyCode(ExecuteCode, Action):
llm=None,
nb=None,
timeout: int = 600,
max_tries: int = 3
):
super().__init__(name, context, llm)
if nb is None:
@ -63,7 +62,6 @@ class ExecutePyCode(ExecuteCode, Action):
else:
self.nb = nb
self.timeout = timeout
self.max_tries = max_tries
self.nb_client = NotebookClient(self.nb, timeout=self.timeout)
self.console = Console()
self.interaction = "ipython" if self.is_ipython() else "terminal"
@ -195,22 +193,15 @@ class ExecutePyCode(ExecuteCode, Action):
# add code to the notebook
self.add_code_cell(code=code)
tries = 0
success = False
outputs = ""
while tries < self.max_tries and not success:
# build code executor
await self.build()
# run code
cell_index = len(self.nb.cells) - 1
success, error_message = await self.run_cell(self.nb.cells[-1], cell_index)
# build code executor
await self.build()
if success:
outputs = self.parse_outputs(self.nb.cells[-1].outputs)
else:
tries += 1
# run code
cell_index = len(self.nb.cells) - 1
success, error_message = await self.run_cell(self.nb.cells[-1], cell_index)
if success:
outputs = self.parse_outputs(self.nb.cells[-1].outputs)
return truncate(remove_escape_and_color_codes(outputs)), True
else:
return error_message, False