Dus Tran commited on
Commit
fbac3e6
·
1 Parent(s): 463f005

chore: remove IS_RENDER and gemini imports

Browse files
Files changed (2) hide show
  1. app/core/config.py +1 -6
  2. 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
- IS_RENDER = os.getenv("RENDER") == "true"
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, IS_RENDER
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