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

Some quick fixes to the CLI behavior to make it consistent

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-28 17:17:46 -07:00
parent f1244f6d9e
commit 3063329dad
5 changed files with 74 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_toolchain/distribution/package.py
View file

@ -28,6 +28,9 @@ class BuildType(Enum):
container = "container"
conda_env = "conda_env"
def descriptor(self) -> str:
return "image" if self == self.container else "env"
class Dependencies(BaseModel):
pip_packages: List[str]
@ -77,12 +80,11 @@ def build_package(
provider = distribution_id if is_stack else api1.provider
api_or_stack = "stack" if is_stack else api1.api.value
build_desc = "image" if build_type == BuildType.container else "env"
build_dir = BUILDS_BASE_DIR / api_or_stack
os.makedirs(build_dir, exist_ok=True)
package_name = f"{build_desc}-{provider}-{name}"
package_name = f"{build_type.descriptor()}-{provider}-{name}"
package_name = package_name.replace("::", "-")
package_file = build_dir / f"{package_name}.yaml"