docent-python 0.1.9a0__tar.gz → 0.1.10a0__tar.gz

{docent_python-0.1.9a0 → docent_python-0.1.10a0}/.gitignore

@@ -135,6 +135,7 @@ celerybeat.pid
 # Environments
 .env
 .env.*
+!.env.template
 .venv
 env/
 venv/

{docent_python-0.1.9a0 → docent_python-0.1.10a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docent-python
-Version: 0.1.9a0
+Version: 0.1.10a0
 Summary: Docent SDK
 Project-URL: Homepage, https://github.com/TransluceAI/docent
 Project-URL: Issues, https://github.com/TransluceAI/docent/issues

{docent_python-0.1.9a0 → docent_python-0.1.10a0}/docent/sdk/client.py

@@ -196,7 +196,7 @@ class Docent:
         response.raise_for_status()
         return response.json()
 
-    def list_searches(self, collection_id: str) -> list[dict[str, Any]]:
+    def list_rubrics(self, collection_id: str) -> list[dict[str, Any]]:
         """List all rubrics for a given collection.
 
         Args:
@@ -213,71 +213,73 @@ class Docent:
         response.raise_for_status()
         return response.json()
 
-    def get_search_results(
-        self, collection_id: str, rubric_id: str, rubric_version: int
-    ) -> list[dict[str, Any]]:
-        """Get rubric results for a given collection, rubric and version.
+    def get_rubric_run_state(self, collection_id: str, rubric_id: str) -> dict[str, Any]:
+        """Get rubric run state for a given collection and rubric.
 
         Args:
             collection_id: ID of the Collection.
-            rubric_id: The ID of the rubric to get results for.
-            rubric_version: The version of the rubric to get results for.
+            rubric_id: The ID of the rubric to get run state for.
 
         Returns:
-            list: List of dictionaries containing rubric result information.
+            dict: Dictionary containing rubric run state with results, job_id, and total_agent_runs.
 
         Raises:
             requests.exceptions.HTTPError: If the API request fails.
         """
-        url = f"{self._server_url}/rubric/{collection_id}/{rubric_id}/results"
-        response = self._session.get(url, params={"rubric_version": rubric_version})
+        url = f"{self._server_url}/rubric/{collection_id}/{rubric_id}/rubric_run_state"
+        response = self._session.get(url)
         response.raise_for_status()
         return response.json()
 
-    def list_search_clusters(
-        self, collection_id: str, rubric_id: str, rubric_version: int | None = None
-    ) -> list[dict[str, Any]]:
-        """List all centroids for a given collection and rubric.
+    def get_clustering_state(self, collection_id: str, rubric_id: str) -> dict[str, Any]:
+        """Get clustering state for a given collection and rubric.
 
         Args:
             collection_id: ID of the Collection.
-            rubric_id: The ID of the rubric to get centroids for.
-            rubric_version: Optional version of the rubric. If not provided, uses latest.
+            rubric_id: The ID of the rubric to get clustering state for.
 
         Returns:
-            list: List of dictionaries containing centroid information.
+            dict: Dictionary containing job_id, centroids, and assignments.
 
         Raises:
             requests.exceptions.HTTPError: If the API request fails.
         """
-        url = f"{self._server_url}/rubric/{collection_id}/{rubric_id}/centroids"
-        params: dict[str, int] = {}
-        if rubric_version is not None:
-            params["rubric_version"] = rubric_version
-        response = self._session.get(url, params=params)
+        url = f"{self._server_url}/rubric/{collection_id}/{rubric_id}/clustering_job"
+        response = self._session.get(url)
         response.raise_for_status()
         return response.json()
 
-    def get_cluster_matches(
-        self, collection_id: str, rubric_id: str, rubric_version: int
-    ) -> list[dict[str, Any]]:
+    def get_cluster_centroids(self, collection_id: str, rubric_id: str) -> list[dict[str, Any]]:
+        """Get centroids for a given collection and rubric.
+
+        Args:
+            collection_id: ID of the Collection.
+            rubric_id: The ID of the rubric to get centroids for.
+
+        Returns:
+            list: List of dictionaries containing centroid information.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        clustering_state = self.get_clustering_state(collection_id, rubric_id)
+        return clustering_state.get("centroids", [])
+
+    def get_cluster_assignments(self, collection_id: str, rubric_id: str) -> dict[str, list[str]]:
         """Get centroid assignments for a given rubric.
 
         Args:
             collection_id: ID of the Collection.
             rubric_id: The ID of the rubric to get assignments for.
-            rubric_version: The version of the rubric to get assignments for.
 
         Returns:
-            list: List of dictionaries containing centroid assignment information.
+            dict: Dictionary mapping centroid IDs to lists of judge result IDs.
 
         Raises:
             requests.exceptions.HTTPError: If the API request fails.
         """
-        url = f"{self._server_url}/rubric/{collection_id}/{rubric_id}/assignments"
-        response = self._session.get(url, params={"rubric_version": rubric_version})
-        response.raise_for_status()
-        return response.json()
+        clustering_state = self.get_clustering_state(collection_id, rubric_id)
+        return clustering_state.get("assignments", {})
 
     def get_agent_run(self, collection_id: str, agent_run_id: str) -> AgentRun | None:
         """Get a specific agent run by its ID.

{docent_python-0.1.9a0 → docent_python-0.1.10a0}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "docent-python"
 description = "Docent SDK"
-version = "0.1.9-alpha"
+version = "0.1.10-alpha"
 authors = [
   { name="Transluce", email="info@transluce.org" },
 ]