Update src/query_engine.py

src/query_engine.py

@@ -163,6 +163,7 @@ def init_query(user_query:str,
         }
 
 def understand_query(
+    user_query: str,
     cfg: Dict
 ) -> dict:
     """Analyze and reformulate a user query for better retrieval.
@@ -202,6 +203,7 @@ def understand_query(
     kb_overview = load_kb_meta(cfg)
 
     prompt = build_query_understanding_prompt(
+        user_query,
         domain,
         sql_schema_summary=schema_summary,
         kb_overview=kb_overview,