Merge pull request #54 from mohit-sheth/rag

slim down image and drop NVIDIA deps

Author: vishnuchalla

requirements.txt

@@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
 aiohappyeyeballs==2.6.1
 aiohttp==3.11.14
 aiosignal==1.3.2
@@ -106,21 +107,6 @@ networkx==3.5
 nltk==3.9.2
 nodeenv==1.9.1
 numpy==2.2.4
-nvidia-cublas-cu12==12.8.4.1
-nvidia-cuda-cupti-cu12==12.8.90
-nvidia-cuda-nvrtc-cu12==12.8.93
-nvidia-cuda-runtime-cu12==12.8.90
-nvidia-cudnn-cu12==9.10.2.21
-nvidia-cufft-cu12==11.3.3.83
-nvidia-cufile-cu12==1.13.1.3
-nvidia-curand-cu12==10.3.9.90
-nvidia-cusolver-cu12==11.7.3.90
-nvidia-cusparse-cu12==12.5.8.93
-nvidia-cusparselt-cu12==0.7.1
-nvidia-nccl-cu12==2.27.5
-nvidia-nvjitlink-cu12==12.8.93
-nvidia-nvshmem-cu12==3.3.20
-nvidia-nvtx-cu12==12.8.90
 ollama==0.6.0
 openai==1.109.1
 orjson==3.10.15
@@ -179,10 +165,9 @@ threadpoolctl==3.6.0
 tiktoken==0.12.0
 tokenizers==0.22.1
 tomlkit==0.13.2
-torch==2.9.0
+torch==2.9.0+cpu
 tqdm==4.67.1
 transformers==4.57.1
-triton==3.5.0
 trove-classifiers==2025.1.6.15
 truststore==0.9.2
 typing-inspect==0.9.0