ZJU_czx
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from app.tool.python_execute import PythonExecute
|
|
|
|
|
|
class NormalPythonExecute(PythonExecute):
|
|
"""A tool for executing Python code with timeout and safety restrictions."""
|
|
|
|
name: str = "common_python_execute"
|
|
description: str = (
|
|
"""Executes Python code strings to tasks such as data process and data report"""
|
|
)
|
|
parameters: dict = {
|
|
"type": "object",
|
|
"properties": {
|
|
"code": {
|
|
"type": "string",
|
|
"description": """The Python code to execute. Note:
|
|
1. Only outputs from print() are visible; function return values are not captured. Use print() statements to display results
|
|
2. Do data process (cleaning / transform) saved in *.csv
|
|
3. Generate a data analysis report in html""",
|
|
},
|
|
"code_type": {
|
|
"description": "code type",
|
|
"type": "string",
|
|
"default": "html",
|
|
"enum": ["process", "report", "others"],
|
|
},
|
|
},
|
|
"required": ["code"],
|
|
}
|
|
|
|
async def execute(self, code: str, code_type: str, timeout=5):
|
|
return await super().execute(code, timeout)
|