mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-06 04:34:57 +00:00
Merge 26a490b7fc
into a7f9ce9a3a
This commit is contained in:
commit
fa6c5a0ea6
95 changed files with 1854 additions and 465 deletions
|
@ -6,6 +6,8 @@
|
|||
|
||||
import importlib.resources
|
||||
import sys
|
||||
import tomllib
|
||||
from pathlib import Path
|
||||
|
||||
from pydantic import BaseModel
|
||||
from termcolor import cprint
|
||||
|
@ -72,8 +74,13 @@ def get_provider_dependencies(
|
|||
external_provider_deps.append(provider_spec.module)
|
||||
else:
|
||||
external_provider_deps.extend(provider_spec.module)
|
||||
if hasattr(provider_spec, "pip_packages"):
|
||||
deps.extend(provider_spec.pip_packages)
|
||||
|
||||
pyproject = Path(provider_spec.module.replace(".", "/")) / "pyproject.toml"
|
||||
with open(pyproject, "rb") as f:
|
||||
data = tomllib.load(f)
|
||||
|
||||
dependencies = data.get("project", {}).get("dependencies", [])
|
||||
deps.extend(dependencies)
|
||||
if hasattr(provider_spec, "container_image") and provider_spec.container_image:
|
||||
raise ValueError("A stack's dependencies cannot have a container image")
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue