Update agents/planner.py

agents/planner.py

@@ -1,36 +1,55 @@
-def detect_intent(message: str) -> str:
+def detect_intent(message: str):
     """
-    Detects whether the user is asking for:
-    - slotting optimization
-    - picking route optimization
-    - full warehouse report
+    A robust, rule-based classifier for warehouse tasks.
+    Returns one of:
+        - "slotting"
+        - "picking"
+        - "report"
     """
-    if not message:
-        return "unknown"
 
     msg = message.lower().strip()
 
-    # --- PICKING FIRST (stronger keywords) ---
-    picking_kw = [
-        "picking", "pick route", "picking route",
-        "optimize picking", "path", "shortest path",
-        "picker travel"
+    # -----------------------------
+    # SLOTING KEYWORDS
+    # -----------------------------
+    slot_keywords = [
+        "slot", "slotting", "putaway", "storage location",
+        "rearrange items", "reorganize racks", "optimize space",
+        "velocity", "sku placement", "bin allocation"
     ]
-    if any(k in msg for k in picking_kw):
-        return "picking"
 
-    # --- SLOTTING ---
-    slotting_kw = [
-        "slotting", "slot", "sku placement",
-        "velocity", "fast mover", "slow mover",
-        "optimize slots", "slot plan"
+    # -----------------------------
+    # PICKING KEYWORDS
+    # -----------------------------
+    pick_keywords = [
+        "pick", "picking", "route", "path", "walk sequence",
+        "shortest path", "aisle", "rack", "order fulfillment",
+        "picklist"
+    ]
+
+    # -----------------------------
+    # FULL REPORT KEYWORDS
+    # -----------------------------
+    report_keywords = [
+        "full report", "overall performance", "warehouse report",
+        "summary", "end to end", "combined", "complete report",
+        "all optimizations"
     ]
-    if any(k in msg for k in slotting_kw):
-        return "slotting"
 
-    # --- FULL REPORT ---
-    report_kw = ["full report", "warehouse report", "summary", "complete analysis"]
-    if any(k in msg for k in report_kw):
+    # -----------------------------
+    # PRIORITY: Full Report → Slotting → Picking
+    # (Prevents overlap confusion)
+    # -----------------------------
+    if any(k in msg for k in report_keywords):
         return "report"
 
-    return "unknown"
+    if any(k in msg for k in slot_keywords):
+        return "slotting"
+
+    if any(k in msg for k in pick_keywords):
+        return "picking"
+
+    # -----------------------------
+    # UNKNOWN → DEFAULT TO REPORT
+    # -----------------------------
+    return "report"