Add Files

Dockerfile

@@ -3,7 +3,8 @@ FROM python:3.10-slim
 
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
-    PIP_NO_CACHE_DIR=1
+    PIP_NO_CACHE_DIR=1 \
+    CHROMA_DB_PATH=/data/chroma
 
 # System deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -13,6 +14,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 WORKDIR /app
 
+# Ensure writable path for Chroma
+RUN mkdir -p "$CHROMA_DB_PATH" && chmod -R 777 "$CHROMA_DB_PATH"
+
 # Copy requirements first for better caching
 COPY requirements.txt ./
 RUN pip install --upgrade pip && pip install -r requirements.txt

process_aware_rag.py

@@ -13,9 +13,16 @@ class ProcessAwareRAG:
         self.legal_graph = LegalProcessGraph()
         self.legal_graph.load_graph('legal_processes.pkl')
         
-        # Initialize vector store
-        client = chromadb.PersistentClient(path="./legal_vector_db")
-        self.vector_collection = client.get_collection("legal_context")
+        # Initialize vector store (use writable path by default)
+        chroma_path = os.getenv('CHROMA_DB_PATH', '/tmp/legal_vector_db')
+        os.makedirs(chroma_path, exist_ok=True)
+        client = chromadb.PersistentClient(path=chroma_path)
+        
+        # Ensure collection exists
+        try:
+            self.vector_collection = client.get_collection("legal_context")
+        except Exception:
+            self.vector_collection = client.create_collection("legal_context")
         
         # Initialize LLM
         genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))