inspect-ai 0.3.98__py3-none-any.whl → 0.3.100__py3-none-any.whl

inspect_ai/_util/working.py

@@ -1,19 +1,14 @@
 import time
 from contextvars import ContextVar
 
-from inspect_ai.util._limit import LimitExceededError
+from inspect_ai.util._limit import check_working_limit, record_waiting_time
 
 
-def init_sample_working_limit(start_time: float, working_limit: float | None) -> None:
-    _sample_working_limit.set(working_limit)
+def init_sample_working_time(start_time: float) -> None:
     _sample_start_time.set(start_time)
     _sample_waiting_time.set(0)
 
 
-def end_sample_working_limit() -> None:
-    _sample_working_limit.set(None)
-
-
 def sample_waiting_time() -> float:
     return _sample_waiting_time.get()
 
@@ -23,37 +18,13 @@ def sample_working_time() -> float:
 
 
 def report_sample_waiting_time(waiting_time: float) -> None:
+    # record and check for scoped limits
+    record_waiting_time(waiting_time)
+    check_working_limit()
+
+    # record sample-level limits
     _sample_waiting_time.set(_sample_waiting_time.get() + waiting_time)
-    check_sample_working_limit()
-
-
-def check_sample_working_limit() -> None:
-    from inspect_ai.log._transcript import SampleLimitEvent, transcript
-
-    # no check if we don't have a limit
-    working_limit = _sample_working_limit.get()
-    if working_limit is None:
-        return
-
-    # are we over the limit?
-    running_time = time.monotonic() - _sample_start_time.get()
-    working_time = running_time - sample_waiting_time()
-    if working_time > working_limit:
-        message = f"Exceeded working time limit ({working_limit:,} seconds)"
-        transcript()._event(
-            SampleLimitEvent(type="working", limit=int(working_limit), message=message)
-        )
-        raise LimitExceededError(
-            type="working",
-            value=int(working_time),
-            limit=int(working_limit),
-            message=message,
-        )
-
-
-_sample_working_limit: ContextVar[float | None] = ContextVar(
-    "sample_working_limit", default=None
-)
+
 
 _sample_start_time: ContextVar[float] = ContextVar("sample_start_time", default=0)
 

inspect_ai/_view/schema.py

@@ -30,6 +30,7 @@ def sync_view_schema() -> None:
         for key in defs.keys():
             defs[key] = schema_to_strict(defs[key])
         f.write(json.dumps(schema, indent=2))
+        f.write("\n")
 
         # generate types w/ json-schema-to-typescript
         subprocess.run(
@@ -44,9 +45,10 @@ def sync_view_schema() -> None:
                 "false",
             ],
             cwd=WWW_DIR,
+            check=True,
         )
 
-        subprocess.run(["yarn", "prettier:write"], cwd=types_path.parent)
+        subprocess.run(["yarn", "prettier:write"], cwd=types_path.parent, check=True)
 
         shutil.copyfile(types_path, vs_code_types_path)
 

inspect_ai/_view/view.py

@@ -30,6 +30,20 @@ def view(
     log_level: str | None = None,
     fs_options: dict[str, Any] = {},
 ) -> None:
+    """Run the Inspect View server.
+
+    Args:
+        log_dir: Directory to view logs from.
+        recursive: Recursively list files in `log_dir`.
+        host: Tcp/ip host (defaults to "127.0.0.1").
+        port: Tcp/ip port (defaults to 7575).
+        authorization: Validate requests by checking for this authorization header.
+        log_level: Level for logging to the console: "debug", "http", "sandbox",
+            "info", "warning", "error", or "critical" (defaults to "warning").
+        fs_options: Additional arguments to pass through to the filesystem provider
+            (e.g. `S3FileSystem`). Use `{"anon": True }` if you are accessing a
+            public S3 bucket with no credentials.
+    """
     init_dotenv()
     init_logger(log_level)
 

inspect_ai/_view/www/CLAUDE.md

@@ -0,0 +1,15 @@
+# CLAUDE.md
+
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
+
+## Build/Lint/Test Commands
+
+- Run type checker, linter, and tests: `yarn check-all`
+- Run type checker: `yarn tsc`
+- Run linter: `yarn lint`
+- Run tests: `yarn test`
+
+## Other Information
+
+- The code in this project is typescript, learn more about the configuration by inspecting package.json.
+- Respect existing code patterns when modifying files. Run linting before committing changes.