4
4
5
5
import httpx
6
6
7
- from ..types import worker_list_params , worker_create_params , worker_update_params
7
+ from ..types import worker_list_params , worker_tools_params , worker_create_params , worker_update_params
8
8
from .._types import NOT_GIVEN , Body , Query , Headers , NoneType , NotGiven
9
9
from .._utils import (
10
10
maybe_transform ,
20
20
)
21
21
from ..pagination import SyncOffsetPage , AsyncOffsetPage
22
22
from .._base_client import AsyncPaginator , make_request_options
23
+ from ..types .tool_definition import ToolDefinition
23
24
from ..types .worker_response import WorkerResponse
24
- from ..types .worker_tools_response import WorkerToolsResponse
25
25
from ..types .worker_health_response import WorkerHealthResponse
26
26
27
27
__all__ = ["WorkersResource" , "AsyncWorkersResource" ]
@@ -281,17 +281,23 @@ def tools(
281
281
self ,
282
282
id : str ,
283
283
* ,
284
+ limit : int | NotGiven = NOT_GIVEN ,
285
+ offset : int | NotGiven = NOT_GIVEN ,
284
286
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
285
287
# The extra values given here take precedence over values defined on the client or passed to this method.
286
288
extra_headers : Headers | None = None ,
287
289
extra_query : Query | None = None ,
288
290
extra_body : Body | None = None ,
289
291
timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
290
- ) -> WorkerToolsResponse :
292
+ ) -> SyncOffsetPage [ ToolDefinition ] :
291
293
"""
292
294
Returns a page of tools
293
295
294
296
Args:
297
+ limit: Number of items to return (default: 25, max: 100)
298
+
299
+ offset: Offset from the start of the list (default: 0)
300
+
295
301
extra_headers: Send extra headers
296
302
297
303
extra_query: Add additional query parameters to the request
@@ -302,12 +308,23 @@ def tools(
302
308
"""
303
309
if not id :
304
310
raise ValueError (f"Expected a non-empty value for `id` but received { id !r} " )
305
- return self ._get (
311
+ return self ._get_api_list (
306
312
f"/v1/workers/{ id } /tools" ,
313
+ page = SyncOffsetPage [ToolDefinition ],
307
314
options = make_request_options (
308
- extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
315
+ extra_headers = extra_headers ,
316
+ extra_query = extra_query ,
317
+ extra_body = extra_body ,
318
+ timeout = timeout ,
319
+ query = maybe_transform (
320
+ {
321
+ "limit" : limit ,
322
+ "offset" : offset ,
323
+ },
324
+ worker_tools_params .WorkerToolsParams ,
325
+ ),
309
326
),
310
- cast_to = WorkerToolsResponse ,
327
+ model = ToolDefinition ,
311
328
)
312
329
313
330
@@ -561,21 +578,27 @@ async def health(
561
578
cast_to = WorkerHealthResponse ,
562
579
)
563
580
564
- async def tools (
581
+ def tools (
565
582
self ,
566
583
id : str ,
567
584
* ,
585
+ limit : int | NotGiven = NOT_GIVEN ,
586
+ offset : int | NotGiven = NOT_GIVEN ,
568
587
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
569
588
# The extra values given here take precedence over values defined on the client or passed to this method.
570
589
extra_headers : Headers | None = None ,
571
590
extra_query : Query | None = None ,
572
591
extra_body : Body | None = None ,
573
592
timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
574
- ) -> WorkerToolsResponse :
593
+ ) -> AsyncPaginator [ ToolDefinition , AsyncOffsetPage [ ToolDefinition ]] :
575
594
"""
576
595
Returns a page of tools
577
596
578
597
Args:
598
+ limit: Number of items to return (default: 25, max: 100)
599
+
600
+ offset: Offset from the start of the list (default: 0)
601
+
579
602
extra_headers: Send extra headers
580
603
581
604
extra_query: Add additional query parameters to the request
@@ -586,12 +609,23 @@ async def tools(
586
609
"""
587
610
if not id :
588
611
raise ValueError (f"Expected a non-empty value for `id` but received { id !r} " )
589
- return await self ._get (
612
+ return self ._get_api_list (
590
613
f"/v1/workers/{ id } /tools" ,
614
+ page = AsyncOffsetPage [ToolDefinition ],
591
615
options = make_request_options (
592
- extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
616
+ extra_headers = extra_headers ,
617
+ extra_query = extra_query ,
618
+ extra_body = extra_body ,
619
+ timeout = timeout ,
620
+ query = maybe_transform (
621
+ {
622
+ "limit" : limit ,
623
+ "offset" : offset ,
624
+ },
625
+ worker_tools_params .WorkerToolsParams ,
626
+ ),
593
627
),
594
- cast_to = WorkerToolsResponse ,
628
+ model = ToolDefinition ,
595
629
)
596
630
597
631
0 commit comments