mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 04:04:14 +00:00
efbe5fd477
- add the `llama stack version` command that accepts modifiers to get additional informations - without any argument it displays the build version as the llama --version command - -o table,json allows to change output format - --dependencies display the dependency list used to build llamastack - --build-info displays the build information - --all display both and build version Signed-off-by: Akram Ben Aissi <akram.benaissi@gmail.com>
53 lines
1.5 KiB
Python
Executable file
53 lines
1.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from setuptools import setup
|
|
from setuptools.command.build_py import build_py
|
|
|
|
|
|
class BuildWithBuildInfo(build_py):
|
|
"""Custom build command that generates build info before building"""
|
|
|
|
def run(self):
|
|
# Generate build info before building
|
|
self.generate_build_info()
|
|
# Run the standard build
|
|
super().run()
|
|
|
|
def generate_build_info(self):
|
|
"""Generate build_info.py with current git information"""
|
|
script_path = Path(__file__).parent / "scripts" / "generate_build_info.py"
|
|
|
|
try:
|
|
# Run the build info generation script
|
|
result = subprocess.run(
|
|
[sys.executable, str(script_path)],
|
|
check=True,
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
print("Build info generation completed successfully")
|
|
if result.stdout:
|
|
print(result.stdout)
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Warning: Failed to generate build info: {e}")
|
|
if e.stderr:
|
|
print(f"Error output: {e.stderr}")
|
|
# Don't fail the build, just continue with default values
|
|
|
|
|
|
if __name__ == "__main__":
|
|
setup(
|
|
cmdclass={
|
|
"build_py": BuildWithBuildInfo,
|
|
},
|
|
)
|