inspect-ai 0.3.62__py3-none-any.whl → 0.3.64__py3-none-any.whl

inspect_ai/_view/www/src/samples/transcript/TranscriptView.mjs

@@ -1,459 +0,0 @@
-// @ts-check
-import { html } from "htm/preact";
-import { useCallback, useState } from "preact/hooks";
-import { SampleInitEventView } from "./SampleInitEventView.mjs";
-import { StateEventView } from "./state/StateEventView.mjs";
-import { StepEventView } from "./StepEventView.mjs";
-import { SubtaskEventView } from "./SubtaskEventView.mjs";
-import { ModelEventView } from "./ModelEventView.mjs";
-import { LoggerEventView } from "./LoggerEventView.mjs";
-import { InfoEventView } from "./InfoEventView.mjs";
-import { ScoreEventView } from "./ScoreEventView.mjs";
-import { ToolEventView } from "./ToolEventView.mjs";
-import { ErrorEventView } from "./ErrorEventView.mjs";
-import { InputEventView } from "./InputEventView.mjs";
-import { ApprovalEventView } from "./ApprovalEventView.mjs";
-import { SampleLimitEventView } from "./SampleLimitEventView.mjs";
-import { FontSize } from "../../appearance/Fonts.mjs";
-import { EventNode } from "./Types.mjs";
-// @ts-ignore
-import { VirtualList } from "../../components/VirtualList.mjs";
-
-/**
- * Renders the TranscriptView component.
- *
- * @param {Object} props - The parameters for the component.
- * @param {string} props.id - The identifier for this view
- * @param {import("../../types/log").Events} props.events - The transcript events to display.
- * @param {number} props.depth - The base depth for this transcript view
- * @returns {import("preact").JSX.Element} The TranscriptView component.
- */
-export const TranscriptView = ({ id, events, depth = 0 }) => {
-  const [transcriptState, setTranscriptState] = useState({});
-  const onTranscriptState = useCallback(
-    (state) => {
-      setTranscriptState(state);
-    },
-    [transcriptState, setTranscriptState],
-  );
-
-  // Normalize Events themselves
-  const resolvedEvents = fixupEventStream(events);
-  const eventNodes = treeifyEvents(resolvedEvents, depth);
-  return html`
-    <${TranscriptComponent}
-      id=${id}
-      eventNodes=${eventNodes}
-      transcriptState=${transcriptState}
-      setTranscriptState=${onTranscriptState}
-    />
-  `;
-};
-
-/**
- * Renders the Transcript component.
- *
- * @param {Object} props - The parameters for the component.
- * @param {string} props.id - The identifier for this view
- * @param {import("../../types/log").Events} props.events - The transcript events to display.
- * @param {Object} props.style - The transcript style to display.
- * @param {number} props.depth - The base depth for this transcript view
- * @param {import("htm/preact").MutableRef<HTMLElement>} props.scrollRef - The scrollable parent element
- * @returns {import("preact").JSX.Element} The TranscriptView component.
- */
-export const TranscriptVirtualList = (props) => {
-  let { id, scrollRef, events, depth, style } = props;
-
-  // Normalize Events themselves
-  const resolvedEvents = fixupEventStream(events);
-  const eventNodes = treeifyEvents(resolvedEvents, depth);
-
-  const [transcriptState, setTranscriptState] = useState({});
-  const onTranscriptState = useCallback(
-    (state) => {
-      setTranscriptState(state);
-    },
-    [transcriptState, setTranscriptState],
-  );
-
-  return html`<${TranscriptVirtualListComponent}
-    id=${id}
-    eventNodes=${eventNodes}
-    style=${style}
-    scrollRef=${scrollRef}
-    transcriptState=${transcriptState}
-    setTranscriptState=${onTranscriptState}
-  />`;
-};
-
-/**
- * Renders the Transcript component.
- *
- * @param {Object} props - The parameters for the component.
- * @param {string} props.id - The identifier for this view
- * @param {EventNode[]} props.eventNodes - The transcript events nodes to display.
- * @param {Object} props.style - The transcript style to display.
- * @param {import("htm/preact").MutableRef<HTMLElement>} props.scrollRef - The scrollable parent element
- * @param {import("./Types.mjs").TranscriptState} props.transcriptState - The state for this transcript
- * @param {(state: import("./Types.mjs").TranscriptState) => void} props.setTranscriptState - Set the transcript state for this transcript
- * @returns {import("preact").JSX.Element} The TranscriptView component.
- */
-export const TranscriptVirtualListComponent = ({
-  id,
-  eventNodes,
-  style,
-  scrollRef,
-  transcriptState,
-  setTranscriptState,
-}) => {
-  const renderRow = (item, index) => {
-    const toggleStyle = {};
-    if (item.depth % 2 == 0) {
-      toggleStyle.backgroundColor = "var(--bs-light-bg-subtle)";
-    } else {
-      toggleStyle.backgroundColor = "var(--bs-body-bg)";
-    }
-
-    let paddingTop = "0";
-    if (index === 0) {
-      paddingTop = ".5em";
-    }
-    const eventId = `${id}-event${index}`;
-    const setEventState = useCallback(
-      (state) => {
-        setTranscriptState({ ...transcriptState, [eventId]: state });
-      },
-      [setTranscriptState, transcriptState],
-    );
-
-    return html`<div style=${{ paddingTop, paddingBottom: ".5em" }}>
-      <${RenderedEventNode}
-        id=${eventId}
-        node=${item}
-        style=${{
-          ...toggleStyle,
-          ...style,
-        }}
-        scrollRef=${scrollRef}
-        eventState=${transcriptState[eventId] || {}}
-        setEventState=${setEventState}
-      />
-    </div>`;
-  };
-
-  return html`<${VirtualList}
-    data=${eventNodes}
-    tabIndex="0"
-    renderRow=${renderRow}
-    scrollRef=${scrollRef}
-    style=${{ width: "100%", marginTop: "1em" }}
-  />`;
-};
-
-/**
- * Renders the Transcript component.
- *
- * @param {Object} props - The parameters for the component.
- * @param {string} props.id - The identifier for this view
- * @param {EventNode[]} props.eventNodes - The transcript events nodes to display.
- * @param {Object} props.style - The transcript style to display.
- * @param {import("./Types.mjs").TranscriptState} props.transcriptState - The state for this transcript
- * @param {(state: import("./Types.mjs").TranscriptState) => void} props.setTranscriptState - Set the transcript state for this transcript
- * @returns {import("preact").JSX.Element} The TranscriptView component.
- */
-export const TranscriptComponent = ({
-  id,
-  transcriptState,
-  setTranscriptState,
-  eventNodes,
-  style,
-}) => {
-  const rows = eventNodes.map((eventNode, index) => {
-    const toggleStyle = {};
-    if (eventNode.depth % 2 == 0) {
-      toggleStyle.backgroundColor = "var(--bs-light-bg-subtle)";
-    } else {
-      toggleStyle.backgroundColor = "var(--bs-body-bg)";
-    }
-    if (index === eventNodes.length - 1) {
-      toggleStyle.marginBottom = "0";
-    } else if (eventNode.depth === 0) {
-      toggleStyle.marginBottom = "1.5em";
-    }
-
-    let paddingBottom = ".5em";
-    if (index === eventNodes.length - 1) {
-      paddingBottom = "0";
-    }
-
-    const eventId = `${id}-event${index}`;
-    const setEventState = useCallback(
-      (state) => {
-        setTranscriptState({ ...transcriptState, [eventId]: state });
-      },
-      [setTranscriptState, transcriptState],
-    );
-
-    const row = html`
-      <div style=${{ paddingBottom }}>
-        <${RenderedEventNode}
-          id=${eventId}
-          node=${eventNode}
-          style=${{
-            ...toggleStyle,
-            ...style,
-          }}
-          eventState=${transcriptState[eventId] || {}}
-          setEventState=${setEventState}
-        />
-      </div>
-    `;
-    return row;
-  });
-
-  return html`<div
-    id=${id}
-    key=${id}
-    style=${{
-      fontSize: FontSize.small,
-      display: "grid",
-      margin: "0.5em 0 0 0",
-      width: "100%",
-    }}
-  >
-    ${rows}
-  </div>`;
-};
-
-/**
- * Renders the event based on its type.
- *
- * @param { Object } props - This event.
- * @param {string} props.id - The id for this event.
- * @param { EventNode } props.node - This event.
- * @param { Object } props.style - The style for this node.
- * @param {import("htm/preact").MutableRef<HTMLElement>} props.scrollRef - The scrollable parent element
- * @param {import("./Types.mjs").TranscriptEventState} props.eventState - The state for this event
- * @param {(state: import("./Types.mjs").TranscriptEventState) => void} props.setEventState - Update the state for this event
- * @returns {import("preact").JSX.Element} The rendered event.
- */
-export const RenderedEventNode = ({
-  id,
-  node,
-  style,
-  scrollRef,
-  eventState,
-  setEventState,
-}) => {
-  switch (node.event.event) {
-    case "sample_init":
-      return html`<${SampleInitEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "sample_limit":
-      return html`<${SampleLimitEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "info":
-      return html`<${InfoEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "logger":
-      return html`<${LoggerEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "model":
-      return html`<${ModelEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "score":
-      return html`<${ScoreEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "state":
-      return html`<${StateEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "step":
-      return html`<${StepEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        children=${node.children}
-        style=${style}
-        scrollRef=${scrollRef}
-      />`;
-
-    case "store":
-      return html`<${StateEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-        isStore=${true}
-      />`;
-
-    case "subtask":
-      return html`<${SubtaskEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-        depth=${node.depth}
-      />`;
-
-    case "tool":
-      return html`<${ToolEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-        depth=${node.depth}
-      />`;
-
-    case "input":
-      return html`<${InputEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "error":
-      return html`<${ErrorEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    case "approval":
-      return html`<${ApprovalEventView}
-        id=${id}
-        event=${node.event}
-        eventState=${eventState}
-        setEventState=${setEventState}
-        style=${style}
-      />`;
-
-    default:
-      return html``;
-  }
-};
-
-/**
- * Normalizes event content
- *
- * @param {import("../../types/log").Events} events - The transcript events to display.
- * @returns {import("../../types/log").Events} Events with resolved content.
- */
-const fixupEventStream = (events) => {
-  const initEventIndex = events.findIndex((e) => {
-    return e.event === "sample_init";
-  });
-  const initEvent = events[initEventIndex];
-
-  const fixedUp = [...events];
-  if (initEvent) {
-    fixedUp.splice(initEventIndex, 0, {
-      timestamp: initEvent.timestamp,
-      event: "step",
-      action: "begin",
-      type: null,
-      name: "sample_init",
-      pending: false,
-    });
-
-    fixedUp.splice(initEventIndex + 2, 0, {
-      timestamp: initEvent.timestamp,
-      event: "step",
-      action: "end",
-      type: null,
-      name: "sample_init",
-      pending: false,
-    });
-  }
-
-  return fixedUp;
-};
-
-/**
- * Gathers events into a hierarchy of EventNodes.
- * @param {import("../../types/log").Events} events - The transcript events to display.
- * @param { number } depth - the base depth
- * @returns {EventNode[]} An array of root EventNodes.
- */
-function treeifyEvents(events, depth) {
-  const rootNodes = [];
-  const stack = [];
-
-  const pushNode = (event) => {
-    const node = new EventNode(event, stack.length + depth);
-    if (stack.length > 0) {
-      const parentNode = stack[stack.length - 1];
-      parentNode.children.push(node);
-    } else {
-      rootNodes.push(node);
-    }
-    return node;
-  };
-
-  events.forEach((event) => {
-    if (event.event === "step" && event.action === "begin") {
-      // Starting a new step
-      const node = pushNode(event);
-      stack.push(node);
-    } else if (event.event === "step" && event.action === "end") {
-      // An ending step
-      if (stack.length > 0) {
-        stack.pop();
-      }
-    } else {
-      // An event
-      pushNode(event);
-    }
-  });
-
-  return rootNodes;
-}

inspect_ai/_view/www/src/samples/transcript/Types.mjs

@@ -1,44 +0,0 @@
-/**
- * @typedef {Object} StateManager
- * @property {string} scope - The scope (name) of this statemanager
- * @property {function(): Object} getState - Retrieves the current state object.
- * @property {function(Object): void} initializeState - Initializes the current state with a new state object.
- * @property {function(import("../../types/log").Changes): Object} applyChanges - Updates the current state with a new state object.*
- */
-
-/**
- * Class representing a node for an event in a tree.
- */
-export class EventNode {
-  /**
-   * Create an EventNode.
-   * @param { import("../../types/log").SampleInitEvent | import("../../types/log").SampleLimitEvent | import("../../types/log").StateEvent | import("../../types/log").StoreEvent | import("../../types/log").ModelEvent | import("../../types/log").LoggerEvent | import("../../types/log").InfoEvent | import("../../types/log").StepEvent | import("../../types/log").SubtaskEvent| import("../../types/log").ScoreEvent | import("../../types/log").ToolEvent | import("../../types/log").InputEvent | import("../../types/log").ErrorEvent | import("../../types/log").ApprovalEvent } event - This event.
-   * @param {number} depth - the depth of this item
-   */
-  constructor(event, depth) {
-    /**
-     * @type { import("../../types/log").SampleInitEvent | import("../../types/log").SampleLimitEvent | import("../../types/log").StateEvent | import("../../types/log").StoreEvent | import("../../types/log").ModelEvent | import("../../types/log").LoggerEvent | import("../../types/log").InfoEvent | import("../../types/log").StepEvent | import("../../types/log").SubtaskEvent| import("../../types/log").ScoreEvent | import("../../types/log").ToolEvent | import("../../types/log").InputEvent | import("../../types/log").ErrorEvent | import("../../types/log").ApprovalEvent } event - This event.
-     */
-    this.event = event;
-
-    /**
-     * @type {EventNode[]} children - An array of child EventNodes.
-     */
-    this.children = [];
-
-    /**
-     * @type {number} depth - the depth of this item
-     */
-    this.depth = depth;
-  }
-}
-
-/**
- * @typedef {Record<string, TranscriptEventState>} TranscriptState
- */
-
-/**
- * @typedef {Object} TranscriptEventState
- * @property {string} [selectedNav] - The selected nav for this event
- * @property {boolean} [collapsed] - The collapse state for this event
- */

inspect_ai/_view/www/src/samples/transcript/state/StateDiffView.mjs

@@ -1,53 +0,0 @@
-// @ts-check
-/// <reference path="../../../types/jsondiffpatch.d.ts" />
-
-import { html } from "htm/preact";
-import { diff } from "jsondiffpatch";
-import { format } from "jsondiffpatch/formatters/html";
-
-/**
- * Renders a view displaying a list of state changes.
- *
- * @param {Object} props - The properties for the component.
- * @param {Object} props.before - The object in its before state
- * @param {Object} props.after - The object in its before state
- * @param {Record<string, string>} [props.style] - Optional custom styles for the view container.
- * @returns {import("preact").JSX.Element | undefined} The component.
- */
-export const StateDiffView = ({ before, after, style }) => {
-  // Diff the objects and render the diff
-  const state_diff = diff(sanitizeKeys(before), sanitizeKeys(after));
-
-  const html_result = format(state_diff) || "Unable to render differences";
-  return html`<div
-    dangerouslySetInnerHTML=${{ __html: unescapeNewlines(html_result) }}
-    style=${{ ...style }}
-  ></div>`;
-};
-function unescapeNewlines(obj) {
-  if (typeof obj === "string") {
-    return obj.replace(/\\n/g, "\n");
-  } else if (typeof obj === "object") {
-    for (let key in obj) {
-      obj[key] = unescapeNewlines(obj[key]);
-    }
-  }
-  return obj;
-}
-
-function sanitizeKeys(obj) {
-  if (typeof obj !== "object" || obj === null) {
-    return obj;
-  }
-
-  if (Array.isArray(obj)) {
-    return obj.map(sanitizeKeys);
-  }
-
-  return Object.fromEntries(
-    Object.entries(obj).map(([key, value]) => [
-      key.replace(/</g, "&lt;").replace(/>/g, "&gt;"),
-      sanitizeKeys(value),
-    ]),
-  );
-}