From e9562240d25fd013396f4c46cfeafa95f2fd0ac8 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Thu, 26 Sep 2024 14:38:06 -0700
Subject: [PATCH] bake run file into container

---
 llama_stack/distribution/build_container.sh | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/llama_stack/distribution/build_container.sh b/llama_stack/distribution/build_container.sh
index fec1e394f..b35be7f63 100755
--- a/llama_stack/distribution/build_container.sh
+++ b/llama_stack/distribution/build_container.sh
@@ -29,6 +29,7 @@ SCRIPT_DIR=$(dirname "$(readlink -f "$0")")
 REPO_DIR=$(dirname $(dirname "$SCRIPT_DIR"))
 DOCKER_BINARY=${DOCKER_BINARY:-docker}
 DOCKER_OPTS=${DOCKER_OPTS:-}
+REPO_CONFIGS_DIR="$REPO_DIR/tmp/configs"
 
 TEMP_DIR=$(mktemp -d)
 
@@ -125,3 +126,21 @@ echo "You can run it with: podman run -p 8000:8000 $image_name"
 
 echo "Checking image builds..."
 $DOCKER_BINARY run $DOCKER_OPTS -it $image_name cat llamastack-build.yaml
+
+container_build_dir="/app/builds"
+
+echo "Configuring default configuration to docker container..."
+$DOCKER_BINARY run $DOCKER_OPTS -it \
+  -v $REPO_CONFIGS_DIR:$container_build_dir \
+  $image_name \
+  llama stack configure ./llamastack-build.yaml --output-dir $container_build_dir
+
+tmp_container_name="$image_name-tmp"
+
+$DOCKER_BINARY create --name $tmp_container_name $image_name
+
+$DOCKER_BINARY cp "$REPO_CONFIGS_DIR/$build_name-run.yaml" $tmp_container_name:/app/
+
+$DOCKER_BINARY commit $tmp_container_name $image_name
+
+echo "Success!"