From fbca51d6da9bce6ed9786a0483173ebfd1dcfd59 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Tue, 17 Dec 2024 12:19:34 -0800
Subject: [PATCH] Fix to conda env build script

---
 llama_stack/distribution/build_conda_env.sh |  4 +++-
 llama_stack/scripts/install_packages.sh     | 15 +++++++++++++++
 2 files changed, 18 insertions(+), 1 deletion(-)
 create mode 100755 llama_stack/scripts/install_packages.sh

diff --git a/llama_stack/distribution/build_conda_env.sh b/llama_stack/distribution/build_conda_env.sh
index 3d582b715..fc1e48665 100755
--- a/llama_stack/distribution/build_conda_env.sh
+++ b/llama_stack/distribution/build_conda_env.sh
@@ -83,7 +83,9 @@ ensure_conda_env_python310() {
     # these packages are damaged in test-pypi, so install them first
     $CONDA_PREFIX/bin/pip install fastapi libcst
     $CONDA_PREFIX/bin/pip install --extra-index-url https://test.pypi.org/simple/ \
-      llama-models==$TEST_PYPI_VERSION llama-stack==$TEST_PYPI_VERSION \
+      llama-models==$TEST_PYPI_VERSION \
+      llama-stack-client==$TEST_PYPI_VERSION \
+      llama-stack==$TEST_PYPI_VERSION \
       $pip_dependencies
     if [ -n "$special_pip_deps" ]; then
       IFS='#' read -ra parts <<<"$special_pip_deps"
diff --git a/llama_stack/scripts/install_packages.sh b/llama_stack/scripts/install_packages.sh
new file mode 100755
index 000000000..151b7b9db
--- /dev/null
+++ b/llama_stack/scripts/install_packages.sh
@@ -0,0 +1,15 @@
+#!/bin/bash
+
+# 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.
+
+VERSION="$1"
+
+set -euo pipefail
+set -x
+
+pip install -U --extra-index-url https://test.pypi.org/simple \
+  llama-stack==$VERSION llama-models==$VERSION llama-stack-client==$VERSION