make agent categorize tasks

Author: SentienceDEV

predicate/__init__.py

@@ -119,6 +119,15 @@
 from .ordinal import OrdinalIntent, boost_ordinal_elements, detect_ordinal_intent, select_by_ordinal
 from .overlay import clear_overlay, show_overlay
 from .permissions import PermissionPolicy
+from .pruning import (
+    CategoryDetectionResult,
+    PrunedSnapshotContext,
+    PruningTaskCategory,
+    SkeletonDomNode,
+    classify_task_category,
+    prune_snapshot_for_task,
+    serialize_pruned_snapshot,
+)
 from .query import find, query
 from .read import extract, extract_async, read, read_best_effort
 from .recorder import Recorder, Trace, TraceStep, record
@@ -313,6 +322,13 @@
     "save_storage_state",
     # Formatting (v0.12.0+)
     "format_snapshot_for_llm",
+    "CategoryDetectionResult",
+    "PrunedSnapshotContext",
+    "PruningTaskCategory",
+    "SkeletonDomNode",
+    "classify_task_category",
+    "prune_snapshot_for_task",
+    "serialize_pruned_snapshot",
     # Agent Config (v0.12.0+)
     "AgentConfig",
     # Enums

predicate/agent_runtime.py

@@ -398,19 +398,23 @@ async def click(self, element_id: int) -> None:
 
         await self.record_action(f"CLICK({element_id})")
 
-    async def type(self, element_id: int, text: str) -> None:
+    async def type(self, element_id: int, text: str, *, delay_ms: float | None = None) -> None:
         """
         Type text into an element.
 
         Args:
             element_id: Element ID from snapshot
             text: Text to type
+            delay_ms: Optional delay between keystrokes in milliseconds
         """
         # First click to focus
         await self.click(element_id)
 
         # Then type
-        await self.backend.type_text(text)
+        if delay_ms is None:
+            await self.backend.type_text(text)
+        else:
+            await self.backend.type_text(text, delay_ms=delay_ms)
         await self.record_action(f"TYPE({element_id}, '{text[:20]}...')" if len(text) > 20 else f"TYPE({element_id}, '{text}')")
 
     async def press(self, key: str) -> None:

predicate/agents/automation_task.py

@@ -179,6 +179,32 @@ class AutomationTask:
     # Domain hints for heuristics (e.g., ["ecommerce", "amazon"])
     domain_hints: tuple[str, ...] = field(default_factory=tuple)
 
+    # Force a specific pruning category (overrides auto-detection)
+    force_pruning_category: str | None = None
+
+    def pruning_category_hint(self):
+        """
+        Return the pruning-oriented category for this task.
+
+        If force_pruning_category is set, returns that category directly.
+        Otherwise, uses rule-based normalization from task text and hints.
+        """
+        from ..pruning import PruningTaskCategory, classify_task_category
+
+        # If a category is forced, return it directly
+        if self.force_pruning_category:
+            try:
+                return PruningTaskCategory(self.force_pruning_category)
+            except ValueError:
+                pass  # Invalid category, fall through to auto-detection
+
+        return classify_task_category(
+            task_text=self.task,
+            current_url=self.starting_url,
+            domain_hints=self.domain_hints,
+            task_category=self.category,
+        ).category
+
     @classmethod
     def from_webbench_task(cls, task: Any) -> "AutomationTask":
         """