Download src/embeddings.py from Devisri515/Agentic_RAG_Knowledge_Search: direct link, hf CLI and curl.
- Browser
- Download file 717 Bytes
-
https://huggingface.co/spaces/Devisri515/Agentic_RAG_Knowledge_Search/resolve/main/src/embeddings.py
- Command line
-
hf download hf://spaces/Devisri515/Agentic_RAG_Knowledge_Search/src/embeddings.py
-
curl -L -o embeddings.py https://huggingface.co/spaces/Devisri515/Agentic_RAG_Knowledge_Search/resolve/main/src/embeddings.py
717 Bytes
| from langchain_huggingface import HuggingFaceEmbeddings | |
| _MODEL_NAME = "all-MiniLM-L6-v2" | |
| _embeddings = None | |
| def get_embeddings() -> HuggingFaceEmbeddings: | |
| """Shared embeddings instance, loaded once and reused by RAG and the evaluator.""" | |
| global _embeddings | |
| if _embeddings is None: | |
| _embeddings = HuggingFaceEmbeddings(model_name=_MODEL_NAME) | |
| return _embeddings | |
| def get_sentence_transformer(): | |
| """The raw SentenceTransformer behind the shared embeddings (used by the evaluator).""" | |
| model = getattr(get_embeddings(), "client", None) | |
| if model is None: | |
| from sentence_transformers import SentenceTransformer | |
| model = SentenceTransformer(_MODEL_NAME) | |
| return model | |