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

workaround list templates command

Browse source
This commit is contained in:
Xi Yan 2024-10-24 23:05:26 -07:00
parent cb43caa2c3
commit 58c3c45f19
2 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/cli/stack/build.py
View file

@ -16,6 +16,12 @@ TEMPLATES_PATH = (
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"
@lru_cache()
def available_templates_specs() -> List[BuildConfig]:

6
setup.py
View file

@ -29,7 +29,11 @@ setup(
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
url="https://github.com/meta-llama/llama-stack",
packages=find_packages(),
# 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"]},
classifiers=[
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",