trismik 0.9.6__tar.gz → 0.9.8__tar.gz

{trismik-0.9.6 → trismik-0.9.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trismik
-Version: 0.9.6
+Version: 0.9.8
 Summary: 
 License-File: LICENSE
 Author: Bartosz Kielczewski

{trismik-0.9.6 → trismik-0.9.8}/pyproject.toml

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

{trismik-0.9.6 → trismik-0.9.8}/src/trismik/_mapper.py

@@ -9,6 +9,7 @@ from trismik.types import (
     TrismikMeResponse,
     TrismikMultipleChoiceTextItem,
     TrismikOrganization,
+    TrismikProject,
     TrismikReplayResponse,
     TrismikResponse,
     TrismikResult,
@@ -336,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.6 → trismik-0.9.8}/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.6 → trismik-0.9.8}/src/trismik/client_async.py

@@ -22,6 +22,7 @@ from trismik.types import (
     TrismikClassicEvalResponse,
     TrismikDataset,
     TrismikMeResponse,
+    TrismikProject,
     TrismikReplayRequest,
     TrismikReplayResponse,
     TrismikRunMetadata,
@@ -71,7 +72,7 @@ class TrismikAsyncClient:
         default_headers = {"x-api-key": self._api_key}
 
         self._http_client = http_client or httpx.AsyncClient(
-            base_url=self._service_url, headers=default_headers
+            base_url=self._service_url, headers=default_headers, timeout=30.0
         )
 
     def _handle_http_error(self, e: httpx.HTTPStatusError) -> Exception:
@@ -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.6 → trismik-0.9.8}/src/trismik/types.py

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