Update master_key.py

master_key.py

@@ -218,37 +218,39 @@ TABLE_SCHEMAS = {
     },
     
     # Vehicle Registration Tables
-    "Vehicle Registration Numbers Maintenance": {
-        "headings": [
-            {"level": 1, "text": "Vehicle Registration Numbers of Records Examined"},
-            {"level": 2, "text": "Maintenance Management"}
-        ],
-        "orientation": "row1",
-        "labels": [
-            "No.", "Registration Number", "Roadworthiness Certificates",
-            "Maintenance Records", "Daily Checks",
-            "Fault Recording/ Reporting", "Fault Repair"
-        ],
-        "priority": 80,
-        "context_keywords": ["maintenance", "vehicle registration"]
-    },
     "Vehicle Registration Numbers Mass": {
-        "headings": [
-            {"level": 1, "text": "Vehicle Registration Numbers of Records Examined"},
-            {"level": 2, "text": "MASS MANAGEMENT"}
-        ],
-        "orientation": "row1", 
-        "labels": [
-            "No.", "Registration Number", "Sub contractor",
-            "Sub-contracted Vehicles Statement of Compliance",
-            "Weight Verification Records",
-            "RFS Suspension Certification #",
-            "Suspension System Maintenance", "Trip Records",
-            "Fault Recording/ Reporting on Suspension System"
-        ],
-        "priority": 80,
-        "context_keywords": ["mass", "vehicle registration"]
-    },
+    "headings": [
+        {"level": 1, "text": "Vehicle Registration Numbers of Records Examined"},
+        {"level": 2, "text": "MASS MANAGEMENT"}
+    ],
+    "orientation": "row1", 
+    "labels": [
+        "No.", "Registration Number", "Sub contractor",
+        "Sub-contracted Vehicles Statement of Compliance",
+        "Weight Verification Records",
+        "RFS Suspension Certification #",
+        "Suspension System Maintenance", "Trip Records",
+        "Fault Recording/ Reporting on Suspension System"
+    ],
+    "priority": 90,  # Higher priority
+    "context_keywords": ["mass", "vehicle registration", "rfs suspension", "weight verification"],
+    "context_exclusions": ["maintenance", "roadworthiness", "daily checks"]  # Exclude maintenance-specific terms
+},
+"Vehicle Registration Numbers Maintenance": {
+    "headings": [
+        {"level": 1, "text": "Vehicle Registration Numbers of Records Examined"},
+        {"level": 2, "text": "Maintenance Management"}
+    ],
+    "orientation": "row1",
+    "labels": [
+        "No.", "Registration Number", "Roadworthiness Certificates",
+        "Maintenance Records", "Daily Checks",
+        "Fault Recording/ Reporting", "Fault Repair"
+    ],
+    "priority": 85,  # Lower priority
+    "context_keywords": ["maintenance", "vehicle registration", "roadworthiness", "daily checks"],
+    "context_exclusions": ["mass", "rfs suspension", "weight verification"]  # Exclude mass-specific terms
+},
     "Driver / Scheduler Records Examined": {
         "headings": [
             {"level": 1, "text": "Driver / Scheduler Records Examined"},