phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-20 03:40:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add postgres deps to starter distro

Browse source
This commit is contained in:
Ashwin Bharambe 2025-06-03 10:57:08 -07:00
parent 3c9a10d2fe
commit 032f92b3e1
8 changed files with 68 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/templates/template.py
View file

@ -154,6 +154,11 @@ class DistributionTemplate(BaseModel):
available_models_by_provider: dict[str, list[ProviderModelEntry]] | None = None
# we may want to specify additional pip packages without necessarily indicating a
# specific "default" inference store (which is what typically used to dictate additional
# pip packages)
additional_pip_packages: list[str] | None = None
def build_config(self) -> BuildConfig:
additional_pip_packages: list[str] = []
for run_config in self.run_configs.values():
@ -161,6 +166,9 @@ class DistributionTemplate(BaseModel):
if run_config_.inference_store:
additional_pip_packages.extend(run_config_.inference_store.pip_packages)
if self.additional_pip_packages:
additional_pip_packages.extend(self.additional_pip_packages)
return BuildConfig(
distribution_spec=DistributionSpec(
description=self.description,