Update core/chatbot/retrieval_chatbot.py

core/chatbot/retrieval_chatbot.py

@@ -87,11 +87,14 @@ class RetrievalChatbot(BaseChatbot):
             return answer
         
     async def subquestion_answerer(self, sub_question: str, image_paths=None, return_logs=False) -> str:
+        time_s = time.time()
         sub_retrieve_reference=""
         sub_retrieve = self.retriever.retrieve(sub_question)
         for ref in sub_retrieve:
             sub_retrieve_reference += "Related research: {ref}\n".format(ref=ref)
         sub_answerer_context = "Sub Question References: {sub_retrieve_reference}\nQuestion: {question}\n".format(sub_retrieve_reference=sub_retrieve_reference, question=sub_question)
         sub_answer = self.answerer.refine(sub_answerer_context, self.memory, image_paths)
+        time_e = time.time()
+        print(f"Time: {time_e - time_s}")
         print(f"Subanswer: {sub_answer}")
         return "Subquestion: {sub_question}\nSubanswer: {sub_answer}\n\n\n".format(sub_question=sub_question, sub_answer=sub_answer)