trismik 0.9.4__tar.gz → 0.9.5__tar.gz

{trismik-0.9.4 → trismik-0.9.5}/PKG-INFO

@@ -1,7 +1,8 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: trismik
-Version: 0.9.4
+Version: 0.9.5
 Summary: 
+License-File: LICENSE
 Author: Bartosz Kielczewski
 Author-email: bk352@cam.ac.uk
 Requires-Python: >=3.9
@@ -11,6 +12,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Provides-Extra: examples
 Requires-Dist: accelerate (>=1.7.0,<2.0.0) ; extra == "examples"
 Requires-Dist: notebook (>=7.4.4,<8.0.0) ; extra == "examples"

{trismik-0.9.4 → trismik-0.9.5}/pyproject.toml

@@ -10,7 +10,7 @@ authors = [
 ]
 readme = "README.md"
 requires-python = ">=3.9"
-version = "0.9.4"
+version = "0.9.5"
 
 [project.optional-dependencies]
 examples = [

{trismik-0.9.4 → trismik-0.9.5}/src/trismik/_mapper.py

@@ -9,6 +9,7 @@ from trismik.types import (
     TrismikMeResponse,
     TrismikMultipleChoiceTextItem,
     TrismikOrganization,
+    TrismikProject,
     TrismikReplayResponse,
     TrismikResponse,
     TrismikResult,
@@ -274,7 +275,7 @@ class TrismikResponseMapper:
             TrismikMeResponse: Me response object.
         """
         user_data = json["user"]
-        organization_data = json["organization"]
+        organizations_data = json["organizations"]
 
         user_info = TrismikUserInfo(
             id=user_data["id"],
@@ -284,16 +285,19 @@ class TrismikResponseMapper:
             createdAt=user_data.get("createdAt"),
         )
 
-        organization = TrismikOrganization(
-            id=organization_data["id"],
-            name=organization_data["name"],
-            type=organization_data["type"],
-            role=organization_data["role"],
-        )
+        organizations = [
+            TrismikOrganization(
+                id=org_data["id"],
+                name=org_data["name"],
+                type=org_data["type"],
+                role=org_data["role"],
+            )
+            for org_data in organizations_data
+        ]
 
         return TrismikMeResponse(
             user=user_info,
-            organization=organization,
+            organizations=organizations,
         )
 
     @staticmethod
@@ -333,3 +337,23 @@ class TrismikResponseMapper:
             user=user_info,
             responseCount=json["responseCount"],
         )
+
+    @staticmethod
+    def to_project(json: Dict[str, Any]) -> TrismikProject:
+        """
+        Convert JSON response to a TrismikProject object.
+
+        Args:
+            json (Dict[str, Any]): JSON response from project creation endpoint.
+
+        Returns:
+            TrismikProject: Project object.
+        """
+        return TrismikProject(
+            id=json["id"],
+            name=json["name"],
+            description=json.get("description"),
+            organizationId=json["organizationId"],
+            createdAt=json["createdAt"],
+            updatedAt=json["updatedAt"],
+        )

{trismik-0.9.4 → trismik-0.9.5}/src/trismik/adaptive_test.py

@@ -22,6 +22,7 @@ from trismik.types import (
     TrismikDataset,
     TrismikItem,
     TrismikMeResponse,
+    TrismikProject,
     TrismikReplayRequest,
     TrismikReplayRequestItem,
     TrismikRunMetadata,
@@ -140,6 +141,59 @@ class AdaptiveTest:
         """
         return await self._client.me()
 
+    def create_project(
+        self,
+        name: str,
+        organization_id: str,
+        description: Optional[str] = None,
+    ) -> TrismikProject:
+        """
+        Create a new project synchronously.
+
+        Args:
+            name (str): Name of the project.
+            organization_id (str): ID of the organization to create the
+                project in.
+            description (Optional[str]): Optional description of the project.
+
+        Returns:
+            TrismikProject: Created project information.
+
+        Raises:
+            TrismikValidationError: If the request fails validation.
+            TrismikApiError: If API request fails.
+        """
+        loop = self._get_loop()
+        return loop.run_until_complete(
+            self.create_project_async(name, organization_id, description)
+        )
+
+    async def create_project_async(
+        self,
+        name: str,
+        organization_id: str,
+        description: Optional[str] = None,
+    ) -> TrismikProject:
+        """
+        Create a new project asynchronously.
+
+        Args:
+            name (str): Name of the project.
+            organization_id (str): ID of the organization to create the
+                project in.
+            description (Optional[str]): Optional description of the project.
+
+        Returns:
+            TrismikProject: Created project information.
+
+        Raises:
+            TrismikValidationError: If the request fails validation.
+            TrismikApiError: If API request fails.
+        """
+        return await self._client.create_project(
+            name, organization_id, description
+        )
+
     @overload
     def run(  # noqa: E704
         self,

{trismik-0.9.4 → trismik-0.9.5}/src/trismik/client_async.py

@@ -22,6 +22,7 @@ from trismik.types import (
     TrismikClassicEvalResponse,
     TrismikDataset,
     TrismikMeResponse,
+    TrismikProject,
     TrismikReplayRequest,
     TrismikReplayResponse,
     TrismikRunMetadata,
@@ -360,3 +361,44 @@ class TrismikAsyncClient:
             raise self._handle_http_error(e) from e
         except httpx.HTTPError as e:
             raise TrismikApiError(str(e)) from e
+
+    async def create_project(
+        self,
+        name: str,
+        organization_id: str,
+        description: Optional[str] = None,
+    ) -> TrismikProject:
+        """
+        Create a new project.
+
+        Args:
+            name (str): Name of the project.
+            organization_id (str): ID of the organization to create the
+                project in.
+            description (Optional[str]): Optional description of the project.
+
+        Returns:
+            TrismikProject: Created project information.
+
+        Raises:
+            TrismikValidationError: If the request fails validation.
+            TrismikApiError: If API request fails.
+        """
+        try:
+            url = "../admin/public/projects"
+
+            body = {"name": name}
+            if description is not None:
+                body["description"] = description
+
+            headers = {"x-organization-id": organization_id}
+            response = await self._http_client.post(
+                url, json=body, headers=headers
+            )
+            response.raise_for_status()
+            json = response.json()
+            return TrismikResponseMapper.to_project(json)
+        except httpx.HTTPStatusError as e:
+            raise self._handle_http_error(e) from e
+        except httpx.HTTPError as e:
+            raise TrismikApiError(str(e)) from e

{trismik-0.9.4 → trismik-0.9.5}/src/trismik/types.py

@@ -247,7 +247,7 @@ class TrismikMeResponse:
     """Response from the /admin/api-keys/me endpoint."""
 
     user: TrismikUserInfo
-    organization: TrismikOrganization
+    organizations: List[TrismikOrganization]
 
 
 @dataclass
@@ -299,3 +299,23 @@ class TrismikClassicEvalResponse:
     createdAt: str
     user: TrismikUserInfo
     responseCount: int
+
+
+@dataclass
+class TrismikProjectRequest:
+    """Request to create a new project."""
+
+    name: str
+    description: Optional[str] = None
+
+
+@dataclass
+class TrismikProject:
+    """Project information."""
+
+    id: str
+    name: str
+    description: Optional[str]
+    organizationId: str
+    createdAt: str
+    updatedAt: str