Spaces:
Sleeping
Sleeping
File size: 662 Bytes
ac7ebac | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | """
src/embedding/base_embedder.py
Provides a factory function and base wrapper for embedding models.
Currently wraps HuggingFaceEmbeddings — swap out for OpenAI, Cohere, etc.
"""
from langchain_huggingface import HuggingFaceEmbeddings
DEFAULT_MODEL = "all-MiniLM-L6-v2"
def get_embeddings(model_name: str = DEFAULT_MODEL) -> HuggingFaceEmbeddings:
"""
Return a LangChain-compatible embeddings instance.
Args:
model_name: HuggingFace model identifier (or path to local model).
Returns:
HuggingFaceEmbeddings instance ready for embed_query / embed_documents.
"""
return HuggingFaceEmbeddings(model_name=model_name)
|