| """Patch chromadb DefaultEmbeddingFunction to avoid downloading ONNX model.""" |
| import re, sys, pathlib |
|
|
| def patch_chromadb(): |
| try: |
| import chromadb.utils.embedding_functions as ef_mod |
| except ImportError: |
| print("[chromadb] not installed; skipping patch") |
| return |
| fp = pathlib.Path(ef_mod.__file__) |
| txt = fp.read_text() |
| if "# patched: default embedding disabled" in txt: |
| print("[chromadb] already patched") |
| return |
| new = re.sub( |
| r"def DefaultEmbeddingFunction\(\)[^:]*:.*?return ONNXMiniLM_L6_V2\(\)", |
| "def DefaultEmbeddingFunction():\n # patched: default embedding disabled\n return None", |
| txt, flags=re.DOTALL, count=1, |
| ) |
| if new != txt: |
| fp.write_text(new) |
| print(f"[chromadb] patched: {fp}") |
| else: |
| print("[chromadb] pattern not found, skipping") |
|
|
| def patch_posthog(): |
| try: |
| import posthog |
| except ImportError: |
| print("[posthog] not installed; skipping") |
| return |
| fp = pathlib.Path(posthog.__file__) |
| txt = fp.read_text() |
| if "# patched: telemetry no-op" in txt: |
| print("[posthog] already patched") |
| return |
| new = re.sub(r"def capture\(([^)]*)\):", r"def capture(\1):\n return None # patched: telemetry no-op", txt, count=1) |
| if new != txt: |
| fp.write_text(new) |
| print(f"[posthog] patched: {fp}") |
|
|
| patch_chromadb() |
| patch_posthog() |
|
|
