phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-04 04:42:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

tmp dockerfile_only flag

Browse source
This commit is contained in:
Xi Yan 2025-01-14 13:59:20 -08:00
parent 91907b714e
commit 4747c844a2
3 changed files with 35 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/distribution/build.py
View file

@ -102,7 +102,9 @@ def print_pip_install_help(providers: Dict[str, List[Provider]]):
print()
def build_image(build_config: BuildConfig, build_file_path: Path):
def build_image(
build_config: BuildConfig, build_file_path: Path, dockerfile_only: bool = False
):
docker_image = build_config.distribution_spec.docker_image or "python:3.10-slim"
normal_deps, special_deps = get_provider_dependencies(
@ -121,6 +123,7 @@ def build_image(build_config: BuildConfig, build_file_path: Path):
str(build_file_path),
str(BUILDS_BASE_DIR / ImageType.docker.value),
" ".join(normal_deps),
dockerfile_only,
]
elif build_config.image_type == ImageType.conda.value:
script = (