mindroot 9.22.0__py3-none-any.whl → 10.14.0__py3-none-any.whl

mindroot/coreplugins/check_list/mod.py.backup

@@ -0,0 +1,378 @@
+"""
+checklist helper for an LLM‑agent system
+—————————————————————————————————————————————————————————————————————————————
+Purpose
+-------
+• Parse a Markdown "##\tChecklist" section where each subtask is
+  written as a task‑list line:
+
+      - [ ] label of subtask
+          arbitrary markdown body… (until next task or heading)
+
+• Store parsed tasks + cursor in `context.data['checklist']`.
+
+• Runtime helpers:
+    load_checklist(md, ctx)   → parse markdown + reset state
+    complete_subtask(subtask_id, context)  → mark subtask done, advance cursor
+    goto_subtask(subtask_id, context)  → move to a specific subtask
+    clear_subtask(subtask_id, context)  → mark a subtask as incomplete
+    get_checklist_status(context)  → show the full checklist status
+
+`complete_subtask` and other commands live at module level so the agent can call them
+as tools. No third‑party deps—only Python's `re` module.
+"""
+
+import re
+from lib.providers.commands import command, command_manager
+from lib.providers.services import service
+import traceback
+
+# ---------- parsing ------------------------------------------------------
+
+# Only match top-level checklist items (no indentation at start of line)
+TASK_RE   = re.compile(r'^-\s*\[(?P<done>[ xX])\]\s*(?P<label>.*?)\s*$')
+HEADER_RE = re.compile(r'^#{1,6}\s')           # any Markdown heading
+CHECKLIST_HEADER_RE = re.compile(r'^#{1,6}\s+Checklist\b', re.IGNORECASE)  # Checklist heading
+
+def _parse(md: str):
+    """Extract tasks as a list of dicts: {label, body, done}."""
+    lines, tasks, i = md.splitlines(), [], 0
+    while i < len(lines):
+        # Skip lines that start with whitespace (nested items)
+        if lines[i].startswith(' ') or lines[i].startswith('\t'):
+            i += 1
+            continue
+        m = TASK_RE.match(lines[i])
+        if not m:                            # not a task‑list line
+            i += 1
+            continue
+
+        done  = m.group("done").strip().lower() in ("x", "✓")
+        label = m.group("label").strip()
+
+        # collect everything until next task or heading
+        body_lines = []
+        i += 1
+        while i < len(lines) and not (
+            # Stop at next top-level task or heading
+            (TASK_RE.match(lines[i]) and not (lines[i].startswith(' ') or lines[i].startswith('\t'))) or 
+            HEADER_RE.match(lines[i])
+        ):
+            body_lines.append(lines[i])
+            i += 1
+
+        tasks.append({
+            "label": label,
+            "body": "\n".join(body_lines).strip(),
+            "done": done,
+        })
+    return tasks
+
+
+def extract_checklist_section(md: str):
+    """Extract checklist items from the entire markdown text.
+    
+    No longer requires a specific 'Checklist' heading - processes the entire text
+    and extracts only top-level checklist items, leaving nested ones intact.
+    """
+    return md  # Return the entire text for processing
+
+
+# ---------- state helpers ------------------------------------------------
+
+def _state(ctx):
+    """Return the mutable checklist state in ctx.data."""
+    return ctx.data.setdefault("checklist", {"tasks": [], "cursor": 0})
+
+def load_checklist(md: str, ctx):
+    """Parse markdown and reset cursor to first unchecked subtask."""
+    st = _state(ctx)
+    st["tasks"]  = _parse(md)
+    st["cursor"] = next(
+        (i for i, t in enumerate(st["tasks"]) if not t["done"]),
+        len(st["tasks"]),
+    )
+
+
+@service()
+async def load_checklist_from_instructions(md: str, context=None):
+    """Load checklist from instructions.
+    
+    Processes the entire text and extracts top-level checklist items.
+    
+    Example:
+    { "load_checklist_from_instructions": { "md": "Full instructions with embedded checklist" } }
+    """
+    checklist_section = extract_checklist_section(md)
+    load_checklist(checklist_section, context)
+    return f"Loaded checklist from instructions.\n\n{_format_checklist_status(context)}"
+
+
+# ---------- helper functions ---------------------------------------------
+
+def _resolve_subtask_id(subtask_id, context):
+    """Resolve a subtask_id to an index (0-based).
+    
+    subtask_id can be:
+    - A number (1-based index, converted to 0-based)
+    - A string matching a subtask label
+    - None/default (-1) to use the current cursor position
+    """
+    st = _state(context)
+    
+    # Default to current cursor position
+    if subtask_id is None or subtask_id == -1:
+        return st["cursor"]
+    
+    # If it's a number, convert from 1-based to 0-based
+    if isinstance(subtask_id, int):
+        idx = subtask_id - 1
+    else:
+        # It's a string, try to find a matching label
+        for i, task in enumerate(st["tasks"]):
+            if task["label"] == subtask_id:
+                return i
+        # No match found
+        return -1
+    
+    return idx
+
+
+def _format_checklist_status(context):
+    """Format the full checklist status as a markdown string."""
+    st = _state(context)
+    if not st["tasks"]:
+        return "_No checklist found. Make sure to include a checklist in your instructions._"
+    
+    lines = ["### Checklist Status\n"]
+    
+    for i, task in enumerate(st["tasks"]):
+        # Determine status indicator
+        if i == st["cursor"]:
+            status = "➡️ " # Current task
+        elif task["done"]:
+            status = "✅ " # Completed
+        else:
+            status = "❌ " # Not completed
+        
+        # Add task line
+        lines.append(f"{status}**Subtask**: {task['label']}")
+    
+    return "\n".join(lines)
+
+
+# ---------- module‑level tool commands ----------------------------------
+
+@command()
+async def complete_subtask(subtask_id=None, context=None):
+    """
+    Mark a subtask complete and return a Markdown status message.
+    
+    Parameters:
+    - subtask_id: Optional. The subtask to complete, specified by:
+                 - The exact subtask label text
+                 - Omit to complete the current subtask
+    
+    Example:
+    { "complete_subtask": {} }  # Complete current subtask
+    { "complete_subtask": { "subtask_id": "Review documents" } }  # Complete by label
+    """
+    if context is None:
+        return "_Context is required._"
+        
+    st = _state(context)
+    if not st["tasks"]:
+        try:
+            print("Loading checklist from instructions...")
+            print("Agent is")
+            print(context.agent)
+            instructions = context.agent["instructions"]
+            await load_checklist_from_instructions(instructions, context)
+        except Exception as e:
+            print(f"Error loading checklist: {e}")
+            trace = traceback.format_exc()
+            print(trace)
+            return "_No checklist found. Make sure to include a checklist in your instructions._"
+    
+    idx = _resolve_subtask_id(subtask_id, context)
+    if idx < 0 or idx >= len(st["tasks"]):
+        return "_Invalid subtask identifier._"
+
+    # mark as done
+    done_task = st["tasks"][idx]
+    done_task["done"] = True
+
+    # advance cursor to next open subtask
+    st["cursor"] = next(
+        (i for i, t in enumerate(st["tasks"][idx + 1:], idx + 1) if not t["done"]),
+        len(st["tasks"]),
+    )
+
+    # build markdown response
+    completed = f"Completed Subtask {idx+1}: - [x] {done_task['label']}"
+    if st["cursor"] >= len(st["tasks"]):
+        return f"{completed}\n\nAll subtasks complete ✅\n\n{_format_checklist_status(context)}"
+
+    next_task = st["tasks"][st['cursor']]
+    return f"""
+            {completed}
+
+            Next subtask (Subtask {st['cursor']+1})
+            - [ ] {next_task['label']}
+            {next_task['body']}
+
+            {_format_checklist_status(context)}
+            """
+
+@command()
+async def goto_subtask(subtask_id, context=None):
+    """
+    Move to a specific subtask without changing its completion status.
+    
+    Parameters:
+    - subtask_id: Required. The subtask to navigate to, specified by:
+                 - The exact subtask label text
+    
+    Example:
+    { "goto_subtask": { "subtask_id": "Data analysis" } }  # Go to by label
+    """
+    if context is None:
+        return "_Context is required._"
+        
+    st = _state(context)
+    if not st["tasks"]:
+        return "_No checklist found. Make sure to include a checklist in your instructions._"
+    
+    idx = _resolve_subtask_id(subtask_id, context)
+    if idx < 0 or idx >= len(st["tasks"]):
+        return "_Invalid subtask identifier._"
+    
+    # Update cursor position
+    st["cursor"] = idx
+    
+    # Get the current task
+    current_task = st["tasks"][idx]
+    
+    # Build response
+    status = "✅" if current_task["done"] else "❌"
+    return (
+        f"Moved to Subtask {idx+1}: {status} {current_task['label']}\n"
+        f"{current_task['body']}\n\n"
+        f"{_format_checklist_status(context)}"
+    )
+
+
+@command()
+async def clear_subtask(subtask_id=None, context=None):
+    """
+    Mark a subtask as incomplete (not done).
+    
+    Parameters:
+    - subtask_id: Optional. The subtask to clear, specified by:
+                 - The exact subtask label text
+                 - Omit to clear the current subtask
+    
+    Example:
+    { "clear_subtask": {} }  # Clear current subtask
+    { "clear_subtask": { "subtask_id": "Review documents" } }  # Clear by label
+    """
+    if context is None:
+        return "_Context is required._"
+        
+    st = _state(context)
+    if not st["tasks"]:
+        return "_No checklist found. Make sure to include a checklist in your instructions._"
+    
+    idx = _resolve_subtask_id(subtask_id, context)
+    if idx < 0 or idx >= len(st["tasks"]):
+        return "_Invalid subtask identifier._"
+    
+    # Mark as not done
+    task = st["tasks"][idx]
+    was_done = task["done"]
+    task["done"] = False
+    
+    # If we cleared the current or earlier task, update cursor
+    if idx <= st["cursor"]:
+        st["cursor"] = idx
+    
+    # Build response
+    action = "Cleared" if was_done else "Already clear"
+    return (
+        f"{action} Subtask {idx+1}: - [ ] {task['label']}\n"
+        f"Current subtask is now Subtask {st['cursor']+1}\n\n"
+        f"{_format_checklist_status(context)}"
+    )
+
+
+@command()
+async def get_checklist_status(context=None):
+    """
+    Show the full status of the checklist.
+    
+    Example:
+    { "get_checklist_status": {} }
+    """
+    if context is None:
+        return "_Context is required._"
+    
+    return _format_checklist_status(context)
+
+
+@command()
+async def delegate_subtask(subtask_id, details:str, agent=None, context=None):
+    """
+    Delegate a subtask to an agent, automatically passing the subtask body as
+    instructions, along with any details you add.
+
+    If agent is not specified, the current agent name will be used for the subtask.
+   
+    Example:
+
+    Given a checklist with subtasks like:
+      
+      - Research 
+         ...
+    
+      - Draft
+         ...
+    
+      - Complete
+         ..
+
+    Suppose you had already created /data/sess_1234.
+    You might initiate work on the first subtask with this command:
+
+    { "delegate_subtask": { "subtask_id": "Research", 
+                            "details": "Session data in /data/sess_1234/" }} 
+
+    Note that you do not need to repeat the text of the subtask item from the checklist
+    in your details.
+    """
+    st = _state(context)
+    if not st["tasks"]:
+        try:
+            print("Loading checklist from instructions...")
+            print("Agent is")
+            print(context.agent)
+            instructions = context.agent["instructions"]
+            await load_checklist_from_instructions(instructions, context)
+        except Exception as e:
+            print(f"Error loading checklist: {e}")
+            trace = traceback.format_exc()
+            print(trace)
+            return "_No checklist found. Make sure to include a checklist in your instructions._"
+    idx = _resolve_subtask_id(subtask_id, context)
+    if idx < 0 or idx >= len(st["tasks"]):
+        return "_Invalid subtask identifier._"
+    current_task = st["tasks"][idx]
+    subtask = current_task["body"]
+    reminder = """Important: you may see instructions for the full process. However, you are to ONLY 
+do the specified part of the process and then return a task result."""
+    instructions = f"You are working as part of a multi-step process. Please complete the following subtask:\n\n{subtask}\n\n{details}\n\n{reminder}\n"
+    if agent is None:
+        agent_name = context.agent["name"]
+    else:
+        agent_name = agent
+    return await command_manager.delegate_task(instructions, agent_name, context=context)
+