src/retrieval/vector_store.py

from langchain_chroma import Chroma
from langchain_huggingface import HuggingFaceEmbeddings

CHROMA_DB_DIR = "chroma_data"
COLLECTION_NAME = "chilicare_kb"

def get_embedder():
    return HuggingFaceEmbeddings(model_name="Qwen/Qwen3-Embedding-0.6B")

def get_vector_store():
    embedder = get_embedder()
    return Chroma(
        persist_directory=CHROMA_DB_DIR, 
        embedding_function=embedder,
        collection_name=COLLECTION_NAME
    )