禁用遥测功能(Telemetry) #25
在调用 Chroma 时,传入 client_settings=Settings(anonymized_telemetry=False)
This commit is contained in:
+4
-2
@@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user