Update modules/railway_rag/railway_base_rag.py

modules/railway_rag/railway_base_rag.py

@@ -1,5 +1,6 @@
 # Railway Rules RAG Module
 from helpers.hf_llm import generate_text
+from modules.railway_rag.retrieval_engine import retrieve_rules
 SYSTEM_PROMPT = (
     "You are a railway rules assistant.\n"
     "You must answer the user question ONLY if the provided railway rules, clearly and directly contain the answer.\n"
@@ -76,12 +77,12 @@ def answer_with_rag(query: str):
         "- Do not add external knowledge.\n"
     )
     prompt = f"""
-    {SYSTEM_PROMPT}
-    
-    {user_prompt}
-    
-    Answer clearly and concisely.
-    """
+{SYSTEM_PROMPT}
+
+{user_prompt}
+
+Answer clearly and concisely.
+"""
     answer_text = generate_text(
         prompt=prompt,
         max_new_tokens=350,