当前位置：首页 > article >正文

GPT Code Interpreter

article 2024/12/25 13:15:16

前几天，找了一些开源项目阅读学习，主要了解现在执行GPT生成的代码一般的做法是什么样的。

unsetunset使用Ipython执行代码unsetunset

我们Google一搜，很容易就搜到：https://e2b.dev/，可以使用它提供的api在e2b的沙箱中执行生成的Python或JavaScript代码，一开始以为是开源的，仓库为：https://github.com/e2b-dev/code-interpreter.git

浏览源码后，开源的只是一个sdk，即你需要提供api key，然后使用e2b cloud sandbox去执行代码，项目关键代码如下：

try:
    with self._client.stream(
        "POST",
        f"{self._jupyter_url}/execute",
        json={
            "code": code,
            "context_id": context_id,
            "language": language,
            "env_vars": envs,
        },
        timeout=(request_timeout, timeout, request_timeout, request_timeout),
    ) as response:
        err = extract_exception(response)
        if err:
            raise err

            execution = Execution()

        for line in response.iter_lines():
            parse_output(
                execution,
                line,
                on_stdout=on_stdout,
                on_stderr=on_stderr,
                on_result=on_result,
                on_error=on_error,
            )

            return execution
except httpx.ReadTimeout:
    raise format_execution_timeout_error()
except httpx.TimeoutException:
    raise format_request_timeout_error()

其中jupyter_url这个变量名引起我的注意，Python仔都知道，Jupyter我们常用来写写数据分析和算法相关的notebook，是可以执行代码的，直接ChatGPT调研，就可以发现，我们可以使用ipython来执行代码。

然后我们发现了这个项目：https://github.com/chapyter/chapyter.git

Jupyter Notebook是基于Ipython的，可以使用Ipython包中的InteractiveShell来执行代码，用法如下：

from IPython.core.interactiveshell import InteractiveShell

# 创建一个 InteractiveShell 实例
shell = InteractiveShell.instance()

# 要执行的代码
code = """
def greet(name):
    return f"Hello, {name}!"

result = greet("Lindee")
print(result)
"""

# 执行代码
shell.run_cell(code)

不负责猜测，e2b可能也用了这种方式来执行。

当然，单纯这样，应该是不行的，因为用户可能会提交恶意代码，即GPT可能会生成自毁代码，所以需要限制执行。

因为e2b这里没有更多信息，所以我们换其他开源项目。

unsetunset使用subprocess.Popenunsetunset

又找到一个code-interpreter：https://github.com/haseeb-heaven/code-interpreter.git

阅读代码，可以找到关键代码如下：

# libs/code_interpreter.py

def execute_code(self, code, language):
        try:
            language = language.lower()
            self.logger.info(f"Running code: {code[:100]} in language: {language}")

            # Check for code and language validity
            ifnot code or len(code.strip()) == 0:
                return"Code is empty. Cannot execute an empty code."
            
            # Check for compilers on the system
            compilers_status = self._check_compilers(language)
            ifnot compilers_status:
                raise Exception("Compilers not found. Please install compilers on your system.")
            
            if language == "python":
                process = subprocess.Popen(["python", "-c", code], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
                stdout, stderr = process.communicate()
                stdout_output = stdout.decode("utf-8")
                stderr_output = stderr.decode("utf-8")
                self.logger.info(f"Python Output execution: {stdout_output}, Errors: {stderr_output}")
                return stdout_output, stderr_output
            
            elif language == "javascript":
                process = subprocess.Popen(["node", "-e", code], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
                stdout, stderr = process.communicate()
                stdout_output = stdout.decode("utf-8")
                stderr_output = stderr.decode("utf-8")
                self.logger.info(f"JavaScript Output execution: {stdout_output}, Errors: {stderr_output}")
                return stdout_output, stderr_output
            
            else:
                self.logger.info("Unsupported language.")
                raise Exception("Unsupported language.")
                
        except Exception as exception:
            self.logger.error(f"Exception in running code: {str(exception)}")
            raise exception

通过subprocess.Popen来执行Python或JavaScript代码，但跟Ipython一样，依旧会有安全性问题。

这些让cursor基于代码上下文，给我一下解决方案：