SalehAhmad commited on
Commit
272b678
·
verified ·
1 Parent(s): 2fb2862

Update data_query.py

Browse files
Files changed (1) hide show
  1. data_query.py +2 -1
data_query.py CHANGED
@@ -76,11 +76,12 @@ class ChatbotDataQuery:
76
 
77
  final_reranked_docs = []
78
  for reranked_doc in relevant_docs:
79
- if reranked_doc['score'] < 0.35:
80
  continue
81
  else:
82
  idx_of_content_in_context_doc = reranked_doc['result_index']
83
  meta_data = context_docs[idx_of_content_in_context_doc].metadata
 
84
  final_reranked_docs.append(Document(page_content=reranked_doc['content'], metadata=meta_data))
85
 
86
  context_docs = [final_reranked_docs[0]]
 
76
 
77
  final_reranked_docs = []
78
  for reranked_doc in relevant_docs:
79
+ if reranked_doc['score'] < 0.50:
80
  continue
81
  else:
82
  idx_of_content_in_context_doc = reranked_doc['result_index']
83
  meta_data = context_docs[idx_of_content_in_context_doc].metadata
84
+ print(f'Reranked_Doc: {meta_data}')
85
  final_reranked_docs.append(Document(page_content=reranked_doc['content'], metadata=meta_data))
86
 
87
  context_docs = [final_reranked_docs[0]]