docent-python 0.1.31a0__tar.gz → 0.1.33a0__tar.gz

{docent_python-0.1.31a0 → docent_python-0.1.33a0}/PKG-INFO

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

{docent_python-0.1.31a0 → docent_python-0.1.33a0}/docent/data_models/__init__.py

@@ -1,5 +1,6 @@
 from docent.data_models.agent_run import AgentRun
 from docent.data_models.citation import Citation
+from docent.data_models.collection import Collection
 from docent.data_models.judge import Label
 from docent.data_models.regex import RegexSnippet
 from docent.data_models.transcript import Transcript, TranscriptGroup
@@ -7,6 +8,7 @@ from docent.data_models.transcript import Transcript, TranscriptGroup
 __all__ = [
     "AgentRun",
     "Citation",
+    "Collection",
     "Label",
     "RegexSnippet",
     "Transcript",

docent_python-0.1.33a0/docent/data_models/collection.py

@@ -0,0 +1,23 @@
+from datetime import datetime
+
+from pydantic import BaseModel
+
+
+class Collection(BaseModel):
+    """Represents a collection of agent runs.
+
+    A Collection is a container for organizing and managing related agent runs.
+
+    Attributes:
+        id: Unique identifier for the collection.
+        name: Human-readable name for the collection.
+        description: Optional description of the collection's purpose.
+        created_by: User ID of the collection creator (if available).
+        created_at: Timestamp when the collection was created.
+    """
+
+    id: str
+    name: str | None = None
+    description: str | None = None
+    created_by: str | None = None
+    created_at: datetime

{docent_python-0.1.31a0 → docent_python-0.1.33a0}/docent/sdk/client.py

@@ -8,6 +8,7 @@ from tqdm import tqdm
 
 from docent._log_util.logger import get_logger
 from docent.data_models.agent_run import AgentRun
+from docent.data_models.collection import Collection
 from docent.data_models.judge import Label
 from docent.judges.util.meta_schema import validate_judge_result_schema
 from docent.loaders import load_inspect
@@ -188,11 +189,11 @@ class Docent:
         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]]:
+    def list_collections(self) -> list[Collection]:
         """Lists all available Collections.
 
         Returns:
-            list: List of dictionaries containing Collection information.
+            list: List of Collection objects.
 
         Raises:
             requests.exceptions.HTTPError: If the API request fails.
@@ -200,7 +201,28 @@ class Docent:
         url = f"{self._server_url}/collections"
         response = self._session.get(url)
         self._handle_response_errors(response)
-        return response.json()
+        return [Collection.model_validate(c) for c in response.json()]
+
+    def get_collection(self, collection_id: str) -> Collection | None:
+        """Get details about a specific Collection.
+
+        Requires READ permission on the collection.
+
+        Args:
+            collection_id: ID of the Collection to retrieve.
+
+        Returns:
+            Collection: Collection object with id, name, description, created_at, and created_by.
+                       Returns None if collection not found.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/{collection_id}/collection_details"
+        response = self._session.get(url)
+        self._handle_response_errors(response)
+        data = response.json()
+        return Collection.model_validate(data) if data is not None else None
 
     def list_rubrics(self, collection_id: str) -> list[dict[str, Any]]:
         """List all rubrics for a given collection.

{docent_python-0.1.31a0 → docent_python-0.1.33a0}/docent/trace.py

@@ -44,6 +44,7 @@ from opentelemetry.sdk.trace.export import (
 )
 from opentelemetry.trace import Span
 from requests import Response
+
 from docent._log_util import get_logger
 
 logger = get_logger(__name__)
@@ -1676,23 +1677,29 @@ def is_package_installed(package_name: str) -> bool:
     return package_name.lower() in installed_packages
 
 
-def get_tracer(caller: str = "get_tracer()") -> Optional[DocentTracer]:
+def get_tracer(
+    caller: str = "get_tracer()", log_error_if_tracer_is_none: bool = True
+) -> Optional[DocentTracer]:
     """
     Get the global Docent tracer if it has been initialized.
 
     Args:
         caller: Human-readable name of the API being invoked. Used for log output.
+        log_error_if_tracer_is_none: Whether to log an error if the tracer is None.
+            NOTE(mengk): when get_tracer is called in is_disabled, I don't want an error logged,
+            since that's what I'm trying to check. In other contexts, it makes sense.
 
     Returns:
         The global Docent tracer, or None if tracing has not been initialized.
     """
     tracer = _global_tracer
     if tracer is None:
-        logger.error(
-            f"{caller} requires initialize_tracing() to be called before use. "
-            "You can also disable tracing by calling set_disabled(True) or by setting "
-            "the DOCENT_DISABLE_TRACING environment variable to 'true'."
-        )
+        if log_error_if_tracer_is_none:
+            logger.error(
+                f"{caller} requires initialize_tracing() to be called before use. "
+                "You can also disable tracing by calling set_disabled(True) or by setting "
+                "the DOCENT_DISABLE_TRACING environment variable to 'true'."
+            )
         return None
 
     if not tracer.is_initialized():
@@ -1743,7 +1750,7 @@ def is_disabled(context_name: str = "Docent tracing") -> bool:
     """
     if _global_tracing_disabled:
         return True
-    tracer = get_tracer(context_name)
+    tracer = get_tracer(context_name, log_error_if_tracer_is_none=False)
     if tracer is None:
         return True
     return tracer.is_disabled()

{docent_python-0.1.31a0 → docent_python-0.1.33a0}/pyproject.toml

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