Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
5407e1f
1
Parent(s):
e8ef27e
remove telementry
Browse files
utils.py
CHANGED
|
@@ -8,6 +8,7 @@ from dotenv import load_dotenv
|
|
| 8 |
|
| 9 |
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
| 10 |
|
|
|
|
| 11 |
|
| 12 |
# Set up logging
|
| 13 |
logging.basicConfig(
|
|
@@ -35,7 +36,10 @@ def get_save_path() -> Literal["chroma/"] | Literal["/data/chroma/"]:
|
|
| 35 |
def get_chroma_client():
|
| 36 |
logger.info("Initializing Chroma client")
|
| 37 |
SAVE_PATH = get_save_path()
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
|
| 41 |
def get_collection(chroma_client, embedding_function, collection_name):
|
|
|
|
| 8 |
|
| 9 |
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
| 10 |
|
| 11 |
+
from chromadb.config import Settings
|
| 12 |
|
| 13 |
# Set up logging
|
| 14 |
logging.basicConfig(
|
|
|
|
| 36 |
def get_chroma_client():
|
| 37 |
logger.info("Initializing Chroma client")
|
| 38 |
SAVE_PATH = get_save_path()
|
| 39 |
+
|
| 40 |
+
return chromadb.PersistentClient(
|
| 41 |
+
path=SAVE_PATH, settings=Settings(anonymized_telemetry=False)
|
| 42 |
+
)
|
| 43 |
|
| 44 |
|
| 45 |
def get_collection(chroma_client, embedding_function, collection_name):
|