mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 14:04:27 +01:00
4a86da95f9
This happens often in PRs so fixing this everywhere will make many PRs mergeable as they won't include irrelevant whitespace fixes
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import docker
|
|
import os
|
|
|
|
|
|
def execute_python_file(file):
|
|
"""Execute a Python file in a Docker container and return the output"""
|
|
workspace_folder = "auto_gpt_workspace"
|
|
|
|
print (f"Executing file '{file}' in workspace '{workspace_folder}'")
|
|
|
|
if not file.endswith(".py"):
|
|
return "Error: Invalid file type. Only .py files are allowed."
|
|
|
|
file_path = os.path.join(workspace_folder, file)
|
|
|
|
if not os.path.isfile(file_path):
|
|
return f"Error: File '{file}' does not exist."
|
|
|
|
try:
|
|
client = docker.from_env()
|
|
|
|
# You can replace 'python:3.8' with the desired Python image/version
|
|
# You can find available Python images on Docker Hub:
|
|
# https://hub.docker.com/_/python
|
|
container = client.containers.run(
|
|
'python:3.10',
|
|
f'python {file}',
|
|
volumes={
|
|
os.path.abspath(workspace_folder): {
|
|
'bind': '/workspace',
|
|
'mode': 'ro'}},
|
|
working_dir='/workspace',
|
|
stderr=True,
|
|
stdout=True,
|
|
detach=True,
|
|
)
|
|
|
|
output = container.wait()
|
|
logs = container.logs().decode('utf-8')
|
|
container.remove()
|
|
|
|
# print(f"Execution complete. Output: {output}")
|
|
# print(f"Logs: {logs}")
|
|
|
|
return logs
|
|
|
|
except Exception as e:
|
|
return f"Error: {str(e)}"
|