trismik 0.9.8__tar.gz → 0.9.9__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: trismik
3
- Version: 0.9.8
3
+ Version: 0.9.9
4
4
  Summary:
5
5
  License-File: LICENSE
6
6
  Author: Bartosz Kielczewski
@@ -15,11 +15,14 @@ Classifier: Programming Language :: Python :: 3.13
15
15
  Classifier: Programming Language :: Python :: 3.14
16
16
  Provides-Extra: examples
17
17
  Requires-Dist: accelerate (>=1.7.0,<2.0.0) ; extra == "examples"
18
+ Requires-Dist: httpx (>=0.27.2,<1.0.0)
19
+ Requires-Dist: nest-asyncio (>=1.6.0,<2.0.0)
18
20
  Requires-Dist: notebook (>=7.4.4,<8.0.0) ; extra == "examples"
19
21
  Requires-Dist: openai (>=1.81.0,<2.0.0) ; extra == "examples"
20
22
  Requires-Dist: torch (>=2.7.0,<3.0.0) ; extra == "examples"
21
23
  Requires-Dist: torchaudio (>=2.7.0,<3.0.0) ; extra == "examples"
22
24
  Requires-Dist: torchvision (>=0.22.0,<1.0.0) ; extra == "examples"
25
+ Requires-Dist: tqdm (>=4.67.1,<5.0.0)
23
26
  Requires-Dist: transformers (>=4.51.3,<5.0.0) ; extra == "examples"
24
27
  Description-Content-Type: text/markdown
25
28
 
@@ -10,7 +10,12 @@ authors = [
10
10
  ]
11
11
  readme = "README.md"
12
12
  requires-python = ">=3.9"
13
- version = "0.9.8"
13
+ dependencies = [
14
+ "httpx (>=0.27.2,<1.0.0)",
15
+ "nest-asyncio (>=1.6.0,<2.0.0)",
16
+ "tqdm (>=4.67.1, <5.0.0)",
17
+ ]
18
+ version = "0.9.9"
14
19
 
15
20
  [project.optional-dependencies]
16
21
  examples = [
@@ -30,10 +35,6 @@ packages = [{ include = "trismik", from = "src" }]
30
35
  [tool.poetry.requires-plugins]
31
36
  poetry-dynamic-versioning = { version = ">=1.0.0,<2.0.0", extras = ["plugin"] }
32
37
 
33
- [tool.poetry.group.main.dependencies]
34
- httpx = "^0.27.2"
35
- nest-asyncio = "^1.6.0"
36
- tqdm = "^4.67.1"
37
38
 
38
39
 
39
40
 
@@ -8,7 +8,6 @@ from trismik.types import (
8
8
  TrismikItem,
9
9
  TrismikMeResponse,
10
10
  TrismikMultipleChoiceTextItem,
11
- TrismikOrganization,
12
11
  TrismikProject,
13
12
  TrismikReplayResponse,
14
13
  TrismikResponse,
@@ -18,6 +17,7 @@ from trismik.types import (
18
17
  TrismikRunResponse,
19
18
  TrismikRunState,
20
19
  TrismikRunSummary,
20
+ TrismikTeam,
21
21
  TrismikTextChoice,
22
22
  TrismikUserInfo,
23
23
  )
@@ -275,7 +275,7 @@ class TrismikResponseMapper:
275
275
  TrismikMeResponse: Me response object.
276
276
  """
277
277
  user_data = json["user"]
278
- organizations_data = json["organizations"]
278
+ teams_data = json["teams"]
279
279
 
280
280
  user_info = TrismikUserInfo(
281
281
  id=user_data["id"],
@@ -283,22 +283,20 @@ class TrismikResponseMapper:
283
283
  firstname=user_data["firstname"],
284
284
  lastname=user_data["lastname"],
285
285
  createdAt=user_data.get("createdAt"),
286
+ account_id=user_data.get("accountId"),
286
287
  )
287
288
 
288
- organizations = [
289
- TrismikOrganization(
290
- id=org_data["id"],
291
- name=org_data["name"],
292
- type=org_data["type"],
293
- role=org_data["role"],
289
+ teams = [
290
+ TrismikTeam(
291
+ id=team_data["id"],
292
+ name=team_data["name"],
293
+ role=team_data["role"],
294
+ account_id=team_data["accountId"],
294
295
  )
295
- for org_data in organizations_data
296
+ for team_data in teams_data
296
297
  ]
297
298
 
298
- return TrismikMeResponse(
299
- user=user_info,
300
- organizations=organizations,
301
- )
299
+ return TrismikMeResponse(user=user_info, teams=teams)
302
300
 
303
301
  @staticmethod
304
302
  def to_classic_eval_response(
@@ -320,11 +318,12 @@ class TrismikResponseMapper:
320
318
  email=user_data["email"],
321
319
  firstname=user_data["firstname"],
322
320
  lastname=user_data["lastname"],
321
+ account_id=user_data.get("accountId"),
323
322
  )
324
323
 
325
324
  return TrismikClassicEvalResponse(
326
325
  id=json["id"],
327
- organizationId=json["organizationId"],
326
+ accountId=json["accountId"],
328
327
  projectId=json["projectId"],
329
328
  experimentId=json["experimentId"],
330
329
  experimentName=json["experimentName"],
@@ -215,7 +215,7 @@ class TrismikAsyncClient:
215
215
  TrismikApiError: If API request fails.
216
216
  """
217
217
  try:
218
- url = f"/runs/{run_id}"
218
+ url = f"/runs/adaptive/{run_id}"
219
219
  response = await self._http_client.get(url)
220
220
  response.raise_for_status()
221
221
  json = response.json()
@@ -222,13 +222,13 @@ class TrismikReplayResponse:
222
222
 
223
223
 
224
224
  @dataclass
225
- class TrismikOrganization:
226
- """Organization information."""
225
+ class TrismikTeam:
226
+ """Team information."""
227
227
 
228
228
  id: str
229
229
  name: str
230
- type: str
231
230
  role: str
231
+ account_id: str
232
232
 
233
233
 
234
234
  @dataclass
@@ -239,6 +239,7 @@ class TrismikUserInfo:
239
239
  email: str
240
240
  firstname: str
241
241
  lastname: str
242
+ account_id: str
242
243
  createdAt: Optional[str] = None
243
244
 
244
245
 
@@ -247,7 +248,7 @@ class TrismikMeResponse:
247
248
  """Response from the /admin/api-keys/me endpoint."""
248
249
 
249
250
  user: TrismikUserInfo
250
- organizations: List[TrismikOrganization]
251
+ teams: List[TrismikTeam]
251
252
 
252
253
 
253
254
  @dataclass
@@ -287,7 +288,7 @@ class TrismikClassicEvalResponse:
287
288
  """Response from a classic evaluation submission."""
288
289
 
289
290
  id: str
290
- organizationId: str
291
+ accountId: str
291
292
  projectId: str
292
293
  experimentId: str
293
294
  experimentName: str
File without changes
File without changes
File without changes
File without changes
File without changes