docent-python 0.1.0a7__tar.gz → 0.1.0a8__tar.gz

{docent_python-0.1.0a7 → docent_python-0.1.0a8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docent-python
-Version: 0.1.0a7
+Version: 0.1.0a8
 Summary: Docent SDK
 Project-URL: Homepage, https://github.com/TransluceAI/docent
 Project-URL: Issues, https://github.com/TransluceAI/docent/issues
@@ -12,3 +12,4 @@ Requires-Python: >=3.11
 Requires-Dist: pydantic>=2.11.7
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: tiktoken>=0.7.0
+Requires-Dist: tqdm>=4.67.1

{docent_python-0.1.0a7 → docent_python-0.1.0a8}/README.md

@@ -9,7 +9,7 @@ The official Python SDK for [Docent](https://github.com/TransluceAI/docent) - a
 
 Docent helps you understand AI agent behavior by providing tools to collect, analyze, and visualize agent execution data. This SDK allows you to programmatically interact with the Docent platform to:
 
-- Create and manage FrameGrids (collections of agent runs)
+- Create and manage collections of agent runs
 - Upload agent execution traces and transcripts
 - Define custom dimensions and filters
 - Perform searches and analyses on agent behavior

{docent_python-0.1.0a7 → docent_python-0.1.0a8}/docent/data_models/metadata.py

@@ -34,7 +34,7 @@ class BaseMetadata(BaseModel):
     """
 
     model_config = ConfigDict(extra="allow")
-    allow_fields_without_descriptions: bool = False
+    allow_fields_without_descriptions: bool = True
 
     # Private attribute to store field descriptions
     _field_descriptions: dict[str, str | None] | None = PrivateAttr(default=None)
@@ -189,3 +189,41 @@ class BaseAgentRunMetadata(BaseMetadata):
     scores: dict[str, int | float | bool | None] = Field(
         description="A dict of score_key -> score_value. Use one key for each metric you're tracking."
     )
+
+
+class InspectAgentRunMetadata(BaseAgentRunMetadata):
+    """Extends BaseAgentRunMetadata with fields specific to Inspect runs.
+
+    Attributes:
+        task_id: The ID of the 'benchmark' or 'set of evals' that the transcript belongs to
+        sample_id: The specific task inside of the `task_id` benchmark that the transcript was run on
+        epoch_id: Each `sample_id` should be run multiple times due to stochasticity; `epoch_id` is the integer index of a specific run.
+        model: The model that was used to generate the transcript
+        scoring_metadata: Additional metadata about the scoring process
+        additional_metadata: Additional metadata about the transcript
+    """
+
+    task_id: str = Field(
+        description="The ID of the 'benchmark' or 'set of evals' that the transcript belongs to"
+    )
+
+    # Identification of this particular run
+    sample_id: str = Field(
+        description="The specific task inside of the `task_id` benchmark that the transcript was run on"
+    )
+    epoch_id: int = Field(
+        description="Each `sample_id` should be run multiple times due to stochasticity; `epoch_id` is the integer index of a specific run."
+    )
+
+    # Parameters for the run
+    model: str = Field(description="The model that was used to generate the transcript")
+
+    # Outcome
+    scoring_metadata: dict[str, Any] | None = Field(
+        description="Additional metadata about the scoring process"
+    )
+
+    # Inspect metadata
+    additional_metadata: dict[str, Any] | None = Field(
+        description="Additional metadata about the transcript"
+    )

{docent_python-0.1.0a7 → docent_python-0.1.0a8}/docent/data_models/transcript.py

@@ -110,6 +110,8 @@ class Transcript(BaseModel):
         A unit of action represents a logical group of messages, such as a system message
         on its own or a user message followed by assistant responses and tool outputs.
 
+        For precise details on how action units are determined, refer to the _compute_units_of_action method implementation.
+
         Returns:
             list[list[int]]: List of units of action, where each unit is a list of message indices.
         """

docent_python-0.1.0a8/docent/sdk/client.py

@@ -0,0 +1,270 @@
+import os
+from typing import Any
+
+import requests
+
+from docent._log_util.logger import get_logger
+from docent.data_models.agent_run import AgentRun
+
+logger = get_logger(__name__)
+
+
+class Docent:
+    """Client for interacting with the Docent API.
+
+    This client provides methods for creating and managing Collections,
+    dimensions, agent runs, and filters in the Docent system.
+
+    Args:
+        server_url: URL of the Docent API server.
+        web_url: URL of the Docent web UI.
+        email: Email address for authentication.
+        password: Password for authentication.
+    """
+
+    def __init__(
+        self,
+        server_url: str = "https://aws-docent-backend.transluce.org",
+        web_url: str = "https://docent-alpha.transluce.org",
+        api_key: str | None = None,
+    ):
+        self._server_url = server_url.rstrip("/") + "/rest"
+        self._web_url = web_url.rstrip("/")
+
+        # Use requests.Session for connection pooling and persistent headers
+        self._session = requests.Session()
+
+        api_key = api_key or os.getenv("DOCENT_API_KEY")
+
+        if api_key is None:
+            raise ValueError(
+                "api_key is required. Please provide an "
+                "api_key or set the DOCENT_API_KEY environment variable."
+            )
+
+        self._login(api_key)
+
+    def _login(self, api_key: str):
+        """Login with email/password to establish session."""
+        self._session.headers.update({"Authorization": f"Bearer {api_key}"})
+
+        url = f"{self._server_url}/api-keys/test"
+        response = self._session.get(url)
+        response.raise_for_status()
+
+        logger.info("Logged in with API key")
+        return
+
+    def create_collection(
+        self,
+        collection_id: str | None = None,
+        name: str | None = None,
+        description: str | None = None,
+    ) -> str:
+        """Creates a new Collection.
+
+        Creates a new Collection and sets up a default MECE dimension
+        for grouping on the homepage.
+
+        Args:
+            collection_id: Optional ID for the new Collection. If not provided, one will be generated.
+            name: Optional name for the Collection.
+            description: Optional description for the Collection.
+
+        Returns:
+            str: The ID of the created Collection.
+
+        Raises:
+            ValueError: If the response is missing the Collection ID.
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/create"
+        payload = {
+            "collection_id": collection_id,
+            "name": name,
+            "description": description,
+        }
+
+        response = self._session.post(url, json=payload)
+        response.raise_for_status()
+
+        response_data = response.json()
+        collection_id = response_data.get("collection_id")
+        if collection_id is None:
+            raise ValueError("Failed to create collection: 'collection_id' missing in response.")
+
+        logger.info(f"Successfully created Collection with id='{collection_id}'")
+
+        logger.info(
+            f"Collection creation complete. Frontend available at: {self._web_url}/dashboard/{collection_id}"
+        )
+        return collection_id
+
+    def set_io_bin_keys(
+        self, collection_id: str, inner_bin_key: str | None, outer_bin_key: str | None
+    ):
+        """Set inner and outer bin keys for a collection."""
+        response = self._session.post(
+            f"{self._server_url}/{collection_id}/set_io_bin_keys",
+            json={"inner_bin_key": inner_bin_key, "outer_bin_key": outer_bin_key},
+        )
+        response.raise_for_status()
+
+    def set_inner_bin_key(self, collection_id: str, dim: str):
+        """Set the inner bin key for a collection."""
+        current_io_bin_keys = self.get_io_bin_keys(collection_id)
+        if current_io_bin_keys is None:
+            current_io_bin_keys = (None, None)
+        self.set_io_bin_keys(collection_id, dim, current_io_bin_keys[1])  # Set inner, keep outer
+
+    def set_outer_bin_key(self, collection_id: str, dim: str):
+        """Set the outer bin key for a collection."""
+        current_io_bin_keys = self.get_io_bin_keys(collection_id)
+        if current_io_bin_keys is None:
+            current_io_bin_keys = (None, None)
+        self.set_io_bin_keys(collection_id, current_io_bin_keys[0], dim)  # Keep inner, set outer
+
+    def get_io_bin_keys(self, collection_id: str) -> tuple[str | None, str | None] | None:
+        """Gets the current inner and outer bin keys for a Collection.
+
+        Args:
+            collection_id: ID of the Collection.
+
+        Returns:
+            tuple: (inner_bin_key | None, outer_bin_key | None)
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/{collection_id}/io_bin_keys"
+        response = self._session.get(url)
+        response.raise_for_status()
+        data = response.json()
+        return (data.get("inner_bin_key"), data.get("outer_bin_key"))
+
+    def add_agent_runs(self, collection_id: str, agent_runs: list[AgentRun]) -> dict[str, Any]:
+        """Adds agent runs to a Collection.
+
+        Agent runs represent execution traces that can be visualized and analyzed.
+        This method batches the insertion in groups of 1,000 for better performance.
+
+        Args:
+            collection_id: ID of the Collection.
+            agent_runs: List of AgentRun objects to add.
+
+        Returns:
+            dict: API response data.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        from tqdm import tqdm
+
+        url = f"{self._server_url}/{collection_id}/agent_runs"
+        batch_size = 1000
+        total_runs = len(agent_runs)
+
+        # Process agent runs in batches
+        with tqdm(total=total_runs, desc="Adding agent runs", unit="runs") as pbar:
+            for i in range(0, total_runs, batch_size):
+                batch = agent_runs[i : i + batch_size]
+                payload = {"agent_runs": [ar.model_dump(mode="json") for ar in batch]}
+
+                response = self._session.post(url, json=payload)
+                response.raise_for_status()
+
+                pbar.update(len(batch))
+
+        url = f"{self._server_url}/{collection_id}/compute_embeddings"
+        response = self._session.post(url)
+        response.raise_for_status()
+
+        logger.info(f"Successfully added {total_runs} agent runs to Collection '{collection_id}'")
+        return {"status": "success", "total_runs_added": total_runs}
+
+    def list_collections(self) -> list[dict[str, Any]]:
+        """Lists all available Collections.
+
+        Returns:
+            list: List of dictionaries containing Collection information.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/collections"
+        response = self._session.get(url)
+        response.raise_for_status()
+        return response.json()
+
+    def list_searches(self, collection_id: str) -> list[dict[str, Any]]:
+        """List all searches for a given collection.
+
+        Args:
+            collection_id: ID of the Collection.
+
+        Returns:
+            list: List of dictionaries containing search query information.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/{collection_id}/list_search_queries"
+        response = self._session.get(url)
+        response.raise_for_status()
+        return response.json()
+
+    def get_search_results(self, collection_id: str, search_query: str) -> list[dict[str, Any]]:
+        """Get search results for a given collection and search query.
+        Pass in either search_query or query_id.
+
+        Args:
+            collection_id: ID of the Collection.
+            search_query: The search query to get results for.
+
+        Returns:
+            list: List of dictionaries containing search result information.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/{collection_id}/get_search_results"
+        response = self._session.post(url, json={"search_query": search_query})
+        response.raise_for_status()
+        return response.json()
+
+    def list_search_clusters(self, collection_id: str, search_query: str) -> list[dict[str, Any]]:
+        """List all search clusters for a given collection.
+        Pass in either search_query or query_id.
+
+        Args:
+            collection_id: ID of the Collection.
+            search_query: The search query to get clusters for.
+
+        Returns:
+            list: List of dictionaries containing search cluster information.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/{collection_id}/list_search_clusters"
+        response = self._session.post(url, json={"search_query": search_query})
+        response.raise_for_status()
+        return response.json()
+
+    def get_cluster_matches(self, collection_id: str, centroid: str) -> list[dict[str, Any]]:
+        """Get the matches for a given cluster.
+
+        Args:
+            collection_id: ID of the Collection.
+            cluster_id: The ID of the cluster to get matches for.
+
+        Returns:
+            list: List of dictionaries containing the search results that match the cluster.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/{collection_id}/get_cluster_matches"
+        response = self._session.post(url, json={"centroid": centroid})
+        response.raise_for_status()
+        return response.json()

{docent_python-0.1.0a7 → docent_python-0.1.0a8}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "docent-python"
 description = "Docent SDK"
-version = "0.1.0-alpha.7"
+version = "0.1.0-alpha.8"
 authors = [
   { name="Transluce", email="info@transluce.org" },
 ]
@@ -13,6 +13,7 @@ dependencies = [
     "pydantic>=2.11.7",
     "pyyaml>=6.0.2",
     "tiktoken>=0.7.0",
+    "tqdm>=4.67.1",
 ]
 
 [build-system]

{docent_python-0.1.0a7 → docent_python-0.1.0a8}/uv.lock

@@ -68,14 +68,24 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/20/94/c5790835a017658cbfabd07f3bfb549140c3ac458cfc196323996b10095a/charset_normalizer-3.4.2-py3-none-any.whl", hash = "sha256:7f56930ab0abd1c45cd15be65cc741c28b1c9a34876ce8c17a2fa107810c0af0", size = 52626, upload-time = "2025-05-02T08:34:40.053Z" },
 ]
 
+[[package]]
+name = "colorama"
+version = "0.4.6"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/d8/53/6f443c9a4a8358a93a6792e2acffb9d9d5cb0a5cfd8802644b7b1c9a02e4/colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44", size = 27697, upload-time = "2022-10-25T02:36:22.414Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6", size = 25335, upload-time = "2022-10-25T02:36:20.889Z" },
+]
+
 [[package]]
 name = "docent-python"
-version = "0.1.0a1"
+version = "0.1.0a7"
 source = { editable = "." }
 dependencies = [
     { name = "pydantic" },
     { name = "pyyaml" },
     { name = "tiktoken" },
+    { name = "tqdm" },
 ]
 
 [package.metadata]
@@ -83,6 +93,7 @@ requires-dist = [
     { name = "pydantic", specifier = ">=2.11.7" },
     { name = "pyyaml", specifier = ">=6.0.2" },
     { name = "tiktoken", specifier = ">=0.7.0" },
+    { name = "tqdm", specifier = ">=4.67.1" },
 ]
 
 [[package]]
@@ -307,6 +318,18 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/de/a8/8f499c179ec900783ffe133e9aab10044481679bb9aad78436d239eee716/tiktoken-0.9.0-cp313-cp313-win_amd64.whl", hash = "sha256:5ea0edb6f83dc56d794723286215918c1cde03712cbbafa0348b33448faf5b95", size = 894669, upload-time = "2025-02-14T06:02:47.341Z" },
 ]
 
+[[package]]
+name = "tqdm"
+version = "4.67.1"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "colorama", marker = "sys_platform == 'win32'" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/a8/4b/29b4ef32e036bb34e4ab51796dd745cdba7ed47ad142a9f4a1eb8e0c744d/tqdm-4.67.1.tar.gz", hash = "sha256:f8aef9c52c08c13a65f30ea34f4e5aac3fd1a34959879d7e59e63027286627f2", size = 169737, upload-time = "2024-11-24T20:12:22.481Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/d0/30/dc54f88dd4a2b5dc8a0279bdd7270e735851848b762aeb1c1184ed1f6b14/tqdm-4.67.1-py3-none-any.whl", hash = "sha256:26445eca388f82e72884e0d580d5464cd801a3ea01e63e5601bdff9ba6a48de2", size = 78540, upload-time = "2024-11-24T20:12:19.698Z" },
+]
+
 [[package]]
 name = "typing-extensions"
 version = "4.14.0"

docent_python-0.1.0a7/docent/sdk/client.py

@@ -1,248 +0,0 @@
-import os
-from typing import Any
-
-import requests
-
-from docent._log_util.logger import get_logger
-from docent.data_models.agent_run import AgentRun
-
-logger = get_logger(__name__)
-
-
-class Docent:
-    """Client for interacting with the Docent API.
-
-    This client provides methods for creating and managing FrameGrids,
-    dimensions, agent runs, and filters in the Docent system.
-
-    Args:
-        server_url: URL of the Docent API server.
-        web_url: URL of the Docent web UI.
-        email: Email address for authentication.
-        password: Password for authentication.
-    """
-
-    def __init__(
-        self,
-        server_url: str = "https://aws-docent-backend.transluce.org",
-        web_url: str = "https://docent-alpha.transluce.org",
-        email: str | None = None,
-        password: str | None = None,
-    ):
-        self._server_url = server_url.rstrip("/") + "/rest"
-        self._web_url = web_url.rstrip("/")
-
-        self._email = email or os.getenv("DOCENT_EMAIL")
-        if self._email is None:
-            raise ValueError(
-                "Email address must be provided through keyword argument or DOCENT_EMAIL environment variable"
-            )
-
-        self._password = password or os.getenv("DOCENT_PASSWORD")
-        if self._password is None:
-            raise ValueError(
-                "Password must be provided through keyword argument or DOCENT_PASSWORD environment variable"
-            )
-
-        # Use requests.Session for connection pooling and persistent headers
-        self._session = requests.Session()
-        self._login()
-
-    def _login(self):
-        """Login with email/password to establish session."""
-        login_url = f"{self._server_url}/login"
-        response = self._session.post(
-            login_url, json={"email": self._email, "password": self._password}
-        )
-
-        if response.status_code == 401:
-            raise ValueError(
-                "Invalid username/password combination. "
-                f"If you don't already have an account, please sign up at {self._web_url}/signup"
-            )
-
-        response.raise_for_status()
-        logger.info(f"Successfully logged in as {self._email}")
-
-    def create_framegrid(
-        self,
-        fg_id: str | None = None,
-        name: str | None = None,
-        description: str | None = None,
-    ) -> str:
-        """Creates a new FrameGrid.
-
-        Creates a new FrameGrid and sets up a default MECE dimension
-        for grouping on the homepage.
-
-        Args:
-            fg_id: Optional ID for the new FrameGrid. If not provided, one will be generated.
-            name: Optional name for the FrameGrid.
-            description: Optional description for the FrameGrid.
-
-        Returns:
-            str: The ID of the created FrameGrid.
-
-        Raises:
-            ValueError: If the response is missing the FrameGrid ID.
-            requests.exceptions.HTTPError: If the API request fails.
-        """
-        url = f"{self._server_url}/create"
-        payload = {
-            "fg_id": fg_id,
-            "name": name,
-            "description": description,
-        }
-
-        response = self._session.post(url, json=payload)
-        response.raise_for_status()
-
-        response_data = response.json()
-        fg_id = response_data.get("fg_id")
-        if fg_id is None:
-            raise ValueError("Failed to create frame grid: 'fg_id' missing in response.")
-
-        logger.info(f"Successfully created FrameGrid with id='{fg_id}'")
-
-        logger.info(f"FrameGrid creation complete. Frontend available at: {self._web_url}/{fg_id}")
-        return fg_id
-
-    def set_io_bin_keys(self, fg_id: str, inner_bin_key: str | None, outer_bin_key: str | None):
-        """Set inner and outer bin keys for a frame grid."""
-        response = self._session.post(
-            f"{self._server_url}/{fg_id}/set_io_bin_keys",
-            json={"inner_bin_key": inner_bin_key, "outer_bin_key": outer_bin_key},
-        )
-        response.raise_for_status()
-
-    def set_inner_bin_key(self, fg_id: str, dim: str):
-        """Set the inner bin key for a frame grid."""
-        current_io_bin_keys = self.get_io_bin_keys(fg_id)
-        if current_io_bin_keys is None:
-            current_io_bin_keys = (None, None)
-        self.set_io_bin_keys(fg_id, dim, current_io_bin_keys[1])  # Set inner, keep outer
-
-    def set_outer_bin_key(self, fg_id: str, dim: str):
-        """Set the outer bin key for a frame grid."""
-        current_io_bin_keys = self.get_io_bin_keys(fg_id)
-        if current_io_bin_keys is None:
-            current_io_bin_keys = (None, None)
-        self.set_io_bin_keys(fg_id, current_io_bin_keys[0], dim)  # Keep inner, set outer
-
-    def get_io_bin_keys(self, fg_id: str) -> tuple[str | None, str | None] | None:
-        """Gets the current inner and outer bin keys for a FrameGrid.
-
-        Args:
-            fg_id: ID of the FrameGrid.
-
-        Returns:
-            tuple: (inner_bin_key | None, outer_bin_key | None)
-
-        Raises:
-            requests.exceptions.HTTPError: If the API request fails.
-        """
-        url = f"{self._server_url}/{fg_id}/io_bin_keys"
-        response = self._session.get(url)
-        response.raise_for_status()
-        data = response.json()
-        return (data.get("inner_bin_key"), data.get("outer_bin_key"))
-
-    def add_agent_runs(self, fg_id: str, agent_runs: list[AgentRun]) -> dict[str, Any]:
-        """Adds agent runs to a FrameGrid.
-
-        Agent runs represent execution traces that can be visualized and analyzed.
-        This method batches the insertion in groups of 5,000 for better performance.
-
-        Args:
-            fg_id: ID of the FrameGrid.
-            agent_runs: List of AgentRun objects to add.
-
-        Returns:
-            dict: API response data.
-
-        Raises:
-            requests.exceptions.HTTPError: If the API request fails.
-        """
-        from tqdm import tqdm
-
-        url = f"{self._server_url}/{fg_id}/agent_runs"
-        batch_size = 1000
-        total_runs = len(agent_runs)
-
-        # Process agent runs in batches
-        with tqdm(total=total_runs, desc="Adding agent runs", unit="runs") as pbar:
-            for i in range(0, total_runs, batch_size):
-                batch = agent_runs[i : i + batch_size]
-                payload = {"agent_runs": [ar.model_dump(mode="json") for ar in batch]}
-
-                response = self._session.post(url, json=payload)
-                response.raise_for_status()
-
-                pbar.update(len(batch))
-
-        url = f"{self._server_url}/{fg_id}/compute_embeddings"
-        response = self._session.post(url)
-        response.raise_for_status()
-
-        logger.info(f"Successfully added {total_runs} agent runs to FrameGrid '{fg_id}'")
-        return {"status": "success", "total_runs_added": total_runs}
-
-    def list_framegrids(self) -> list[dict[str, Any]]:
-        """Lists all available FrameGrids.
-
-        Returns:
-            list: List of dictionaries containing FrameGrid information.
-
-        Raises:
-            requests.exceptions.HTTPError: If the API request fails.
-        """
-        url = f"{self._server_url}/framegrids"
-        response = self._session.get(url)
-        response.raise_for_status()
-        return response.json()
-
-    def get_dimensions(self, fg_id: str, dim_ids: list[str] | None = None) -> list[dict[str, Any]]:
-        """Retrieves dimensions for a FrameGrid.
-
-        Args:
-            fg_id: ID of the FrameGrid.
-            dim_ids: Optional list of dimension IDs to retrieve. If None, retrieves all dimensions.
-
-        Returns:
-            list: List of dictionaries containing dimension information.
-
-        Raises:
-            requests.exceptions.HTTPError: If the API request fails.
-        """
-        url = f"{self._server_url}/{fg_id}/get_dimensions"
-        payload = {
-            "dim_ids": dim_ids,
-        }
-        response = self._session.post(url, json=payload)
-        response.raise_for_status()
-        return response.json()
-
-    def list_attribute_searches(
-        self, fg_id: str, base_data_only: bool = True
-    ) -> list[dict[str, Any]]:
-        """Lists available attribute searches for a FrameGrid.
-
-        Attribute searches allow finding frames with specific metadata attributes.
-
-        Args:
-            fg_id: ID of the FrameGrid.
-            base_data_only: If True, returns only basic search information.
-
-        Returns:
-            list: List of dictionaries containing attribute search information.
-
-        Raises:
-            requests.exceptions.HTTPError: If the API request fails.
-        """
-        url = f"{self._server_url}/{fg_id}/attribute_searches"
-        params = {
-            "base_data_only": base_data_only,
-        }
-        response = self._session.get(url, params=params)
-        response.raise_for_status()
-        return response.json()