labnotes 0.2.6__tar.gz → 0.2.7__tar.gz

{labnotes-0.2.6 → labnotes-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: labnotes
-Version: 0.2.6
+Version: 0.2.7
 Summary: Browse experiment designs, notes, and run data from a folder.
 Project-URL: Repository, https://github.com/thekevinscott/labnotes
 Author-email: Kevin Scott <me@thekevinscott.com>

{labnotes-0.2.6 → labnotes-0.2.7}/src/labnotes/cli.py

@@ -2,6 +2,7 @@ from pathlib import Path
 
 import click
 
+from labnotes.command_line_group import CommandLineGroup
 from labnotes.context_guide import context_guide
 from labnotes.experiments import experiments
 from labnotes.init import init
@@ -11,6 +12,7 @@ from labnotes.server import server
 
 
 @click.group(
+    cls=CommandLineGroup,
     invoke_without_command=True,
     help="Manage experiments in a folder. `labnotes context` prints a guide for"
     " coding agents.",

labnotes-0.2.7/src/labnotes/command_line_group.py

@@ -0,0 +1,20 @@
+import shlex
+from typing import Any
+
+import click
+
+COMMAND_LINE = "labnotes.command_line"
+
+
+class CommandLineGroup(click.Group):
+    def make_context(
+        self,
+        info_name: str | None,
+        args: list[str],
+        parent: click.Context | None = None,
+        **extra: Any,
+    ) -> click.Context:
+        command_line = shlex.join(["labnotes", *args])
+        ctx = super().make_context(info_name, args, parent, **extra)
+        ctx.meta[COMMAND_LINE] = command_line
+        return ctx

{labnotes-0.2.6 → labnotes-0.2.7}/src/labnotes/experiments.py

@@ -4,6 +4,7 @@ from pathlib import Path
 
 import click
 
+from labnotes.command_line_group import COMMAND_LINE
 from labnotes.create_experiment import create_experiment
 from labnotes.runner.start_experiment import start_experiment
 from labnotes.utils.format_bytes import format_bytes
@@ -163,6 +164,7 @@ def experiments(
                 passthrough or None,
                 setup,
                 teardown,
+                ctx.meta.get(COMMAND_LINE),
             )
             try:
                 if sys.stdout.isatty() and sys.stderr.isatty():

{labnotes-0.2.6 → labnotes-0.2.7}/src/labnotes/runner/execute_run.py

@@ -5,6 +5,7 @@ import os
 import time
 from collections.abc import Callable, Sequence
 from datetime import UTC, datetime
+from importlib.metadata import version
 from pathlib import Path
 
 import psutil
@@ -43,6 +44,7 @@ async def execute_run(
     environment: dict[str, str] | None = None,
     git_info: dict[str, str | bool] | None = None,
     continues: bool = False,
+    command_line: str | None = None,
 ) -> dict[str, object]:
     resolved = shell_command(command, folder, params, args)
     git = git_info if git_info is not None else await asyncio.to_thread(run_git, root)
@@ -56,6 +58,8 @@ async def execute_run(
         "ended_by": None,
         "signal": None,
         "git": git,
+        "labnotes_version": version("labnotes"),
+        "labnotes_command": command_line,
     }
     if args:
         run["args"] = list(args)
@@ -76,6 +80,8 @@ async def execute_run(
             "error": None,
             "git": None if git is None else git["commit"],
             "args": list(args),
+            "labnotes_version": run["labnotes_version"],
+            "labnotes_command": command_line,
             **logs,
         }
         record = folder / "run.json"

{labnotes-0.2.6 → labnotes-0.2.7}/src/labnotes/runner/first_invocation.py

@@ -8,6 +8,8 @@ def first_invocation(run: dict[str, object]) -> dict[str, object]:
         "error": run.get("error"),
         "git": git.get("commit") if isinstance(git, dict) else None,
         "args": run.get("args", []),
+        "labnotes_version": run.get("labnotes_version"),
+        "labnotes_command": run.get("labnotes_command"),
         "stdout": "stdout.log",
         "stderr": "stderr.log",
     }

{labnotes-0.2.6 → labnotes-0.2.7}/src/labnotes/runner/run_experiment.py

@@ -53,6 +53,7 @@ async def run_experiment(
     args: list[str] | None = None,
     environment: dict[str, str] | None = None,
     snapshot: ProjectSnapshot | None = None,
+    command_line: str | None = None,
 ) -> list[dict[str, object]]:
     command_root = snapshot.root if snapshot is not None else config.root
     inherited = dict(os.environ) if environment is None else environment
@@ -323,6 +324,7 @@ async def run_experiment(
                 else None
             ),
             continues=experiment.continues,
+            command_line=command_line,
         )
         del running[run], trees[run]
         report({"event": "ended", "run": run, **result})

{labnotes-0.2.6 → labnotes-0.2.7}/src/labnotes/runner/start_experiment.py

@@ -17,6 +17,7 @@ def start_experiment(
     args: list[str] | None,
     setup: str | None,
     teardown: str | None,
+    command_line: str | None = None,
 ) -> Start:
     async def start(report: Report, listen: Listen) -> list[dict[str, object]]:
         experiment, limit = await prepare_experiment(
@@ -35,6 +36,7 @@ def start_experiment(
                 only=only,
                 args=args,
                 snapshot=snapshot,
+                command_line=command_line,
             )
 
     return start