2020 StepsResourceWithStreamingResponse ,
2121 AsyncStepsResourceWithStreamingResponse ,
2222)
23- from ...types import agent_create_params
23+ from ...types import agent_list_params , agent_create_params
2424from .session import (
2525 SessionResource ,
2626 AsyncSessionResource ,
4040 async_to_streamed_response_wrapper ,
4141)
4242from ..._base_client import make_request_options
43+ from ...types .agent_list_response import AgentListResponse
4344from ...types .agent_create_response import AgentCreateResponse
45+ from ...types .agent_retrieve_response import AgentRetrieveResponse
4446from ...types .shared_params .agent_config import AgentConfig
4547
4648__all__ = ["AgentsResource" , "AsyncAgentsResource" ]
@@ -112,6 +114,85 @@ def create(
112114 cast_to = AgentCreateResponse ,
113115 )
114116
117+ def retrieve (
118+ self ,
119+ agent_id : str ,
120+ * ,
121+ # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
122+ # The extra values given here take precedence over values defined on the client or passed to this method.
123+ extra_headers : Headers | None = None ,
124+ extra_query : Query | None = None ,
125+ extra_body : Body | None = None ,
126+ timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
127+ ) -> AgentRetrieveResponse :
128+ """
129+ Describe an agent by its ID.
130+
131+ Args:
132+ extra_headers: Send extra headers
133+
134+ extra_query: Add additional query parameters to the request
135+
136+ extra_body: Add additional JSON properties to the request
137+
138+ timeout: Override the client-level default timeout for this request, in seconds
139+ """
140+ if not agent_id :
141+ raise ValueError (f"Expected a non-empty value for `agent_id` but received { agent_id !r} )
142+ return self ._get (
143+ f"/v1/agents/{ agent_id } ,
144+ options = make_request_options (
145+ extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
146+ ),
147+ cast_to = AgentRetrieveResponse ,
148+ )
149+
150+ def list (
151+ self ,
152+ * ,
153+ limit : int | NotGiven = NOT_GIVEN ,
154+ start_index : int | NotGiven = NOT_GIVEN ,
155+ # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
156+ # The extra values given here take precedence over values defined on the client or passed to this method.
157+ extra_headers : Headers | None = None ,
158+ extra_query : Query | None = None ,
159+ extra_body : Body | None = None ,
160+ timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
161+ ) -> AgentListResponse :
162+ """
163+ List all agents.
164+
165+ Args:
166+ limit: The number of agents to return.
167+
168+ start_index: The index to start the pagination from.
169+
170+ extra_headers: Send extra headers
171+
172+ extra_query: Add additional query parameters to the request
173+
174+ extra_body: Add additional JSON properties to the request
175+
176+ timeout: Override the client-level default timeout for this request, in seconds
177+ """
178+ return self ._get (
179+ "/v1/agents" ,
180+ options = make_request_options (
181+ extra_headers = extra_headers ,
182+ extra_query = extra_query ,
183+ extra_body = extra_body ,
184+ timeout = timeout ,
185+ query = maybe_transform (
186+ {
187+ "limit" : limit ,
188+ "start_index" : start_index ,
189+ },
190+ agent_list_params .AgentListParams ,
191+ ),
192+ ),
193+ cast_to = AgentListResponse ,
194+ )
195+
115196 def delete (
116197 self ,
117198 agent_id : str ,
@@ -213,6 +294,85 @@ async def create(
213294 cast_to = AgentCreateResponse ,
214295 )
215296
297+ async def retrieve (
298+ self ,
299+ agent_id : str ,
300+ * ,
301+ # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
302+ # The extra values given here take precedence over values defined on the client or passed to this method.
303+ extra_headers : Headers | None = None ,
304+ extra_query : Query | None = None ,
305+ extra_body : Body | None = None ,
306+ timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
307+ ) -> AgentRetrieveResponse :
308+ """
309+ Describe an agent by its ID.
310+
311+ Args:
312+ extra_headers: Send extra headers
313+
314+ extra_query: Add additional query parameters to the request
315+
316+ extra_body: Add additional JSON properties to the request
317+
318+ timeout: Override the client-level default timeout for this request, in seconds
319+ """
320+ if not agent_id :
321+ raise ValueError (f"Expected a non-empty value for `agent_id` but received { agent_id !r} )
322+ return await self ._get (
323+ f"/v1/agents/{ agent_id } ,
324+ options = make_request_options (
325+ extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
326+ ),
327+ cast_to = AgentRetrieveResponse ,
328+ )
329+
330+ async def list (
331+ self ,
332+ * ,
333+ limit : int | NotGiven = NOT_GIVEN ,
334+ start_index : int | NotGiven = NOT_GIVEN ,
335+ # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
336+ # The extra values given here take precedence over values defined on the client or passed to this method.
337+ extra_headers : Headers | None = None ,
338+ extra_query : Query | None = None ,
339+ extra_body : Body | None = None ,
340+ timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
341+ ) -> AgentListResponse :
342+ """
343+ List all agents.
344+
345+ Args:
346+ limit: The number of agents to return.
347+
348+ start_index: The index to start the pagination from.
349+
350+ extra_headers: Send extra headers
351+
352+ extra_query: Add additional query parameters to the request
353+
354+ extra_body: Add additional JSON properties to the request
355+
356+ timeout: Override the client-level default timeout for this request, in seconds
357+ """
358+ return await self ._get (
359+ "/v1/agents" ,
360+ options = make_request_options (
361+ extra_headers = extra_headers ,
362+ extra_query = extra_query ,
363+ extra_body = extra_body ,
364+ timeout = timeout ,
365+ query = await async_maybe_transform (
366+ {
367+ "limit" : limit ,
368+ "start_index" : start_index ,
369+ },
370+ agent_list_params .AgentListParams ,
371+ ),
372+ ),
373+ cast_to = AgentListResponse ,
374+ )
375+
216376 async def delete (
217377 self ,
218378 agent_id : str ,
@@ -255,6 +415,12 @@ def __init__(self, agents: AgentsResource) -> None:
255415 self .create = to_raw_response_wrapper (
256416 agents .create ,
257417 )
418+ self .retrieve = to_raw_response_wrapper (
419+ agents .retrieve ,
420+ )
421+ self .list = to_raw_response_wrapper (
422+ agents .list ,
423+ )
258424 self .delete = to_raw_response_wrapper (
259425 agents .delete ,
260426 )
@@ -279,6 +445,12 @@ def __init__(self, agents: AsyncAgentsResource) -> None:
279445 self .create = async_to_raw_response_wrapper (
280446 agents .create ,
281447 )
448+ self .retrieve = async_to_raw_response_wrapper (
449+ agents .retrieve ,
450+ )
451+ self .list = async_to_raw_response_wrapper (
452+ agents .list ,
453+ )
282454 self .delete = async_to_raw_response_wrapper (
283455 agents .delete ,
284456 )
@@ -303,6 +475,12 @@ def __init__(self, agents: AgentsResource) -> None:
303475 self .create = to_streamed_response_wrapper (
304476 agents .create ,
305477 )
478+ self .retrieve = to_streamed_response_wrapper (
479+ agents .retrieve ,
480+ )
481+ self .list = to_streamed_response_wrapper (
482+ agents .list ,
483+ )
306484 self .delete = to_streamed_response_wrapper (
307485 agents .delete ,
308486 )
@@ -327,6 +505,12 @@ def __init__(self, agents: AsyncAgentsResource) -> None:
327505 self .create = async_to_streamed_response_wrapper (
328506 agents .create ,
329507 )
508+ self .retrieve = async_to_streamed_response_wrapper (
509+ agents .retrieve ,
510+ )
511+ self .list = async_to_streamed_response_wrapper (
512+ agents .list ,
513+ )
330514 self .delete = async_to_streamed_response_wrapper (
331515 agents .delete ,
332516 )
![stainless-app[bot]](https://avatars.githubusercontent.com/in/378072?v=4&size=40){kind=avatar}
0 commit comments