phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 12:07:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add build yaml to Dockerfile

Browse source
This commit is contained in:
Xi Yan 2024-09-14 23:18:50 -07:00
parent f639e7f80e
commit b9f454f613
4 changed files with 27 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_toolchain/core/package.py
View file

@ -21,6 +21,8 @@ from pydantic import BaseModel
from termcolor import cprint
from llama_toolchain.core.datatypes import * # noqa: F403
from pathlib import Path
from llama_toolchain.core.distribution import api_providers, SERVER_DEPENDENCIES
@ -39,7 +41,7 @@ class ApiInput(BaseModel):
provider: str
def build_package(build_config: BuildConfig):
def build_package(build_config: BuildConfig, build_file_path: Path):
package_deps = Dependencies(
docker_image=build_config.distribution_spec.docker_image or "python:3.10-slim",
pip_packages=SERVER_DEPENDENCIES,
@ -67,6 +69,7 @@ def build_package(build_config: BuildConfig):
script,
build_config.name,
package_deps.docker_image,
str(build_file_path),
" ".join(package_deps.pip_packages),
]
else: