禁用遥测功能(Telemetry) #25

在调用 Chroma 时,传入 client_settings=Settings(anonymized_telemetry=False)
This commit is contained in:
YILING0013
2025-02-03 20:53:37 +08:00
parent 90b7b77147
commit dd871741ed
+4 -2
View File
@@ -10,6 +10,7 @@ from typing import List, Optional
from langchain_openai import ChatOpenAI from langchain_openai import ChatOpenAI
from langchain_openai import OpenAIEmbeddings from langchain_openai import OpenAIEmbeddings
from langchain_community.vectorstores import Chroma from langchain_community.vectorstores import Chroma
from chromadb.config import Settings
from langchain.docstore.document import Document from langchain.docstore.document import Document
# nltk、sentence_transformers 及文本处理相关 # nltk、sentence_transformers 及文本处理相关
@@ -185,7 +186,8 @@ def init_vector_store(
vectorstore = Chroma.from_documents( vectorstore = Chroma.from_documents(
documents, documents,
embedding=embeddings, embedding=embeddings,
persist_directory=VECTOR_STORE_DIR persist_directory=VECTOR_STORE_DIR,
client_settings=Settings(anonymized_telemetry=False)
) )
vectorstore.persist() vectorstore.persist()
return vectorstore return vectorstore
@@ -213,7 +215,7 @@ def load_vector_store(
interface_format=interface_format, interface_format=interface_format,
embedding_model_name=embedding_model_name embedding_model_name=embedding_model_name
) )
return Chroma(persist_directory=VECTOR_STORE_DIR, embedding_function=embeddings) return Chroma(persist_directory=VECTOR_STORE_DIR, embedding_function=embeddings,client_settings=Settings(anonymized_telemetry=False))
def update_vector_store( def update_vector_store(