mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-06-27 18:50:41 +00:00
parent
04592b9590
commit
ee6feaa2d5
1 changed files with 0 additions and 22 deletions
|
@ -15,11 +15,6 @@ from pathlib import Path
|
||||||
|
|
||||||
from rich.progress import Progress, SpinnerColumn, TextColumn
|
from rich.progress import Progress, SpinnerColumn, TextColumn
|
||||||
|
|
||||||
from llama_stack.distribution.build import (
|
|
||||||
SERVER_DEPENDENCIES,
|
|
||||||
get_provider_dependencies,
|
|
||||||
)
|
|
||||||
|
|
||||||
REPO_ROOT = Path(__file__).parent.parent
|
REPO_ROOT = Path(__file__).parent.parent
|
||||||
|
|
||||||
|
|
||||||
|
@ -90,23 +85,6 @@ def check_for_changes(change_tracker: ChangedPathTracker) -> bool:
|
||||||
return has_changes
|
return has_changes
|
||||||
|
|
||||||
|
|
||||||
def collect_template_dependencies(template_dir: Path) -> tuple[str | None, list[str]]:
|
|
||||||
try:
|
|
||||||
module_name = f"llama_stack.templates.{template_dir.name}"
|
|
||||||
module = importlib.import_module(module_name)
|
|
||||||
|
|
||||||
if template_func := getattr(module, "get_distribution_template", None):
|
|
||||||
template = template_func()
|
|
||||||
normal_deps, special_deps = get_provider_dependencies(template)
|
|
||||||
# Combine all dependencies in order: normal deps, special deps, server deps
|
|
||||||
all_deps = sorted(set(normal_deps + SERVER_DEPENDENCIES)) + sorted(set(special_deps))
|
|
||||||
|
|
||||||
return template.name, all_deps
|
|
||||||
except Exception:
|
|
||||||
return None, []
|
|
||||||
return None, []
|
|
||||||
|
|
||||||
|
|
||||||
def pre_import_templates(template_dirs: list[Path]) -> None:
|
def pre_import_templates(template_dirs: list[Path]) -> None:
|
||||||
# Pre-import all template modules to avoid deadlocks.
|
# Pre-import all template modules to avoid deadlocks.
|
||||||
for template_dir in template_dirs:
|
for template_dir in template_dirs:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue