diff --git a/docker/main/rootfs/etc/s6-overlay/s6-rc.d/chroma/run b/docker/main/rootfs/etc/s6-overlay/s6-rc.d/chroma/run index 45b44779b..bf28a56b4 100755 --- a/docker/main/rootfs/etc/s6-overlay/s6-rc.d/chroma/run +++ b/docker/main/rootfs/etc/s6-overlay/s6-rc.d/chroma/run @@ -13,4 +13,4 @@ echo "[INFO] Starting ChromaDB..." # Replace the bash process with the Frigate process, redirecting stderr to stdout exec 2>&1 -exec /usr/local/chroma run --path /config/chroma --host 127.0.0.1 +exec /usr/local/chroma run --path /config/chroma --host 0.0.0.0 diff --git a/frigate/embeddings/processor.py b/frigate/embeddings/processor.py index b58bfdb4e..15908c9f3 100644 --- a/frigate/embeddings/processor.py +++ b/frigate/embeddings/processor.py @@ -53,7 +53,9 @@ class EmbeddingProcessor(threading.Thread): return # Create the database - self.chroma = ChromaClient(settings=Settings(anonymized_telemetry=False)) + self.chroma = ChromaClient( + host="127.0.0.1", settings=Settings(anonymized_telemetry=False) + ) # Create/Load the collection(s) self.thumbnail = self.chroma.get_or_create_collection( diff --git a/frigate/http.py b/frigate/http.py index 5cded11b6..2be2a3961 100644 --- a/frigate/http.py +++ b/frigate/http.py @@ -108,7 +108,9 @@ def create_app( app.plus_api = plus_api app.camera_error_image = None app.hwaccel_errors = [] - app.chroma = ChromaClient(settings=Settings(anonymized_telemetry=False)) + app.chroma = ChromaClient( + host="127.0.0.1", settings=Settings(anonymized_telemetry=False) + ) app.thumbnail_collection = app.chroma.get_or_create_collection( name="event_thumbnail", embedding_function=ClipEmbedding() ) diff --git a/frigate/test/__init__.py b/frigate/test/__init__.py index e69de29bb..11f933bdd 100644 --- a/frigate/test/__init__.py +++ b/frigate/test/__init__.py @@ -0,0 +1,5 @@ +# This is required to satisfy the chromadb dependency +__import__("pysqlite3") +import sys # noqa: E402 + +sys.modules["sqlite3"] = sys.modules.pop("pysqlite3")