models/embedder.py

from sentence_transformers import SentenceTransformer 
import chromadb 
 
class Embedder: 
    def __init__(self, model_name="all-MiniLM-L6-v2"): 
        self.model = SentenceTransformer(model_name) 
        self.client = chromadb.Client() 
        self.collection = self.client.create_collection(name="documents") 
 
    def embed(self, texts): 
        return self.model.encode(texts).tolist()