arize 8.0.0a14__py3-none-any.whl → 8.0.0a16__py3-none-any.whl

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.
Files changed (65) hide show
  1. arize/__init__.py +70 -1
  2. arize/_flight/client.py +163 -43
  3. arize/_flight/types.py +1 -0
  4. arize/_generated/api_client/__init__.py +5 -1
  5. arize/_generated/api_client/api/datasets_api.py +6 -6
  6. arize/_generated/api_client/api/experiments_api.py +924 -61
  7. arize/_generated/api_client/api_client.py +1 -1
  8. arize/_generated/api_client/configuration.py +1 -1
  9. arize/_generated/api_client/exceptions.py +1 -1
  10. arize/_generated/api_client/models/__init__.py +3 -1
  11. arize/_generated/api_client/models/dataset.py +2 -2
  12. arize/_generated/api_client/models/dataset_version.py +1 -1
  13. arize/_generated/api_client/models/datasets_create_request.py +3 -3
  14. arize/_generated/api_client/models/datasets_list200_response.py +1 -1
  15. arize/_generated/api_client/models/datasets_list_examples200_response.py +1 -1
  16. arize/_generated/api_client/models/error.py +1 -1
  17. arize/_generated/api_client/models/experiment.py +6 -6
  18. arize/_generated/api_client/models/experiments_create_request.py +98 -0
  19. arize/_generated/api_client/models/experiments_list200_response.py +1 -1
  20. arize/_generated/api_client/models/experiments_runs_list200_response.py +92 -0
  21. arize/_generated/api_client/rest.py +1 -1
  22. arize/_generated/api_client/test/test_dataset.py +2 -1
  23. arize/_generated/api_client/test/test_dataset_version.py +1 -1
  24. arize/_generated/api_client/test/test_datasets_api.py +1 -1
  25. arize/_generated/api_client/test/test_datasets_create_request.py +2 -1
  26. arize/_generated/api_client/test/test_datasets_list200_response.py +1 -1
  27. arize/_generated/api_client/test/test_datasets_list_examples200_response.py +1 -1
  28. arize/_generated/api_client/test/test_error.py +1 -1
  29. arize/_generated/api_client/test/test_experiment.py +6 -1
  30. arize/_generated/api_client/test/test_experiments_api.py +23 -2
  31. arize/_generated/api_client/test/test_experiments_create_request.py +61 -0
  32. arize/_generated/api_client/test/test_experiments_list200_response.py +1 -1
  33. arize/_generated/api_client/test/test_experiments_runs_list200_response.py +56 -0
  34. arize/_generated/api_client_README.md +13 -8
  35. arize/client.py +19 -2
  36. arize/config.py +50 -3
  37. arize/constants/config.py +8 -2
  38. arize/constants/openinference.py +14 -0
  39. arize/constants/pyarrow.py +1 -0
  40. arize/datasets/__init__.py +0 -70
  41. arize/datasets/client.py +106 -19
  42. arize/datasets/errors.py +61 -0
  43. arize/datasets/validation.py +46 -0
  44. arize/experiments/client.py +455 -0
  45. arize/experiments/evaluators/__init__.py +0 -0
  46. arize/experiments/evaluators/base.py +255 -0
  47. arize/experiments/evaluators/exceptions.py +10 -0
  48. arize/experiments/evaluators/executors.py +502 -0
  49. arize/experiments/evaluators/rate_limiters.py +277 -0
  50. arize/experiments/evaluators/types.py +122 -0
  51. arize/experiments/evaluators/utils.py +198 -0
  52. arize/experiments/functions.py +920 -0
  53. arize/experiments/tracing.py +276 -0
  54. arize/experiments/types.py +394 -0
  55. arize/models/client.py +4 -1
  56. arize/spans/client.py +16 -20
  57. arize/utils/arrow.py +4 -3
  58. arize/utils/openinference_conversion.py +56 -0
  59. arize/utils/proto.py +13 -0
  60. arize/utils/size.py +22 -0
  61. arize/version.py +1 -1
  62. {arize-8.0.0a14.dist-info → arize-8.0.0a16.dist-info}/METADATA +3 -1
  63. {arize-8.0.0a14.dist-info → arize-8.0.0a16.dist-info}/RECORD +65 -44
  64. {arize-8.0.0a14.dist-info → arize-8.0.0a16.dist-info}/WHEEL +0 -0
  65. {arize-8.0.0a14.dist-info → arize-8.0.0a16.dist-info}/licenses/LICENSE.md +0 -0
@@ -3,7 +3,7 @@
3
3
  """
4
4
  Arize REST API
5
5
 
6
- API specification for the backend data server. The API is hosted globally at https://app.arize.com/api/v1 or in your own environment. You can access the OpenAPI spec for this API at https://app.arize.com/api/v1/spec.yaml
6
+ API specification for the backend data server. The API is hosted globally at https://api.arize.com/v2 or in your own environment. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml
7
7
 
8
8
  The version of the OpenAPI document: 0.0.1
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -19,7 +19,10 @@ from typing_extensions import Annotated
19
19
  from pydantic import Field, StrictInt, StrictStr
20
20
  from typing import Optional
21
21
  from typing_extensions import Annotated
22
+ from arize._generated.api_client.models.experiment import Experiment
23
+ from arize._generated.api_client.models.experiments_create_request import ExperimentsCreateRequest
22
24
  from arize._generated.api_client.models.experiments_list200_response import ExperimentsList200Response
25
+ from arize._generated.api_client.models.experiments_runs_list200_response import ExperimentsRunsList200Response
23
26
 
24
27
  from arize._generated.api_client.api_client import ApiClient, RequestSerialized
25
28
  from arize._generated.api_client.api_response import ApiResponse
@@ -39,6 +42,295 @@ class ExperimentsApi:
39
42
  self.api_client = api_client
40
43
 
41
44
 
45
+ @validate_call
46
+ def experiments_create(
47
+ self,
48
+ experiments_create_request: Annotated[ExperimentsCreateRequest, Field(description="Body containing experiment creation parameters")],
49
+ _request_timeout: Union[
50
+ None,
51
+ Annotated[StrictFloat, Field(gt=0)],
52
+ Tuple[
53
+ Annotated[StrictFloat, Field(gt=0)],
54
+ Annotated[StrictFloat, Field(gt=0)]
55
+ ]
56
+ ] = None,
57
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
58
+ _content_type: Optional[StrictStr] = None,
59
+ _headers: Optional[Dict[StrictStr, Any]] = None,
60
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
61
+ ) -> Experiment:
62
+ """Create a new experiment with run data
63
+
64
+
65
+ :param experiments_create_request: Body containing experiment creation parameters (required)
66
+ :type experiments_create_request: ExperimentsCreateRequest
67
+ :param _request_timeout: timeout setting for this request. If one
68
+ number provided, it will be total request
69
+ timeout. It can also be a pair (tuple) of
70
+ (connection, read) timeouts.
71
+ :type _request_timeout: int, tuple(int, int), optional
72
+ :param _request_auth: set to override the auth_settings for an a single
73
+ request; this effectively ignores the
74
+ authentication in the spec for a single request.
75
+ :type _request_auth: dict, optional
76
+ :param _content_type: force content-type for the request.
77
+ :type _content_type: str, Optional
78
+ :param _headers: set to override the headers for a single
79
+ request; this effectively ignores the headers
80
+ in the spec for a single request.
81
+ :type _headers: dict, optional
82
+ :param _host_index: set to override the host_index for a single
83
+ request; this effectively ignores the host_index
84
+ in the spec for a single request.
85
+ :type _host_index: int, optional
86
+ :return: Returns the result object.
87
+ """ # noqa: E501
88
+
89
+ _param = self._experiments_create_serialize(
90
+ experiments_create_request=experiments_create_request,
91
+ _request_auth=_request_auth,
92
+ _content_type=_content_type,
93
+ _headers=_headers,
94
+ _host_index=_host_index
95
+ )
96
+
97
+ _response_types_map: Dict[str, Optional[str]] = {
98
+ '201': "Experiment",
99
+ '400': "Error",
100
+ '401': "Error",
101
+ '403': "Error",
102
+ '409': "Error",
103
+ '422': "Error",
104
+ '429': "Error",
105
+ }
106
+ response_data = self.api_client.call_api(
107
+ *_param,
108
+ _request_timeout=_request_timeout
109
+ )
110
+ response_data.read()
111
+ return self.api_client.response_deserialize(
112
+ response_data=response_data,
113
+ response_types_map=_response_types_map,
114
+ ).data
115
+
116
+
117
+ @validate_call
118
+ def experiments_create_with_http_info(
119
+ self,
120
+ experiments_create_request: Annotated[ExperimentsCreateRequest, Field(description="Body containing experiment creation parameters")],
121
+ _request_timeout: Union[
122
+ None,
123
+ Annotated[StrictFloat, Field(gt=0)],
124
+ Tuple[
125
+ Annotated[StrictFloat, Field(gt=0)],
126
+ Annotated[StrictFloat, Field(gt=0)]
127
+ ]
128
+ ] = None,
129
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
130
+ _content_type: Optional[StrictStr] = None,
131
+ _headers: Optional[Dict[StrictStr, Any]] = None,
132
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
133
+ ) -> ApiResponse[Experiment]:
134
+ """Create a new experiment with run data
135
+
136
+
137
+ :param experiments_create_request: Body containing experiment creation parameters (required)
138
+ :type experiments_create_request: ExperimentsCreateRequest
139
+ :param _request_timeout: timeout setting for this request. If one
140
+ number provided, it will be total request
141
+ timeout. It can also be a pair (tuple) of
142
+ (connection, read) timeouts.
143
+ :type _request_timeout: int, tuple(int, int), optional
144
+ :param _request_auth: set to override the auth_settings for an a single
145
+ request; this effectively ignores the
146
+ authentication in the spec for a single request.
147
+ :type _request_auth: dict, optional
148
+ :param _content_type: force content-type for the request.
149
+ :type _content_type: str, Optional
150
+ :param _headers: set to override the headers for a single
151
+ request; this effectively ignores the headers
152
+ in the spec for a single request.
153
+ :type _headers: dict, optional
154
+ :param _host_index: set to override the host_index for a single
155
+ request; this effectively ignores the host_index
156
+ in the spec for a single request.
157
+ :type _host_index: int, optional
158
+ :return: Returns the result object.
159
+ """ # noqa: E501
160
+
161
+ _param = self._experiments_create_serialize(
162
+ experiments_create_request=experiments_create_request,
163
+ _request_auth=_request_auth,
164
+ _content_type=_content_type,
165
+ _headers=_headers,
166
+ _host_index=_host_index
167
+ )
168
+
169
+ _response_types_map: Dict[str, Optional[str]] = {
170
+ '201': "Experiment",
171
+ '400': "Error",
172
+ '401': "Error",
173
+ '403': "Error",
174
+ '409': "Error",
175
+ '422': "Error",
176
+ '429': "Error",
177
+ }
178
+ response_data = self.api_client.call_api(
179
+ *_param,
180
+ _request_timeout=_request_timeout
181
+ )
182
+ response_data.read()
183
+ return self.api_client.response_deserialize(
184
+ response_data=response_data,
185
+ response_types_map=_response_types_map,
186
+ )
187
+
188
+
189
+ @validate_call
190
+ def experiments_create_without_preload_content(
191
+ self,
192
+ experiments_create_request: Annotated[ExperimentsCreateRequest, Field(description="Body containing experiment creation parameters")],
193
+ _request_timeout: Union[
194
+ None,
195
+ Annotated[StrictFloat, Field(gt=0)],
196
+ Tuple[
197
+ Annotated[StrictFloat, Field(gt=0)],
198
+ Annotated[StrictFloat, Field(gt=0)]
199
+ ]
200
+ ] = None,
201
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
202
+ _content_type: Optional[StrictStr] = None,
203
+ _headers: Optional[Dict[StrictStr, Any]] = None,
204
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
205
+ ) -> RESTResponseType:
206
+ """Create a new experiment with run data
207
+
208
+
209
+ :param experiments_create_request: Body containing experiment creation parameters (required)
210
+ :type experiments_create_request: ExperimentsCreateRequest
211
+ :param _request_timeout: timeout setting for this request. If one
212
+ number provided, it will be total request
213
+ timeout. It can also be a pair (tuple) of
214
+ (connection, read) timeouts.
215
+ :type _request_timeout: int, tuple(int, int), optional
216
+ :param _request_auth: set to override the auth_settings for an a single
217
+ request; this effectively ignores the
218
+ authentication in the spec for a single request.
219
+ :type _request_auth: dict, optional
220
+ :param _content_type: force content-type for the request.
221
+ :type _content_type: str, Optional
222
+ :param _headers: set to override the headers for a single
223
+ request; this effectively ignores the headers
224
+ in the spec for a single request.
225
+ :type _headers: dict, optional
226
+ :param _host_index: set to override the host_index for a single
227
+ request; this effectively ignores the host_index
228
+ in the spec for a single request.
229
+ :type _host_index: int, optional
230
+ :return: Returns the result object.
231
+ """ # noqa: E501
232
+
233
+ _param = self._experiments_create_serialize(
234
+ experiments_create_request=experiments_create_request,
235
+ _request_auth=_request_auth,
236
+ _content_type=_content_type,
237
+ _headers=_headers,
238
+ _host_index=_host_index
239
+ )
240
+
241
+ _response_types_map: Dict[str, Optional[str]] = {
242
+ '201': "Experiment",
243
+ '400': "Error",
244
+ '401': "Error",
245
+ '403': "Error",
246
+ '409': "Error",
247
+ '422': "Error",
248
+ '429': "Error",
249
+ }
250
+ response_data = self.api_client.call_api(
251
+ *_param,
252
+ _request_timeout=_request_timeout
253
+ )
254
+ return response_data.response
255
+
256
+
257
+ def _experiments_create_serialize(
258
+ self,
259
+ experiments_create_request,
260
+ _request_auth,
261
+ _content_type,
262
+ _headers,
263
+ _host_index,
264
+ ) -> RequestSerialized:
265
+
266
+ _host = None
267
+
268
+ _collection_formats: Dict[str, str] = {
269
+ }
270
+
271
+ _path_params: Dict[str, str] = {}
272
+ _query_params: List[Tuple[str, str]] = []
273
+ _header_params: Dict[str, Optional[str]] = _headers or {}
274
+ _form_params: List[Tuple[str, str]] = []
275
+ _files: Dict[
276
+ str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
277
+ ] = {}
278
+ _body_params: Optional[bytes] = None
279
+
280
+ # process the path parameters
281
+ # process the query parameters
282
+ # process the header parameters
283
+ # process the form parameters
284
+ # process the body parameter
285
+ if experiments_create_request is not None:
286
+ _body_params = experiments_create_request
287
+
288
+
289
+ # set the HTTP header `Accept`
290
+ if 'Accept' not in _header_params:
291
+ _header_params['Accept'] = self.api_client.select_header_accept(
292
+ [
293
+ 'application/json'
294
+ ]
295
+ )
296
+
297
+ # set the HTTP header `Content-Type`
298
+ if _content_type:
299
+ _header_params['Content-Type'] = _content_type
300
+ else:
301
+ _default_content_type = (
302
+ self.api_client.select_header_content_type(
303
+ [
304
+ 'application/json'
305
+ ]
306
+ )
307
+ )
308
+ if _default_content_type is not None:
309
+ _header_params['Content-Type'] = _default_content_type
310
+
311
+ # authentication setting
312
+ _auth_settings: List[str] = [
313
+ 'ApiKeyAuth'
314
+ ]
315
+
316
+ return self.api_client.param_serialize(
317
+ method='POST',
318
+ resource_path='/v2/experiments',
319
+ path_params=_path_params,
320
+ query_params=_query_params,
321
+ header_params=_header_params,
322
+ body=_body_params,
323
+ post_params=_form_params,
324
+ files=_files,
325
+ auth_settings=_auth_settings,
326
+ collection_formats=_collection_formats,
327
+ _host=_host,
328
+ _request_auth=_request_auth
329
+ )
330
+
331
+
332
+
333
+
42
334
  @validate_call
43
335
  def experiments_delete(
44
336
  self,
@@ -55,8 +347,428 @@ class ExperimentsApi:
55
347
  _content_type: Optional[StrictStr] = None,
56
348
  _headers: Optional[Dict[StrictStr, Any]] = None,
57
349
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
58
- ) -> None:
59
- """Delete an experiment by ID
350
+ ) -> None:
351
+ """Delete an experiment by ID
352
+
353
+
354
+ :param experiment_id: The unique identifier of the experiment (required)
355
+ :type experiment_id: str
356
+ :param _request_timeout: timeout setting for this request. If one
357
+ number provided, it will be total request
358
+ timeout. It can also be a pair (tuple) of
359
+ (connection, read) timeouts.
360
+ :type _request_timeout: int, tuple(int, int), optional
361
+ :param _request_auth: set to override the auth_settings for an a single
362
+ request; this effectively ignores the
363
+ authentication in the spec for a single request.
364
+ :type _request_auth: dict, optional
365
+ :param _content_type: force content-type for the request.
366
+ :type _content_type: str, Optional
367
+ :param _headers: set to override the headers for a single
368
+ request; this effectively ignores the headers
369
+ in the spec for a single request.
370
+ :type _headers: dict, optional
371
+ :param _host_index: set to override the host_index for a single
372
+ request; this effectively ignores the host_index
373
+ in the spec for a single request.
374
+ :type _host_index: int, optional
375
+ :return: Returns the result object.
376
+ """ # noqa: E501
377
+
378
+ _param = self._experiments_delete_serialize(
379
+ experiment_id=experiment_id,
380
+ _request_auth=_request_auth,
381
+ _content_type=_content_type,
382
+ _headers=_headers,
383
+ _host_index=_host_index
384
+ )
385
+
386
+ _response_types_map: Dict[str, Optional[str]] = {
387
+ '204': None,
388
+ '400': "Error",
389
+ '401': "Error",
390
+ '403': "Error",
391
+ '404': "Error",
392
+ '422': "Error",
393
+ '429': "Error",
394
+ }
395
+ response_data = self.api_client.call_api(
396
+ *_param,
397
+ _request_timeout=_request_timeout
398
+ )
399
+ response_data.read()
400
+ return self.api_client.response_deserialize(
401
+ response_data=response_data,
402
+ response_types_map=_response_types_map,
403
+ ).data
404
+
405
+
406
+ @validate_call
407
+ def experiments_delete_with_http_info(
408
+ self,
409
+ experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
410
+ _request_timeout: Union[
411
+ None,
412
+ Annotated[StrictFloat, Field(gt=0)],
413
+ Tuple[
414
+ Annotated[StrictFloat, Field(gt=0)],
415
+ Annotated[StrictFloat, Field(gt=0)]
416
+ ]
417
+ ] = None,
418
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
419
+ _content_type: Optional[StrictStr] = None,
420
+ _headers: Optional[Dict[StrictStr, Any]] = None,
421
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
422
+ ) -> ApiResponse[None]:
423
+ """Delete an experiment by ID
424
+
425
+
426
+ :param experiment_id: The unique identifier of the experiment (required)
427
+ :type experiment_id: str
428
+ :param _request_timeout: timeout setting for this request. If one
429
+ number provided, it will be total request
430
+ timeout. It can also be a pair (tuple) of
431
+ (connection, read) timeouts.
432
+ :type _request_timeout: int, tuple(int, int), optional
433
+ :param _request_auth: set to override the auth_settings for an a single
434
+ request; this effectively ignores the
435
+ authentication in the spec for a single request.
436
+ :type _request_auth: dict, optional
437
+ :param _content_type: force content-type for the request.
438
+ :type _content_type: str, Optional
439
+ :param _headers: set to override the headers for a single
440
+ request; this effectively ignores the headers
441
+ in the spec for a single request.
442
+ :type _headers: dict, optional
443
+ :param _host_index: set to override the host_index for a single
444
+ request; this effectively ignores the host_index
445
+ in the spec for a single request.
446
+ :type _host_index: int, optional
447
+ :return: Returns the result object.
448
+ """ # noqa: E501
449
+
450
+ _param = self._experiments_delete_serialize(
451
+ experiment_id=experiment_id,
452
+ _request_auth=_request_auth,
453
+ _content_type=_content_type,
454
+ _headers=_headers,
455
+ _host_index=_host_index
456
+ )
457
+
458
+ _response_types_map: Dict[str, Optional[str]] = {
459
+ '204': None,
460
+ '400': "Error",
461
+ '401': "Error",
462
+ '403': "Error",
463
+ '404': "Error",
464
+ '422': "Error",
465
+ '429': "Error",
466
+ }
467
+ response_data = self.api_client.call_api(
468
+ *_param,
469
+ _request_timeout=_request_timeout
470
+ )
471
+ response_data.read()
472
+ return self.api_client.response_deserialize(
473
+ response_data=response_data,
474
+ response_types_map=_response_types_map,
475
+ )
476
+
477
+
478
+ @validate_call
479
+ def experiments_delete_without_preload_content(
480
+ self,
481
+ experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
482
+ _request_timeout: Union[
483
+ None,
484
+ Annotated[StrictFloat, Field(gt=0)],
485
+ Tuple[
486
+ Annotated[StrictFloat, Field(gt=0)],
487
+ Annotated[StrictFloat, Field(gt=0)]
488
+ ]
489
+ ] = None,
490
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
491
+ _content_type: Optional[StrictStr] = None,
492
+ _headers: Optional[Dict[StrictStr, Any]] = None,
493
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
494
+ ) -> RESTResponseType:
495
+ """Delete an experiment by ID
496
+
497
+
498
+ :param experiment_id: The unique identifier of the experiment (required)
499
+ :type experiment_id: str
500
+ :param _request_timeout: timeout setting for this request. If one
501
+ number provided, it will be total request
502
+ timeout. It can also be a pair (tuple) of
503
+ (connection, read) timeouts.
504
+ :type _request_timeout: int, tuple(int, int), optional
505
+ :param _request_auth: set to override the auth_settings for an a single
506
+ request; this effectively ignores the
507
+ authentication in the spec for a single request.
508
+ :type _request_auth: dict, optional
509
+ :param _content_type: force content-type for the request.
510
+ :type _content_type: str, Optional
511
+ :param _headers: set to override the headers for a single
512
+ request; this effectively ignores the headers
513
+ in the spec for a single request.
514
+ :type _headers: dict, optional
515
+ :param _host_index: set to override the host_index for a single
516
+ request; this effectively ignores the host_index
517
+ in the spec for a single request.
518
+ :type _host_index: int, optional
519
+ :return: Returns the result object.
520
+ """ # noqa: E501
521
+
522
+ _param = self._experiments_delete_serialize(
523
+ experiment_id=experiment_id,
524
+ _request_auth=_request_auth,
525
+ _content_type=_content_type,
526
+ _headers=_headers,
527
+ _host_index=_host_index
528
+ )
529
+
530
+ _response_types_map: Dict[str, Optional[str]] = {
531
+ '204': None,
532
+ '400': "Error",
533
+ '401': "Error",
534
+ '403': "Error",
535
+ '404': "Error",
536
+ '422': "Error",
537
+ '429': "Error",
538
+ }
539
+ response_data = self.api_client.call_api(
540
+ *_param,
541
+ _request_timeout=_request_timeout
542
+ )
543
+ return response_data.response
544
+
545
+
546
+ def _experiments_delete_serialize(
547
+ self,
548
+ experiment_id,
549
+ _request_auth,
550
+ _content_type,
551
+ _headers,
552
+ _host_index,
553
+ ) -> RequestSerialized:
554
+
555
+ _host = None
556
+
557
+ _collection_formats: Dict[str, str] = {
558
+ }
559
+
560
+ _path_params: Dict[str, str] = {}
561
+ _query_params: List[Tuple[str, str]] = []
562
+ _header_params: Dict[str, Optional[str]] = _headers or {}
563
+ _form_params: List[Tuple[str, str]] = []
564
+ _files: Dict[
565
+ str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
566
+ ] = {}
567
+ _body_params: Optional[bytes] = None
568
+
569
+ # process the path parameters
570
+ if experiment_id is not None:
571
+ _path_params['experimentId'] = experiment_id
572
+ # process the query parameters
573
+ # process the header parameters
574
+ # process the form parameters
575
+ # process the body parameter
576
+
577
+
578
+ # set the HTTP header `Accept`
579
+ if 'Accept' not in _header_params:
580
+ _header_params['Accept'] = self.api_client.select_header_accept(
581
+ [
582
+ 'application/json'
583
+ ]
584
+ )
585
+
586
+
587
+ # authentication setting
588
+ _auth_settings: List[str] = [
589
+ 'ApiKeyAuth'
590
+ ]
591
+
592
+ return self.api_client.param_serialize(
593
+ method='DELETE',
594
+ resource_path='/v2/experiments/{experimentId}',
595
+ path_params=_path_params,
596
+ query_params=_query_params,
597
+ header_params=_header_params,
598
+ body=_body_params,
599
+ post_params=_form_params,
600
+ files=_files,
601
+ auth_settings=_auth_settings,
602
+ collection_formats=_collection_formats,
603
+ _host=_host,
604
+ _request_auth=_request_auth
605
+ )
606
+
607
+
608
+
609
+
610
+ @validate_call
611
+ def experiments_get(
612
+ self,
613
+ experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
614
+ _request_timeout: Union[
615
+ None,
616
+ Annotated[StrictFloat, Field(gt=0)],
617
+ Tuple[
618
+ Annotated[StrictFloat, Field(gt=0)],
619
+ Annotated[StrictFloat, Field(gt=0)]
620
+ ]
621
+ ] = None,
622
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
623
+ _content_type: Optional[StrictStr] = None,
624
+ _headers: Optional[Dict[StrictStr, Any]] = None,
625
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
626
+ ) -> Experiment:
627
+ """Get experiment by ID
628
+
629
+
630
+ :param experiment_id: The unique identifier of the experiment (required)
631
+ :type experiment_id: str
632
+ :param _request_timeout: timeout setting for this request. If one
633
+ number provided, it will be total request
634
+ timeout. It can also be a pair (tuple) of
635
+ (connection, read) timeouts.
636
+ :type _request_timeout: int, tuple(int, int), optional
637
+ :param _request_auth: set to override the auth_settings for an a single
638
+ request; this effectively ignores the
639
+ authentication in the spec for a single request.
640
+ :type _request_auth: dict, optional
641
+ :param _content_type: force content-type for the request.
642
+ :type _content_type: str, Optional
643
+ :param _headers: set to override the headers for a single
644
+ request; this effectively ignores the headers
645
+ in the spec for a single request.
646
+ :type _headers: dict, optional
647
+ :param _host_index: set to override the host_index for a single
648
+ request; this effectively ignores the host_index
649
+ in the spec for a single request.
650
+ :type _host_index: int, optional
651
+ :return: Returns the result object.
652
+ """ # noqa: E501
653
+
654
+ _param = self._experiments_get_serialize(
655
+ experiment_id=experiment_id,
656
+ _request_auth=_request_auth,
657
+ _content_type=_content_type,
658
+ _headers=_headers,
659
+ _host_index=_host_index
660
+ )
661
+
662
+ _response_types_map: Dict[str, Optional[str]] = {
663
+ '200': "Experiment",
664
+ '400': "Error",
665
+ '401': "Error",
666
+ '403': "Error",
667
+ '404': "Error",
668
+ '422': "Error",
669
+ '429': "Error",
670
+ }
671
+ response_data = self.api_client.call_api(
672
+ *_param,
673
+ _request_timeout=_request_timeout
674
+ )
675
+ response_data.read()
676
+ return self.api_client.response_deserialize(
677
+ response_data=response_data,
678
+ response_types_map=_response_types_map,
679
+ ).data
680
+
681
+
682
+ @validate_call
683
+ def experiments_get_with_http_info(
684
+ self,
685
+ experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
686
+ _request_timeout: Union[
687
+ None,
688
+ Annotated[StrictFloat, Field(gt=0)],
689
+ Tuple[
690
+ Annotated[StrictFloat, Field(gt=0)],
691
+ Annotated[StrictFloat, Field(gt=0)]
692
+ ]
693
+ ] = None,
694
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
695
+ _content_type: Optional[StrictStr] = None,
696
+ _headers: Optional[Dict[StrictStr, Any]] = None,
697
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
698
+ ) -> ApiResponse[Experiment]:
699
+ """Get experiment by ID
700
+
701
+
702
+ :param experiment_id: The unique identifier of the experiment (required)
703
+ :type experiment_id: str
704
+ :param _request_timeout: timeout setting for this request. If one
705
+ number provided, it will be total request
706
+ timeout. It can also be a pair (tuple) of
707
+ (connection, read) timeouts.
708
+ :type _request_timeout: int, tuple(int, int), optional
709
+ :param _request_auth: set to override the auth_settings for an a single
710
+ request; this effectively ignores the
711
+ authentication in the spec for a single request.
712
+ :type _request_auth: dict, optional
713
+ :param _content_type: force content-type for the request.
714
+ :type _content_type: str, Optional
715
+ :param _headers: set to override the headers for a single
716
+ request; this effectively ignores the headers
717
+ in the spec for a single request.
718
+ :type _headers: dict, optional
719
+ :param _host_index: set to override the host_index for a single
720
+ request; this effectively ignores the host_index
721
+ in the spec for a single request.
722
+ :type _host_index: int, optional
723
+ :return: Returns the result object.
724
+ """ # noqa: E501
725
+
726
+ _param = self._experiments_get_serialize(
727
+ experiment_id=experiment_id,
728
+ _request_auth=_request_auth,
729
+ _content_type=_content_type,
730
+ _headers=_headers,
731
+ _host_index=_host_index
732
+ )
733
+
734
+ _response_types_map: Dict[str, Optional[str]] = {
735
+ '200': "Experiment",
736
+ '400': "Error",
737
+ '401': "Error",
738
+ '403': "Error",
739
+ '404': "Error",
740
+ '422': "Error",
741
+ '429': "Error",
742
+ }
743
+ response_data = self.api_client.call_api(
744
+ *_param,
745
+ _request_timeout=_request_timeout
746
+ )
747
+ response_data.read()
748
+ return self.api_client.response_deserialize(
749
+ response_data=response_data,
750
+ response_types_map=_response_types_map,
751
+ )
752
+
753
+
754
+ @validate_call
755
+ def experiments_get_without_preload_content(
756
+ self,
757
+ experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
758
+ _request_timeout: Union[
759
+ None,
760
+ Annotated[StrictFloat, Field(gt=0)],
761
+ Tuple[
762
+ Annotated[StrictFloat, Field(gt=0)],
763
+ Annotated[StrictFloat, Field(gt=0)]
764
+ ]
765
+ ] = None,
766
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
767
+ _content_type: Optional[StrictStr] = None,
768
+ _headers: Optional[Dict[StrictStr, Any]] = None,
769
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
770
+ ) -> RESTResponseType:
771
+ """Get experiment by ID
60
772
 
61
773
 
62
774
  :param experiment_id: The unique identifier of the experiment (required)
@@ -83,7 +795,7 @@ class ExperimentsApi:
83
795
  :return: Returns the result object.
84
796
  """ # noqa: E501
85
797
 
86
- _param = self._experiments_delete_serialize(
798
+ _param = self._experiments_get_serialize(
87
799
  experiment_id=experiment_id,
88
800
  _request_auth=_request_auth,
89
801
  _content_type=_content_type,
@@ -92,7 +804,7 @@ class ExperimentsApi:
92
804
  )
93
805
 
94
806
  _response_types_map: Dict[str, Optional[str]] = {
95
- '204': None,
807
+ '200': "Experiment",
96
808
  '400': "Error",
97
809
  '401': "Error",
98
810
  '403': "Error",
@@ -104,6 +816,141 @@ class ExperimentsApi:
104
816
  *_param,
105
817
  _request_timeout=_request_timeout
106
818
  )
819
+ return response_data.response
820
+
821
+
822
+ def _experiments_get_serialize(
823
+ self,
824
+ experiment_id,
825
+ _request_auth,
826
+ _content_type,
827
+ _headers,
828
+ _host_index,
829
+ ) -> RequestSerialized:
830
+
831
+ _host = None
832
+
833
+ _collection_formats: Dict[str, str] = {
834
+ }
835
+
836
+ _path_params: Dict[str, str] = {}
837
+ _query_params: List[Tuple[str, str]] = []
838
+ _header_params: Dict[str, Optional[str]] = _headers or {}
839
+ _form_params: List[Tuple[str, str]] = []
840
+ _files: Dict[
841
+ str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
842
+ ] = {}
843
+ _body_params: Optional[bytes] = None
844
+
845
+ # process the path parameters
846
+ if experiment_id is not None:
847
+ _path_params['experimentId'] = experiment_id
848
+ # process the query parameters
849
+ # process the header parameters
850
+ # process the form parameters
851
+ # process the body parameter
852
+
853
+
854
+ # set the HTTP header `Accept`
855
+ if 'Accept' not in _header_params:
856
+ _header_params['Accept'] = self.api_client.select_header_accept(
857
+ [
858
+ 'application/json'
859
+ ]
860
+ )
861
+
862
+
863
+ # authentication setting
864
+ _auth_settings: List[str] = [
865
+ 'ApiKeyAuth'
866
+ ]
867
+
868
+ return self.api_client.param_serialize(
869
+ method='GET',
870
+ resource_path='/v2/experiments/{experimentId}',
871
+ path_params=_path_params,
872
+ query_params=_query_params,
873
+ header_params=_header_params,
874
+ body=_body_params,
875
+ post_params=_form_params,
876
+ files=_files,
877
+ auth_settings=_auth_settings,
878
+ collection_formats=_collection_formats,
879
+ _host=_host,
880
+ _request_auth=_request_auth
881
+ )
882
+
883
+
884
+
885
+
886
+ @validate_call
887
+ def experiments_list(
888
+ self,
889
+ dataset_id: Annotated[Optional[StrictStr], Field(description="Filter experiments to a particular dataset ID")] = None,
890
+ limit: Annotated[Optional[StrictInt], Field(description="Maximum number of items to return")] = None,
891
+ _request_timeout: Union[
892
+ None,
893
+ Annotated[StrictFloat, Field(gt=0)],
894
+ Tuple[
895
+ Annotated[StrictFloat, Field(gt=0)],
896
+ Annotated[StrictFloat, Field(gt=0)]
897
+ ]
898
+ ] = None,
899
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
900
+ _content_type: Optional[StrictStr] = None,
901
+ _headers: Optional[Dict[StrictStr, Any]] = None,
902
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
903
+ ) -> ExperimentsList200Response:
904
+ """List experiments
905
+
906
+
907
+ :param dataset_id: Filter experiments to a particular dataset ID
908
+ :type dataset_id: str
909
+ :param limit: Maximum number of items to return
910
+ :type limit: int
911
+ :param _request_timeout: timeout setting for this request. If one
912
+ number provided, it will be total request
913
+ timeout. It can also be a pair (tuple) of
914
+ (connection, read) timeouts.
915
+ :type _request_timeout: int, tuple(int, int), optional
916
+ :param _request_auth: set to override the auth_settings for an a single
917
+ request; this effectively ignores the
918
+ authentication in the spec for a single request.
919
+ :type _request_auth: dict, optional
920
+ :param _content_type: force content-type for the request.
921
+ :type _content_type: str, Optional
922
+ :param _headers: set to override the headers for a single
923
+ request; this effectively ignores the headers
924
+ in the spec for a single request.
925
+ :type _headers: dict, optional
926
+ :param _host_index: set to override the host_index for a single
927
+ request; this effectively ignores the host_index
928
+ in the spec for a single request.
929
+ :type _host_index: int, optional
930
+ :return: Returns the result object.
931
+ """ # noqa: E501
932
+
933
+ _param = self._experiments_list_serialize(
934
+ dataset_id=dataset_id,
935
+ limit=limit,
936
+ _request_auth=_request_auth,
937
+ _content_type=_content_type,
938
+ _headers=_headers,
939
+ _host_index=_host_index
940
+ )
941
+
942
+ _response_types_map: Dict[str, Optional[str]] = {
943
+ '200': "ExperimentsList200Response",
944
+ '400': "Error",
945
+ '401': "Error",
946
+ '403': "Error",
947
+ '422': "Error",
948
+ '429': "Error",
949
+ }
950
+ response_data = self.api_client.call_api(
951
+ *_param,
952
+ _request_timeout=_request_timeout
953
+ )
107
954
  response_data.read()
108
955
  return self.api_client.response_deserialize(
109
956
  response_data=response_data,
@@ -112,9 +959,10 @@ class ExperimentsApi:
112
959
 
113
960
 
114
961
  @validate_call
115
- def experiments_delete_with_http_info(
962
+ def experiments_list_with_http_info(
116
963
  self,
117
- experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
964
+ dataset_id: Annotated[Optional[StrictStr], Field(description="Filter experiments to a particular dataset ID")] = None,
965
+ limit: Annotated[Optional[StrictInt], Field(description="Maximum number of items to return")] = None,
118
966
  _request_timeout: Union[
119
967
  None,
120
968
  Annotated[StrictFloat, Field(gt=0)],
@@ -127,12 +975,14 @@ class ExperimentsApi:
127
975
  _content_type: Optional[StrictStr] = None,
128
976
  _headers: Optional[Dict[StrictStr, Any]] = None,
129
977
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
130
- ) -> ApiResponse[None]:
131
- """Delete an experiment by ID
978
+ ) -> ApiResponse[ExperimentsList200Response]:
979
+ """List experiments
132
980
 
133
981
 
134
- :param experiment_id: The unique identifier of the experiment (required)
135
- :type experiment_id: str
982
+ :param dataset_id: Filter experiments to a particular dataset ID
983
+ :type dataset_id: str
984
+ :param limit: Maximum number of items to return
985
+ :type limit: int
136
986
  :param _request_timeout: timeout setting for this request. If one
137
987
  number provided, it will be total request
138
988
  timeout. It can also be a pair (tuple) of
@@ -155,8 +1005,9 @@ class ExperimentsApi:
155
1005
  :return: Returns the result object.
156
1006
  """ # noqa: E501
157
1007
 
158
- _param = self._experiments_delete_serialize(
159
- experiment_id=experiment_id,
1008
+ _param = self._experiments_list_serialize(
1009
+ dataset_id=dataset_id,
1010
+ limit=limit,
160
1011
  _request_auth=_request_auth,
161
1012
  _content_type=_content_type,
162
1013
  _headers=_headers,
@@ -164,11 +1015,10 @@ class ExperimentsApi:
164
1015
  )
165
1016
 
166
1017
  _response_types_map: Dict[str, Optional[str]] = {
167
- '204': None,
1018
+ '200': "ExperimentsList200Response",
168
1019
  '400': "Error",
169
1020
  '401': "Error",
170
1021
  '403': "Error",
171
- '404': "Error",
172
1022
  '422': "Error",
173
1023
  '429': "Error",
174
1024
  }
@@ -184,9 +1034,10 @@ class ExperimentsApi:
184
1034
 
185
1035
 
186
1036
  @validate_call
187
- def experiments_delete_without_preload_content(
1037
+ def experiments_list_without_preload_content(
188
1038
  self,
189
- experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
1039
+ dataset_id: Annotated[Optional[StrictStr], Field(description="Filter experiments to a particular dataset ID")] = None,
1040
+ limit: Annotated[Optional[StrictInt], Field(description="Maximum number of items to return")] = None,
190
1041
  _request_timeout: Union[
191
1042
  None,
192
1043
  Annotated[StrictFloat, Field(gt=0)],
@@ -200,11 +1051,13 @@ class ExperimentsApi:
200
1051
  _headers: Optional[Dict[StrictStr, Any]] = None,
201
1052
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
202
1053
  ) -> RESTResponseType:
203
- """Delete an experiment by ID
1054
+ """List experiments
204
1055
 
205
1056
 
206
- :param experiment_id: The unique identifier of the experiment (required)
207
- :type experiment_id: str
1057
+ :param dataset_id: Filter experiments to a particular dataset ID
1058
+ :type dataset_id: str
1059
+ :param limit: Maximum number of items to return
1060
+ :type limit: int
208
1061
  :param _request_timeout: timeout setting for this request. If one
209
1062
  number provided, it will be total request
210
1063
  timeout. It can also be a pair (tuple) of
@@ -227,8 +1080,9 @@ class ExperimentsApi:
227
1080
  :return: Returns the result object.
228
1081
  """ # noqa: E501
229
1082
 
230
- _param = self._experiments_delete_serialize(
231
- experiment_id=experiment_id,
1083
+ _param = self._experiments_list_serialize(
1084
+ dataset_id=dataset_id,
1085
+ limit=limit,
232
1086
  _request_auth=_request_auth,
233
1087
  _content_type=_content_type,
234
1088
  _headers=_headers,
@@ -236,11 +1090,10 @@ class ExperimentsApi:
236
1090
  )
237
1091
 
238
1092
  _response_types_map: Dict[str, Optional[str]] = {
239
- '204': None,
1093
+ '200': "ExperimentsList200Response",
240
1094
  '400': "Error",
241
1095
  '401': "Error",
242
1096
  '403': "Error",
243
- '404': "Error",
244
1097
  '422': "Error",
245
1098
  '429': "Error",
246
1099
  }
@@ -251,9 +1104,10 @@ class ExperimentsApi:
251
1104
  return response_data.response
252
1105
 
253
1106
 
254
- def _experiments_delete_serialize(
1107
+ def _experiments_list_serialize(
255
1108
  self,
256
- experiment_id,
1109
+ dataset_id,
1110
+ limit,
257
1111
  _request_auth,
258
1112
  _content_type,
259
1113
  _headers,
@@ -275,9 +1129,15 @@ class ExperimentsApi:
275
1129
  _body_params: Optional[bytes] = None
276
1130
 
277
1131
  # process the path parameters
278
- if experiment_id is not None:
279
- _path_params['experimentId'] = experiment_id
280
1132
  # process the query parameters
1133
+ if dataset_id is not None:
1134
+
1135
+ _query_params.append(('datasetId', dataset_id))
1136
+
1137
+ if limit is not None:
1138
+
1139
+ _query_params.append(('limit', limit))
1140
+
281
1141
  # process the header parameters
282
1142
  # process the form parameters
283
1143
  # process the body parameter
@@ -298,8 +1158,8 @@ class ExperimentsApi:
298
1158
  ]
299
1159
 
300
1160
  return self.api_client.param_serialize(
301
- method='DELETE',
302
- resource_path='/api/v1/experiments/{experimentId}',
1161
+ method='GET',
1162
+ resource_path='/v2/experiments',
303
1163
  path_params=_path_params,
304
1164
  query_params=_query_params,
305
1165
  header_params=_header_params,
@@ -316,9 +1176,9 @@ class ExperimentsApi:
316
1176
 
317
1177
 
318
1178
  @validate_call
319
- def experiments_list(
1179
+ def experiments_runs_list(
320
1180
  self,
321
- dataset_id: Annotated[StrictStr, Field(description="The unique identifier of the dataset")],
1181
+ experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
322
1182
  limit: Annotated[Optional[StrictInt], Field(description="Maximum number of items to return")] = None,
323
1183
  _request_timeout: Union[
324
1184
  None,
@@ -332,12 +1192,12 @@ class ExperimentsApi:
332
1192
  _content_type: Optional[StrictStr] = None,
333
1193
  _headers: Optional[Dict[StrictStr, Any]] = None,
334
1194
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
335
- ) -> ExperimentsList200Response:
336
- """List experiments for a given dataset
1195
+ ) -> ExperimentsRunsList200Response:
1196
+ """List experiment runs for a specific experiment
337
1197
 
338
1198
 
339
- :param dataset_id: The unique identifier of the dataset (required)
340
- :type dataset_id: str
1199
+ :param experiment_id: The unique identifier of the experiment (required)
1200
+ :type experiment_id: str
341
1201
  :param limit: Maximum number of items to return
342
1202
  :type limit: int
343
1203
  :param _request_timeout: timeout setting for this request. If one
@@ -362,8 +1222,8 @@ class ExperimentsApi:
362
1222
  :return: Returns the result object.
363
1223
  """ # noqa: E501
364
1224
 
365
- _param = self._experiments_list_serialize(
366
- dataset_id=dataset_id,
1225
+ _param = self._experiments_runs_list_serialize(
1226
+ experiment_id=experiment_id,
367
1227
  limit=limit,
368
1228
  _request_auth=_request_auth,
369
1229
  _content_type=_content_type,
@@ -372,10 +1232,11 @@ class ExperimentsApi:
372
1232
  )
373
1233
 
374
1234
  _response_types_map: Dict[str, Optional[str]] = {
375
- '200': "ExperimentsList200Response",
1235
+ '200': "ExperimentsRunsList200Response",
376
1236
  '400': "Error",
377
1237
  '401': "Error",
378
1238
  '403': "Error",
1239
+ '404': "Error",
379
1240
  '422': "Error",
380
1241
  '429': "Error",
381
1242
  }
@@ -391,9 +1252,9 @@ class ExperimentsApi:
391
1252
 
392
1253
 
393
1254
  @validate_call
394
- def experiments_list_with_http_info(
1255
+ def experiments_runs_list_with_http_info(
395
1256
  self,
396
- dataset_id: Annotated[StrictStr, Field(description="The unique identifier of the dataset")],
1257
+ experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
397
1258
  limit: Annotated[Optional[StrictInt], Field(description="Maximum number of items to return")] = None,
398
1259
  _request_timeout: Union[
399
1260
  None,
@@ -407,12 +1268,12 @@ class ExperimentsApi:
407
1268
  _content_type: Optional[StrictStr] = None,
408
1269
  _headers: Optional[Dict[StrictStr, Any]] = None,
409
1270
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
410
- ) -> ApiResponse[ExperimentsList200Response]:
411
- """List experiments for a given dataset
1271
+ ) -> ApiResponse[ExperimentsRunsList200Response]:
1272
+ """List experiment runs for a specific experiment
412
1273
 
413
1274
 
414
- :param dataset_id: The unique identifier of the dataset (required)
415
- :type dataset_id: str
1275
+ :param experiment_id: The unique identifier of the experiment (required)
1276
+ :type experiment_id: str
416
1277
  :param limit: Maximum number of items to return
417
1278
  :type limit: int
418
1279
  :param _request_timeout: timeout setting for this request. If one
@@ -437,8 +1298,8 @@ class ExperimentsApi:
437
1298
  :return: Returns the result object.
438
1299
  """ # noqa: E501
439
1300
 
440
- _param = self._experiments_list_serialize(
441
- dataset_id=dataset_id,
1301
+ _param = self._experiments_runs_list_serialize(
1302
+ experiment_id=experiment_id,
442
1303
  limit=limit,
443
1304
  _request_auth=_request_auth,
444
1305
  _content_type=_content_type,
@@ -447,10 +1308,11 @@ class ExperimentsApi:
447
1308
  )
448
1309
 
449
1310
  _response_types_map: Dict[str, Optional[str]] = {
450
- '200': "ExperimentsList200Response",
1311
+ '200': "ExperimentsRunsList200Response",
451
1312
  '400': "Error",
452
1313
  '401': "Error",
453
1314
  '403': "Error",
1315
+ '404': "Error",
454
1316
  '422': "Error",
455
1317
  '429': "Error",
456
1318
  }
@@ -466,9 +1328,9 @@ class ExperimentsApi:
466
1328
 
467
1329
 
468
1330
  @validate_call
469
- def experiments_list_without_preload_content(
1331
+ def experiments_runs_list_without_preload_content(
470
1332
  self,
471
- dataset_id: Annotated[StrictStr, Field(description="The unique identifier of the dataset")],
1333
+ experiment_id: Annotated[StrictStr, Field(description="The unique identifier of the experiment")],
472
1334
  limit: Annotated[Optional[StrictInt], Field(description="Maximum number of items to return")] = None,
473
1335
  _request_timeout: Union[
474
1336
  None,
@@ -483,11 +1345,11 @@ class ExperimentsApi:
483
1345
  _headers: Optional[Dict[StrictStr, Any]] = None,
484
1346
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
485
1347
  ) -> RESTResponseType:
486
- """List experiments for a given dataset
1348
+ """List experiment runs for a specific experiment
487
1349
 
488
1350
 
489
- :param dataset_id: The unique identifier of the dataset (required)
490
- :type dataset_id: str
1351
+ :param experiment_id: The unique identifier of the experiment (required)
1352
+ :type experiment_id: str
491
1353
  :param limit: Maximum number of items to return
492
1354
  :type limit: int
493
1355
  :param _request_timeout: timeout setting for this request. If one
@@ -512,8 +1374,8 @@ class ExperimentsApi:
512
1374
  :return: Returns the result object.
513
1375
  """ # noqa: E501
514
1376
 
515
- _param = self._experiments_list_serialize(
516
- dataset_id=dataset_id,
1377
+ _param = self._experiments_runs_list_serialize(
1378
+ experiment_id=experiment_id,
517
1379
  limit=limit,
518
1380
  _request_auth=_request_auth,
519
1381
  _content_type=_content_type,
@@ -522,10 +1384,11 @@ class ExperimentsApi:
522
1384
  )
523
1385
 
524
1386
  _response_types_map: Dict[str, Optional[str]] = {
525
- '200': "ExperimentsList200Response",
1387
+ '200': "ExperimentsRunsList200Response",
526
1388
  '400': "Error",
527
1389
  '401': "Error",
528
1390
  '403': "Error",
1391
+ '404': "Error",
529
1392
  '422': "Error",
530
1393
  '429': "Error",
531
1394
  }
@@ -536,9 +1399,9 @@ class ExperimentsApi:
536
1399
  return response_data.response
537
1400
 
538
1401
 
539
- def _experiments_list_serialize(
1402
+ def _experiments_runs_list_serialize(
540
1403
  self,
541
- dataset_id,
1404
+ experiment_id,
542
1405
  limit,
543
1406
  _request_auth,
544
1407
  _content_type,
@@ -561,8 +1424,8 @@ class ExperimentsApi:
561
1424
  _body_params: Optional[bytes] = None
562
1425
 
563
1426
  # process the path parameters
564
- if dataset_id is not None:
565
- _path_params['datasetId'] = dataset_id
1427
+ if experiment_id is not None:
1428
+ _path_params['experimentId'] = experiment_id
566
1429
  # process the query parameters
567
1430
  if limit is not None:
568
1431
 
@@ -589,7 +1452,7 @@ class ExperimentsApi:
589
1452
 
590
1453
  return self.api_client.param_serialize(
591
1454
  method='GET',
592
- resource_path='/api/v1/datasets/{datasetId}/experiments',
1455
+ resource_path='/v2/experiments/{experimentId}/runs',
593
1456
  path_params=_path_params,
594
1457
  query_params=_query_params,
595
1458
  header_params=_header_params,