From f29df84301ff2894e26f520c488310750c5b4f7c Mon Sep 17 00:00:00 2001 From: Pierre Marais Date: Thu, 2 Nov 2023 12:45:48 +0100 Subject: [PATCH] Disable chromaDB anonymous information collection (#1144) See https://docs.trychroma.com/telemetry --- .../components/vector_store/vector_store_component.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/private_gpt/components/vector_store/vector_store_component.py b/private_gpt/components/vector_store/vector_store_component.py index 5d173fb57..ce7010bb9 100644 --- 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"