agent-harness / tasks /patches /TASK_CR_007_source.patch
cuber12's picture
Publish agent harness research code and paper artifacts
d61821a verified
Raw
History Blame Contribute Delete
1.92 kB
diff --git a/functions/concrete/builder/builder.go b/functions/concrete/builder/builder.go
index 89828cb4..b480902a 100644
--- a/functions/concrete/builder/builder.go
+++ b/functions/concrete/builder/builder.go
@@ -60,6 +60,7 @@ func Build(job spec.Job, vars variables.Provider, options ...Option) ([]byte, er
config.AfterScriptIgnoreErrors = variables.DefaultBool(
b.variables, "AFTER_SCRIPT_IGNORE_ERRORS", true,
)
+ config.TraceSections = b.meta.Features.TraceSections
config.CacheArchive, err = b.buildCacheArchive()
if err != nil {
diff --git a/functions/concrete/run/runner.go b/functions/concrete/run/runner.go
index 5088598c..8e50b3c3 100644
--- a/functions/concrete/run/runner.go
+++ b/functions/concrete/run/runner.go
@@ -49,6 +49,7 @@ type Config struct {
ScriptTimeout time.Duration `json:"script_timeout,omitempty"`
AfterScriptTimeout time.Duration `json:"after_script_timeout,omitempty"`
AfterScriptIgnoreErrors bool `json:"after_script_ignore_errors,omitempty"`
+ TraceSections bool `json:"trace_sections,omitempty"`
ID int64 `json:"id,omitempty"`
Token string `json:"token,omitempty"`
BaseURL string `json:"base_url,omitempty"`
@@ -368,8 +369,10 @@ func (r *Runner) withTimeout(parent context.Context, d time.Duration) (context.C
}
func (r *Runner) section(ctx context.Context, name string, fn func(context.Context, *env.Env) error) error {
- fmt.Fprintf(r.env.Stdout, "section_start:%d:%s\r\033[0K", time.Now().Unix(), name)
- defer fmt.Fprintf(r.env.Stdout, "section_end:%d:%s\r\033[0K", time.Now().Unix(), name)
+ if r.config.TraceSections {
+ fmt.Fprintf(r.env.Stdout, "section_start:%d:%s\r\033[0K", time.Now().Unix(), name)
+ defer fmt.Fprintf(r.env.Stdout, "section_end:%d:%s\r\033[0K", time.Now().Unix(), name)
+ }
return fn(ctx, r.env)
}