aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2026-02-23 15:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix issues with execute_python_code responses (#4738)

Browse Source 
Co-authored-by: merwanehamadi <merwanehamadi@gmail.com>
This commit is contained in:
Erik Peterson
2023-06-18 20:30:08 -07:00
committed by GitHub
parent ee7d04775e
commit 9f737274b7
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
autogpt/commands/execute_code.py
View File

@@ -32,13 +32,13 @@ DENYLIST_CONTROL = "denylist"
},
},
)
def execute_python_code(code: str, basename: str, agent: Agent) -> str:
def execute_python_code(code: str, name: str, agent: Agent) -> str:
"""Create and execute a Python file in a Docker container and return the STDOUT of the
executed code. If there is any data that needs to be captured use a print statement
Args:
code (str): The Python code to run
basename (str): A name to be given to the Python file
name (str): A name to be given to the Python file
Returns:
str: The STDOUT captured from the code when it ran
@@ -47,10 +47,10 @@ def execute_python_code(code: str, basename: str, agent: Agent) -> str:
directory = os.path.join(agent.config.workspace_path, ai_name, "executed_code")
os.makedirs(directory, exist_ok=True)
if not basename.endswith(".py"):
basename = basename + ".py"
if not name.endswith(".py"):
name = name + ".py"
path = os.path.join(directory, basename)
path = os.path.join(directory, name)
try:
with open(path, "w+", encoding="utf-8") as f:

3
autogpt/json_utils/utilities.py
View File

@@ -22,7 +22,8 @@ def extract_json_from_response(response_content: str) -> dict:
try:
return ast.literal_eval(response_content)
except BaseException as e:
logger.error(f"Error parsing JSON response with literal_eval {e}")
logger.info(f"Error parsing JSON response with literal_eval {e}")
logger.debug(f"Invalid JSON received in response: {response_content}")
# TODO: How to raise an error here without causing the program to exit?
return {}