Disable chromaDB anonymous information collection (#1144)

See https://docs.trychroma.com/telemetry
zylon-ai · Nov 2, 2023 · f29df84 · f29df84
1 parent a517a58
commit f29df84
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/private_gpt/components/vector_store/vector_store_component.py b/private_gpt/components/vector_store/vector_store_component.py
@@ -1,6 +1,7 @@
 import typing
 
 import chromadb
+from chromadb.config import Settings
 from injector import inject, singleton
 from llama_index import VectorStoreIndex
 from llama_index.indices.vector_store import VectorIndexRetriever
@@ -36,8 +37,10 @@ class VectorStoreComponent:
 
     @inject
     def __init__(self) -> None:
+        chroma_settings = Settings(anonymized_telemetry=False)
         chroma_client = chromadb.PersistentClient(
-            path=str((local_data_path / "chroma_db").absolute())
+            path=str((local_data_path / "chroma_db").absolute()),
+            settings=chroma_settings,
         )
         chroma_collection = chroma_client.get_or_create_collection(
             "make_this_parameterizable_per_api_call"