inspect-ai 0.3.57__py3-none-any.whl → 0.3.59__py3-none-any.whl

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

@@ -1,70 +0,0 @@
-import { applyPatch } from "fast-json-patch";
-
-/**
- * Manages the state, providing methods to retrieve and update it.
- *
- * @param {string} scope - The name identifier for the state manager.
- * @returns {import("./Types.mjs").StateManager} An object containing `getState` and `onState` methods for managing state.
- */
-export const initStateManager = (scope) => {
-  /** @type {Object} */
-  let state = {};
-  return {
-    scope: scope,
-
-    /**
-     * Retrieves the current state object.
-     *
-     * @returns {Object} The current state object.
-     */
-    getState: () => {
-      return state;
-    },
-    /**
-     * Updates the current state with a new state object.
-     *
-     * @param {Object} newState - The new state object to update with.
-     */
-    initializeState: (newState) => {
-      state = newState;
-    },
-    /**
-     * Updates the current state with a new state object.
-     *
-     * @param {import("../../types/log").Changes} changes - The new state object to update with.
-     */
-    applyChanges: (changes) => {
-      try {
-        state = applyPatch(
-          structuredClone(state),
-          structuredClone(changes).map(ensureValidChange),
-          true,
-        ).newDocument;
-      } catch (ex) {
-        const ops = changes.reduce((prev, change) => {
-          if (!Object.keys(prev).includes(change.op)) {
-            prev[change.op] = [];
-          }
-          prev[change.op].push(change.path);
-          return prev;
-        }, {});
-        const message = `${ex.name}\nFailed to apply patch:\n${JSON.stringify(ops, undefined, 2)}`;
-        console.error(message);
-      }
-    },
-  };
-};
-
-/**
- * Ensures that the change is valid (provides default values)
- * If the operation is "add" and `value` is not present, it assigns `null` to `value`.
- *
- * @param { import("../../types/log").JsonChange } change - The change object containing the operation and value.
- * @returns {Object} The modified change object with a guaranteed `value` property.
- */
-const ensureValidChange = (change) => {
-  if (change.op === "add" && !change.value) {
-    change.value = null;
-  }
-  return change;
-};

inspect_ai/model/_trace.py

@@ -1,48 +0,0 @@
-from rich.console import RenderableType
-from rich.text import Text
-
-from inspect_ai._util.rich import lines_display
-from inspect_ai._util.transcript import transcript_markdown
-from inspect_ai.util._trace import trace_enabled, trace_panel
-
-from ._chat_message import ChatMessage, ChatMessageAssistant, ChatMessageTool
-from ._render import messages_preceding_assistant, render_tool_calls
-
-MESSAGE_TITLE = "Message"
-
-
-def trace_tool_mesage(message: ChatMessageTool) -> None:
-    if trace_enabled():
-        # truncate output to 100 lines
-        output = message.error.message if message.error else message.text.strip()
-        content = lines_display(output, 100)
-
-        trace_panel(
-            title=f"Tool Output: {message.function}",
-            content=content,
-        )
-
-
-def trace_assistant_message(
-    input: list[ChatMessage], message: ChatMessageAssistant
-) -> None:
-    if trace_enabled():
-        # print precding messages that aren't tool or assistant
-        for m in messages_preceding_assistant(input):
-            trace_panel(
-                title=m.role.capitalize(),
-                content=transcript_markdown(m.text, escape=True),
-            )
-
-        # start with assistant content
-        content: list[RenderableType] = (
-            [transcript_markdown(message.text, escape=True)] if message.text else []
-        )
-
-        # print tool calls
-        if message.tool_calls:
-            content.append(Text())
-            content.extend(render_tool_calls(message.tool_calls))
-
-        # print the assistant message
-        trace_panel(title="Assistant", content=content)