trismik 0.9.4__py3-none-any.whl → 0.9.5__py3-none-any.whl

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/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/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/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

{trismik-0.9.4.dist-info → trismik-0.9.5.dist-info}/METADATA

@@ -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.5.dist-info/RECORD

@@ -0,0 +1,12 @@
+trismik/__init__.py,sha256=20SwXrda9YsgykaoPohwz6foj2FkraniPA-GTQS9m00,197
+trismik/_mapper.py,sha256=9LtEtekXas6tLP_BETl5geDeLKmTGrjdGwkU3bkgmHk,10995
+trismik/_utils.py,sha256=WZ7x0EaG7PdXdFZMIs1wzgwqiPQm59QDDmQzBKYGCEg,3753
+trismik/adaptive_test.py,sha256=Fmc6PJZg_SKQjV_xQF29ZJQwPxpGwfKuvi2VznAmOB0,21646
+trismik/client_async.py,sha256=Z0tmePA86nO3Ul_55eZdT_52LSgRESANYqqrX4hH9Us,13883
+trismik/exceptions.py,sha256=2wb4_K7GdDf00s3xUaiSfw6718ZV3Eaa4M2lYbiEZl4,1945
+trismik/settings.py,sha256=FCP-d8ZEiYWUTWoa9nOVzSwTOLvg8y0pU08dAseiOwY,412
+trismik/types.py,sha256=DQ88f_PcJmvmSAQPFDxEmtJSrQQDTSvhxbdrzzRHd-4,6571
+trismik-0.9.5.dist-info/METADATA,sha256=rMF2N0d9MaoA2uIA9CL9LuoURTXs2ZCPSb4ZNZ2yZHc,6692
+trismik-0.9.5.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
+trismik-0.9.5.dist-info/licenses/LICENSE,sha256=tgetRhapGLh7ZxfknW6Mm-WobfziPd64nAK52X5XKaw,1077
+trismik-0.9.5.dist-info/RECORD,,

{trismik-0.9.4.dist-info → trismik-0.9.5.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.2.0
 Root-Is-Purelib: true
 Tag: py3-none-any

trismik-0.9.4.dist-info/RECORD

@@ -1,12 +0,0 @@
-trismik/__init__.py,sha256=20SwXrda9YsgykaoPohwz6foj2FkraniPA-GTQS9m00,197
-trismik/_mapper.py,sha256=A1mik1jlF7hm9oyq5Of-BZtA16MVhVJv1AI_20PKv2k,10315
-trismik/_utils.py,sha256=WZ7x0EaG7PdXdFZMIs1wzgwqiPQm59QDDmQzBKYGCEg,3753
-trismik/adaptive_test.py,sha256=hLGdf0jgLmJJ55I9c5QjaEd1bCA8QNfhJJoOBkF38ik,20002
-trismik/client_async.py,sha256=ZOvAH-nTnh_S3mrpLuprxtByZBasTnNCysBwUPwXMb0,12523
-trismik/exceptions.py,sha256=2wb4_K7GdDf00s3xUaiSfw6718ZV3Eaa4M2lYbiEZl4,1945
-trismik/settings.py,sha256=FCP-d8ZEiYWUTWoa9nOVzSwTOLvg8y0pU08dAseiOwY,412
-trismik/types.py,sha256=OS0npQV6FgN5DRE1Ebd4tY1Z8_R_K6SSzRdHvKdosug,6240
-trismik-0.9.4.dist-info/LICENSE,sha256=tgetRhapGLh7ZxfknW6Mm-WobfziPd64nAK52X5XKaw,1077
-trismik-0.9.4.dist-info/METADATA,sha256=2N19pz_qSvCRolCF_2Cw7TaScnXRopj7gy87KlXiyoQ,6619
-trismik-0.9.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-trismik-0.9.4.dist-info/RECORD,,