RAG_Chatbot / embeddings /embedding_manager.py
nikhildsst's picture
Upload 6 files
1638245 verified
Raw
History Blame Contribute Delete
380 Bytes
# 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)