From df9bd6a2bae16c264b1e3f60b58a88f211d688c8 Mon Sep 17 00:00:00 2001
From: Keming <kemingyang@tensorchord.ai>
Date: Fri, 29 Jul 2022 12:06:59 +0800
Subject: [PATCH] chore: skip starship prompt for customized image (#687)

Signed-off-by: Keming <kemingyang@tensorchord.ai>
---
 pkg/lang/ir/shell.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pkg/lang/ir/shell.go b/pkg/lang/ir/shell.go
index fe18bd5a6..065426b89 100644
--- a/pkg/lang/ir/shell.go
+++ b/pkg/lang/ir/shell.go
@@ -51,6 +51,10 @@ func (g *Graph) compileShell(root llb.State) (llb.State, error) {
 }
 
 func (g *Graph) compilePrompt(root llb.State) llb.State {
+	// skip this for customized image
+	if g.Image != nil {
+		return root
+	}
 	// starship config
 	config := root.
 		File(llb.Mkdir(defaultConfigDir, 0755, llb.WithParents(true)),