claude-task-master 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

claude_task_master/cli_commands/workflow.py

@@ -13,6 +13,7 @@ from ..core.context_accumulator import ContextAccumulator
 from ..core.credentials import CredentialManager
 from ..core.logger import LogFormat, LogLevel, TaskLogger
 from ..core.orchestrator import WorkLoopOrchestrator
+from ..core.plan_updater import PlanUpdater
 from ..core.planner import Planner
 from ..core.state import StateManager, StateResumeValidationError, TaskOptions
 from ..webhooks import WebhookClient
@@ -276,6 +277,12 @@ def start(
 
 
 def resume(
+    message: Annotated[
+        str | None,
+        typer.Argument(
+            help="Optional change request to update the plan before resuming",
+        ),
+    ] = None,
     force: Annotated[
         bool,
         typer.Option("--force", "-f", help="Force resume from failed/blocked state"),
@@ -288,11 +295,16 @@ def resume(
     - Interrupted by Ctrl+C
     - Blocked and waiting for intervention
 
+    Optionally provide a message to update the plan before resuming.
+    This is useful when requirements change mid-task.
+
     Use --force to recover from a failed state.
 
     Examples:
         claudetm resume
         claudetm resume --force  # Recover from failed state
+        claudetm resume "Add authentication to the API"
+        claudetm resume "Fix the bug in the login form instead"
     """
     console.print("[bold blue]Resuming task...[/bold blue]")
 
@@ -383,6 +395,31 @@ def resume(
             state_manager.save_state(state)
             console.print("\n[yellow]Attempting to resume blocked task...[/yellow]")
 
+        # If a message was provided, update the plan first
+        if message:
+            console.print("\n[bold cyan]Updating plan with change request...[/bold cyan]")
+            console.print(
+                f"[dim]Message: {message[:100]}{'...' if len(message) > 100 else ''}[/dim]"
+            )
+
+            plan_updater = PlanUpdater(agent, state_manager, logger=logger)
+            try:
+                update_result = plan_updater.update_plan(message)
+                if update_result["changes_made"]:
+                    console.print("[green]Plan updated successfully[/green]")
+                    # Display a brief summary of the updated plan
+                    plan = state_manager.load_plan()
+                    if plan:
+                        # Count tasks
+                        completed = plan.count("- [x]")
+                        pending = plan.count("- [ ]")
+                        console.print(f"[dim]Tasks: {completed} completed, {pending} pending[/dim]")
+                else:
+                    console.print("[yellow]No changes needed to plan[/yellow]")
+            except Exception as e:
+                console.print(f"[red]Error updating plan: {e}[/red]")
+                console.print("[yellow]Continuing with existing plan...[/yellow]")
+
         # Create webhook client if URL was configured
         wh_client: WebhookClient | None = None
         if state.options.webhook_url:

claude_task_master/core/__init__.py

@@ -120,6 +120,7 @@ from claude_task_master.core.parallel import (
     TaskResult,
     TaskStatus,
 )
+from claude_task_master.core.plan_updater import PlanUpdater
 from claude_task_master.core.pr_context import PRContextManager
 from claude_task_master.core.progress_tracker import (
     ExecutionTracker,
@@ -132,6 +133,7 @@ from claude_task_master.core.prompts import (
     PromptSection,
     build_context_extraction_prompt,
     build_error_recovery_prompt,
+    build_plan_update_prompt,
     build_planning_prompt,
     build_task_completion_check_prompt,
     build_verification_prompt,
@@ -274,6 +276,8 @@ __all__ = [
     "MaxSessionsReachedError",
     # Orchestrator classes
     "WorkLoopOrchestrator",
+    # Plan updater classes
+    "PlanUpdater",
     # Task runner exceptions
     "TaskRunnerError",
     "NoPlanFoundError",
@@ -309,6 +313,7 @@ __all__ = [
     "PromptBuilder",
     "PromptSection",
     "build_planning_prompt",
+    "build_plan_update_prompt",
     "build_work_prompt",
     "build_verification_prompt",
     "build_task_completion_check_prompt",

claude_task_master/core/agent_phases.py

@@ -22,7 +22,7 @@ if TYPE_CHECKING:
 T = TypeVar("T")
 
 
-def run_async_with_cleanup(coro: Coroutine[Any, Any, T]) -> T:
+def run_async_with_cleanup[T](coro: Coroutine[Any, Any, T]) -> T:
     """Run async coroutine with proper cleanup on KeyboardInterrupt.
 
     This ensures that when Ctrl+C is pressed, all pending tasks are cancelled

claude_task_master/core/config.py

@@ -109,8 +109,8 @@ class ToolsConfig(BaseModel):
     """
 
     planning: list[str] = Field(
-        default_factory=lambda: ["Read", "Glob", "Grep", "Bash"],
-        description="Tools available during planning phase (read-only + bash for checks).",
+        default_factory=lambda: ["Read", "Glob", "Grep", "Bash", "WebFetch", "WebSearch"],
+        description="Tools available during planning phase (read-only + bash for checks + web tools for research).",
     )
     verification: list[str] = Field(
         default_factory=lambda: ["Read", "Glob", "Grep", "Bash"],
@@ -152,7 +152,7 @@ class ClaudeTaskMasterConfig(BaseModel):
         "auto_push": true
       },
       "tools": {
-        "planning": ["Read", "Glob", "Grep", "Bash"],
+        "planning": ["Read", "Glob", "Grep", "Bash", "WebFetch", "WebSearch"],
         "verification": ["Read", "Glob", "Grep", "Bash"],
         "working": []
       }