58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
|
import os
|
||
|
|
import subprocess
|
||
|
|
from google.genai import types
|
||
|
|
|
||
|
|
def run_python_file(working_directory, file_path, args=[]):
|
||
|
|
abs_working_dir = os.path.abspath(working_directory)
|
||
|
|
abs_file_path = os.path.abspath(os.path.join(working_directory, file_path))
|
||
|
|
if not abs_file_path.startswith(abs_working_dir):
|
||
|
|
return f'Error: Cannot execute "{file_path}" as it is outside the permitted working directory'
|
||
|
|
if not os.path.isfile(abs_file_path):
|
||
|
|
return f'Error: File "{file_path}" not found'
|
||
|
|
if not abs_file_path.endswith(".py"):
|
||
|
|
return f'Error: "{file_path}" is not a Python file.'
|
||
|
|
|
||
|
|
try:
|
||
|
|
completed = subprocess.run(args=["python3", abs_file_path]+args, capture_output=True, cwd=abs_working_dir, timeout=30)
|
||
|
|
|
||
|
|
if completed == None:
|
||
|
|
raise Exception("Subprocess holder is empty")
|
||
|
|
|
||
|
|
if completed.stdout == None or completed.stdout == "":
|
||
|
|
return f"No output produced."
|
||
|
|
|
||
|
|
message = f"STDOUT: {completed.stdout.decode()}. STDERR: {completed.stderr.decode()}."
|
||
|
|
if completed.returncode != 0:
|
||
|
|
message += f" Error: Process exited with code {completed.returncode}"
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
return f"Error: executing Python file: {e}"
|
||
|
|
|
||
|
|
return message
|
||
|
|
|
||
|
|
schema_run_python_file = types.FunctionDeclaration(
|
||
|
|
name="run_python_file",
|
||
|
|
description="Run the python file",
|
||
|
|
parameters=types.Schema(
|
||
|
|
type=types.Type.OBJECT,
|
||
|
|
properties={
|
||
|
|
"working_directory": types.Schema(
|
||
|
|
type=types.Type.STRING,
|
||
|
|
description="The directory to read the file from. If not provided, lists files in the working directory itself.",
|
||
|
|
),
|
||
|
|
"file_path": types.Schema(
|
||
|
|
type=types.Type.STRING,
|
||
|
|
description="The file to execute. If not provided, the function will not work."
|
||
|
|
),
|
||
|
|
"args": types.Schema(
|
||
|
|
type=types.Type.ARRAY,
|
||
|
|
items=types.Schema(
|
||
|
|
type=types.Type.STRING,
|
||
|
|
description="Optional arguments to pass to the Python file.",
|
||
|
|
),
|
||
|
|
description="Arguments to run the python file with."
|
||
|
|
)
|
||
|
|
},
|
||
|
|
),
|
||
|
|
)
|