Method Execute

pkg/tools/bridge/runtime.go:111–186

(ctx context.Context, code string, input map[string]any)

}

func (r *PythonRuntime) Execute(ctx context.Context, code string, input map[string]any) (*ExecutionResult, error) {
	start := time.Now()

	// 创建临时文件
	tmpFile, err := os.CreateTemp(r.config.WorkDir, "aster_*.py")
	if err != nil {
		return nil, fmt.Errorf("create temp file: %w", err)
	}
	defer func() { _ = os.Remove(tmpFile.Name()) }()

	// 包装代码以处理输入和输出
	wrappedCode := r.wrapCode(code, input)
	if _, err := tmpFile.WriteString(wrappedCode); err != nil {
		return nil, fmt.Errorf("write code: %w", err)
	}
	_ = tmpFile.Close()

	// 创建带超时的 context
	execCtx, cancel := context.WithTimeout(ctx, r.config.Timeout)
	defer cancel()

	// 执行 Python
	cmd := exec.CommandContext(execCtx, r.pythonPath, tmpFile.Name())
	cmd.Dir = r.config.WorkDir

	// 设置环境变量
	cmd.Env = os.Environ()
	for k, v := range r.config.Env {
		cmd.Env = append(cmd.Env, fmt.Sprintf("%s=%s", k, v))
	}

	var stdout, stderr bytes.Buffer
	cmd.Stdout = &stdout
	cmd.Stderr = &stderr

	err = cmd.Run()
	duration := time.Since(start).Milliseconds()

	result := &ExecutionResult{
		Stdout:   truncateOutput(stdout.String(), r.config.MaxOutput),
		Stderr:   truncateOutput(stderr.String(), r.config.MaxOutput),
		Duration: duration,
	}

	if err != nil {
		if execCtx.Err() == context.DeadlineExceeded {
			result.Error = "execution timeout"
			result.ExitCode = -1
		} else if exitErr := (&exec.ExitError{}); errors.As(err, &exitErr) {
			result.ExitCode = exitErr.ExitCode()
			result.Error = stderr.String()
		} else {
			result.Error = err.Error()
			result.ExitCode = -1
		}
		return result, nil
	}