phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

[CLI] remove dependency on CONDA_PREFIX in CLI (#144)

Browse source 
* remove dependency on CONDA_PREFIX in CLI

* lint

* typo

* more robust
This commit is contained in:
Xi Yan 2024-09-28 16:46:47 -07:00 • committed by GitHub
parent fe460ba103
commit 6a8c2ae1df
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 26 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

23
llama_stack/cli/stack/configure.py
View file

@ -65,18 +65,27 @@ class StackConfigure(Subcommand):
f"Could not find {build_config_file}. Trying conda build name instead...",
color="green",
)
if os.getenv("CONDA_PREFIX"):
if os.getenv("CONDA_PREFIX", ""):
conda_dir = (
Path(os.getenv("CONDA_PREFIX")).parent / f"llamastack-{args.config}"
)
build_config_file = Path(conda_dir) / f"{args.config}-build.yaml"
else:
cprint(
"Cannot find CONDA_PREFIX. Trying default conda path ~/.conda/envs...",
color="green",
)
conda_dir = (
Path(os.path.expanduser("~/.conda/envs")) / f"llamastack-{args.config}"
)
if build_config_file.exists():
with open(build_config_file, "r") as f:
build_config = BuildConfig(**yaml.safe_load(f))
build_config_file = Path(conda_dir) / f"{args.config}-build.yaml"
self._configure_llama_distribution(build_config, args.output_dir)
return
if build_config_file.exists():
with open(build_config_file, "r") as f:
build_config = BuildConfig(**yaml.safe_load(f))
self._configure_llama_distribution(build_config, args.output_dir)
return
# if we get here, we need to try to find the docker image
cprint(