From dd871741ed677bed383014af52216fdd1a00fc50 Mon Sep 17 00:00:00 2001 From: YILING0013 Date: Mon, 3 Feb 2025 20:53:37 +0800 Subject: [PATCH] =?UTF-8?q?=E7=A6=81=E7=94=A8=E9=81=A5=E6=B5=8B=E5=8A=9F?= =?UTF-8?q?=E8=83=BD(Telemetry)=20#25?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 在调用 Chroma 时，传入 client_settings=Settings(anonymized_telemetry=False) --- novel_generator.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/novel_generator.py b/novel_generator.py index 0fcadfd..f394186 100644 --- a/novel_generator.py +++ b/novel_generator.py @@ -10,6 +10,7 @@ from typing import List, Optional from langchain_openai import ChatOpenAI from langchain_openai import OpenAIEmbeddings from langchain_community.vectorstores import Chroma +from chromadb.config import Settings from langchain.docstore.document import Document # nltk、sentence_transformers 及文本处理相关 @@ -185,7 +186,8 @@ def init_vector_store( vectorstore = Chroma.from_documents( documents, embedding=embeddings, - persist_directory=VECTOR_STORE_DIR + persist_directory=VECTOR_STORE_DIR, + client_settings=Settings(anonymized_telemetry=False) ) vectorstore.persist() return vectorstore @@ -213,7 +215,7 @@ def load_vector_store( interface_format=interface_format, 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(