Spaces:
Sleeping
Sleeping
Priyansh Saxena commited on
Commit ·
9c92c35
1
Parent(s): 3240afb
Migration Phase 7: Retrieval Augmented Generation components
Browse files- rag/embeddings.py +3 -5
rag/embeddings.py
CHANGED
|
@@ -1,9 +1,7 @@
|
|
| 1 |
from langchain_openai import OpenAIEmbeddings
|
| 2 |
-
from langchain_huggingface import HuggingFaceEmbeddings
|
| 3 |
-
import os
|
| 4 |
|
| 5 |
def get_embeddings():
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
return OpenAIEmbeddings()
|
|
|
|
| 1 |
from langchain_openai import OpenAIEmbeddings
|
|
|
|
|
|
|
| 2 |
|
| 3 |
def get_embeddings():
|
| 4 |
+
"""
|
| 5 |
+
Returns the embedding model used for the RAG pipeline.
|
| 6 |
+
"""
|
| 7 |
return OpenAIEmbeddings()
|