dlab-cli 0.1.2__tar.gz → 0.2.0__tar.gz

{dlab_cli-0.1.2 → dlab_cli-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dlab-cli
-Version: 0.1.2
+Version: 0.2.0
 Summary: A harness for agentic data science — run coding agents with domain skills, parallel subagents, and frozen Docker environments
 Author: DecisionAI
 License: Apache-2.0
@@ -18,6 +18,8 @@ Requires-Dist: pyyaml>=6.0
 Requires-Dist: textual>=2.0
 Requires-Dist: modal>=0.70
 Requires-Dist: dhub-cli>=0.10
+Requires-Dist: fastapi>=0.100
+Requires-Dist: uvicorn[standard]>=0.20
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.23; extra == "dev"

{dlab_cli-0.1.2 → dlab_cli-0.2.0}/dlab/__init__.py

@@ -3,4 +3,4 @@ dlab: CLI wrapper that runs opencode in automated mode, sandboxed with Docker,
 and with parallel-agent capability.
 """
 
-__version__ = "0.1.2"
+__version__ = "0.1.3"

{dlab_cli-0.1.2 → dlab_cli-0.2.0}/dlab/cli.py

@@ -273,6 +273,34 @@ def create_parser() -> argparse.ArgumentParser:
         help="Path to session work directory (default: cwd if it has _opencode_logs)",
     )
 
+    # View subcommand (browser-based session viewer)
+    view_parser = subparsers.add_parser(
+        "view",
+        help="Open browser-based session viewer with DAG visualization",
+    )
+    view_parser.add_argument(
+        "work_dir",
+        metavar="WORK_DIR",
+        help="Path to session work directory",
+    )
+    view_parser.add_argument(
+        "--port",
+        type=int,
+        default=0,
+        help="Port for the viewer server (default: auto-select)",
+    )
+    view_parser.add_argument(
+        "--no-open",
+        action="store_true",
+        help="Start server without opening browser",
+    )
+    view_parser.add_argument(
+        "--export",
+        metavar="FILE",
+        default=None,
+        help="Export self-contained HTML file instead of starting server",
+    )
+
     # Create decision-pack subcommand
     create_dpack_parser = subparsers.add_parser(
         "create-dpack",
@@ -401,13 +429,18 @@ def cmd_run(args: argparse.Namespace) -> int:
     if continue_mode:
         continue_dir = Path(args.continue_dir).resolve()
         if not continue_dir.exists():
-            raise ValueError(f"Continue directory not found: {args.continue_dir}")
+            console.print(f"Oops, continue directory [bold]{args.continue_dir}[/bold] not found.")
+            return 1
 
         if args.work_dir:
             # Copy continue-dir to work-dir, then continue from there
             work_path = Path(args.work_dir).resolve()
             if work_path.exists():
-                raise ValueError(f"Work directory already exists: {args.work_dir}")
+                console.print(
+                    f"Oops, work directory [bold]{args.work_dir}[/bold] already exists.\n"
+                    f"You can remove it with: [cyan]rm -rf {args.work_dir}[/cyan]"
+                )
+                return 1
             shutil.copytree(continue_dir, work_path)
             work_dir = str(work_path)
             print(f"Copied {continue_dir} to {work_dir}")
@@ -423,10 +456,9 @@ def cmd_run(args: argparse.Namespace) -> int:
         # Overwrite .opencode with latest from decision-pack (agent prompts may have changed)
         opencode_dir = Path(work_dir) / ".opencode"
         if opencode_dir.exists():
-            if no_sandboxing:
-                # Local mode: files are user-owned
+            try:
                 shutil.rmtree(opencode_dir)
-            else:
+            except PermissionError:
                 # Docker mode: files may be root-owned (e.g. node_modules/)
                 subprocess.run(
                     ["sudo", "rm", "-rf", str(opencode_dir)],
@@ -893,6 +925,47 @@ def cmd_create_parallel_agent(args: argparse.Namespace) -> int:
     return 0
 
 
+def cmd_view(args: argparse.Namespace) -> int:
+    """
+    Handle view mode - browser-based session viewer with DAG visualization.
+
+    Parameters
+    ----------
+    args : argparse.Namespace
+        Parsed command-line arguments.
+
+    Returns
+    -------
+    int
+        Exit code (0 for success, non-zero for failure).
+    """
+    work_dir: Path = Path(args.work_dir).resolve()
+
+    if not work_dir.exists():
+        print(f"Error: Work directory not found: {work_dir}", file=sys.stderr)
+        return 1
+
+    logs_dir: Path = work_dir / "_opencode_logs"
+    if not logs_dir.exists():
+        print(f"Error: No logs directory found: {logs_dir}", file=sys.stderr)
+        print("Make sure this is a valid dlab session directory.", file=sys.stderr)
+        return 1
+
+    # Export mode — no server dependencies needed
+    if args.export:
+        from dlab.viewer.server import export_viewer
+        output_path: Path = Path(args.export)
+        return export_viewer(work_dir, output_path)
+
+    from dlab.viewer import run_viewer
+
+    return run_viewer(
+        work_dir,
+        port=args.port,
+        open_browser=not args.no_open,
+    )
+
+
 def cmd_timeline(args: argparse.Namespace) -> int:
     """
     Handle timeline mode - display execution timeline for a session.
@@ -1062,6 +1135,8 @@ def main() -> None:
         exit_code = cmd_create_dpack(args)
     elif args.command == "timeline":
         exit_code = cmd_timeline(args)
+    elif args.command == "view":
+        exit_code = cmd_view(args)
     elif args.dpack or args.data or args.prompt or args.prompt_file or args.continue_dir:
         exit_code = cmd_run(args)
     else:

{dlab_cli-0.1.2 → dlab_cli-0.2.0}/dlab/create_dpack.py

@@ -884,8 +884,16 @@ export default tool({
   args: {
     input: tool.schema.string().describe("Input to process"),
   },
-  async run({ input }) {
-    return `Processed: ${input}`
+  async execute(args) {
+    // Run a CLI command with Bun shell (e.g. Python, bash, etc.)
+    const result = await Bun.$`echo "Processing: ${args.input}"`.nothrow()
+    const stdout = result.stdout.toString()
+    const stderr = result.stderr.toString()
+
+    if (result.exitCode !== 0) {
+      return `ERROR (exit code ${result.exitCode}):\\n${stderr}`
+    }
+    return stdout.trim()
   },
 })
 """

{dlab_cli-0.1.2 → dlab_cli-0.2.0}/dlab/docker.py

@@ -527,7 +527,8 @@ def build_runner_script(
     return f'''#!/bin/bash
 set -o pipefail
 prompt=$(cat {prompt_file})
-opencode run --format json --log-level DEBUG --model "{model}" "$prompt" 2>&1 | tee /_opencode_logs/{log_prefix}.log
+echo "$prompt" | python3 -c "import json,sys; print(json.dumps({{'type':'dlab_start','timestamp':int(__import__('time').time()*1000),'model':'{model}','agent':'{log_prefix}','prompt':sys.stdin.read().strip()}}))" > /_opencode_logs/{log_prefix}.log
+opencode run --format json --log-level DEBUG --model "{model}" "$prompt" 2>&1 | tee -a /_opencode_logs/{log_prefix}.log
 '''
 
 

{dlab_cli-0.1.2 → dlab_cli-0.2.0}/dlab/js/parallel-agents.ts

@@ -269,9 +269,11 @@ CRITICAL OUTPUT RULES:
 `
       const fullPrompt = subagentContext + args.prompts[i] + "\n\n" + (config.subagent_suffix_prompt || "")
 
-      // Spawn - log file starts with JSON directly (no header)
       const logFile = join(logsDir, `instance-${i + 1}.log`)
 
+      // Write dlab_start event as first line (model, agent, prompt)
+      writeFileSync(logFile, JSON.stringify({type: "dlab_start", timestamp: Date.now(), model, agent: args.agent, prompt: fullPrompt}) + "\n")
+
       const proc = Bun.spawn(["opencode", "run", "--format", "json", "--log-level", "DEBUG", "--model", model, fullPrompt], {
         cwd: instanceDir,
         stdout: "pipe",
@@ -355,6 +357,9 @@ RULES:
 
       const consLogFile = join(logsDir, "consolidator.log")
 
+      // Write dlab_start event as first line
+      writeFileSync(consLogFile, JSON.stringify({type: "dlab_start", timestamp: Date.now(), model: consolidatorModel, agent: "consolidator", prompt: consolidatorPrompt}) + "\n")
+
       const consProc = Bun.spawn(["opencode", "run", "--format", "json", "--log-level", "DEBUG", "--model", consolidatorModel, consolidatorPrompt], {
         cwd: runDir,
         stdout: "pipe",

{dlab_cli-0.1.2 → dlab_cli-0.2.0}/dlab/local.py

@@ -251,7 +251,8 @@ def run_opencode_local(
     runner_script: str = f'''#!/bin/bash
 set -o pipefail
 prompt=$(cat "{prompt_file}")
-opencode run --format json --log-level DEBUG --model "{model}" "$prompt" 2>&1 | tee "{log_path}"
+echo "$prompt" | python3 -c "import json,sys; print(json.dumps({{'type':'dlab_start','timestamp':int(__import__('time').time()*1000),'model':'{model}','agent':'{log_prefix}','prompt':sys.stdin.read().strip()}}))" > "{log_path}"
+opencode run --format json --log-level DEBUG --model "{model}" "$prompt" 2>&1 | tee -a "{log_path}"
 '''
     runner_file: Path = work_path / ".run_opencode.sh"
     runner_file.write_text(runner_script)