Update data_query.py

data_query.py

@@ -83,7 +83,7 @@ class ChatbotDataQuery:
                         meta_data = context_docs[idx_of_content_in_context_doc].metadata
                         final_reranked_docs.append(Document(page_content=reranked_doc['content'], metadata=meta_data))
                 
-                context_docs = final_reranked_docs[0]
+                context_docs = [final_reranked_docs[0]]
 
             prompt = ChatPromptTemplate.from_template(
                 "You are a helpful assistant that only answers questions about the context. "