singhankur01 commited on
Commit
b5046cc
·
verified ·
1 Parent(s): d90ea5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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("gemini_api_key3")
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": 16 ,"lambda_mult": 0.7} )
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": 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"]
 
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"]