From 3b4351062f1fb3fe2f18419849f7f540663766af Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Wed, 18 Sep 2024 11:04:11 -0700
Subject: [PATCH] move build.yaml

---
 llama_stack/cli/stack/build.py     | 5 ++++-
 llama_stack/cli/stack/configure.py | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py
index 70654a67e..112a5d74e 100644
--- a/llama_stack/cli/stack/build.py
+++ b/llama_stack/cli/stack/build.py
@@ -63,7 +63,10 @@ class StackBuild(Subcommand):
                 llama_stack_path / "configs/distributions" / build_config.image_type
             )
         else:
-            build_dir = Path(os.getenv("CONDA_PREFIX")).parent
+            build_dir = (
+                Path(os.getenv("CONDA_PREFIX")).parent
+                / f"llamastack-{build_config.name}"
+            )
 
         os.makedirs(build_dir, exist_ok=True)
         build_file_path = build_dir / f"{build_config.name}-build.yaml"
diff --git a/llama_stack/cli/stack/configure.py b/llama_stack/cli/stack/configure.py
index 364345669..8c7f6dc37 100644
--- a/llama_stack/cli/stack/configure.py
+++ b/llama_stack/cli/stack/configure.py
@@ -52,7 +52,7 @@ class StackConfigure(Subcommand):
         from llama_stack.distribution.build import ImageType
 
         docker_image = None
-        conda_dir = Path(os.getenv("CONDA_PREFIX")).parent
+        conda_dir = Path(os.getenv("CONDA_PREFIX")).parent / f"llamastack-{args.config}"
         build_config_file = Path(conda_dir) / f"{args.config}-build.yaml"
 
         if not build_config_file.exists():