zwarm 2.3.5__py3-none-any.whl → 3.0__py3-none-any.whl

zwarm/cli/main.py

@@ -79,6 +79,7 @@ app = typer.Typer(
     [cyan]init[/]         Initialize zwarm (creates .zwarm/ with config)
     [cyan]reset[/]        Reset state and optionally config files
     [cyan]orchestrate[/]  Start orchestrator to delegate tasks to executors
+    [cyan]pilot[/]        Conversational orchestrator REPL (interactive)
     [cyan]exec[/]         Run a single executor directly (for testing)
     [cyan]status[/]       Show current state (sessions, tasks, events)
     [cyan]history[/]      Show event history log
@@ -274,6 +275,112 @@ def orchestrate(
         sys.exit(1)
 
 
+class PilotLM(str, Enum):
+    """LM options for pilot mode."""
+    gpt5_mini = "gpt5-mini"      # GPT5MiniTester - fast, cheap, good for testing
+    gpt5 = "gpt5"                # GPT5Large - standard
+    gpt5_verbose = "gpt5-verbose"  # GPT5LargeVerbose - with extended thinking
+
+
+@app.command()
+def pilot(
+    task: Annotated[Optional[str], typer.Option("--task", "-t", help="Initial task (optional)")] = None,
+    task_file: Annotated[Optional[Path], typer.Option("--task-file", "-f", help="Read task from file")] = None,
+    config: Annotated[Optional[Path], typer.Option("--config", "-c", help="Path to config YAML")] = None,
+    overrides: Annotated[Optional[list[str]], typer.Option("--set", help="Override config (key=value)")] = None,
+    working_dir: Annotated[Path, typer.Option("--working-dir", "-w", help="Working directory")] = Path("."),
+    instance: Annotated[Optional[str], typer.Option("--instance", "-i", help="Instance ID (for isolation)")] = None,
+    instance_name: Annotated[Optional[str], typer.Option("--name", "-n", help="Human-readable instance name")] = None,
+    model: Annotated[PilotLM, typer.Option("--model", "-m", help="LM to use")] = PilotLM.gpt5_verbose,
+):
+    """
+    Interactive conversational orchestrator REPL.
+
+    Like 'orchestrate' but conversational: give instructions, watch the
+    orchestrator work, course-correct in real-time, time-travel to checkpoints.
+
+    [bold]Features:[/]
+        - Streaming display of orchestrator thinking and tool calls
+        - Turn-by-turn execution with checkpoints
+        - Time travel (:goto T1) to return to previous states
+        - Session visibility (:sessions) and state inspection (:state)
+
+    [bold]Commands:[/]
+        :help              Show help
+        :history [N|all]   Show turn checkpoints
+        :goto <turn|root>  Time travel (e.g., :goto T1)
+        :state             Show orchestrator state
+        :sessions          Show active executor sessions
+        :reasoning on|off  Toggle reasoning display
+        :quit              Exit
+
+    [bold]LM Options:[/]
+        gpt5-mini     GPT5MiniTester - fast/cheap, good for testing
+        gpt5          GPT5Large - standard model
+        gpt5-verbose  GPT5LargeVerbose - with extended thinking (default)
+
+    [bold]Examples:[/]
+        [dim]# Start fresh, give instructions interactively[/]
+        $ zwarm pilot
+
+        [dim]# Start with an initial task[/]
+        $ zwarm pilot --task "Build user authentication"
+
+        [dim]# Use faster model for testing[/]
+        $ zwarm pilot --model gpt5-mini
+
+        [dim]# Named instance[/]
+        $ zwarm pilot --name my-feature
+    """
+    from zwarm.cli.pilot import run_pilot, build_pilot_orchestrator
+
+    # Resolve task (optional for pilot)
+    resolved_task = _resolve_task(task, task_file)
+
+    console.print(f"[bold]Starting pilot session...[/]")
+    console.print(f"  Working dir: {working_dir.absolute()}")
+    console.print(f"  Model: {model.value}")
+    if resolved_task:
+        console.print(f"  Initial task: {resolved_task[:60]}...")
+    if instance:
+        console.print(f"  Instance: {instance}" + (f" ({instance_name})" if instance_name else ""))
+    console.print()
+
+    orchestrator = None
+    try:
+        orchestrator = build_pilot_orchestrator(
+            config_path=config,
+            working_dir=working_dir.absolute(),
+            overrides=list(overrides or []),
+            instance_id=instance,
+            instance_name=instance_name,
+            lm_choice=model.value,
+        )
+
+        # Show instance ID if auto-generated
+        if orchestrator.instance_id and not instance:
+            console.print(f"  [dim]Instance: {orchestrator.instance_id[:8]}[/]")
+
+        # Run the pilot REPL
+        run_pilot(orchestrator, initial_task=resolved_task)
+
+        # Save state on exit
+        orchestrator.save_state()
+        console.print("\n[dim]State saved.[/]")
+
+    except KeyboardInterrupt:
+        console.print("\n\n[yellow]Interrupted.[/]")
+        if orchestrator:
+            orchestrator.save_state()
+            console.print("[dim]State saved.[/]")
+        sys.exit(1)
+    except Exception as e:
+        console.print(f"\n[red]Error:[/] {e}")
+        import traceback
+        traceback.print_exc()
+        sys.exit(1)
+
+
 @app.command()
 def exec(
     task: Annotated[str, typer.Option("--task", "-t", help="Task to execute")],