add llb log

Signed-off-by: Keming <kemingyang@tensorchord.ai>
tensorchord · Sep 26, 2022 · 16ca22f · 16ca22f
1 parent 7327c13
commit 16ca22f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pkg/lang/ir/system.go b/pkg/lang/ir/system.go
@@ -152,7 +152,7 @@ func (g *Graph) preparePythonBase() llb.State {
 	// envd-sshd
 	sshd := base.File(llb.Copy(
 		llb.Image(types.EnvdSshdImage), "/usr/bin/envd-sshd", "/var/envd/bin/envd-sshd",
-		&llb.CopyInfo{CreateDestPath: true}))
+		&llb.CopyInfo{CreateDestPath: true}), llb.WithCustomName("[internal] add envd-sshd"))
 
 	// apt packages
 	var sb strings.Builder
@@ -169,7 +169,7 @@ func (g *Graph) preparePythonBase() llb.State {
 	cacheDir := "/var/cache/apt"
 	cacheLibDir := "/var/lib/apt"
 
-	run := sshd.Run(llb.Shlex(fmt.Sprintf("bash -c \"%s\"", sb.String())), llb.WithCustomNamef(sb.String()))
+	run := sshd.Run(llb.Shlex(fmt.Sprintf("bash -c \"%s\"", sb.String())), llb.WithCustomName("[internal] system packages"))
 	run.AddMount(cacheDir, llb.Scratch(),
 		llb.AsPersistentCacheDir(g.CacheID(cacheDir), llb.CacheMountShared))
 	run.AddMount(cacheLibDir, llb.Scratch(),

diff --git a/pkg/types/envd.go b/pkg/types/envd.go
@@ -36,8 +36,7 @@ const DefaultPathEnvWindows = system.DefaultPathEnvWindows
 
 const PythonBaseImage = "ubuntu:20.04"
 
-var EnvdSshdImage = fmt.Sprintf(
-	"tensorchord/envd-sshd-from-scratch:%s", version.GetEnvdVersion())
+var EnvdSshdImage = fmt.Sprintf("tensorchord/envd-sshd-from-scratch:%s", version.GetEnvdVersion())
 
 var BaseEnvironment = []struct {
 	Name  string