Skip to content

Commit f5e4fd2

Browse files
feat(api): api update
1 parent 2e6f4b2 commit f5e4fd2

19 files changed

+147
-10
lines changed

.stats.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 34
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-03963937874221f124e7a52b2640b8eb69d5ad3e0dafea5d4e65dd3ebb4c06b3.yml
3-
openapi_spec_hash: db2c598ad7459cbeb3d5dbf812b0ff6f
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-6803f43832a4ba37348dfc1daefe5f60f2472b76c73c286257ca370248d64e75.yml
3+
openapi_spec_hash: 44d9bed4803ac9f98419df9e0ad1c485
44
config_hash: 6481ea6b42040f435dedcb00a98f35f8

src/agentex/resources/agents.py

Lines changed: 28 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,8 @@ def retrieve(
9292
def list(
9393
self,
9494
*,
95+
limit: int | Omit = omit,
96+
page_number: int | Omit = omit,
9597
task_id: Optional[str] | Omit = omit,
9698
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
9799
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -104,6 +106,10 @@ def list(
104106
List all registered agents, optionally filtered by query parameters.
105107
106108
Args:
109+
limit: Limit
110+
111+
page_number: Page number
112+
107113
task_id: Task ID
108114
109115
extra_headers: Send extra headers
@@ -121,7 +127,14 @@ def list(
121127
extra_query=extra_query,
122128
extra_body=extra_body,
123129
timeout=timeout,
124-
query=maybe_transform({"task_id": task_id}, agent_list_params.AgentListParams),
130+
query=maybe_transform(
131+
{
132+
"limit": limit,
133+
"page_number": page_number,
134+
"task_id": task_id,
135+
},
136+
agent_list_params.AgentListParams,
137+
),
125138
),
126139
cast_to=AgentListResponse,
127140
)
@@ -639,6 +652,8 @@ async def retrieve(
639652
async def list(
640653
self,
641654
*,
655+
limit: int | Omit = omit,
656+
page_number: int | Omit = omit,
642657
task_id: Optional[str] | Omit = omit,
643658
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
644659
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -651,6 +666,10 @@ async def list(
651666
List all registered agents, optionally filtered by query parameters.
652667
653668
Args:
669+
limit: Limit
670+
671+
page_number: Page number
672+
654673
task_id: Task ID
655674
656675
extra_headers: Send extra headers
@@ -668,7 +687,14 @@ async def list(
668687
extra_query=extra_query,
669688
extra_body=extra_body,
670689
timeout=timeout,
671-
query=await async_maybe_transform({"task_id": task_id}, agent_list_params.AgentListParams),
690+
query=await async_maybe_transform(
691+
{
692+
"limit": limit,
693+
"page_number": page_number,
694+
"task_id": task_id,
695+
},
696+
agent_list_params.AgentListParams,
697+
),
672698
),
673699
cast_to=AgentListResponse,
674700
)

src/agentex/resources/messages/messages.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,8 @@ def list(
180180
self,
181181
*,
182182
task_id: str,
183-
limit: Optional[int] | Omit = omit,
183+
limit: int | Omit = omit,
184+
page_number: int | Omit = omit,
184185
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
185186
# The extra values given here take precedence over values defined on the client or passed to this method.
186187
extra_headers: Headers | None = None,
@@ -213,6 +214,7 @@ def list(
213214
{
214215
"task_id": task_id,
215216
"limit": limit,
217+
"page_number": page_number,
216218
},
217219
message_list_params.MessageListParams,
218220
),
@@ -367,7 +369,8 @@ async def list(
367369
self,
368370
*,
369371
task_id: str,
370-
limit: Optional[int] | Omit = omit,
372+
limit: int | Omit = omit,
373+
page_number: int | Omit = omit,
371374
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
372375
# The extra values given here take precedence over values defined on the client or passed to this method.
373376
extra_headers: Headers | None = None,
@@ -400,6 +403,7 @@ async def list(
400403
{
401404
"task_id": task_id,
402405
"limit": limit,
406+
"page_number": page_number,
403407
},
404408
message_list_params.MessageListParams,
405409
),

src/agentex/resources/spans.py

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,8 @@ def update(
222222
def list(
223223
self,
224224
*,
225+
limit: int | Omit = omit,
226+
page_number: int | Omit = omit,
225227
trace_id: Optional[str] | Omit = omit,
226228
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
227229
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -249,7 +251,14 @@ def list(
249251
extra_query=extra_query,
250252
extra_body=extra_body,
251253
timeout=timeout,
252-
query=maybe_transform({"trace_id": trace_id}, span_list_params.SpanListParams),
254+
query=maybe_transform(
255+
{
256+
"limit": limit,
257+
"page_number": page_number,
258+
"trace_id": trace_id,
259+
},
260+
span_list_params.SpanListParams,
261+
),
253262
),
254263
cast_to=SpanListResponse,
255264
)
@@ -452,6 +461,8 @@ async def update(
452461
async def list(
453462
self,
454463
*,
464+
limit: int | Omit = omit,
465+
page_number: int | Omit = omit,
455466
trace_id: Optional[str] | Omit = omit,
456467
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
457468
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -479,7 +490,14 @@ async def list(
479490
extra_query=extra_query,
480491
extra_body=extra_body,
481492
timeout=timeout,
482-
query=await async_maybe_transform({"trace_id": trace_id}, span_list_params.SpanListParams),
493+
query=await async_maybe_transform(
494+
{
495+
"limit": limit,
496+
"page_number": page_number,
497+
"trace_id": trace_id,
498+
},
499+
span_list_params.SpanListParams,
500+
),
483501
),
484502
cast_to=SpanListResponse,
485503
)

src/agentex/resources/states.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,8 @@ def list(
166166
self,
167167
*,
168168
agent_id: Optional[str] | Omit = omit,
169+
limit: int | Omit = omit,
170+
page_number: int | Omit = omit,
169171
task_id: Optional[str] | Omit = omit,
170172
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
171173
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -180,6 +182,10 @@ def list(
180182
Args:
181183
agent_id: Agent ID
182184
185+
limit: Limit
186+
187+
page_number: Page number
188+
183189
task_id: Task ID
184190
185191
extra_headers: Send extra headers
@@ -200,6 +206,8 @@ def list(
200206
query=maybe_transform(
201207
{
202208
"agent_id": agent_id,
209+
"limit": limit,
210+
"page_number": page_number,
203211
"task_id": task_id,
204212
},
205213
state_list_params.StateListParams,
@@ -384,6 +392,8 @@ async def list(
384392
self,
385393
*,
386394
agent_id: Optional[str] | Omit = omit,
395+
limit: int | Omit = omit,
396+
page_number: int | Omit = omit,
387397
task_id: Optional[str] | Omit = omit,
388398
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
389399
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -398,6 +408,10 @@ async def list(
398408
Args:
399409
agent_id: Agent ID
400410
411+
limit: Limit
412+
413+
page_number: Page number
414+
401415
task_id: Task ID
402416
403417
extra_headers: Send extra headers
@@ -418,6 +432,8 @@ async def list(
418432
query=await async_maybe_transform(
419433
{
420434
"agent_id": agent_id,
435+
"limit": limit,
436+
"page_number": page_number,
421437
"task_id": task_id,
422438
},
423439
state_list_params.StateListParams,

src/agentex/resources/tasks.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,8 @@ def list(
8484
*,
8585
agent_id: Optional[str] | Omit = omit,
8686
agent_name: Optional[str] | Omit = omit,
87+
limit: int | Omit = omit,
88+
page_number: int | Omit = omit,
8789
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
8890
# The extra values given here take precedence over values defined on the client or passed to this method.
8991
extra_headers: Headers | None = None,
@@ -114,6 +116,8 @@ def list(
114116
{
115117
"agent_id": agent_id,
116118
"agent_name": agent_name,
119+
"limit": limit,
120+
"page_number": page_number,
117121
},
118122
task_list_params.TaskListParams,
119123
),
@@ -349,6 +353,8 @@ async def list(
349353
*,
350354
agent_id: Optional[str] | Omit = omit,
351355
agent_name: Optional[str] | Omit = omit,
356+
limit: int | Omit = omit,
357+
page_number: int | Omit = omit,
352358
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
353359
# The extra values given here take precedence over values defined on the client or passed to this method.
354360
extra_headers: Headers | None = None,
@@ -379,6 +385,8 @@ async def list(
379385
{
380386
"agent_id": agent_id,
381387
"agent_name": agent_name,
388+
"limit": limit,
389+
"page_number": page_number,
382390
},
383391
task_list_params.TaskListParams,
384392
),

src/agentex/resources/tracker.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,8 @@ def list(
131131
self,
132132
*,
133133
agent_id: Optional[str] | Omit = omit,
134+
limit: int | Omit = omit,
135+
page_number: int | Omit = omit,
134136
task_id: Optional[str] | Omit = omit,
135137
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
136138
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -145,6 +147,10 @@ def list(
145147
Args:
146148
agent_id: Agent ID
147149
150+
limit: Limit
151+
152+
page_number: Page number
153+
148154
task_id: Task ID
149155
150156
extra_headers: Send extra headers
@@ -165,6 +171,8 @@ def list(
165171
query=maybe_transform(
166172
{
167173
"agent_id": agent_id,
174+
"limit": limit,
175+
"page_number": page_number,
168176
"task_id": task_id,
169177
},
170178
tracker_list_params.TrackerListParams,
@@ -281,6 +289,8 @@ async def list(
281289
self,
282290
*,
283291
agent_id: Optional[str] | Omit = omit,
292+
limit: int | Omit = omit,
293+
page_number: int | Omit = omit,
284294
task_id: Optional[str] | Omit = omit,
285295
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
286296
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -295,6 +305,10 @@ async def list(
295305
Args:
296306
agent_id: Agent ID
297307
308+
limit: Limit
309+
310+
page_number: Page number
311+
298312
task_id: Task ID
299313
300314
extra_headers: Send extra headers
@@ -315,6 +329,8 @@ async def list(
315329
query=await async_maybe_transform(
316330
{
317331
"agent_id": agent_id,
332+
"limit": limit,
333+
"page_number": page_number,
318334
"task_id": task_id,
319335
},
320336
tracker_list_params.TrackerListParams,

src/agentex/types/agent_list_params.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,11 @@
99

1010

1111
class AgentListParams(TypedDict, total=False):
12+
limit: int
13+
"""Limit"""
14+
15+
page_number: int
16+
"""Page number"""
17+
1218
task_id: Optional[str]
1319
"""Task ID"""

src/agentex/types/message_list_params.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
from __future__ import annotations
44

5-
from typing import Optional
65
from typing_extensions import Required, TypedDict
76

87
__all__ = ["MessageListParams"]
@@ -12,4 +11,6 @@ class MessageListParams(TypedDict, total=False):
1211
task_id: Required[str]
1312
"""The task ID"""
1413

15-
limit: Optional[int]
14+
limit: int
15+
16+
page_number: int

src/agentex/types/span_list_params.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,8 @@
99

1010

1111
class SpanListParams(TypedDict, total=False):
12+
limit: int
13+
14+
page_number: int
15+
1216
trace_id: Optional[str]

0 commit comments

Comments
 (0)