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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dlab-cli
-Version: 0.1.3
+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.3 → 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",
@@ -897,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.
@@ -1066,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.3 → 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.3 → 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.3 → 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)