Update main.py

main.py

@@ -857,24 +857,32 @@ async def chat_with_ai(input_data: ChatInput):
                 raw_actions = set((kb_results.get("actions") or []))
                 msg_low2 = (input_data.user_message or "").lower()
                 if not raw_actions and ("creation" in msg_low2 or "create" in msg_low2 or "set up" in msg_low2 or "setup" in msg_low2):
-                    raw_actions = {"create"}
+                   raw_actions = {"create"}
                 elif not raw_actions and ("update" in msg_low2 or "modify" in msg_low2 or "edit" in msg_low2 or "change" in msg_low2):
-                    raw_actions = {"update"}
+                   raw_actions = {"update"}
                 elif not raw_actions and ("delete" in msg_low2 or "remove" in msg_low2 or "cancel" in msg_low2 or "void" in msg_low2):
-                    raw_actions = {"delete"}
-
+                   raw_actions = {"delete"}
+                sec_title_low = ((top_meta or {}).get("section") or "").strip().lower()
+                section_is_create = any(k in sec_title_low for k in ("create", "creation"))
+                section_is_update = any(k in sec_title_low for k in ("update", "updation"))
+                section_is_delete = any(k in sec_title_low for k in ("delete", "removal", "cancellation"))
+                skip_action_filter = (
+                    ("create" in raw_actions and section_is_create) or
+                    ("update" in raw_actions and section_is_update) or
+                    ("delete" in raw_actions and section_is_delete)
+)
                 wanted, exclude = set(), set()
-                if "create" in raw_actions and not ({"update", "delete"} & raw_actions):
-                    wanted, exclude = {"create"}, {"update", "delete"}
-                elif "update" in raw_actions and not ({"create", "delete"} & raw_actions):
-                    wanted, exclude = {"update"}, {"create", "delete"}
-                elif "delete" in raw_actions and not ({"create", "update"} & raw_actions):
-                    wanted, exclude = {"delete"}, {"create", "update"}
-
-                if wanted or exclude:
-                    numbered_full = _filter_numbered_steps_by_actions(numbered_full, wanted=wanted, exclude=exclude)
-
-                # --- NEW: keyword-free anchor-based next-step resolver ---
+                if not skip_action_filter:
+                   if "create" in raw_actions and not ({"update", "delete"} & raw_actions):
+                       wanted, exclude = {"create"}, {"update", "delete"}
+                   elif "update" in raw_actions and not ({"create", "delete"} & raw_actions):
+                       wanted, exclude = {"update"}, {"create", "delete"}
+                   elif "delete" in raw_actions and not ({"create", "update"} & raw_actions):
+                       wanted, exclude = {"delete"}, {"create", "update"}
+                if (wanted or exclude) and not skip_action_filter:
+    numbered_full = _filter_numbered_steps_by_actions(numbered_full, wanted=wanted, exclude=exclude)
+
+                 # --- NEW: keyword-free anchor-based next-step resolver ---
                 next_only = _anchor_next_steps(input_data.user_message, numbered_full, max_next=6)
 
                 if next_only is not None: