From 0dc08e205b42eed7b89f6c1a3f80bbde100ff9f0 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: Thu, 18 Jun 2026 10:50:37 +0000 Subject: [PATCH] =?UTF-8?q?=D0=9F=D1=83=D0=B1=D0=BB=D0=B8=D0=BA=D0=B0?= =?UTF-8?q?=D1=86=D0=B8=D1=8F=20=D1=80=D0=B5=D1=88=D0=B5=D0=BD=D0=B8=D1=8F?= =?UTF-8?q?=20=D0=B4=D0=BB=D1=8F=20taskId=206a1d75c5fd30e81cf3126ae7:=20up?= =?UTF-8?q?date=20src/utils.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/utils.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/utils.py b/src/utils.py index 0f0559a..2237382 100644 --- a/src/utils.py +++ b/src/utils.py @@ -9,6 +9,9 @@ from langchain_ollama import OllamaEmbeddings # Chroma collection name COLLECTION_NAME = "faq" +# Persistence directory from config +from src.config import CHROMA_PERSIST_DIR + def load_faq_to_chroma(md_dir: str = "data") -> None: """Load all .md files from md_dir into a Chroma vector store. @@ -22,21 +25,23 @@ def load_faq_to_chroma(md_dir: str = "data") -> None: all_docs.extend(docs) splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50) docs = splitter.split_documents(all_docs) - embeddings = OllamaEmbeddings(model="nomic-embed-text") - # Create or update Chroma collection + embeddings = OllamaEmbeddings("nomic-embed-text") + # Create or update Chroma collection with persistence Chroma.from_documents( docs, embeddings, collection_name=COLLECTION_NAME, + persist_directory=CHROMA_PERSIST_DIR, ) def search_course_docs(query: str, k: int = 3) -> List[str]: """Return top k document snippets from the persisted Chroma store.""" - embeddings = OllamaEmbeddings(model="nomic-embed-text") + embeddings = OllamaEmbeddings("nomic-embed-text") store = Chroma( collection_name=COLLECTION_NAME, embedding_function=embeddings, + persist_directory=CHROMA_PERSIST_DIR, ) results = store.similarity_search(query, k=k) return [doc.page_content for doc in results]