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()