Update agents/planner.py

agents/planner.py

@@ -1,41 +1,58 @@
 # ---------------------------------------------------------
-# INTENT DETECTOR FOR AUTO-GPT WAREHOUSE AGENT
+# PLANNER: CLASSIFIES USER INTENT AND SUB-SCENARIOS
 # ---------------------------------------------------------
 
 def detect_intent(message: str) -> str:
     """
-    Classifies user intent for warehouse tasks.
+    Classifies the high-level warehouse task:
+    - slotting
+    - picking
+    - full_report
+    - unknown
     """
+    msg = message.lower()
 
-    if not message:
-        return "unknown"
+    if any(k in msg for k in ["slot", "re-slot", "sku", "velocity", "fast mover", "slow mover"]):
+        return "slotting"
 
-    msg = message.lower()
+    if any(k in msg for k in ["pick", "picking", "pick path", "walking route", "picker route"]):
+        return "picking"
 
-    # Slotting-related
-    slotting_keywords = [
-        "slot", "slotting", "sku placement", "velocity",
-        "fast mover", "medium mover", "slow mover",
-        "optimize slotting", "re-slot"
-    ]
-    if any(k in msg for k in slotting_keywords):
-        return "slotting_analysis"
-
-    # Picking path related
-    picking_keywords = [
-        "pick", "picking", "pick path", "route",
-        "picking route", "walking path", "optimize picking"
-    ]
-    if any(k in msg for k in picking_keywords):
-        return "picking_route"
-
-    # Full assessment
-    overall_keywords = [
-        "full report", "warehouse report", "assessment",
-        "complete analysis", "operations report",
-        "full warehouse", "360"
-    ]
-    if any(k in msg for k in overall_keywords):
-        return "full_warehouse_assessment"
+    if any(k in msg for k in ["report", "summary", "analysis", "warehouse report"]):
+        return "report"
 
     return "unknown"
+
+
+def detect_slotting_scenario(message: str) -> str:
+    """ Detects which slotting scenario the agent should optimize for. """
+    msg = message.lower()
+
+    if "congestion" in msg or "traffic" in msg:
+        return "reduce_congestion"
+    if "fast mover" in msg or "priority" in msg:
+        return "prioritize_fast"
+    if "slow mover" in msg or "free space" in msg or "prime space" in msg:
+        return "move_slow_out"
+    if "balance" in msg or "even" in msg:
+        return "balance_zones"
+    if "efficiency" in msg or "walking" in msg:
+        return "improve_efficiency"
+
+    return "generic_slotting"
+
+
+def detect_picking_scenario(message: str) -> str:
+    """ Detects specialized picking scenarios for adaptive behavior. """
+    msg = message.lower()
+
+    if "inefficient" in msg or "better path" in msg:
+        return "fix_inefficiency"
+    if "dispatch" in msg or "start from" in msg:
+        return "new_dispatch_location"
+    if "avoid" in msg or "block" in msg:
+        return "avoid_aisles"
+    if "batch" in msg or "multi-order" in msg:
+        return "batch_picking"
+
+    return "generic_picking"