This commit is contained in:
Xi Yan 2024-10-25 10:43:22 -07:00
parent 58c3c45f19
commit 81ed0327f3
11 changed files with 12 additions and 7 deletions

View file

@ -13,14 +13,14 @@ from functools import lru_cache
from pathlib import Path
TEMPLATES_PATH = (
Path(os.path.relpath(__file__)).parent.parent.parent.parent / "distributions"
Path(os.path.relpath(__file__)).parent.parent.parent.parent / "distributions----"
)
# build.yaml templates exist in the llama-stack/distributions while wheel installs llama-stack/llama_stack
# we copied the distributions folder to llama-stack/llama_stack/cli/distributions for wheel builds,
# so we need to check both locations
if not TEMPLATES_PATH.exists():
TEMPLATES_PATH = Path(os.path.relpath(__file__)).parent.parent / "distributions"
TEMPLATES_PATH = Path(os.path.relpath(__file__)).parent.parent.parent / "templates"
@lru_cache()

View file

@ -0,0 +1 @@
../../distributions/bedrock/build.yaml

View file

@ -0,0 +1 @@
../../distributions/fireworks/build.yaml

View file

@ -0,0 +1 @@
../../distributions/hf-endpoint/build.yaml

View file

@ -0,0 +1 @@
../../distributions/hf-serverless/build.yaml

View file

@ -0,0 +1 @@
../../distributions/meta-reference-gpu/build.yaml

View file

@ -0,0 +1 @@
../../distributions/ollama/build.yaml

View file

@ -0,0 +1 @@
../../distributions/tgi/build.yaml

View file

@ -0,0 +1 @@
../../distributions/together/build.yaml

View file

@ -0,0 +1 @@
../../distributions/vllm/build.yaml

View file

@ -29,11 +29,7 @@ setup(
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
url="https://github.com/meta-llama/llama-stack",
# workaround to include distributions/*/build.yaml files in the package
# https://stackoverflow.com/questions/62550952/including-package-data-python-from-top-level-when-package-is-in-subdirectory
packages=find_packages() + ["llama_stack.cli.distributions"],
package_dir={"llama_stack.cli.distributions": "distributions"},
package_data={"llama_stack.cli.distributions": ["distributions/*/build.yaml"]},
packages=find_packages(),
classifiers=[
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",