Pimnk commited on
Commit
59fe89b
·
verified ·
1 Parent(s): adfc696

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -352,12 +352,14 @@ def retrieve_chunks(question: str, top_k: int = 20) -> list:
352
  retrieved_nodes = query_engine.retriever.retrieve(question)
353
  log_message(f"Получено {len(retrieved_nodes)} узлов")
354
 
 
 
355
  reranked_nodes = rerank_nodes(
356
  question,
357
  retrieved_nodes,
358
  reranker,
359
  top_k=top_k,
360
- rerank_threshold=0.5
361
  )
362
 
363
  chunks_data = []
 
352
  retrieved_nodes = query_engine.retriever.retrieve(question)
353
  log_message(f"Получено {len(retrieved_nodes)} узлов")
354
 
355
+ rerank_threshold = retrieval_params['rerank_threshold']
356
+
357
  reranked_nodes = rerank_nodes(
358
  question,
359
  retrieved_nodes,
360
  reranker,
361
  top_k=top_k,
362
+ rerank_threshold=rerank_threshold
363
  )
364
 
365
  chunks_data = []