trismik 0.9.6__py3-none-any.whl → 0.9.9__py3-none-any.whl

trismik/_mapper.py

@@ -8,7 +8,7 @@ from trismik.types import (
     TrismikItem,
     TrismikMeResponse,
     TrismikMultipleChoiceTextItem,
-    TrismikOrganization,
+    TrismikProject,
     TrismikReplayResponse,
     TrismikResponse,
     TrismikResult,
@@ -17,6 +17,7 @@ from trismik.types import (
     TrismikRunResponse,
     TrismikRunState,
     TrismikRunSummary,
+    TrismikTeam,
     TrismikTextChoice,
     TrismikUserInfo,
 )
@@ -274,7 +275,7 @@ class TrismikResponseMapper:
             TrismikMeResponse: Me response object.
         """
         user_data = json["user"]
-        organizations_data = json["organizations"]
+        teams_data = json["teams"]
 
         user_info = TrismikUserInfo(
             id=user_data["id"],
@@ -282,22 +283,20 @@ class TrismikResponseMapper:
             firstname=user_data["firstname"],
             lastname=user_data["lastname"],
             createdAt=user_data.get("createdAt"),
+            account_id=user_data.get("accountId"),
         )
 
-        organizations = [
-            TrismikOrganization(
-                id=org_data["id"],
-                name=org_data["name"],
-                type=org_data["type"],
-                role=org_data["role"],
+        teams = [
+            TrismikTeam(
+                id=team_data["id"],
+                name=team_data["name"],
+                role=team_data["role"],
+                account_id=team_data["accountId"],
             )
-            for org_data in organizations_data
+            for team_data in teams_data
         ]
 
-        return TrismikMeResponse(
-            user=user_info,
-            organizations=organizations,
-        )
+        return TrismikMeResponse(user=user_info, teams=teams)
 
     @staticmethod
     def to_classic_eval_response(
@@ -319,11 +318,12 @@ class TrismikResponseMapper:
             email=user_data["email"],
             firstname=user_data["firstname"],
             lastname=user_data["lastname"],
+            account_id=user_data.get("accountId"),
         )
 
         return TrismikClassicEvalResponse(
             id=json["id"],
-            organizationId=json["organizationId"],
+            accountId=json["accountId"],
             projectId=json["projectId"],
             experimentId=json["experimentId"],
             experimentName=json["experimentName"],
@@ -336,3 +336,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,
@@ -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:
@@ -214,7 +215,7 @@ class TrismikAsyncClient:
             TrismikApiError: If API request fails.
         """
         try:
-            url = f"/runs/{run_id}"
+            url = f"/runs/adaptive/{run_id}"
             response = await self._http_client.get(url)
             response.raise_for_status()
             json = response.json()
@@ -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

@@ -222,13 +222,13 @@ class TrismikReplayResponse:
 
 
 @dataclass
-class TrismikOrganization:
-    """Organization information."""
+class TrismikTeam:
+    """Team information."""
 
     id: str
     name: str
-    type: str
     role: str
+    account_id: str
 
 
 @dataclass
@@ -239,6 +239,7 @@ class TrismikUserInfo:
     email: str
     firstname: str
     lastname: str
+    account_id: str
     createdAt: Optional[str] = None
 
 
@@ -247,7 +248,7 @@ class TrismikMeResponse:
     """Response from the /admin/api-keys/me endpoint."""
 
     user: TrismikUserInfo
-    organizations: List[TrismikOrganization]
+    teams: List[TrismikTeam]
 
 
 @dataclass
@@ -287,7 +288,7 @@ class TrismikClassicEvalResponse:
     """Response from a classic evaluation submission."""
 
     id: str
-    organizationId: str
+    accountId: str
     projectId: str
     experimentId: str
     experimentName: str
@@ -299,3 +300,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.6.dist-info → trismik-0.9.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trismik
-Version: 0.9.6
+Version: 0.9.9
 Summary: 
 License-File: LICENSE
 Author: Bartosz Kielczewski
@@ -15,11 +15,14 @@ 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: httpx (>=0.27.2,<1.0.0)
+Requires-Dist: nest-asyncio (>=1.6.0,<2.0.0)
 Requires-Dist: notebook (>=7.4.4,<8.0.0) ; extra == "examples"
 Requires-Dist: openai (>=1.81.0,<2.0.0) ; extra == "examples"
 Requires-Dist: torch (>=2.7.0,<3.0.0) ; extra == "examples"
 Requires-Dist: torchaudio (>=2.7.0,<3.0.0) ; extra == "examples"
 Requires-Dist: torchvision (>=0.22.0,<1.0.0) ; extra == "examples"
+Requires-Dist: tqdm (>=4.67.1,<5.0.0)
 Requires-Dist: transformers (>=4.51.3,<5.0.0) ; extra == "examples"
 Description-Content-Type: text/markdown
 

trismik-0.9.9.dist-info/RECORD

@@ -0,0 +1,12 @@
+trismik/__init__.py,sha256=20SwXrda9YsgykaoPohwz6foj2FkraniPA-GTQS9m00,197
+trismik/_mapper.py,sha256=wHrwmoTx8Z5B1BQH2_NUPaX3xCpBLKkidp245I4feno,11004
+trismik/_utils.py,sha256=WZ7x0EaG7PdXdFZMIs1wzgwqiPQm59QDDmQzBKYGCEg,3753
+trismik/adaptive_test.py,sha256=Fmc6PJZg_SKQjV_xQF29ZJQwPxpGwfKuvi2VznAmOB0,21646
+trismik/client_async.py,sha256=7NYcYomIfETjL5BLGQnJWhL249GBS67otGLH9Rue0sI,13906
+trismik/exceptions.py,sha256=2wb4_K7GdDf00s3xUaiSfw6718ZV3Eaa4M2lYbiEZl4,1945
+trismik/settings.py,sha256=FCP-d8ZEiYWUTWoa9nOVzSwTOLvg8y0pU08dAseiOwY,412
+trismik/types.py,sha256=-HHnOTzMeXfryKJCuZ7LHAPyS0G6uzsOJFUtIU0VB3k,6560
+trismik-0.9.9.dist-info/METADATA,sha256=gZtOCSIn4VaEIsKo-MugwvEjatgpAYRkwl3gWf8Y2Ic,6814
+trismik-0.9.9.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+trismik-0.9.9.dist-info/licenses/LICENSE,sha256=tgetRhapGLh7ZxfknW6Mm-WobfziPd64nAK52X5XKaw,1077
+trismik-0.9.9.dist-info/RECORD,,

{trismik-0.9.6.dist-info → trismik-0.9.9.dist-info}/WHEEL

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

trismik-0.9.6.dist-info/RECORD

@@ -1,12 +0,0 @@
-trismik/__init__.py,sha256=20SwXrda9YsgykaoPohwz6foj2FkraniPA-GTQS9m00,197
-trismik/_mapper.py,sha256=8iQ-cvKG5IkKoAHhmb9PSjB4Xwim5fEU2btgaw8aPsw,10375
-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=CNjvMQGnRM-ghjiZ-OWi6ZBI8RiyphuT_ABt0lhrtIA,6247
-trismik-0.9.6.dist-info/METADATA,sha256=LIfHvXxhQ7RftF7pMf7WuTW9sc6X9cC7A2i6AMbvZGc,6692
-trismik-0.9.6.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
-trismik-0.9.6.dist-info/licenses/LICENSE,sha256=tgetRhapGLh7ZxfknW6Mm-WobfziPd64nAK52X5XKaw,1077
-trismik-0.9.6.dist-info/RECORD,,