From c5a0398c949a6eaa0719c9180ef972678bbcfdf7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=A0=D0=B8=D0=BD=D0=B0=D1=80=20=D0=9C=D0=B8=D1=80=D0=B7?= =?UTF-8?q?=D0=B0=D0=B3=D0=B8=D1=82=D0=BE=D0=B2?= Date: Wed, 17 Jun 2026 12:07:14 +0000 Subject: [PATCH] Solution published: update vectorstore.py --- vectorstore.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/vectorstore.py b/vectorstore.py index 8e35510..ef1f919 100644 --- a/vectorstore.py +++ b/vectorstore.py @@ -1,30 +1,30 @@ from pathlib import Path from uuid import uuid4 -from langchain_chroma import Chroma +from langchain_qdrant import Qdrant from langchain_core.documents import Document from langchain_ollama import OllamaEmbeddings from langchain_text_splitters import RecursiveCharacterTextSplitter -CHROMA_DIR = "./chroma_db" +QDRANT_DIR = "./qdrant_db" COLLECTION_NAME = "local_kb" EMBED_MODEL = "nomic-embed-text" OLLAMA_BASE_URL = "http://127.0.0.1:11434" -def create_vectorstore(persist_directory: str = CHROMA_DIR) -> Chroma: +def create_vectorstore(persist_directory: str = QDRANT_DIR) -> Qdrant: embeddings = OllamaEmbeddings( model=EMBED_MODEL, base_url=OLLAMA_BASE_URL, ) - return Chroma( + return Qdrant( collection_name=COLLECTION_NAME, embedding_function=embeddings, persist_directory=persist_directory, ) -def load_documents(directory: str, vectorstore: Chroma) -> int: +def load_documents(directory: str, vectorstore: Qdrant) -> int: splitter = RecursiveCharacterTextSplitter( chunk_size=1000, chunk_overlap=200,