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):
|