inspect-ai 0.3.69__py3-none-any.whl → 0.3.70__py3-none-any.whl

inspect_ai/_util/rich.py

@@ -2,23 +2,22 @@ from rich.console import RenderableType
 from rich.style import Style
 from rich.text import Text
 
+from inspect_ai._util.text import truncate_lines
+
 
 def lines_display(
     text: str, max_lines: int = 100, style: str | Style = ""
 ) -> list[RenderableType]:
-    lines = text.splitlines()
-    if len(lines) > max_lines:
-        content: list[RenderableType] = [
-            Text("\n".join(lines[0:max_lines]), style=style)
-        ]
+    lines, truncated = truncate_lines(text, max_lines)
+
+    content: list[RenderableType] = [Text(lines, style=style)]
+    if truncated is not None:
         content.append(Text())
         content.append(
             Text.from_markup(
-                f"[italic]Output truncated ({len(lines) - max_lines} additional lines)...[/italic]",
+                f"[italic]Output truncated ({truncated} additional lines)...[/italic]",
                 style=style,
             )
         )
-    else:
-        content = [Text(text, style=style)]
 
     return content

inspect_ai/_util/text.py

@@ -134,6 +134,19 @@ def truncate(text: str, length: int, overflow: str = "...", pad: bool = True) ->
     return truncated
 
 
+def truncate_lines(
+    text: str, max_lines: int = 100, max_characters: int | None = 100 * 100
+) -> tuple[str, int | None]:
+    if max_characters is not None:
+        text = truncate(text, max_characters)
+    lines = text.splitlines()
+    if len(lines) > max_lines:
+        output = "\n".join(lines[0:max_lines])
+        return output, len(lines) - max_lines
+    else:
+        return text, None
+
+
 def generate_large_text(target_tokens: int) -> str:
     """Generate a large amount of text with approximately the target number of tokens"""
     generated_text = []

inspect_ai/_util/transcript.py

@@ -122,8 +122,16 @@ def transcript_reasoning(reasoning: str) -> list[RenderableType]:
     return content
 
 
-def transcript_separator(title: str, color: str) -> RenderableType:
-    return Rule(title=title, style=f"{color} bold", align="center", end="\n\n")
+def transcript_separator(
+    title: str, color: str, characters: str = "─"
+) -> RenderableType:
+    return Rule(
+        title=title,
+        characters=characters,
+        style=f"{color} bold",
+        align="center",
+        end="\n\n",
+    )
 
 
 def transcript_function(function: str, arguments: dict[str, Any]) -> RenderableType:

inspect_ai/_util/working.py

@@ -0,0 +1,46 @@
+import time
+from contextvars import ContextVar
+
+
+def init_sample_working_limit(start_time: float, working_limit: float | None) -> None:
+    _sample_working_limit.set(working_limit)
+    _sample_start_time.set(start_time)
+    _sample_waiting_time.set(0)
+
+
+def sample_waiting_time() -> float:
+    return _sample_waiting_time.get()
+
+
+def report_sample_waiting_time(waiting_time: float) -> None:
+    _sample_waiting_time.set(_sample_waiting_time.get() + waiting_time)
+    check_sample_working_limit()
+
+
+def check_sample_working_limit() -> None:
+    # 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:
+        from inspect_ai.solver._limit import SampleLimitExceededError
+
+        raise SampleLimitExceededError(
+            type="working",
+            value=int(working_time),
+            limit=int(working_limit),
+            message=f"Exceeded working time limit ({working_limit:,} seconds)",
+        )
+
+
+_sample_working_limit: ContextVar[float | None] = ContextVar(
+    "sample_working_limit", default=None
+)
+
+_sample_start_time: ContextVar[float] = ContextVar("sample_start_time", default=0)
+
+_sample_waiting_time: ContextVar[float] = ContextVar("sample_waiting_time", default=0)

inspect_ai/_view/www/dist/assets/index.css

@@ -16317,35 +16317,43 @@ ul.jsondiffpatch-textdiff {
   width: 100%;
   margin-top: 1em;
 }
-._tabPanel_14odp_1 {
+._tabPanel_1isha_1 {
   padding-bottom: 1em;
 }
 
-._fullWidth_14odp_5 {
+._fullWidth_1isha_5 {
   width: 100%;
 }
 
-._metadataPanel_14odp_9 {
+._metadataPanel_1isha_9 {
   display: flex;
   flex-wrap: wrap;
-  align-items: flex-start;
+  align-items: stretch;
   gap: 1em;
   padding-left: 0;
   margin-top: 0.5em;
 }
 
-._padded_14odp_18 {
+._padded_1isha_18 {
   padding-left: 0.8em;
   margin-top: 0.4em;
 }
 
-._ansi_14odp_23 {
+._ansi_1isha_23 {
   margin: 1em 0;
 }
 
-._noTop_14odp_27 {
+._noTop_1isha_27 {
   margin-top: 0;
 }
+
+._timePanel_1isha_31 {
+  display: grid;
+  grid-template-columns: max-content max-content;
+  grid-template-rows: auto;
+  column-gap: 0.5em;
+  min-width: 200px;
+}
 ._flatBody_gk2ju_1 {
   color: var(--bs-danger);
   display: grid;
@@ -16365,15 +16373,15 @@ ul.jsondiffpatch-textdiff {
   line-height: var(--inspect-font-size-base);
   height: var(--inspect-font-size-base);
 }
-._target_yamz4_1 {
+._target_9qy4e_1 {
   padding-left: 0;
 }
 
-._answer_yamz4_5 {
+._answer_9qy4e_5 {
   padding-left: 0;
 }
 
-._grid_yamz4_9 {
+._grid_9qy4e_9 {
   display: grid;
   grid-column-gap: 0.5em;
   border-bottom: solid var(--bs-border-color) 1px;
@@ -16381,14 +16389,18 @@ ul.jsondiffpatch-textdiff {
   padding: 0em 1em 1em 1em;
 }
 
-._centerLabel_yamz4_17 {
+._centerLabel_9qy4e_17 {
   display: flex;
   justify-content: center;
 }
 
-._wrap_yamz4_22 {
+._wrap_9qy4e_22 {
   word-wrap: anywhere;
 }
+
+._titled_9qy4e_26:hover {
+  cursor: pointer;
+}
 ._title_19l1b_1 {
   margin-left: 0.5em;
   display: grid;
@@ -19070,6 +19082,38 @@ span.ap-marker-container:hover span.ap-marker {
   padding: 0;
   margin-bottom: 0;
 }
+._contents_iwnfd_1 {
+  margin-top: 0.5em;
+}
+
+._contents_iwnfd_1 > :last-child {
+  margin-bottom: 0;
+}
+
+._twoColumn_iwnfd_9 {
+  display: grid;
+  grid-template-columns: auto 1fr;
+  column-gap: 1.5em;
+}
+
+._exec_iwnfd_15 {
+  margin-top: 0.5em;
+}
+
+._result_iwnfd_19 {
+  margin-top: 0.5em;
+}
+
+._fileLabel_iwnfd_23 {
+  margin-top: 0;
+  margin-bottom: 0;
+}
+
+._wrapPre_iwnfd_28 {
+  white-space: pre-wrap;
+  word-wrap: break-word;
+  overflow-wrap: break-word;
+}
 ._darkenedBg_c8m1t_1 {
   background-color: var(--bs-light-bg-subtle);
 }