Skip to content

Commit d699e24

Browse files
feat(api): add task state update methods
1 parent 543c83b commit d699e24

24 files changed

+1915
-231
lines changed

.stats.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
configured_endpoints: 35
1+
configured_endpoints: 45
22
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-5fa3cb3c867281c804913c7c3e6d2143b5606d4924d42119f4b2b246f33e3db3.yml
33
openapi_spec_hash: 8ec711692f3ed7cd34a7a3b9d3e33f7c
4-
config_hash: 32eb65911c08ac84d117cecdf2759869
4+
config_hash: d65d8b7a05f302e164ef9ba7fe72b277

api.md

Lines changed: 21 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,13 @@ from agentex.types import (
1515
AgentRpcRequest,
1616
AgentRpcResponse,
1717
AgentRpcResult,
18+
CancelTaskRequest,
19+
CreateTaskRequest,
1820
DataDelta,
1921
ReasoningContentDelta,
2022
ReasoningSummaryDelta,
23+
SendEventRequest,
24+
SendMessageRequest,
2125
TaskMessageContent,
2226
TaskMessageDelta,
2327
TaskMessageUpdate,
@@ -43,18 +47,32 @@ Methods:
4347
Types:
4448

4549
```python
46-
from agentex.types import Task, TaskRetrieveResponse, TaskListResponse, TaskRetrieveByNameResponse
50+
from agentex.types import (
51+
Task,
52+
TaskRetrieveResponse,
53+
TaskListResponse,
54+
TaskQueryWorkflowResponse,
55+
TaskRetrieveByNameResponse,
56+
)
4757
```
4858

4959
Methods:
5060

5161
- <code title="get /tasks/{task_id}">client.tasks.<a href="./src/agentex/resources/tasks.py">retrieve</a>(task_id, \*\*<a href="src/agentex/types/task_retrieve_params.py">params</a>) -> <a href="./src/agentex/types/task_retrieve_response.py">TaskRetrieveResponse</a></code>
5262
- <code title="get /tasks">client.tasks.<a href="./src/agentex/resources/tasks.py">list</a>(\*\*<a href="src/agentex/types/task_list_params.py">params</a>) -> <a href="./src/agentex/types/task_list_response.py">TaskListResponse</a></code>
5363
- <code title="delete /tasks/{task_id}">client.tasks.<a href="./src/agentex/resources/tasks.py">delete</a>(task_id) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
64+
- <code title="post /tasks/{task_id}/cancel">client.tasks.<a href="./src/agentex/resources/tasks.py">cancel</a>(task_id, \*\*<a href="src/agentex/types/task_cancel_params.py">params</a>) -> <a href="./src/agentex/types/task.py">Task</a></code>
65+
- <code title="post /tasks/{task_id}/complete">client.tasks.<a href="./src/agentex/resources/tasks.py">complete</a>(task_id, \*\*<a href="src/agentex/types/task_complete_params.py">params</a>) -> <a href="./src/agentex/types/task.py">Task</a></code>
5466
- <code title="delete /tasks/name/{task_name}">client.tasks.<a href="./src/agentex/resources/tasks.py">delete_by_name</a>(task_name) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
67+
- <code title="post /tasks/{task_id}/fail">client.tasks.<a href="./src/agentex/resources/tasks.py">fail</a>(task_id, \*\*<a href="src/agentex/types/task_fail_params.py">params</a>) -> <a href="./src/agentex/types/task.py">Task</a></code>
68+
- <code title="get /tasks/{task_id}/query/{query_name}">client.tasks.<a href="./src/agentex/resources/tasks.py">query_workflow</a>(query_name, \*, task_id) -> <a href="./src/agentex/types/task_query_workflow_response.py">TaskQueryWorkflowResponse</a></code>
5569
- <code title="get /tasks/name/{task_name}">client.tasks.<a href="./src/agentex/resources/tasks.py">retrieve_by_name</a>(task_name, \*\*<a href="src/agentex/types/task_retrieve_by_name_params.py">params</a>) -> <a href="./src/agentex/types/task_retrieve_by_name_response.py">TaskRetrieveByNameResponse</a></code>
5670
- <code title="get /tasks/{task_id}/stream">client.tasks.<a href="./src/agentex/resources/tasks.py">stream_events</a>(task_id) -> object</code>
5771
- <code title="get /tasks/name/{task_name}/stream">client.tasks.<a href="./src/agentex/resources/tasks.py">stream_events_by_name</a>(task_name) -> object</code>
72+
- <code title="post /tasks/{task_id}/terminate">client.tasks.<a href="./src/agentex/resources/tasks.py">terminate</a>(task_id, \*\*<a href="src/agentex/types/task_terminate_params.py">params</a>) -> <a href="./src/agentex/types/task.py">Task</a></code>
73+
- <code title="post /tasks/{task_id}/timeout">client.tasks.<a href="./src/agentex/resources/tasks.py">timeout</a>(task_id, \*\*<a href="src/agentex/types/task_timeout_params.py">params</a>) -> <a href="./src/agentex/types/task.py">Task</a></code>
74+
- <code title="put /tasks/{task_id}">client.tasks.<a href="./src/agentex/resources/tasks.py">update_by_id</a>(task_id, \*\*<a href="src/agentex/types/task_update_by_id_params.py">params</a>) -> <a href="./src/agentex/types/task.py">Task</a></code>
75+
- <code title="put /tasks/name/{task_name}">client.tasks.<a href="./src/agentex/resources/tasks.py">update_by_name</a>(task_name, \*\*<a href="src/agentex/types/task_update_by_name_params.py">params</a>) -> <a href="./src/agentex/types/task.py">Task</a></code>
5876

5977
# Messages
6078

@@ -160,10 +178,10 @@ Methods:
160178
Types:
161179

162180
```python
163-
from agentex.types import DeploymentHistory, DeploymentHistoryListResponse
181+
from agentex.types import DeploymentHistoryRetrieveResponse, DeploymentHistoryListResponse
164182
```
165183

166184
Methods:
167185

168-
- <code title="get /deployment-history/{deployment_id}">client.deployment_history.<a href="./src/agentex/resources/deployment_history.py">retrieve</a>(deployment_id) -> <a href="./src/agentex/types/deployment_history.py">DeploymentHistory</a></code>
186+
- <code title="get /deployment-history/{deployment_id}">client.deployment_history.<a href="./src/agentex/resources/deployment_history.py">retrieve</a>(deployment_id) -> <a href="./src/agentex/types/deployment_history_retrieve_response.py">DeploymentHistoryRetrieveResponse</a></code>
169187
- <code title="get /deployment-history">client.deployment_history.<a href="./src/agentex/resources/deployment_history.py">list</a>(\*\*<a href="src/agentex/types/deployment_history_list_params.py">params</a>) -> <a href="./src/agentex/types/deployment_history_list_response.py">DeploymentHistoryListResponse</a></code>

src/agentex/resources/deployment_history.py

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
from ..types import deployment_history_list_params
1010
from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
11-
from .._utils import maybe_transform, async_maybe_transform
11+
from .._utils import path_template, maybe_transform, async_maybe_transform
1212
from .._compat import cached_property
1313
from .._resource import SyncAPIResource, AsyncAPIResource
1414
from .._response import (
@@ -18,8 +18,8 @@
1818
async_to_streamed_response_wrapper,
1919
)
2020
from .._base_client import make_request_options
21-
from ..types.deployment_history import DeploymentHistory
2221
from ..types.deployment_history_list_response import DeploymentHistoryListResponse
22+
from ..types.deployment_history_retrieve_response import DeploymentHistoryRetrieveResponse
2323

2424
__all__ = ["DeploymentHistoryResource", "AsyncDeploymentHistoryResource"]
2525

@@ -54,7 +54,7 @@ def retrieve(
5454
extra_query: Query | None = None,
5555
extra_body: Body | None = None,
5656
timeout: float | httpx.Timeout | None | NotGiven = not_given,
57-
) -> DeploymentHistory:
57+
) -> DeploymentHistoryRetrieveResponse:
5858
"""
5959
Get a deployment record by its unique ID.
6060
@@ -70,11 +70,11 @@ def retrieve(
7070
if not deployment_id:
7171
raise ValueError(f"Expected a non-empty value for `deployment_id` but received {deployment_id!r}")
7272
return self._get(
73-
f"/deployment-history/{deployment_id}",
73+
path_template("/deployment-history/{deployment_id}", deployment_id=deployment_id),
7474
options=make_request_options(
7575
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
7676
),
77-
cast_to=DeploymentHistory,
77+
cast_to=DeploymentHistoryRetrieveResponse,
7878
)
7979

8080
def list(
@@ -83,6 +83,8 @@ def list(
8383
agent_id: Optional[str] | Omit = omit,
8484
agent_name: Optional[str] | Omit = omit,
8585
limit: int | Omit = omit,
86+
order_by: Optional[str] | Omit = omit,
87+
order_direction: str | Omit = omit,
8688
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.
@@ -115,6 +117,8 @@ def list(
115117
"agent_id": agent_id,
116118
"agent_name": agent_name,
117119
"limit": limit,
120+
"order_by": order_by,
121+
"order_direction": order_direction,
118122
"page_number": page_number,
119123
},
120124
deployment_history_list_params.DeploymentHistoryListParams,
@@ -154,7 +158,7 @@ async def retrieve(
154158
extra_query: Query | None = None,
155159
extra_body: Body | None = None,
156160
timeout: float | httpx.Timeout | None | NotGiven = not_given,
157-
) -> DeploymentHistory:
161+
) -> DeploymentHistoryRetrieveResponse:
158162
"""
159163
Get a deployment record by its unique ID.
160164
@@ -170,11 +174,11 @@ async def retrieve(
170174
if not deployment_id:
171175
raise ValueError(f"Expected a non-empty value for `deployment_id` but received {deployment_id!r}")
172176
return await self._get(
173-
f"/deployment-history/{deployment_id}",
177+
path_template("/deployment-history/{deployment_id}", deployment_id=deployment_id),
174178
options=make_request_options(
175179
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
176180
),
177-
cast_to=DeploymentHistory,
181+
cast_to=DeploymentHistoryRetrieveResponse,
178182
)
179183

180184
async def list(
@@ -183,6 +187,8 @@ async def list(
183187
agent_id: Optional[str] | Omit = omit,
184188
agent_name: Optional[str] | Omit = omit,
185189
limit: int | Omit = omit,
190+
order_by: Optional[str] | Omit = omit,
191+
order_direction: str | Omit = omit,
186192
page_number: int | Omit = omit,
187193
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
188194
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -215,6 +221,8 @@ async def list(
215221
"agent_id": agent_id,
216222
"agent_name": agent_name,
217223
"limit": limit,
224+
"order_by": order_by,
225+
"order_direction": order_direction,
218226
"page_number": page_number,
219227
},
220228
deployment_history_list_params.DeploymentHistoryListParams,

0 commit comments

Comments
 (0)