基于 Function Calling 构建代码执行工具时遇到的问题

{  "name": "execute_python_code",  "arguments": {    "code": "print(2+2)"  }}

$available_functions = [    [        "type" => "function",        "function" => [            "name" => "search_web",            "description" => "在网络上搜索最新信息，获取权威内容 (Search the web for the latest information and authoritative content)",            "parameters" => [                "type" => "object",                "properties" => [                    "query" => [                        "type" => "string",                        "description" => "搜索查询内容 (The search query)"                    ]                ],                "required" => ["query"]            ]        ]    ],    [        "type" => "function",        "function" => [            "name" => "execute_python_code",            "description" => "运行一段 Python 代码，并返回标准输出结果 (Execute a snippet of Python code and return the standard output)",            "parameters" => [                "type" => "object",                "properties" => [                    "code" => [                        "type" => "string",                        "description" => "要执行的 Python 代码 (The Python code to execute, e.g., print(2+2))"                    ]                ],                "required" => ["code"]            ]        ]    ]];

2 + 2
## 或者result = 2+2result

import ast
class AutoPrintTransformer(ast.NodeTransformer):    """    自动为最后一个表达式添加 print() 的 AST 转换器    """    def __init__(self):        super().__init__()        self.modified = False
    def visit_Module(self, node):        """        处理模块节点，为最后一个表达式添加 print()        """        self.generic_visit(node)
        last_expr_index = -1        for i in range(len(node.body) - 1, -1, -1):            stmt = node.body[i]
            # 判断是否为 print 语句            is_print_stmt = (                isinstance(stmt, ast.Expr) and                 isinstance(stmt.value, ast.Call) and                 isinstance(stmt.value.func, ast.Name) and stmt.value.func.id == 'print'            )
            is_docstring = (                isinstance(stmt, ast.Expr) and                 isinstance(stmt.value, ast.Constant) and                 isinstance(stmt.value.value, str) and i == 0            )
            if not is_print_stmt and not is_docstring:                last_expr_index = i                break
        # 找到了需要添加 print 的表达式        if last_expr_index != -1:            expr_node = node.body[last_expr_index]            print_call = ast.Call(func=ast.Name(id='print', ctx=ast.Load()), args=[expr_node.value], keywords=[])            print_stmt = ast.Expr(value=print_call)            ast.copy_location(print_stmt, expr_node)            node.body[last_expr_index] = print_stmt            self.modified = True
        return node

$b64 = base64_encode($user_python_code);

import base64, ast, traceback
class AutoPrintTransformer(ast.NodeTransformer):    def __init__(self):        self.has_modified = False
    def visit_Module(self, node):        self.generic_visit(node)        last_expr = None        for i in range(len(node.body) - 1, -1, -1):            stmt = node.body[i]            if isinstance(stmt, ast.Expr) and not (                isinstance(stmt.value, ast.Constant) and isinstance(stmt.value.value, str)            ):                last_expr = (i, stmt)                break        if last_expr:            idx, expr = last_expr            is_print = (                isinstance(expr.value, ast.Call) and                isinstance(expr.value.func, ast.Name) and                expr.value.func.id == 'print'            )            if not is_print:                print_node = ast.Expr(                    value=ast.Call(                        func=ast.Name(id='print', ctx=ast.Load()),                        args=[expr.value],                        keywords=[]                    )                )                ast.copy_location(print_node, expr)                node.body[idx] = print_node                self.has_modified = True        return node
def transform_code(src):    try:        tree = ast.parse(src)        transformer = AutoPrintTransformer()        new_tree = transformer.visit(tree)        ast.fix_missing_locations(new_tree)        if transformer.has_modified:            exec(compile(new_tree, '<string>', 'exec'), globals())        else:            exec(src, globals())    except Exception as e:        print(f"处理代码时出错: {e}")        traceback.print_exc()        exec(src, globals())
# 从 Base64 解码恢复原始用户代码user_code = base64.b64decode('<Base64_encoded_string>').decode('utf-8')transform_code(user_code)

import osimport uuidimport matplotlibmatplotlib.use('Agg')  # 使用非交互式后端import matplotlib.pyplot as plt
fig_list = []  # 用于跟踪所有创建的图orig_figure = plt.figure  # 保存原始的 plt.figure 方法
def track_figure(*args, **kwargs):    """    寻找生成的图形。    """    fig = orig_figure(*args, **kwargs)    fig_list.append(fig)    return fig
plt.figure = track_figure

...
saved_paths = []
if 'fig_list' in globals() and fig_list:    output_dir = "/output"  # 输出目录    os.makedirs(output_dir, exist_ok=True)    for i, fig in enumerate(fig_list):        unique_id = str(uuid.uuid4())[:8]  # 生成唯一 ID        filename = os.path.join(output_dir, f"chart_{i}_{unique_id}.png")  # 构建文件名        try:            fig.tight_layout()        except Exception:            pass        try:            fig.savefig(filename, dpi=100, bbox_inches='tight')  # 保存图像            saved_paths.append(filename)  # 添加路径到列表        except Exception as e:            print("出错！！")        finally:            plt.close(fig)  # 关闭图形
if saved_paths:    for img_path in saved_paths:        print(f"###IMAGE_FILE_PATH###{img_path}")else:    print("没有图表。")
plt.close('all'

![图表1](https://chatgpt.wljay.cn/sandbox/matplotlib/chart_xxxxx_12345.png)