Daniel McKnight
Add configuration support (#3)
b7dce4c unverified
Raw
History Blame Contribute Delete
392 Bytes
from sentence_transformers import SentenceTransformer
from app.config import get_settings
settings = get_settings()
# Using a compact, fast model good for semantic search — model name from config
model = SentenceTransformer(settings.rag.embedding_model)
def get_embedding(text: str) -> list[float]:
embedding = model.encode(text, convert_to_numpy=True)
return embedding.tolist()