mindroot 9.24.0__py3-none-any.whl → 10.0.0__py3-none-any.whl

mindroot/coreplugins/check_list/mod.py.backup2

@@ -0,0 +1,506 @@
+"""
+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
+from collections import Counter
+
+# ---------- simple parsing -----------------------------------------------
+
+def _parse(text):
+    """Simple, robust checklist parser.
+    
+    Finds checklist items (- [ ] or - [x]) at any indentation level,
+    but only processes items at the same indentation as the first one found.
+    
+    Returns list of {label, body, done} dicts.
+    """
+    print("parsing:")
+    print("---------------------------------------------")
+    print(text)
+    print('---------------------------------------------')
+    lines = text.splitlines()
+    tasks = []
+    i = 0
+    first_task_indent = None
+    
+    # Find all task lines and their indentations first
+    task_indents = []
+    for line in lines:
+        stripped = line.lstrip()
+        if stripped.startswith('- [ ]') or stripped.startswith('- [x]') or stripped.startswith('- [X]'):
+            indent = len(line) - len(stripped)
+            task_indents.append(indent)
+    
+    if not task_indents:
+        return tasks
+    
+    # Use the SMALLEST indentation level as reference (top-level tasks)
+    reference_indent = min(task_indents)
+    print("reference indent:", reference_indent)
+    while i < len(lines):
+        line = lines[i]
+        stripped = line.lstrip()
+        print(i)
+        # Check if this line is a checklist item
+        if stripped.startswith('- [ ]') or stripped.startswith('- [x]') or stripped.startswith('- [X]'):
+            # Calculate indentation
+            current_indent = len(line) - len(stripped)
+            print("found task at indent:", current_indent)
+            print("reference indent:", reference_indent)
+            # Only process tasks at the reference indentation level
+            if current_indent == reference_indent:
+                # Extract task info
+                done = stripped.startswith('- [x]') or stripped.startswith('- [X]')
+                
+                # Extract label (everything after the checkbox)
+                if stripped.startswith('- [ ]'):
+                    label = stripped[5:].strip()
+                else:  # - [x] or - [X]
+                    label = stripped[5:].strip()
+                
+                # Collect body lines until next task at same level or end
+                body_lines = []
+                i += 1
+                
+                while i < len(lines):
+                    print("checking line for body:",i, lines[i])
+                    next_line = lines[i]
+                    next_stripped = next_line.lstrip()
+                    next_indent = len(next_line) - len(next_stripped)
+                    print("next_stripped:", next_stripped)
+                    print("next_indent:", next_indent)
+                    print("reference_indent:", reference_indent)
+                    # Stop if we hit another task at the same indentation level
+                    if ((next_stripped.startswith('- [ ]') or 
+                         next_stripped.startswith('- [x]') or 
+                         next_stripped.startswith('- [X]')) and 
+                        next_indent <= reference_indent):
+                        break
+                    
+                    # Stop if we hit a markdown heading
+                    if next_stripped.startswith('#'):
+                        break
+                        
+                    body_lines.append(next_line)
+                    i += 1
+                
+                tasks.append({
+                    'label': label,
+                    'body': '\n'.join(body_lines), #.strip(),
+                    'done': done
+                })
+                continue
+        
+        i += 1
+    
+    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
+
+
+# ---------- 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 get_parsed_subtasks(subtask_id=None, context=None):
+    """
+    Return parsed subtasks with their name/id and body for verification.
+    
+    Parameters:
+    - subtask_id: Optional. If provided, parse subtasks from the body of this specific subtask.
+                 If omitted, returns all top-level subtasks from the main checklist.
+    
+    Returns a list of dictionaries with:
+    - label: The subtask name/label
+    - body: The subtask body content
+    - done: Whether the subtask is marked as complete
+    - index: The 0-based index of the subtask
+    
+    Example:
+    { "get_parsed_subtasks": {} }  # Get all top-level subtasks
+    { "get_parsed_subtasks": { "subtask_id": "Research phase" } }  # Get nested subtasks from "Research phase"
+    """
+    if context is None:
+        return "_Context is required._"
+    
+    st = _state(context)
+    if not st["tasks"]:
+        try:
+            print("Loading checklist from instructions...")
+            instructions = context.agent["instructions"]
+            await load_checklist_from_instructions(instructions, context)
+            st = _state(context)  # Refresh state after loading
+        except Exception as e:
+            print(f"Error loading checklist: {e}")
+            return "_No checklist found. Make sure to include a checklist in your instructions._"
+    
+    # If no subtask_id provided, return all top-level subtasks
+    if subtask_id is None:
+        result = []
+        for i, task in enumerate(st["tasks"]):
+            result.append({
+                "index": i,
+                "label": task["label"],
+                "body": task["body"],
+                "done": task["done"]
+            })
+        return {
+            "source": "top-level checklist",
+            "subtasks": result
+        }
+    
+    # Find the specified subtask and parse its body for nested subtasks
+    idx = _resolve_subtask_id(subtask_id, context)
+    if idx < 0 or idx >= len(st["tasks"]):
+        return "_Invalid subtask identifier._"
+    
+    parent_task = st["tasks"][idx]
+    print("Parsing nested subtasks from:", subtask_id)
+    nested_tasks = _parse(parent_task["body"])
+    
+    result = []
+    for i, task in enumerate(nested_tasks):
+        result.append({
+            "index": i,
+            "label": task["label"],
+            "body": task["body"],
+            "done": task["done"]
+        })
+    
+    return {
+        "source": f"nested subtasks from '{parent_task['label']}'",
+        "parent_task": {
+            "index": idx,
+            "label": parent_task["label"],
+            "done": parent_task["done"]
+        },
+        "subtasks": result
+    }
+
+
+@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)
+