Update main.py

main.py

@@ -449,6 +449,60 @@ def _format_steps_as_numbered(steps: list) -> str:
     return "\n".join(out)
 
 
+# --- NEW: sentence-level action filter for mixed paragraphs ---
+def _filter_sentences_by_actions(numbered_text: str,
+                                 wanted: set[str],
+                                 exclude: set[str]) -> str:
+    """
+    Works on already-numbered text. Splits each line into sentences and
+    keeps only sentences that match wanted actions; drops sentences matching exclude actions.
+    Falls back gracefully if no wanted match is found.
+    """
+    ACTION_SYNONYMS = {
+        "create": ("create", "creation", "add", "new", "generate", "setup", "set up", "register"),
+        "update": ("update", "modify", "change", "edit", "amend"),
+        "delete": ("delete", "remove", "cancel", "void"),
+        "navigate": ("navigate", "go to", "open"),
+    }
+
+    def _has_any(text: str, keys: set[str]) -> bool:
+        low = (text or "").lower()
+        for k in keys:
+            for syn in ACTION_SYNONYMS.get(k, (k,)):
+                if syn in low:
+                    return True
+        return False
+
+    def _split_sentences(block: str) -> list:
+        # robust split on ., !, ? followed by space
+        import re
+        parts = [s.strip() for s in re.split(r"(?<=[.!?])\s+", block or "") if s.strip()]
+        return parts if parts else ([block.strip()] if block.strip() else [])
+
+    out_lines = []
+    for ln in (numbered_text or "").splitlines():
+        sent_keep = []
+        sentences = _split_sentences(ln)
+        for s in sentences:
+            if exclude and _has_any(s, exclude):
+                continue
+            if wanted:
+                if _has_any(s, wanted):
+                    sent_keep.append(s)
+            else:
+                # no explicit wanted → keep unless excluded
+                sent_keep.append(s)
+
+        # If nothing matched wanted but line mentions wanted elsewhere, keep the whole line
+        if wanted and not sent_keep:
+            if _has_any(ln, wanted) and not _has_any(ln, exclude):
+                sent_keep = sentences
+
+        if sent_keep:
+            out_lines.append(" ".join(sent_keep))
+
+    return "\n".join(out_lines).strip() or (numbered_text or "").strip()
+
 def _filter_context_for_query(context: str, query: str) -> Tuple[str, Dict[str, Any]]:
     STRICT_OVERLAP = 3
     MAX_SENTENCES_STRICT = 4
@@ -980,25 +1034,29 @@ async def chat_with_ai(input_data: ChatInput):
 
             if full_steps:
                 numbered_full = _ensure_numbering(full_steps)
-                wanted_actions = set((kb_results.get("actions") or []))
-        # If user specifically asked for creation, prefer only creation and exclude update/delete
-                if "create" in wanted_actions and not ({"update", "delete"} & wanted_actions):
-                    numbered_full = _filter_numbered_steps_by_actions(
-                       numbered_full,
-                       wanted={"create"},
-                       exclude={"update", "delete"}
-            )
-                elif "update" in wanted_actions and "delete" not in wanted_actions:
-                   numbered_full = _filter_numbered_steps_by_actions(
-                      numbered_full, 
-                      wanted={"update"}, 
-                      exclude={"create", "delete"}
+                raw_actions = set((kb_results.get("actions") or []))
+                msg_low = (input_data.user_message or "").lower()
+# Fallbacks if extractor missed it
+                if not raw_actions and ("creation" in msg_low or "create" in msg_low or "set up" in msg_low or "setup" in msg_low):
+                    raw_actions = {"create"}
+                elif not raw_actions and ("update" in msg_low or "modify" in msg_low or "edit" in msg_low or "change" in msg_low):
+                    raw_actions = {"update"}
+                elif not raw_actions and ("delete" in msg_low or "remove" in msg_low or "cancel" in msg_low or "void" in msg_low):
+                    raw_actions = {"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:
+            # Sentence-level filtering for mixed paragraphs
+                   numbered_full = _filter_sentences_by_actions(
+                      numbered_full, wanted=wanted, exclude=exclude
             )
-                elif "delete" in wanted_actions and "update" not in wanted_actions:
-                   numbered_full = _filter_numbered_steps_by_actions(
-                      numbered_full, wanted={"delete"}, exclude={"create", "update"}
-                   )
-                    
+
+                     
                 next_only = _resolve_next_steps(
                     input_data.user_message,
                     numbered_full,