Allow users to build RAG versus Docling images

Signed-off-by: Daniel J Walsh <[email protected]>
containers · Feb 5, 2025 · b42085c · b42085c
1 parent 032efae
commit b42085c
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/container-images/pragmatic/build_pragmatic.sh b/container-images/pragmatic/build_pragmatic.sh
@@ -27,14 +27,19 @@ install_pytorch() {
 }
 
 clone_and_build_pragmatic() {
-if [ "$2" == "docling" ]; then
-   ${PYTHON_VERSION} pip install docling --extra-index-url https://download.pytorch.org/whl/$1
-fi
   git clone https://github.com/redhat-et/PRAGmatic
   cd PRAGmatic
   git submodule update --init --recursive
 
-  ${PYTHON_VERSION} pip install -r requirements.txt --prefix=/usr
+  if [ "$2" == "docling" ]; then
+      ${PYTHON_VERSION} pip install docling --extra-index-url https://download.pytorch.org/whl/$1
+      ${PYTHON_VERSION} pip install -r requirements.txt --prefix=/usr
+  else
+      ${PYTHON_VERSION} pip install torch --extra-index-url https://download.pytorch.org/whl/$1
+      grep -v docling requirements.txt > /tmp/nodocling.txt
+      cat /tmp/nodocling.txt
+      ${PYTHON_VERSION} pip install -r /tmp/nodocling.txt --prefix=/usr
+  fi
   ${PYTHON_VERSION} pip install --prefix=/usr .
   cd ..
 }