File size: 380 Bytes
1638245
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
# embeddings/embedding_manager.py
from sentence_transformers import SentenceTransformer
from config import Config

class EmbeddingManager:
    def __init__(self):
        self.model = SentenceTransformer(Config.EMBEDDING_MODEL)

    def get_embedding(self, text):
        return self.model.encode(text)

    def get_embeddings(self, texts):
        return self.model.encode(texts)