Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -68,7 +68,7 @@ async def lifespan(app: FastAPI):
|
|
| 68 |
print("🚀 Initializing models and prompt template...")
|
| 69 |
|
| 70 |
try:
|
| 71 |
-
GOOGLE_API_KEY = os.getenv("
|
| 72 |
print("🔑 gemini_api_key:", "FOUND" if GOOGLE_API_KEY else "NOT FOUND")
|
| 73 |
|
| 74 |
if not GOOGLE_API_KEY:
|
|
@@ -239,15 +239,15 @@ async def run_hackrx(req: RunRequest):
|
|
| 239 |
# end_time2 = time.time() - start_time2
|
| 240 |
# print(f"vector done: {end_time2}")
|
| 241 |
# dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
|
| 242 |
-
dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k":
|
| 243 |
# dense_retriever = vectorstore.as_retriever(search_type="similarity" ,search_kwargs={"k": 11} )
|
| 244 |
|
| 245 |
|
| 246 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|
| 247 |
keyword_retriever = BM25Retriever.from_documents(chunks)
|
| 248 |
-
keyword_retriever.k = 11
|
| 249 |
# dense_retriever = Chroma.from_documents(documents=chunks, embedding=ml_models["embedder"]).as_retriever()
|
| 250 |
-
ensemble_retriever = EnsembleRetriever(retrievers=[keyword_retriever, dense_retriever], weights=[0.3, 0.7],search_kwargs={"k":
|
| 251 |
### to make it faster we are now using our built reranker thats why commenting the code below
|
| 252 |
# compression_retriever = ContextualCompressionRetriever(
|
| 253 |
# base_retriever=ensemble_retriever, base_compressor=ml_models["reranker_compressor"]
|
|
|
|
| 68 |
print("🚀 Initializing models and prompt template...")
|
| 69 |
|
| 70 |
try:
|
| 71 |
+
GOOGLE_API_KEY = os.getenv("gemini_api_key")
|
| 72 |
print("🔑 gemini_api_key:", "FOUND" if GOOGLE_API_KEY else "NOT FOUND")
|
| 73 |
|
| 74 |
if not GOOGLE_API_KEY:
|
|
|
|
| 239 |
# end_time2 = time.time() - start_time2
|
| 240 |
# print(f"vector done: {end_time2}")
|
| 241 |
# dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
|
| 242 |
+
dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 14 ,"lambda_mult": 0.7} ) # prev 16
|
| 243 |
# dense_retriever = vectorstore.as_retriever(search_type="similarity" ,search_kwargs={"k": 11} )
|
| 244 |
|
| 245 |
|
| 246 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|
| 247 |
keyword_retriever = BM25Retriever.from_documents(chunks)
|
| 248 |
+
keyword_retriever.k = 9 #prev 11
|
| 249 |
# dense_retriever = Chroma.from_documents(documents=chunks, embedding=ml_models["embedder"]).as_retriever()
|
| 250 |
+
ensemble_retriever = EnsembleRetriever(retrievers=[keyword_retriever, dense_retriever], weights=[0.3, 0.7],search_kwargs={"k": 14}) #prev 16
|
| 251 |
### to make it faster we are now using our built reranker thats why commenting the code below
|
| 252 |
# compression_retriever = ContextualCompressionRetriever(
|
| 253 |
# base_retriever=ensemble_retriever, base_compressor=ml_models["reranker_compressor"]
|