src/embedding.py

# File: src/embedding.py
import os
from sentence_transformers import SentenceTransformer
from config.settings import Settings

# Chỉ định thư mục cache
os.environ["HF_HOME"] = "/tmp/cache"

class Embedding:
    def __init__(self):
        self.model = SentenceTransformer(Settings.MODEL_NAME)
    
    def generate(self, text):
        return self.model.encode(text, convert_to_numpy=True)