Spaces:
Runtime error
Runtime error
Update data_query.py
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
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]]
|