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

Update data_query.py

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