PythonREPL是python提供的 “Read–Eval–Print Loop”(读取-求值-打印-循环)的缩写,它是一种简单的、交互式的编程环境,我们之前在用AI生成代码后执行就使用这个方法,例如
def py_repl_tool(self, realcode: str):py_repl = PythonREPL()...res = py_repl.run(realcode)
之前使用的挺好,但最近,生成带函数的代码时,就出现了执行报错:
Python REPL can execute arbitrary code. Use with caution.
NameError(“name ‘getReportDataLocal’ is not defined”)
需要执行的代码如下:
from akshare import index_stock_cons
from pythonRepository import getReportDataLocal, CalAnnualRatiodef genHS300EPS(report_date):# 获取指定日期的业绩报表df_report = getReportDataLocal(report_date)# 计算前一年的报表日期......# 打印并返回处理后的stock_listprint(stock_list)return stock_list# 示例调用print(genHS300EPS("20231231"))
代码本身是完整且正确的
但执行结果报错
查了一下报错信息,发现这种报错是,没有引入import库和方法,从而提示X方法无定义。 我看一下PythonREPL()方法,发现实际执行的是exec(command, globals, locals)
我搜索后发现,猜想可能是全局变量和局部变量作用域的相关问题,于是尝试自己写一个exec(command,globals)函数
def py_repl_tool_1(self, realcode: str):# py_repl = PythonREPL()...# res = py_repl.run(realcode)try:result[i] = exec(realcode, globals())print('执行成功')except Exception as e:print("执行失败: %s, %s" % e % result[i])# print(traceback.format_exception(e))traceback.print_exc()
成功了