Upload engine.py

engine.py

@@ -143,21 +143,24 @@ def build_join_sql(base_table, steps):
 # =========================
 
 def parse_intent(question):
-    # 1️⃣ Load metadata
     meta = load_metadata()
 
-    # 2️⃣ Build schema description for the LLM
+    # ✅ Build schema safely (skip empty modules)
     schema_description = "\n".join([
-        f"{module}: {', '.join(meta['modules'][module]['fields'])}"
+        f"{module}: {', '.join(fields)}"
         for module in meta["modules"]
+        if (fields := [
+            f for f in meta["fields"]
+            if meta["fields"][f]["module"] == module
+        ])
     ])
 
-    # 3️⃣ Prompt with schema awareness
     prompt = f"""
 You are a SQL query planner.
 
 You MUST only use fields listed below.
-If a field does not exist, choose the closest valid one.
+If a field does not exist, choose the closest valid field.
+Do NOT invent column names.
 
 Available schema:
 {schema_description}
@@ -165,17 +168,9 @@ Available schema:
 Extract:
 - module
 - filters (field, operator, value)
-- selected fields (list of fields)
+- selected fields
 
-Return JSON only.
-
-Example:
-{{
-  "module": "employees",
-  "filters": [
-    {{ "field": "start_date", "operator": "greater_or_equal", "value": "2024-01-01" }}
-  ]
-}}
+Return ONLY valid JSON.
 
 User question:
 {question}
@@ -187,7 +182,11 @@ User question:
         temperature=0
     )
 
-    return json.loads(res.choices[0].message.content)
+    # ✅ Safe JSON parsing
+    try:
+        return json.loads(res.choices[0].message.content)
+    except json.JSONDecodeError:
+        raise ValueError("LLM returned invalid JSON")
 
 
 # =========================