Add Files

Dockerfile

@@ -21,8 +21,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 WORKDIR /app
 
 # Ensure writable paths for Chroma and model caches
-RUN mkdir -p "$CHROMA_DB_PATH" "$CACHE_ROOT" "$HF_HOME" "$TRANSFORMERS_CACHE" "$SENTENCE_TRANSFORMERS_HOME" "$HOME/.cache" \
-    && chmod -R 777 "$CHROMA_DB_PATH" "$CACHE_ROOT"
+RUN mkdir -p "$CHROMA_DB_PATH" "$CACHE_ROOT" "$HF_HOME" "$TRANSFORMERS_CACHE" "$SENTENCE_TRANSFORMERS_HOME" "$HOME/.cache" "$HOME/.cache/chroma" \
+    && chmod -R 777 "$CHROMA_DB_PATH" "$CACHE_ROOT" "$HOME/.cache"
 
 # Copy requirements first for better caching
 COPY requirements.txt ./

process_aware_rag.py

@@ -24,6 +24,7 @@ class ProcessAwareRAG:
         os.environ.setdefault('HOME', '/data')
         os.makedirs(default_cache_root, exist_ok=True)
         os.makedirs(os.path.join(os.environ['HOME'], '.cache'), exist_ok=True)
+        os.makedirs(os.path.join(os.environ['HOME'], '.cache', 'chroma'), exist_ok=True)
         os.environ.setdefault('HF_HOME', os.path.join(default_cache_root, 'hf'))
         os.environ.setdefault('TRANSFORMERS_CACHE', os.path.join(default_cache_root, 'transformers'))
         os.environ.setdefault('SENTENCE_TRANSFORMERS_HOME', os.path.join(default_cache_root, 'sentence-transformers'))

vector_store_builder.py

@@ -17,6 +17,7 @@ def build_vector_store():
     os.environ.setdefault('HOME', os.path.abspath('.'))
     os.makedirs(default_cache_root, exist_ok=True)
     os.makedirs(os.path.join(os.environ['HOME'], '.cache'), exist_ok=True)
+    os.makedirs(os.path.join(os.environ['HOME'], '.cache', 'chroma'), exist_ok=True)
     os.environ.setdefault('HF_HOME', os.path.join(default_cache_root, 'hf'))
     os.environ.setdefault('TRANSFORMERS_CACHE', os.path.join(default_cache_root, 'transformers'))
     os.environ.setdefault('SENTENCE_TRANSFORMERS_HOME', os.path.join(default_cache_root, 'sentence-transformers'))