Merge pull request #236 from marerem/fix/silence-chromadb-telemetry-warnings
fix: silence ChromaDB telemetry warnings and CoreML segfault on Apple Silicon
This commit is contained in:
+17
-2
@@ -1,6 +1,21 @@
|
|||||||
"""MemPalace — Give your AI a memory. No API key required."""
|
"""MemPalace — Give your AI a memory. No API key required."""
|
||||||
|
|
||||||
from .cli import main
|
import logging
|
||||||
from .version import __version__
|
import os
|
||||||
|
import platform
|
||||||
|
|
||||||
|
from .cli import main # noqa: E402
|
||||||
|
from .version import __version__ # noqa: E402
|
||||||
|
|
||||||
|
# ChromaDB 0.6.x ships a Posthog telemetry client whose capture() signature is
|
||||||
|
# incompatible with the bundled posthog library, producing noisy stderr warnings
|
||||||
|
# on every client operation ("Failed to send telemetry event … capture() takes
|
||||||
|
# 1 positional argument but 3 were given"). Silence just that logger.
|
||||||
|
logging.getLogger("chromadb.telemetry.product.posthog").setLevel(logging.CRITICAL)
|
||||||
|
|
||||||
|
# ONNX Runtime's CoreML provider segfaults during vector queries on Apple Silicon.
|
||||||
|
# Force CPU execution unless the user has explicitly set a preference.
|
||||||
|
if platform.machine() == "arm64" and platform.system() == "Darwin":
|
||||||
|
os.environ.setdefault("ORT_DISABLE_COREML", "1")
|
||||||
|
|
||||||
__all__ = ["main", "__version__"]
|
__all__ = ["main", "__version__"]
|
||||||
|
|||||||
Reference in New Issue
Block a user