Spaces:
Sleeping
Sleeping
Dus Tran commited on
Commit ·
fbac3e6
1
Parent(s): 463f005
chore: remove IS_RENDER and gemini imports
Browse files- app/core/config.py +1 -6
- app/service/retriever.py +1 -1
app/core/config.py
CHANGED
|
@@ -4,12 +4,7 @@ from langchain_groq import ChatGroq
|
|
| 4 |
|
| 5 |
load_dotenv()
|
| 6 |
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
if not IS_RENDER:
|
| 10 |
-
from langchain_huggingface import HuggingFaceEmbeddings
|
| 11 |
-
else:
|
| 12 |
-
from langchain_google_genai import GoogleGenerativeAIEmbeddings
|
| 13 |
|
| 14 |
BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
| 15 |
DATA_PATH = os.path.join(BASE_DIR, "data")
|
|
|
|
| 4 |
|
| 5 |
load_dotenv()
|
| 6 |
|
| 7 |
+
from langchain_huggingface import HuggingFaceEmbeddings
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
| 10 |
DATA_PATH = os.path.join(BASE_DIR, "data")
|
app/service/retriever.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from qdrant_client import QdrantClient
|
| 2 |
from qdrant_client.http import models
|
| 3 |
-
from app.core.config import QDRANT_PATH, RERANK_MODEL_NAME, get_embeddings, COLLECTION_NAME
|
| 4 |
|
| 5 |
_rerank_model = None
|
| 6 |
_qdrant_client = None
|
|
|
|
| 1 |
from qdrant_client import QdrantClient
|
| 2 |
from qdrant_client.http import models
|
| 3 |
+
from app.core.config import QDRANT_PATH, RERANK_MODEL_NAME, get_embeddings, COLLECTION_NAME
|
| 4 |
|
| 5 |
_rerank_model = None
|
| 6 |
_qdrant_client = None
|