diff --git a/app/service/search_image_with_text/service.py b/app/service/search_image_with_text/service.py index edd4d93..6d4f490 100644 --- a/app/service/search_image_with_text/service.py +++ b/app/service/search_image_with_text/service.py @@ -6,7 +6,7 @@ from chromadb.config import Settings from chromadb.utils.embedding_functions.ollama_embedding_function import OllamaEmbeddingFunction from tqdm import tqdm -from app.core.config import OLLAMA_URL +from app.core.config import OLLAMA_URL, CHROMADB_PATH # 读取 csv 文件 # csv_file_path = r'D:/Files/csv/output/output.csv' @@ -15,7 +15,7 @@ from app.core.config import OLLAMA_URL # df = pd.read_csv(csv_file_path, encoding='Windows-1252') # 创建 Chroma 客户端 -client = chromadb.Client(Settings(is_persistent=True, persist_directory="/vector_db")) +client = chromadb.Client(Settings(is_persistent=True, persist_directory=CHROMADB_PATH)) # client = chromadb.Client(Settings(is_persistent=True, persist_directory="./service/search_image_with_text/vector_db")) # client = chromadb.Client(Settings(is_persistent=True, persist_directory="D:/workspace/AiDLab/vector_db")) # 创建集合