prompt update

qa_engine/qa_engine.py

@@ -304,23 +304,20 @@ class QAEngine():
 
         logger.info('Running LLM chain')
         question_processed = QAEngine._preprocess_question(question)
-        
-        answer = self.llm_chain.run(question=question_processed, context=context)
-        
-        answer_step_2 = self.llm_model(
+        question_processed = self.llm_model(
 f"""
 <|im_start|>system                                                             
-Given the question provided below, review the answer, and update it based on common sense if needed. Ensure that the updated answer is clear, concise, and directly addresses the question. Your primary objective is to return only the updated answer. <|im_end|>
+Given a user's query that is vague or broad, your task is to refine and rephrase the query to make it more specific and targeted, ultimately returning only the refined question. <|im_end|>
 <|im_end|>
 <|im_start|>user
 <question>{question_processed}</question>
-<answer>{answer}</answer>
 <|im_end|>
 <|im_start|>assistant
 """
         )
-        
-        answer_postprocessed = QAEngine._postprocess_answer(answer_step_2)
+
+        answer = self.llm_chain.run(question=question_processed, context=context)
+        answer_postprocessed = QAEngine._postprocess_answer(answer)
         response.set_answer(answer_postprocessed)
         logger.info('Received answer')
 
@@ -329,9 +326,9 @@ Given the question provided below, review the answer, and update it based on com
             sep = '\n' + '-' * 100
             logger.info(f'question len: {len(question)} {sep}')
             logger.info(f'question: {question} {sep}')
+            logger.info(f'question processed: {question} {sep}')
             logger.info(f'answer len: {len(response.get_answer())} {sep}')
             logger.info(f'answer original: {answer} {sep}')
-            logger.info(f'answer step 2: {answer_step_2} {sep}')
             logger.info(f'answer postprocessed: {response.get_answer()} {sep}')
             logger.info(f'{response.get_sources_as_text()} {sep}')
             logger.info(f'messages_contex: {messages_context} {sep}')