Function run_python

autoagent/tools/terminal_tools.py:355–419

Run a python script. Args: code_path: The absolute or relative path (the relative path is from the root of the workplace `/workplace`) to the python script file. cwd: The working directory of the python script. If not provided, will regard the directory of the script as the

(
    context_variables,
    code_path: str,
    cwd: str = None,
    env_vars: Optional[Dict[str, str]] = None,
)

@register_tool("run_python")
@process_terminal_response
def run_python(
    context_variables,
    code_path: str,
    cwd: str = None,
    env_vars: Optional[Dict[str, str]] = None,
) -> str:
    """
    Run a python script. 
    Args:
        code_path: The absolute or relative path (the relative path is from the root of the workplace `/workplace`) to the python script file.
        cwd: The working directory of the python script. If not provided, will regard the directory of the script as the working directory. If there is a command `cd ...` in the instruction for running the script, you should provide the cwd and not use the default value. (Optional)
        env_vars: The environment variables to be set before running the python script. (Optional)
    Returns:
        A string representation of the exit code and output of the python script.
    """
    env: Union[DockerEnv, LocalEnv] = context_variables.get("code_env", LocalEnv())
    try:
        # 转换为绝对路径
        # abs_path = str(Path(code_path).resolve())
        if Path(code_path).is_absolute():
            if env.run_command(f"ls {code_path}")['status'] != 0: return f"[ERROR] File {code_path} does not exist"
            code_abs_path = code_path
        else: 
            code_abs_path = f"{env.docker_workplace}/{code_path}"
            if env.run_command(f"ls {code_abs_path}")['status'] != 0: return f'[ERROR] You use a relative path, so we regard the `{env.docker_workplace}` as the root of the workplace, but `{code_abs_path}` does not exist'
        
        
        if cwd:
            # 使用指定的项目根目录
            if Path(cwd).is_absolute():
                if env.run_command(f"ls {cwd}")['status'] != 0: return f"[ERROR] Working directory {cwd} does not exist"
            else: 
                cwd = f"{env.docker_workplace}/{cwd}"
                if env.run_command(f"ls {cwd}")['status'] != 0: return f"[ERROR] You use a relative path for `cwd`, so we regard the `{env.docker_workplace}` as the working directory, but `{cwd}` does not exist"
        else:
            cwd = str(Path(code_abs_path).parent)
            
        
        # 设置PYTHONPATH
        pythonpath = str(cwd)
        
        # 获取Python解释器路径
        env_str = f"PYTHONPATH={pythonpath}"
        
        if env_vars:
            env_str += " " + " ".join([f"{k}={v}" for k, v in env_vars.items()])
        # print(env_str)
        
        # 构建相对模块路径
        try:
            rel_path = Path(code_abs_path).relative_to(cwd)
            module_path = str(rel_path.with_suffix('')).replace(os.sep, '.')
            
            command = f"cd {cwd} && {env_str} python -m {module_path}"
        except ValueError:
            # 如果无法构建相对路径，使用完整路径
            command = f"cd {cwd} && {env_str} python {code_path}"