docent-python 0.1.0a9__tar.gz → 0.1.2a0__tar.gz

{docent_python-0.1.0a9 → docent_python-0.1.2a0}/.gitignore

@@ -133,6 +133,7 @@ celerybeat.pid
 
 # Environments
 .env
+.env.*
 .venv
 env/
 venv/

docent_python-0.1.2a0/PKG-INFO

@@ -0,0 +1,24 @@
+Metadata-Version: 2.4
+Name: docent-python
+Version: 0.1.2a0
+Summary: Docent SDK
+Project-URL: Homepage, https://github.com/TransluceAI/docent
+Project-URL: Issues, https://github.com/TransluceAI/docent/issues
+Project-URL: Docs, https://transluce-docent.readthedocs-hosted.com/en/latest
+Author-email: Transluce <info@transluce.org>
+License-Expression: MIT
+License-File: LICENSE.md
+Requires-Python: >=3.11
+Requires-Dist: opentelemetry-api>=1.35.0
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.35.0
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.35.0
+Requires-Dist: opentelemetry-instrumentation-anthropic>=0.42.0
+Requires-Dist: opentelemetry-instrumentation-bedrock>=0.42.0
+Requires-Dist: opentelemetry-instrumentation-langchain>=0.43.1
+Requires-Dist: opentelemetry-instrumentation-openai>=0.42.0
+Requires-Dist: opentelemetry-instrumentation-threading>=0.56b0
+Requires-Dist: opentelemetry-sdk>=1.35.0
+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.0a9 → docent_python-0.1.2a0}/docent/data_models/transcript.py

@@ -151,7 +151,9 @@ class Transcript(BaseModel):
 
             # System messages are their own unit
             if role == "system":
-                assert not current_unit, "System message should be the first message"
+                # Start a new unit if there's a current unit in progress
+                if current_unit:
+                    _start_new_unit()
                 units.append([i])
 
             # User message always starts a new unit UNLESS the previous message was a user message

{docent_python-0.1.0a9 → docent_python-0.1.2a0}/docent/sdk/client.py

@@ -24,8 +24,8 @@ class Docent:
 
     def __init__(
         self,
-        server_url: str = "https://aws-docent-backend.transluce.org",
-        web_url: str = "https://docent-alpha.transluce.org",
+        server_url: str = "https://api.docent.transluce.org",
+        web_url: str = "https://docent.transluce.org",
         api_key: str | None = None,
     ):
         self._server_url = server_url.rstrip("/") + "/rest"
@@ -291,3 +291,50 @@ class Docent:
             # We do this to avoid metadata validation failing
             # TODO(mengk): kinda hacky
             return AgentRunWithoutMetadataValidator.model_validate(response.json())
+
+    def make_collection_public(self, collection_id: str) -> dict[str, Any]:
+        """Make a collection publicly accessible to anyone with the link.
+
+        Args:
+            collection_id: ID of the Collection to make public.
+
+        Returns:
+            dict: API response data.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/{collection_id}/make_public"
+        response = self._session.post(url)
+        response.raise_for_status()
+
+        logger.info(f"Successfully made Collection '{collection_id}' public")
+        return response.json()
+
+    def share_collection_with_email(self, collection_id: str, email: str) -> dict[str, Any]:
+        """Share a collection with a specific user by email address.
+
+        Args:
+            collection_id: ID of the Collection to share.
+            email: Email address of the user to share with.
+
+        Returns:
+            dict: API response data.
+
+        Raises:
+            requests.exceptions.HTTPError: If the API request fails.
+        """
+        url = f"{self._server_url}/{collection_id}/share_with_email"
+        payload = {"email": email}
+        response = self._session.post(url, json=payload)
+
+        try:
+            response.raise_for_status()
+        except requests.exceptions.HTTPError:
+            if response.status_code == 404:
+                raise ValueError(f"The user you are trying to share with ({email}) does not exist.")
+            else:
+                raise  # Re-raise the original exception
+
+        logger.info(f"Successfully shared Collection '{collection_id}' with {email}")
+        return response.json()