Upload engine.py

engine.py

@@ -171,13 +171,19 @@ def extract_relevant_tables(question, max_tables=4):
                 score += 5
 
         # 5️⃣ Only add if meets minimum threshold (prevents low-quality matches)
-        # Increased threshold from 3 to 4 for better precision
-        if score >= 4:
+        # Use lower threshold for small schemas (more lenient)
+        # Increased threshold from 3 to 4 for better precision, but lower to 2 for small schemas
+        threshold = 2 if len(schema) <= 5 else 4
+        if score >= threshold:
             matched.append((table, score))
 
     # Sort by relevance
     matched.sort(key=lambda x: x[1], reverse=True)
 
+    # If no matches but schema is very small, return all tables (with lower confidence)
+    if not matched and len(schema) <= 3:
+        return list(schema.keys())[:max_tables]
+
     return [t[0] for t in matched[:max_tables]]
 
 
@@ -317,14 +323,20 @@ def is_question_supported(question):
 
 def build_prompt(question):
     matched = extract_relevant_tables(question)
+    full_schema = load_ai_schema()
 
     if matched:
-        schema = {t: load_ai_schema()[t] for t in matched}
+        schema = {t: full_schema[t] for t in matched}
     else:
-        # 🚫 Don't send all 100+ tables! Return a helpful error instead
+        # 🚫 Don't send all 100+ tables! Return a helpful error with available tables
+        available_tables = list(full_schema.keys())[:10]  # Show first 10 tables
+        tables_list = "\n".join(f"- {t}" for t in available_tables)
+        if len(full_schema) > 10:
+            tables_list += f"\n... and {len(full_schema) - 10} more tables"
         raise ValueError(
-            "I couldn't find any relevant tables for your question. "
-            "Please try mentioning a specific table name or use 'what data' to see available tables."
+            f"I couldn't find any relevant tables for your question.\n\n"
+            f"Available tables:\n{tables_list}\n\n"
+            f"Please try mentioning a specific table name or use 'what data' to see all available tables."
         )
 
     prompt = """