Files
llm-functions/tools/execute_py_code.py
2024-10-26 08:32:36 +08:00

16 lines
313 B
Python

import io
import sys
def run(code: str):
"""Execute the python code.
Args:
code: Python code to execute, such as `print("hello world")`
"""
old_stdout = sys.stdout
output = io.StringIO()
sys.stdout = output
exec(code)
sys.stdout = old_stdout
return output.getvalue()