From 83f71d3b504bd2d26c004699487ae194181c21cc Mon Sep 17 00:00:00 2001
From: Will Sackfield <sackfield@replicate.com>
Date: Thu, 8 May 2025 17:11:56 -0400
Subject: [PATCH] =?UTF-8?q?Send=20the=20migration=20status=E2=80=99s=20int?=
 =?UTF-8?q?o=20the=20log?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/coglog/client.go | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/pkg/coglog/client.go b/pkg/coglog/client.go
index 14224631ab..29a964533c 100644
--- a/pkg/coglog/client.go
+++ b/pkg/coglog/client.go
@@ -34,9 +34,13 @@ type pushLog struct {
 }
 
 type migrateLog struct {
-	DurationMs float32 `json:"length_ms"`
-	BuildError *string `json:"error"`
-	Accept     bool    `json:"accept"`
+	DurationMs          float32 `json:"length_ms"`
+	BuildError          *string `json:"error"`
+	Accept              bool    `json:"accept"`
+	PythonPackageStatus string  `json:"python_package_status"`
+	RunStatus           string  `json:"run_status"`
+	PythonPredictStatus string  `json:"python_predict_status"`
+	PythonTrainStatus   string  `json:"python_train_status"`
 }
 
 func NewClient(client *http.Client) *Client {
@@ -131,9 +135,13 @@ func (c *Client) EndMigrate(ctx context.Context, err error, logContext *MigrateL
 		errorStr = &errStr
 	}
 	migrateLog := migrateLog{
-		DurationMs: float32(time.Now().Sub(logContext.started).Milliseconds()),
-		BuildError: errorStr,
-		Accept:     logContext.accept,
+		DurationMs:          float32(time.Now().Sub(logContext.started).Milliseconds()),
+		BuildError:          errorStr,
+		Accept:              logContext.accept,
+		PythonPackageStatus: logContext.PythonPackageStatus,
+		RunStatus:           logContext.RunStatus,
+		PythonPredictStatus: logContext.PythonPredictStatus,
+		PythonTrainStatus:   logContext.PythonTrainStatus,
 	}
 
 	jsonData, err := json.Marshal(migrateLog)