yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-11-11 07:14:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Chi Wang 2023-09-17 20:18:13 +00:00
parent 188ddc19a5
commit bc5e91e5aa
1 changed files with 3 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
autogen/code_utils.py
View File

@ -256,6 +256,7 @@ def execute_code(
cmd, cmd,
cwd=work_dir, cwd=work_dir,
capture_output=True, capture_output=True,
text=True,
) )
signal.alarm(0) signal.alarm(0)
except TimeoutError: except TimeoutError:
@ -268,10 +269,10 @@ def execute_code(
else: else:
abs_path = str(pathlib.Path(work_dir).absolute()) + "/" abs_path = str(pathlib.Path(work_dir).absolute()) + "/"
if result.returncode: if result.returncode:
logs = result.stderr.decode("utf-8") logs = result.stderr
logs = logs.replace(str(abs_path), "") logs = logs.replace(str(abs_path), "")
else: else:
logs = result.stdout.decode("utf-8") logs = result.stdout
return result.returncode, logs, None return result.returncode, logs, None
# create a docker client # create a docker client
@ -299,9 +300,6 @@ def execute_code(
# get a randomized str based on current time to wrap the exit code # get a randomized str based on current time to wrap the exit code
exit_code_str = f"exitcode{time.time()}" exit_code_str = f"exitcode{time.time()}"
abs_path = pathlib.Path(work_dir).absolute() abs_path = pathlib.Path(work_dir).absolute()
# if sys.platform == "win32":
# abs_path = str(abs_path).replace("\\", "/")
# abs_path = f"/{abs_path[0].lower()}{abs_path[2:]}"
cmd = [ cmd = [
"sh", "sh",
"-c", "-c",
@ -326,14 +324,6 @@ def execute_code(
if original_filename is None: if original_filename is None:
os.remove(filepath) os.remove(filepath)
return 1, TIMEOUT_MSG, image return 1, TIMEOUT_MSG, image
# try:
# container.wait(timeout=timeout)
# except (ReadTimeout, ConnectionError):
# container.stop()
# container.remove()
# if original_filename is None:
# os.remove(filepath)
# return 1, "Timeout"
# get the container logs # get the container logs
logs = container.logs().decode("utf-8").rstrip() logs = container.logs().decode("utf-8").rstrip()
# commit the image # commit the image