Spaces:
Sleeping
Sleeping
File size: 1,808 Bytes
34cb740 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | """Shared task text utilities for hooks and status bar."""
import re
# System-generated prefixes that aren't real tasks
_SKIP_PREFIXES = (
"MEMORY HANDOFF",
"[Request interrupted",
"Implement the following plan",
"Execute this plan",
"Follow this plan",
"Here is the plan",
)
# Short non-task messages
_SKIP_EXACT = frozenset({
"proceed", "continue", "yes", "no", "ok", "go ahead", "lgtm",
})
def clean_task_text(prompt: str) -> str:
"""Extract a clean, short task description from a user prompt.
Strips XML tags, system prefixes, markdown headers. Splits at the first
sentence boundary (letter followed by `. `) and caps at 60 chars.
Returns empty string if the prompt isn't a real task.
"""
text = prompt.strip()
if not text:
return ""
# Strip XML-like tags
text = re.sub(r"<[a-zA-Z_-]+>[^<]*</[a-zA-Z_-]+>", "", text)
text = re.sub(r"</?[a-zA-Z_-]+>", "", text)
text = text.strip()
# Skip system-generated / non-task messages
for pfx in _SKIP_PREFIXES:
if text.startswith(pfx):
return ""
if text.lower() in _SKIP_EXACT:
return ""
# Strip markdown heading markers
text = re.sub(r"^#{1,4}\s*", "", text)
# Strip "Resume:" prefix
text = re.sub(r"^[Rr]esume\s*:\s*", "", text)
# First line only (multi-line prompts: first line is usually the task)
text = text.split("\n")[0].strip()
# Sentence split: only at letter + punctuation + space (not "v2." or "87.")
text = re.sub(r"([a-zA-Z])[.!?]\s.*", r"\1", text)
text = text.strip()
# Cap at 60 chars, cut at last space if mid-word
if len(text) > 60:
text = text[:60]
if " " in text:
text = text[:text.rfind(" ")]
return text if len(text) >= 8 else ""
|