3535)
3636from ...pagination import SyncBenchmarksCursorIDPage , AsyncBenchmarksCursorIDPage
3737from ..._base_client import AsyncPaginator , make_request_options
38- from ...types .benchmark_view import BenchmarkView
3938from ...types .benchmark_run_view import BenchmarkRunView
39+ from ...types .benchmark_list_response import BenchmarkListResponse
40+ from ...types .benchmark_create_response import BenchmarkCreateResponse
41+ from ...types .benchmark_retrieve_response import BenchmarkRetrieveResponse
42+ from ...types .benchmark_list_public_response import BenchmarkListPublicResponse
4043
4144__all__ = ["BenchmarksResource" , "AsyncBenchmarksResource" ]
4245
@@ -78,7 +81,7 @@ def create(
7881 extra_body : Body | None = None ,
7982 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
8083 idempotency_key : str | None = None ,
81- ) -> BenchmarkView :
84+ ) -> BenchmarkCreateResponse :
8285 """
8386 Create a Benchmark with a set of Scenarios.
8487
@@ -116,7 +119,7 @@ def create(
116119 timeout = timeout ,
117120 idempotency_key = idempotency_key ,
118121 ),
119- cast_to = BenchmarkView ,
122+ cast_to = BenchmarkCreateResponse ,
120123 )
121124
122125 def retrieve (
@@ -129,7 +132,7 @@ def retrieve(
129132 extra_query : Query | None = None ,
130133 extra_body : Body | None = None ,
131134 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
132- ) -> BenchmarkView :
135+ ) -> BenchmarkRetrieveResponse :
133136 """
134137 Get a previously created Benchmark.
135138
@@ -149,7 +152,7 @@ def retrieve(
149152 options = make_request_options (
150153 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
151154 ),
152- cast_to = BenchmarkView ,
155+ cast_to = BenchmarkRetrieveResponse ,
153156 )
154157
155158 def list (
@@ -163,7 +166,7 @@ def list(
163166 extra_query : Query | None = None ,
164167 extra_body : Body | None = None ,
165168 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
166- ) -> SyncBenchmarksCursorIDPage [BenchmarkView ]:
169+ ) -> SyncBenchmarksCursorIDPage [BenchmarkListResponse ]:
167170 """
168171 List all Benchmarks matching filter.
169172
@@ -182,7 +185,7 @@ def list(
182185 """
183186 return self ._get_api_list (
184187 "/v1/benchmarks" ,
185- page = SyncBenchmarksCursorIDPage [BenchmarkView ],
188+ page = SyncBenchmarksCursorIDPage [BenchmarkListResponse ],
186189 options = make_request_options (
187190 extra_headers = extra_headers ,
188191 extra_query = extra_query ,
@@ -196,7 +199,7 @@ def list(
196199 benchmark_list_params .BenchmarkListParams ,
197200 ),
198201 ),
199- model = BenchmarkView ,
202+ model = BenchmarkListResponse ,
200203 )
201204
202205 def list_public (
@@ -210,7 +213,7 @@ def list_public(
210213 extra_query : Query | None = None ,
211214 extra_body : Body | None = None ,
212215 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
213- ) -> SyncBenchmarksCursorIDPage [BenchmarkView ]:
216+ ) -> SyncBenchmarksCursorIDPage [BenchmarkListPublicResponse ]:
214217 """
215218 List all public benchmarks matching filter.
216219
@@ -229,7 +232,7 @@ def list_public(
229232 """
230233 return self ._get_api_list (
231234 "/v1/benchmarks/list_public" ,
232- page = SyncBenchmarksCursorIDPage [BenchmarkView ],
235+ page = SyncBenchmarksCursorIDPage [BenchmarkListPublicResponse ],
233236 options = make_request_options (
234237 extra_headers = extra_headers ,
235238 extra_query = extra_query ,
@@ -243,7 +246,7 @@ def list_public(
243246 benchmark_list_public_params .BenchmarkListPublicParams ,
244247 ),
245248 ),
246- model = BenchmarkView ,
249+ model = BenchmarkListPublicResponse ,
247250 )
248251
249252 def start_run (
@@ -338,7 +341,7 @@ async def create(
338341 extra_body : Body | None = None ,
339342 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
340343 idempotency_key : str | None = None ,
341- ) -> BenchmarkView :
344+ ) -> BenchmarkCreateResponse :
342345 """
343346 Create a Benchmark with a set of Scenarios.
344347
@@ -376,7 +379,7 @@ async def create(
376379 timeout = timeout ,
377380 idempotency_key = idempotency_key ,
378381 ),
379- cast_to = BenchmarkView ,
382+ cast_to = BenchmarkCreateResponse ,
380383 )
381384
382385 async def retrieve (
@@ -389,7 +392,7 @@ async def retrieve(
389392 extra_query : Query | None = None ,
390393 extra_body : Body | None = None ,
391394 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
392- ) -> BenchmarkView :
395+ ) -> BenchmarkRetrieveResponse :
393396 """
394397 Get a previously created Benchmark.
395398
@@ -409,7 +412,7 @@ async def retrieve(
409412 options = make_request_options (
410413 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
411414 ),
412- cast_to = BenchmarkView ,
415+ cast_to = BenchmarkRetrieveResponse ,
413416 )
414417
415418 def list (
@@ -423,7 +426,7 @@ def list(
423426 extra_query : Query | None = None ,
424427 extra_body : Body | None = None ,
425428 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
426- ) -> AsyncPaginator [BenchmarkView , AsyncBenchmarksCursorIDPage [BenchmarkView ]]:
429+ ) -> AsyncPaginator [BenchmarkListResponse , AsyncBenchmarksCursorIDPage [BenchmarkListResponse ]]:
427430 """
428431 List all Benchmarks matching filter.
429432
@@ -442,7 +445,7 @@ def list(
442445 """
443446 return self ._get_api_list (
444447 "/v1/benchmarks" ,
445- page = AsyncBenchmarksCursorIDPage [BenchmarkView ],
448+ page = AsyncBenchmarksCursorIDPage [BenchmarkListResponse ],
446449 options = make_request_options (
447450 extra_headers = extra_headers ,
448451 extra_query = extra_query ,
@@ -456,7 +459,7 @@ def list(
456459 benchmark_list_params .BenchmarkListParams ,
457460 ),
458461 ),
459- model = BenchmarkView ,
462+ model = BenchmarkListResponse ,
460463 )
461464
462465 def list_public (
@@ -470,7 +473,7 @@ def list_public(
470473 extra_query : Query | None = None ,
471474 extra_body : Body | None = None ,
472475 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
473- ) -> AsyncPaginator [BenchmarkView , AsyncBenchmarksCursorIDPage [BenchmarkView ]]:
476+ ) -> AsyncPaginator [BenchmarkListPublicResponse , AsyncBenchmarksCursorIDPage [BenchmarkListPublicResponse ]]:
474477 """
475478 List all public benchmarks matching filter.
476479
@@ -489,7 +492,7 @@ def list_public(
489492 """
490493 return self ._get_api_list (
491494 "/v1/benchmarks/list_public" ,
492- page = AsyncBenchmarksCursorIDPage [BenchmarkView ],
495+ page = AsyncBenchmarksCursorIDPage [BenchmarkListPublicResponse ],
493496 options = make_request_options (
494497 extra_headers = extra_headers ,
495498 extra_query = extra_query ,
@@ -503,7 +506,7 @@ def list_public(
503506 benchmark_list_public_params .BenchmarkListPublicParams ,
504507 ),
505508 ),
506- model = BenchmarkView ,
509+ model = BenchmarkListPublicResponse ,
507510 )
508511
509512 async def start_run (
0 commit comments