From 5cbb6cc3f69505747b8ca6281ab99a29431db4b3 Mon Sep 17 00:00:00 2001
From: 0xb79 <0xb79orch@gmail.com>
Date: Fri, 28 Jun 2024 16:15:04 -0500
Subject: [PATCH] add MOCK_PIPELINE env var to runner if exists

---
 worker/docker.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/worker/docker.go b/worker/docker.go
index e09e1726..344b6841 100644
--- a/worker/docker.go
+++ b/worker/docker.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"log/slog"
+	"os"
 	"strings"
 	"sync"
 	"time"
@@ -164,6 +165,12 @@ func (m *DockerManager) createContainer(ctx context.Context, pipeline string, mo
 		"PIPELINE=" + pipeline,
 		"MODEL_ID=" + modelID,
 	}
+	_, ok := os.LookupEnv("MOCK_PIPELINE")
+	if ok {
+		envVars = append(envVars, "MOCK_PIPELINE=true")
+		slog.Info("MOCK_PIPELINE set, passing to runner container")
+	}
+
 	for key, value := range optimizationFlags {
 		envVars = append(envVars, key+"="+value.String())
 	}