install jupyter-ai packages with onyxia init-script (#310)

Author: ihiverlet

base/scripts/onyxia-init.sh

@@ -163,6 +163,15 @@ if [[ "$DARK_MODE" == "true" ]]; then
     fi
 fi
 
+# install packages for jupyter-ai
+if [[ "$ENABLE_JUPYTER_AI_EXTENSION" == "true" ]]; then
+    if command -v uv &>/dev/null; then
+        uv pip install --system --no-cache jupyter-ai langchain-ollama langchain-openai
+    else
+        pip install --no-cache-dir jupyter-ai langchain-ollama langchain-openai
+    fi
+fi
+
 # inject proxy variables
 env_vars=("NO_PROXY" "no_proxy" "HTTP_PROXY" "http_proxy" "HTTPS_PROXY" "https_proxy")
 

jupyter/scripts/install-jupyterlab.sh

@@ -10,14 +10,8 @@ function apt_install() {
     fi
 }
 
-PACKAGES=("jupyterlab")
-
-if [[ "$ENABLE_JUPYTER_AI_EXTENSION" == "true" ]]; then
-    PACKAGES+=("jupyter-ai" "langchain-ollama" "langchain-openai")
-fi
-
 if command -v uv &>/dev/null; then
-    uv pip install --system --no-cache "${PACKAGES[@]}"
+    uv pip install --system --no-cache jupyterlab
 else
-    pip install --no-cache-dir "${PACKAGES[@]}"
+    pip install --no-cache-dir jupyterlab
 fi