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

inspect_ai/_util/content.py

@@ -25,5 +25,27 @@ class ContentImage(BaseModel):
     """
 
 
-Content = Union[ContentText, ContentImage]
+class ContentAudio(BaseModel):
+    type: Literal["audio"] = Field(default="audio")
+    """Type."""
+
+    audio: str
+    """Audio file path or base64 encoded data URL."""
+
+    format: Literal["wav", "mp3"]
+    """Format of audio data ('mp3' or 'wav')"""
+
+
+class ContentVideo(BaseModel):
+    type: Literal["video"] = Field(default="video")
+    """Type."""
+
+    video: str
+    """Audio file path or base64 encoded data URL."""
+
+    format: Literal["mp4", "mpeg", "mov"]
+    """Format of video data ('mp4', 'mpeg', or 'mov')"""
+
+
+Content = Union[ContentText, ContentImage, ContentAudio, ContentVideo]
 """Content sent to or received from a model."""

inspect_ai/_util/images.py

@@ -3,7 +3,7 @@ import mimetypes
 
 import httpx
 
-from .file import file
+from .file import file as open_file
 from .url import (
     data_uri_mime_type,
     data_uri_to_base64,
@@ -12,34 +12,37 @@ from .url import (
 )
 
 
-async def image_as_data(image: str) -> tuple[bytes, str]:
-    if is_data_uri(image):
+async def file_as_data(file: str) -> tuple[bytes, str]:
+    if is_data_uri(file):
         # resolve mime type and base64 content
-        mime_type = data_uri_mime_type(image) or "image/png"
-        image_base64 = data_uri_to_base64(image)
-        image_bytes = base64.b64decode(image_base64)
+        mime_type = data_uri_mime_type(file) or "image/png"
+        file_base64 = data_uri_to_base64(file)
+        file_bytes = base64.b64decode(file_base64)
     else:
         # guess mime type; need strict=False for webp images
-        type, _ = mimetypes.guess_type(image, strict=False)
+        type, _ = mimetypes.guess_type(file, strict=False)
         if type:
             mime_type = type
         else:
             mime_type = "image/png"
 
         # handle url or file
-        if is_http_url(image):
+        if is_http_url(file):
             client = httpx.AsyncClient()
-            image_bytes = (await client.get(image)).content
+            file_bytes = (await client.get(file)).content
         else:
-            with file(image, "rb") as f:
-                image_bytes = f.read()
+            with open_file(file, "rb") as f:
+                file_bytes = f.read()
 
     # return bytes and type
-    return image_bytes, mime_type
+    return file_bytes, mime_type
 
 
-async def image_as_data_uri(image: str) -> str:
-    bytes, mime_type = await image_as_data(image)
-    base64_image = base64.b64encode(bytes).decode("utf-8")
-    image = f"data:{mime_type};base64,{base64_image}"
-    return image
+async def file_as_data_uri(file: str) -> str:
+    if is_data_uri(file):
+        return file
+    else:
+        bytes, mime_type = await file_as_data(file)
+        base64_file = base64.b64encode(bytes).decode("utf-8")
+        file = f"data:{mime_type};base64,{base64_file}"
+        return file

inspect_ai/_util/kvstore.py

@@ -0,0 +1,73 @@
+import sqlite3
+from contextlib import AbstractContextManager
+from typing import Any, Optional, cast
+
+from .appdirs import inspect_data_dir
+
+
+class KVStore(AbstractContextManager["KVStore"]):
+    def __init__(self, filename: str, max_entries: int | None = None):
+        self.filename = filename
+        self.max_entries = max_entries
+
+    def __enter__(self) -> "KVStore":
+        self.conn = sqlite3.connect(self.filename)
+        self.conn.execute("""
+            CREATE TABLE IF NOT EXISTS kv_store (
+                key TEXT PRIMARY KEY,
+                value TEXT,
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+            )
+        """)
+        self.conn.commit()
+        return self
+
+    def __exit__(self, *excinfo: Any) -> None:
+        self.conn.close()
+
+    def put(self, key: str, value: str) -> None:
+        # Insert or update the value
+        self.conn.execute(
+            """
+            INSERT OR REPLACE INTO kv_store (key, value, created_at)
+            VALUES (?, ?, CURRENT_TIMESTAMP)
+        """,
+            (key, value),
+        )
+
+        # If we have a max_entries limit, remove oldest entries
+        if self.max_entries:
+            count = self.count()
+            if count > self.max_entries:
+                self.conn.execute(
+                    """
+                    DELETE FROM kv_store
+                    WHERE key IN (
+                        SELECT key FROM kv_store
+                        ORDER BY created_at ASC
+                        LIMIT ?
+                    )
+                    """,
+                    (max(0, count - self.max_entries),),
+                )
+
+        self.conn.commit()
+
+    def get(self, key: str) -> Optional[str]:
+        cursor = self.conn.execute("SELECT value FROM kv_store WHERE key = ?", (key,))
+        result = cursor.fetchone()
+        return result[0] if result else None
+
+    def delete(self, key: str) -> bool:
+        cursor = self.conn.execute("DELETE FROM kv_store WHERE key = ?", (key,))
+        self.conn.commit()
+        return cursor.rowcount > 0
+
+    def count(self) -> int:
+        cursor = self.conn.execute("SELECT COUNT(*) FROM kv_store")
+        return cast(int, cursor.fetchone()[0])
+
+
+def inspect_kvstore(name: str, max_entries: int | None = None) -> KVStore:
+    filename = inspect_data_dir("kvstore") / f"{name}.db"
+    return KVStore(filename.as_posix(), max_entries=max_entries)

inspect_ai/_util/notgiven.py

@@ -0,0 +1,18 @@
+# Sentinel class used until PEP 0661 is accepted
+from typing import Literal
+
+from typing_extensions import override
+
+
+class NotGiven:
+    """A sentinel singleton class used to distinguish omitted keyword arguments from those passed in with the value None (which may have different behavior)."""
+
+    def __bool__(self) -> Literal[False]:
+        return False
+
+    @override
+    def __repr__(self) -> str:
+        return "NOT_GIVEN"
+
+
+NOT_GIVEN = NotGiven()

inspect_ai/_util/port_names.py

@@ -0,0 +1,61 @@
+from typing import Literal
+
+
+def get_service_by_port(port: int, protocol: Literal["tcp", "udp"]) -> str | None:
+    """
+    Returns the likely service running on a given port number.
+
+    Args:
+        port (int): The port number to look up
+        protocol (str): Either 'tcp' or 'udp'
+
+    Returns:
+        str: Description of the likely service, or None if not found
+    """
+    # Common port mappings based on IANA assignments and common usage
+    port_mappings = {
+        "tcp": {
+            20: "FTP (Data)",
+            21: "FTP (Control)",
+            22: "SSH",
+            23: "Telnet",
+            25: "SMTP",
+            53: "DNS",
+            80: "HTTP",
+            110: "POP3",
+            143: "IMAP",
+            443: "HTTPS",
+            445: "Microsoft-DS (SMB)",
+            587: "SMTP (Submission)",
+            993: "IMAPS",
+            995: "POP3S",
+            1433: "Microsoft SQL Server",
+            1521: "Oracle Database",
+            3306: "MySQL",
+            3389: "RDP (Remote Desktop)",
+            5432: "PostgreSQL",
+            5900: "VNC",
+            5901: "VNC Display :1",
+            5902: "VNC Display :2",
+            6080: "noVNC",
+            8080: "HTTP Alternate",
+            8443: "HTTPS Alternate",
+            27017: "MongoDB",
+            27018: "MongoDB Shard",
+            27019: "MongoDB Config Server",
+        },
+        "udp": {
+            53: "DNS",
+            67: "DHCP Server",
+            68: "DHCP Client",
+            69: "TFTP",
+            123: "NTP",
+            161: "SNMP",
+            162: "SNMP Trap",
+            514: "Syslog",
+            1194: "OpenVPN",
+            5353: "mDNS",
+        },
+    }
+
+    return port_mappings.get(protocol, {}).get(port, None)

inspect_ai/_util/text.py

@@ -108,3 +108,26 @@ def str_to_float(s: str) -> float:
         exponent = 1  # Default exponent is 1 if no superscript is present
 
     return base**exponent
+
+
+def truncate(text: str, length: int, overflow: str = "...", pad: bool = True) -> str:
+    """
+    Truncate text to specified length with optional padding and overflow indicator.
+
+    Args:
+        text (str): Text to truncate
+        length (int): Maximum length including overflow indicator
+        overflow (str): String to indicate truncation (defaults to '...')
+        pad (bool): Whether to pad the result to full length (defaults to padding)
+
+    Returns:
+        Truncated string, padded if requested
+
+    """
+    if len(text) <= length:
+        return text + (" " * (length - len(text))) if pad else text
+
+    overflow_length = len(overflow)
+    truncated = text[: length - overflow_length] + overflow
+
+    return truncated

inspect_ai/_util/thread.py

@@ -0,0 +1,5 @@
+import threading
+
+
+def is_main_thread() -> bool:
+    return threading.current_thread() is threading.main_thread()

inspect_ai/_view/www/App.css

@@ -9,6 +9,12 @@
   --inspect-input-border: var(--bs-light-border-subtle);
   --inspect-diff-add-color: #dafbe1;
   --inspect-diff-remove-color: #ffebe9;
+  --inspect-inactive-selection-background: var(--vscode-editor-inactiveSelectionBackground, #d9d9d9);
+  --inspect-active-selection-background: var(--vscode-editor-selectionBackground, #d7d4f0);
+  --inspect-focus-border-color: #86b7fe;
+  --inspect-focus-border-shadow: 0 0 0 0.25rem rgba(var(--bs-primary-rgb), 0.25);
+  --inspect-focus-border-gray-color: #808080;
+  --inspect-focus-border-gray-shadow: 0 0 0 0.25rem rgba(48, 48, 48, 0.25);
 }
 
 body:not([class^="vscode-"]) button {
@@ -650,6 +656,30 @@ table.table.table-sm td {
   height: auto !important;
 }
 
+[data-tooltip] {
+  position: relative;
+}
+[data-tooltip]:hover::after {
+  content: attr(data-tooltip);
+  position: absolute;
+  line-height: 1.25;
+  background: var(--bs-light);
+  color: var(--bs-body-color);
+  opacity: 1;
+  padding: 4px 8px;
+  border-radius: 4px;
+  border: 1px solid var(--bs-border-color);
+  box-shadow: 0 2px 10px 0 rgba(0, 0, 0, 0.25);
+  white-space: pre-wrap;
+  width: max-content;
+  max-width: 400px;
+  z-index: 1000;
+}
+[data-tooltip][data-tooltip-position="bottom-left"]:hover::after {
+  right: 0%;
+  top: 100%;
+}
+
 /* ANSI Coloring */
 .ansi-display {
   font-family: monospace;
@@ -725,7 +755,7 @@ pre[class*="language-"].tool-output {
   background: none !important;
   border: none !important;
   box-shadow: none !important;
-  border-radius: var(--bs-border-radius) !important;  
+  border-radius: var(--bs-border-radius) !important;
 }
 
 /* lightbox styles */

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

@@ -14282,6 +14282,12 @@ pre[class*="language-"] {
   --inspect-input-border: var(--bs-light-border-subtle);
   --inspect-diff-add-color: #dafbe1;
   --inspect-diff-remove-color: #ffebe9;
+  --inspect-inactive-selection-background: var(--vscode-editor-inactiveSelectionBackground, #d9d9d9);
+  --inspect-active-selection-background: var(--vscode-editor-selectionBackground, #d7d4f0);
+  --inspect-focus-border-color: #86b7fe;
+  --inspect-focus-border-shadow: 0 0 0 0.25rem rgba(var(--bs-primary-rgb), 0.25);
+  --inspect-focus-border-gray-color: #808080;
+  --inspect-focus-border-gray-shadow: 0 0 0 0.25rem rgba(48, 48, 48, 0.25);
 }
 
 body:not([class^="vscode-"]) button {
@@ -14923,6 +14929,30 @@ table.table.table-sm td {
   height: auto !important;
 }
 
+[data-tooltip] {
+  position: relative;
+}
+[data-tooltip]:hover::after {
+  content: attr(data-tooltip);
+  position: absolute;
+  line-height: 1.25;
+  background: var(--bs-light);
+  color: var(--bs-body-color);
+  opacity: 1;
+  padding: 4px 8px;
+  border-radius: 4px;
+  border: 1px solid var(--bs-border-color);
+  box-shadow: 0 2px 10px 0 rgba(0, 0, 0, 0.25);
+  white-space: pre-wrap;
+  width: max-content;
+  max-width: 400px;
+  z-index: 1000;
+}
+[data-tooltip][data-tooltip-position="bottom-left"]:hover::after {
+  right: 0%;
+  top: 100%;
+}
+
 /* ANSI Coloring */
 .ansi-display {
   font-family: monospace;
@@ -14998,7 +15028,7 @@ pre[class*="language-"].tool-output {
   background: none !important;
   border: none !important;
   box-shadow: none !important;
-  border-radius: var(--bs-border-radius) !important;  
+  border-radius: var(--bs-border-radius) !important;
 }
 
 /* lightbox styles */