Fix llama stack build docker creation to have correct entrypoint

This commit is contained in:
Ashwin Bharambe 2025-01-22 16:53:54 -08:00
parent 08dcb9e31e
commit a8345f5f76
2 changed files with 11 additions and 6 deletions

View file

@ -119,12 +119,15 @@ def build_image(
normal_deps += SERVER_DEPENDENCIES
if build_config.image_type == ImageType.container.value:
if not template_name:
raise ValueError("template_name is required for container builds")
script = str(
importlib.resources.files("llama_stack") / "distribution/build_container.sh"
)
args = [
script,
image_name,
template_name,
container_image,
str(build_file_path),
str(BUILDS_BASE_DIR / ImageType.container.value),