Merge pull request #25 from zhangw/chroma-telemetry-disable

lightislost · web-flow · commit c14b41ececfb · 2024-03-01T14:07:44.000+08:00
disable the posthog telemetry mechnism that may raise the connection error
diff --git a/coagent/db_handler/vector_db_handler/chroma_handler.py b/coagent/db_handler/vector_db_handler/chroma_handler.py
@@ -16,7 +16,11 @@ def __init__(self, path: str, collection_name: str = ''):
         @param path: path of data
         @collection_name: name of collection
         '''
-        self.client = chromadb.PersistentClient(path)
+        settings = chromadb.get_settings()
+        # disable the posthog telemetry mechnism that may raise the connection error, such as
+        # "requests.exceptions.ConnectTimeout: HTTPSConnectionPool(host='us-api.i.posthog.com', port 443)"
+        settings.anonymized_telemetry = False
+        self.client = chromadb.PersistentClient(path, settings)
         self.client.heartbeat()
 
         if collection_name:
