Update rag.py

rag.py

@@ -220,7 +220,7 @@ def generate_answer(query: str, top_k: int = 5, candidate_k: int = 50, alpha: fl
     try:
         if not validate_query(query):
             logger.warning("Query rejected: Not finance-related.")
-            return "Query rejected: Please ask finance-related questions.", []
+            return "Query rejected: Please ask finance-related questions."
 
         cand_ids = hybrid_candidates(query, candidate_k=candidate_k, alpha=alpha)
         logger.info(f"Hybrid candidates retrieved: {cand_ids}")
@@ -242,9 +242,9 @@ def generate_answer(query: str, top_k: int = 5, candidate_k: int = 50, alpha: fl
             # Pass top 5 chunks as context
             context_text = "\n".join([d["content"] for d in reranked])
             answer = get_mistral_answer(query, context_text)
-        final_answer = answer #validate_output(answer, reranked)
+        final_answer = answer
         logger.info(f"Final Answer: {final_answer}")
         return final_answer
     except Exception as e:
         logger.error(f"Error in RAG pipeline: {e}")
-        return f"Error in RAG pipeline: {e}", []
+        return f"Error in RAG pipeline: {e}"