Rivalcoder commited on
Commit
dff2a93
·
1 Parent(s): 0ec83d3
Files changed (3) hide show
  1. Dockerfile +2 -0
  2. process_aware_rag.py +6 -1
  3. vector_store_builder.py +4 -0
Dockerfile CHANGED
@@ -7,6 +7,8 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
7
  CHROMA_DB_PATH=/data/chroma \
8
  CACHE_ROOT=/data/cache \
9
  HOME=/data \
 
 
10
  HF_HOME=/data/cache/hf \
11
  TRANSFORMERS_CACHE=/data/cache/transformers \
12
  SENTENCE_TRANSFORMERS_HOME=/data/cache/sentence-transformers \
 
7
  CHROMA_DB_PATH=/data/chroma \
8
  CACHE_ROOT=/data/cache \
9
  HOME=/data \
10
+ POSTHOG_DISABLED=true \
11
+ CHROMA_TELEMETRY_DISABLED=true \
12
  HF_HOME=/data/cache/hf \
13
  TRANSFORMERS_CACHE=/data/cache/transformers \
14
  SENTENCE_TRANSFORMERS_HOME=/data/cache/sentence-transformers \
process_aware_rag.py CHANGED
@@ -1,11 +1,16 @@
1
  # process_aware_rag.py
 
 
 
 
 
 
2
  import chromadb
3
  from chromadb.config import Settings
4
  from chromadb.utils import embedding_functions
5
  import google.generativeai as genai
6
  from query_classifier import QueryClassifier
7
  from graph_builder import LegalProcessGraph
8
- import os
9
  from typing import Dict, List, Any
10
 
11
  class ProcessAwareRAG:
 
1
  # process_aware_rag.py
2
+ import os
3
+
4
+ # Disable telemetry before importing chromadb
5
+ os.environ.setdefault("POSTHOG_DISABLED", "true")
6
+ os.environ.setdefault("CHROMA_TELEMETRY_DISABLED", "true")
7
+
8
  import chromadb
9
  from chromadb.config import Settings
10
  from chromadb.utils import embedding_functions
11
  import google.generativeai as genai
12
  from query_classifier import QueryClassifier
13
  from graph_builder import LegalProcessGraph
 
14
  from typing import Dict, List, Any
15
 
16
  class ProcessAwareRAG:
vector_store_builder.py CHANGED
@@ -1,5 +1,9 @@
1
  # vector_store_builder.py
2
  import os
 
 
 
 
3
  import chromadb
4
  from chromadb.config import Settings
5
  from chromadb.utils import embedding_functions
 
1
  # vector_store_builder.py
2
  import os
3
+
4
+ # Disable telemetry before importing chromadb
5
+ os.environ.setdefault("POSTHOG_DISABLED", "true")
6
+ os.environ.setdefault("CHROMA_TELEMETRY_DISABLED", "true")
7
  import chromadb
8
  from chromadb.config import Settings
9
  from chromadb.utils import embedding_functions