Correct Logging

The mirror logging to stdout/err was including the context prefix from
python of `[pid=<prediction_id]` this PR eliminates that problem and
restores proper logging to console.

Author: tempusfrangit

internal/runner/runner.go

@@ -468,14 +468,28 @@ func (r *Runner) setupLogCapture() error {
 func (r *Runner) logStdout(line string) {
 	r.captureLogLine(line)
 
-	_, _ = fmt.Fprintln(os.Stdout, line) //nolint:forbidigo // mirror log to stdout
+	// Strip [pid=xxxxx] prefix before mirroring to stdout
+	mirrorLine := line
+	if LogRegex.MatchString(line) {
+		if m := LogRegex.FindStringSubmatch(line); m != nil {
+			mirrorLine = m[2] // Extract message without pid prefix
+		}
+	}
+	_, _ = fmt.Fprintln(os.Stdout, mirrorLine) //nolint:forbidigo // mirror log to stdout
 }
 
 // logStderr captures a line from stderr and mirrors to stderr
 func (r *Runner) logStderr(line string) {
 	r.captureLogLine(line)
 
-	_, _ = fmt.Fprintln(os.Stderr, line) //nolint:forbidigo // mirror log to stderr
+	// Strip [pid=xxxxx] prefix before mirroring to stderr
+	mirrorLine := line
+	if LogRegex.MatchString(line) {
+		if m := LogRegex.FindStringSubmatch(line); m != nil {
+			mirrorLine = m[2] // Extract message without pid prefix
+		}
+	}
+	_, _ = fmt.Fprintln(os.Stderr, mirrorLine) //nolint:forbidigo // mirror log to stderr
 }
 
 // captureLogLine handles routing log lines like the old implementation

script/test-setuid-cleanup.sh

@@ -38,14 +38,14 @@ python3 -m cog.server.http \
 EOF
 
 # Start Docker container
-docker run -it --detach \
+docker run -it --rm --detach \
     --name "$name" \
     --entrypoint /bin/bash \
     --publish "$port:$port" \
     --volume "$PWD":/src python:latest \
     -c "$SCRIPT"
 
-#trap 'docker stop $name' EXIT
+trap 'docker stop $name' EXIT
 
 # Wait for server to be ready
 sleep 3