This commit is contained in:
Ashwin Bharambe 2025-07-30 19:41:57 -07:00
parent e3e9945dbb
commit 88260a8c0a

View file

@ -122,7 +122,7 @@ def build_image(
normal_deps.extend(api_spec.pip_packages) normal_deps.extend(api_spec.pip_packages)
if build_config.image_type == LlamaStackImageType.CONTAINER.value: if build_config.image_type == LlamaStackImageType.CONTAINER.value:
script = str(importlib.resources.files("llama_stack") / "distribution/build_container.sh") script = str(importlib.resources.files("llama_stack") / "core/build_container.sh")
args = [ args = [
script, script,
"--template-or-config", "--template-or-config",
@ -139,7 +139,7 @@ def build_image(
if run_config is not None: if run_config is not None:
args.extend(["--run-config", run_config]) args.extend(["--run-config", run_config])
elif build_config.image_type == LlamaStackImageType.CONDA.value: elif build_config.image_type == LlamaStackImageType.CONDA.value:
script = str(importlib.resources.files("llama_stack") / "distribution/build_conda_env.sh") script = str(importlib.resources.files("llama_stack") / "core/build_conda_env.sh")
args = [ args = [
script, script,
"--env-name", "--env-name",
@ -150,7 +150,7 @@ def build_image(
" ".join(normal_deps), " ".join(normal_deps),
] ]
elif build_config.image_type == LlamaStackImageType.VENV.value: elif build_config.image_type == LlamaStackImageType.VENV.value:
script = str(importlib.resources.files("llama_stack") / "distribution/build_venv.sh") script = str(importlib.resources.files("llama_stack") / "core/build_venv.sh")
args = [ args = [
script, script,
"--env-name", "--env-name",