docent-python 0.1.8a0__py3-none-any.whl → 0.1.10a0__py3-none-any.whl

docent/data_models/chat/__init__.py

@@ -7,7 +7,7 @@ from docent.data_models.chat.message import (
     UserMessage,
     parse_chat_message,
 )
-from docent.data_models.chat.tool import ToolCall, ToolInfo, ToolParams
+from docent.data_models.chat.tool import ToolCall, ToolCallContent, ToolInfo, ToolParams
 
 __all__ = [
     "ChatMessage",
@@ -19,6 +19,7 @@ __all__ = [
     "ContentReasoning",
     "ContentText",
     "ToolCall",
+    "ToolCallContent",
     "ToolInfo",
     "ToolParams",
     "parse_chat_message",

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/trace.py

@@ -200,7 +200,6 @@ class DocentTracer:
     def _signal_handler(self, signum: int, frame: Optional[object]):
         """Handle shutdown signals."""
         self.cleanup()
-        sys.exit(0)
 
     def _init_spans_exporter(self, endpoint: str) -> Optional[Union[HTTPExporter, GRPCExporter]]:
         """Initialize the appropriate span exporter based on endpoint."""
@@ -463,6 +462,10 @@ class DocentTracer:
         except Exception as e:
             logger.error(f"Error during flush: {e}")
 
+    def is_disabled(self) -> bool:
+        """Check if tracing is disabled."""
+        return self._disabled
+
     def set_disabled(self, disabled: bool) -> None:
         """Enable or disable tracing."""
         self._disabled = disabled
@@ -616,9 +619,10 @@ class DocentTracer:
         Returns:
             Dictionary of headers including Authorization
         """
+
         return {
             "Content-Type": "application/json",
-            "Authorization": f"Bearer {self.headers.get('Authorization', '').replace('Bearer ', '')}",
+            "Authorization": self.headers.get("Authorization", ""),
         }
 
     def _post_json(self, path: str, data: Dict[str, Any]) -> None:
@@ -1163,6 +1167,13 @@ def verify_initialized() -> bool:
     return _global_tracer.verify_initialized()
 
 
+def is_disabled() -> bool:
+    """Check if global tracing is disabled."""
+    if _global_tracer:
+        return _global_tracer.is_disabled()
+    return True
+
+
 def set_disabled(disabled: bool) -> None:
     """Enable or disable global tracing."""
     if _global_tracer:
@@ -1180,6 +1191,8 @@ def agent_run_score(name: str, score: float, attributes: Optional[Dict[str, Any]
     """
     try:
         tracer: DocentTracer = get_tracer()
+        if tracer.is_disabled():
+            return
         agent_run_id = tracer.get_current_agent_run_id()
 
         if not agent_run_id:
@@ -1216,6 +1229,8 @@ def agent_run_metadata(metadata: Dict[str, Any]) -> None:
     """
     try:
         tracer = get_tracer()
+        if tracer.is_disabled():
+            return
         agent_run_id = tracer.get_current_agent_run_id()
         if not agent_run_id:
             logger.warning("No active agent run context. Metadata will not be sent.")
@@ -1248,6 +1263,8 @@ def transcript_metadata(
     """
     try:
         tracer = get_tracer()
+        if tracer.is_disabled():
+            return
         transcript_id = tracer.get_current_transcript_id()
         if not transcript_id:
             logger.warning("No active transcript context. Metadata will not be sent.")

{docent_python-0.1.8a0.dist-info → docent_python-0.1.10a0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docent-python
-Version: 0.1.8a0
+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.8a0.dist-info → docent_python-0.1.10a0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 docent/__init__.py,sha256=J2BbO6rzilfw9WXRUeolr439EGFezqbMU_kCpCCryRA,59
 docent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-docent/trace.py,sha256=jEhmjMr_OX8_KKwMW3FLsu_SLovlxJJdRAIk_3R7vx0,66142
+docent/trace.py,sha256=-3QbmAjZJfKZp9pmhLne0EIB-akbz2XbEDNqpsROEco,66513
 docent/trace_temp.py,sha256=Z0lAPwVzXjFvxpiU-CuvfWIslq9Q4alNkZMoQ77Xudk,40711
 docent/_log_util/__init__.py,sha256=3HXXrxrSm8PxwG4llotrCnSnp7GuroK1FNHsdg6f7aE,73
 docent/_log_util/logger.py,sha256=kwM0yRW1IJd6-XTorjWn48B4l8qvD2ZM6VDjY5eskQI,4422
@@ -12,7 +12,7 @@ docent/data_models/metadata.py,sha256=r0SYC4i2x096dXMLfw_rAMtcJQCsoV6EOMPZuEngbG
 docent/data_models/regex.py,sha256=0ciIerkrNwb91bY5mTcyO5nDWH67xx2tZYObV52fmBo,1684
 docent/data_models/shared_types.py,sha256=jjm-Dh5S6v7UKInW7SEqoziOsx6Z7Uu4e3VzgCbTWvc,225
 docent/data_models/transcript.py,sha256=0iF2ujcWhTss8WkkpNMeIKJyKOfMEsiMoAQMGwY4ing,15753
-docent/data_models/chat/__init__.py,sha256=O04XQ2NmO8GTWqkkB_Iydj8j_CucZuLhoyMVTxJN_cs,570
+docent/data_models/chat/__init__.py,sha256=GleyRzYqKRkwwSRm_tQJw5BudCbgu9WRSa71Fntz0L0,610
 docent/data_models/chat/content.py,sha256=Co-jO8frQa_DSP11wJuhPX0s-GpJk8yqtKqPeiAIZ_U,1672
 docent/data_models/chat/message.py,sha256=iAo38kbV6wYbFh8S23cxLy6HY4C_i3PzQ6RpSQG5dxM,3861
 docent/data_models/chat/tool.py,sha256=x7NKINswPe0Kqvcx4ubjHzB-n0-i4DbFodvaBb2vitk,3042
@@ -22,8 +22,8 @@ docent/samples/load.py,sha256=ZGE07r83GBNO4A0QBh5aQ18WAu3mTWA1vxUoHd90nrM,207
 docent/samples/log.eval,sha256=orrW__9WBfANq7NwKsPSq9oTsQRcG6KohG5tMr_X_XY,397708
 docent/samples/tb_airline.json,sha256=eR2jFFRtOw06xqbEglh6-dPewjifOk-cuxJq67Dtu5I,47028
 docent/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-docent/sdk/client.py,sha256=fLdniy8JzMLoZpaS9SP2pHban_ToavgtI8VeHZLMNZo,12773
-docent_python-0.1.8a0.dist-info/METADATA,sha256=q0ck3nwEmLlEt1qnRDdljMk5X28qps9wvWo55DA878Q,1037
-docent_python-0.1.8a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-docent_python-0.1.8a0.dist-info/licenses/LICENSE.md,sha256=vOHzq3K4Ndu0UV9hPrtXvlD7pHOjyDQmGjHuLSIkRQY,1087
-docent_python-0.1.8a0.dist-info/RECORD,,
+docent/sdk/client.py,sha256=D4_4nFS0Ih3ZPfDDmOG-TQ_VUnl1EnsMAPJgYirke5A,12768
+docent_python-0.1.10a0.dist-info/METADATA,sha256=AxER6mbCg-7cusC20tBqw3d15eTClWJZaPcS8U74iSw,1038
+docent_python-0.1.10a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+docent_python-0.1.10a0.dist-info/licenses/LICENSE.md,sha256=vOHzq3K4Ndu0UV9hPrtXvlD7pHOjyDQmGjHuLSIkRQY,1087
+docent_python-0.1.10a0.dist-info/RECORD,,