3232)
3333from ...pagination import SyncBenchmarksCursorIDPage , AsyncBenchmarksCursorIDPage
3434from ..._base_client import AsyncPaginator , make_request_options
35+ from ...types .benchmark_view import BenchmarkView
3536from ...types .benchmark_run_view import BenchmarkRunView
36- from ...types .benchmark_list_response import BenchmarkListResponse
37- from ...types .benchmark_create_response import BenchmarkCreateResponse
38- from ...types .benchmark_retrieve_response import BenchmarkRetrieveResponse
39- from ...types .benchmark_list_public_response import BenchmarkListPublicResponse
4037
4138__all__ = ["BenchmarksResource" , "AsyncBenchmarksResource" ]
4239
@@ -79,7 +76,7 @@ def create(
7976 extra_body : Body | None = None ,
8077 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
8178 idempotency_key : str | None = None ,
82- ) -> BenchmarkCreateResponse :
79+ ) -> BenchmarkView :
8380 """
8481 Create a Benchmark with a set of Scenarios.
8582
@@ -120,7 +117,7 @@ def create(
120117 timeout = timeout ,
121118 idempotency_key = idempotency_key ,
122119 ),
123- cast_to = BenchmarkCreateResponse ,
120+ cast_to = BenchmarkView ,
124121 )
125122
126123 def retrieve (
@@ -133,7 +130,7 @@ def retrieve(
133130 extra_query : Query | None = None ,
134131 extra_body : Body | None = None ,
135132 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
136- ) -> BenchmarkRetrieveResponse :
133+ ) -> BenchmarkView :
137134 """
138135 Get a previously created Benchmark.
139136
@@ -153,7 +150,7 @@ def retrieve(
153150 options = make_request_options (
154151 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
155152 ),
156- cast_to = BenchmarkRetrieveResponse ,
153+ cast_to = BenchmarkView ,
157154 )
158155
159156 def list (
@@ -167,7 +164,7 @@ def list(
167164 extra_query : Query | None = None ,
168165 extra_body : Body | None = None ,
169166 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
170- ) -> SyncBenchmarksCursorIDPage [BenchmarkListResponse ]:
167+ ) -> SyncBenchmarksCursorIDPage [BenchmarkView ]:
171168 """
172169 List all Benchmarks matching filter.
173170
@@ -186,7 +183,7 @@ def list(
186183 """
187184 return self ._get_api_list (
188185 "/v1/benchmarks" ,
189- page = SyncBenchmarksCursorIDPage [BenchmarkListResponse ],
186+ page = SyncBenchmarksCursorIDPage [BenchmarkView ],
190187 options = make_request_options (
191188 extra_headers = extra_headers ,
192189 extra_query = extra_query ,
@@ -200,7 +197,7 @@ def list(
200197 benchmark_list_params .BenchmarkListParams ,
201198 ),
202199 ),
203- model = BenchmarkListResponse ,
200+ model = BenchmarkView ,
204201 )
205202
206203 def list_public (
@@ -214,7 +211,7 @@ def list_public(
214211 extra_query : Query | None = None ,
215212 extra_body : Body | None = None ,
216213 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
217- ) -> SyncBenchmarksCursorIDPage [BenchmarkListPublicResponse ]:
214+ ) -> SyncBenchmarksCursorIDPage [BenchmarkView ]:
218215 """
219216 List all public benchmarks matching filter.
220217
@@ -233,7 +230,7 @@ def list_public(
233230 """
234231 return self ._get_api_list (
235232 "/v1/benchmarks/list_public" ,
236- page = SyncBenchmarksCursorIDPage [BenchmarkListPublicResponse ],
233+ page = SyncBenchmarksCursorIDPage [BenchmarkView ],
237234 options = make_request_options (
238235 extra_headers = extra_headers ,
239236 extra_query = extra_query ,
@@ -247,7 +244,7 @@ def list_public(
247244 benchmark_list_public_params .BenchmarkListPublicParams ,
248245 ),
249246 ),
250- model = BenchmarkListPublicResponse ,
247+ model = BenchmarkView ,
251248 )
252249
253250 def start_run (
@@ -343,7 +340,7 @@ async def create(
343340 extra_body : Body | None = None ,
344341 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
345342 idempotency_key : str | None = None ,
346- ) -> BenchmarkCreateResponse :
343+ ) -> BenchmarkView :
347344 """
348345 Create a Benchmark with a set of Scenarios.
349346
@@ -384,7 +381,7 @@ async def create(
384381 timeout = timeout ,
385382 idempotency_key = idempotency_key ,
386383 ),
387- cast_to = BenchmarkCreateResponse ,
384+ cast_to = BenchmarkView ,
388385 )
389386
390387 async def retrieve (
@@ -397,7 +394,7 @@ async def retrieve(
397394 extra_query : Query | None = None ,
398395 extra_body : Body | None = None ,
399396 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
400- ) -> BenchmarkRetrieveResponse :
397+ ) -> BenchmarkView :
401398 """
402399 Get a previously created Benchmark.
403400
@@ -417,7 +414,7 @@ async def retrieve(
417414 options = make_request_options (
418415 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
419416 ),
420- cast_to = BenchmarkRetrieveResponse ,
417+ cast_to = BenchmarkView ,
421418 )
422419
423420 def list (
@@ -431,7 +428,7 @@ def list(
431428 extra_query : Query | None = None ,
432429 extra_body : Body | None = None ,
433430 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
434- ) -> AsyncPaginator [BenchmarkListResponse , AsyncBenchmarksCursorIDPage [BenchmarkListResponse ]]:
431+ ) -> AsyncPaginator [BenchmarkView , AsyncBenchmarksCursorIDPage [BenchmarkView ]]:
435432 """
436433 List all Benchmarks matching filter.
437434
@@ -450,7 +447,7 @@ def list(
450447 """
451448 return self ._get_api_list (
452449 "/v1/benchmarks" ,
453- page = AsyncBenchmarksCursorIDPage [BenchmarkListResponse ],
450+ page = AsyncBenchmarksCursorIDPage [BenchmarkView ],
454451 options = make_request_options (
455452 extra_headers = extra_headers ,
456453 extra_query = extra_query ,
@@ -464,7 +461,7 @@ def list(
464461 benchmark_list_params .BenchmarkListParams ,
465462 ),
466463 ),
467- model = BenchmarkListResponse ,
464+ model = BenchmarkView ,
468465 )
469466
470467 def list_public (
@@ -478,7 +475,7 @@ def list_public(
478475 extra_query : Query | None = None ,
479476 extra_body : Body | None = None ,
480477 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
481- ) -> AsyncPaginator [BenchmarkListPublicResponse , AsyncBenchmarksCursorIDPage [BenchmarkListPublicResponse ]]:
478+ ) -> AsyncPaginator [BenchmarkView , AsyncBenchmarksCursorIDPage [BenchmarkView ]]:
482479 """
483480 List all public benchmarks matching filter.
484481
@@ -497,7 +494,7 @@ def list_public(
497494 """
498495 return self ._get_api_list (
499496 "/v1/benchmarks/list_public" ,
500- page = AsyncBenchmarksCursorIDPage [BenchmarkListPublicResponse ],
497+ page = AsyncBenchmarksCursorIDPage [BenchmarkView ],
501498 options = make_request_options (
502499 extra_headers = extra_headers ,
503500 extra_query = extra_query ,
@@ -511,7 +508,7 @@ def list_public(
511508 benchmark_list_public_params .BenchmarkListPublicParams ,
512509 ),
513510 ),
514- model = BenchmarkListPublicResponse ,
511+ model = BenchmarkView ,
515512 )
516513
517514 async def start_run (
0 commit comments