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

Merge branch 'main' into patch-1

Browse source
This commit is contained in:
Yuan Tang 2025-01-17 23:49:05 -05:00 • committed by GitHub
commit f5edd07b29
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
91 changed files with 995 additions and 632 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/templates/template.py
View file

@ -37,7 +37,7 @@ class RunConfigSettings(BaseModel):
self,
name: str,
providers: Dict[str, List[str]],
docker_image: Optional[str] = None,
container_image: Optional[str] = None,
) -> StackRunConfig:
provider_registry = get_provider_registry()
@ -83,7 +83,7 @@ class RunConfigSettings(BaseModel):
return StackRunConfig(
image_name=name,
docker_image=docker_image,
container_image=container_image,
apis=apis,
providers=provider_configs,
metadata_store=SqliteKVStoreConfig.sample_run_config(
@ -112,7 +112,7 @@ class DistributionTemplate(BaseModel):
# Optional configuration
run_config_env_vars: Optional[Dict[str, Tuple[str, str]]] = None
docker_image: Optional[str] = None
container_image: Optional[str] = None
default_models: Optional[List[ModelInput]] = None
@ -121,7 +121,7 @@ class DistributionTemplate(BaseModel):
name=self.name,
distribution_spec=DistributionSpec(
description=self.description,
docker_image=self.docker_image,
container_image=self.container_image,
providers=self.providers,
),
image_type="conda", # default to conda, can be overridden
@ -169,7 +169,7 @@ class DistributionTemplate(BaseModel):
for yaml_pth, settings in self.run_configs.items():
run_config = settings.run_config(
self.name, self.providers, self.docker_image
self.name, self.providers, self.container_image
)
with open(yaml_output_dir / yaml_pth, "w") as f:
yaml.safe_dump(