Skip to content

Commit 53b5c36

Browse files
feat(api): add schedule, checkpoints, and deployment endpoints
1 parent 795424c commit 53b5c36

40 files changed

Lines changed: 5569 additions & 29 deletions

.stats.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
configured_endpoints: 45
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp/agentex-sdk-9ab4b375245291b8e37dd1cbc054fa65f17b7e7db28729126ea9f1289dc99214.yml
1+
configured_endpoints: 63
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp/agentex-sdk-5400cbfee7eab6b5ace17d760b4997fd68f8d169470ab5040cf268a185250a0b.yml
33
openapi_spec_hash: d31d828c46635cbc20165177c7187a70
4-
config_hash: fb079ef7936611b032568661b8165f19
4+
config_hash: 81470e0e689fe06fa3e013ec01a7f84f

api.md

Lines changed: 70 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -30,13 +30,60 @@ from agentex.types import (
3030

3131
Methods:
3232

33-
- <code title="get /agents/{agent_id}">client.agents.<a href="./src/agentex/resources/agents.py">retrieve</a>(agent_id) -> <a href="./src/agentex/types/agent.py">Agent</a></code>
34-
- <code title="get /agents">client.agents.<a href="./src/agentex/resources/agents.py">list</a>(\*\*<a href="src/agentex/types/agent_list_params.py">params</a>) -> <a href="./src/agentex/types/agent_list_response.py">AgentListResponse</a></code>
35-
- <code title="delete /agents/{agent_id}">client.agents.<a href="./src/agentex/resources/agents.py">delete</a>(agent_id) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
36-
- <code title="delete /agents/name/{agent_name}">client.agents.<a href="./src/agentex/resources/agents.py">delete_by_name</a>(agent_name) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
37-
- <code title="get /agents/name/{agent_name}">client.agents.<a href="./src/agentex/resources/agents.py">retrieve_by_name</a>(agent_name) -> <a href="./src/agentex/types/agent.py">Agent</a></code>
38-
- <code title="post /agents/{agent_id}/rpc">client.agents.<a href="./src/agentex/resources/agents.py">rpc</a>(agent_id, \*\*<a href="src/agentex/types/agent_rpc_params.py">params</a>) -> <a href="./src/agentex/types/agent_rpc_response.py">AgentRpcResponse</a></code>
39-
- <code title="post /agents/name/{agent_name}/rpc">client.agents.<a href="./src/agentex/resources/agents.py">rpc_by_name</a>(agent_name, \*\*<a href="src/agentex/types/agent_rpc_by_name_params.py">params</a>) -> <a href="./src/agentex/types/agent_rpc_response.py">AgentRpcResponse</a></code>
33+
- <code title="get /agents/{agent_id}">client.agents.<a href="./src/agentex/resources/agents/agents.py">retrieve</a>(agent_id) -> <a href="./src/agentex/types/agent.py">Agent</a></code>
34+
- <code title="get /agents">client.agents.<a href="./src/agentex/resources/agents/agents.py">list</a>(\*\*<a href="src/agentex/types/agent_list_params.py">params</a>) -> <a href="./src/agentex/types/agent_list_response.py">AgentListResponse</a></code>
35+
- <code title="delete /agents/{agent_id}">client.agents.<a href="./src/agentex/resources/agents/agents.py">delete</a>(agent_id) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
36+
- <code title="delete /agents/name/{agent_name}">client.agents.<a href="./src/agentex/resources/agents/agents.py">delete_by_name</a>(agent_name) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
37+
- <code title="get /agents/name/{agent_name}">client.agents.<a href="./src/agentex/resources/agents/agents.py">retrieve_by_name</a>(agent_name) -> <a href="./src/agentex/types/agent.py">Agent</a></code>
38+
- <code title="post /agents/{agent_id}/rpc">client.agents.<a href="./src/agentex/resources/agents/agents.py">rpc</a>(agent_id, \*\*<a href="src/agentex/types/agent_rpc_params.py">params</a>) -> <a href="./src/agentex/types/agent_rpc_response.py">AgentRpcResponse</a></code>
39+
- <code title="post /agents/name/{agent_name}/rpc">client.agents.<a href="./src/agentex/resources/agents/agents.py">rpc_by_name</a>(agent_name, \*\*<a href="src/agentex/types/agent_rpc_by_name_params.py">params</a>) -> <a href="./src/agentex/types/agent_rpc_response.py">AgentRpcResponse</a></code>
40+
41+
## Deployments
42+
43+
Types:
44+
45+
```python
46+
from agentex.types.agents import (
47+
DeploymentCreateResponse,
48+
DeploymentRetrieveResponse,
49+
DeploymentListResponse,
50+
DeploymentPromoteResponse,
51+
)
52+
```
53+
54+
Methods:
55+
56+
- <code title="post /agents/{agent_id}/deployments">client.agents.deployments.<a href="./src/agentex/resources/agents/deployments.py">create</a>(agent_id, \*\*<a href="src/agentex/types/agents/deployment_create_params.py">params</a>) -> <a href="./src/agentex/types/agents/deployment_create_response.py">DeploymentCreateResponse</a></code>
57+
- <code title="get /agents/{agent_id}/deployments/{deployment_id}">client.agents.deployments.<a href="./src/agentex/resources/agents/deployments.py">retrieve</a>(deployment_id, \*, agent_id) -> <a href="./src/agentex/types/agents/deployment_retrieve_response.py">DeploymentRetrieveResponse</a></code>
58+
- <code title="get /agents/{agent_id}/deployments">client.agents.deployments.<a href="./src/agentex/resources/agents/deployments.py">list</a>(agent_id, \*\*<a href="src/agentex/types/agents/deployment_list_params.py">params</a>) -> <a href="./src/agentex/types/agents/deployment_list_response.py">DeploymentListResponse</a></code>
59+
- <code title="delete /agents/{agent_id}/deployments/{deployment_id}">client.agents.deployments.<a href="./src/agentex/resources/agents/deployments.py">delete</a>(deployment_id, \*, agent_id) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
60+
- <code title="post /agents/{agent_id}/deployments/{deployment_id}/rpc">client.agents.deployments.<a href="./src/agentex/resources/agents/deployments.py">preview_rpc</a>(deployment_id, \*, agent_id, \*\*<a href="src/agentex/types/agents/deployment_preview_rpc_params.py">params</a>) -> <a href="./src/agentex/types/agent_rpc_response.py">AgentRpcResponse</a></code>
61+
- <code title="post /agents/{agent_id}/deployments/{deployment_id}/promote">client.agents.deployments.<a href="./src/agentex/resources/agents/deployments.py">promote</a>(deployment_id, \*, agent_id) -> <a href="./src/agentex/types/agents/deployment_promote_response.py">DeploymentPromoteResponse</a></code>
62+
63+
## Schedules
64+
65+
Types:
66+
67+
```python
68+
from agentex.types.agents import (
69+
ScheduleCreateResponse,
70+
ScheduleRetrieveResponse,
71+
ScheduleListResponse,
72+
SchedulePauseResponse,
73+
ScheduleTriggerResponse,
74+
ScheduleUnpauseResponse,
75+
)
76+
```
77+
78+
Methods:
79+
80+
- <code title="post /agents/{agent_id}/schedules">client.agents.schedules.<a href="./src/agentex/resources/agents/schedules.py">create</a>(agent_id, \*\*<a href="src/agentex/types/agents/schedule_create_params.py">params</a>) -> <a href="./src/agentex/types/agents/schedule_create_response.py">ScheduleCreateResponse</a></code>
81+
- <code title="get /agents/{agent_id}/schedules/{schedule_name}">client.agents.schedules.<a href="./src/agentex/resources/agents/schedules.py">retrieve</a>(schedule_name, \*, agent_id) -> <a href="./src/agentex/types/agents/schedule_retrieve_response.py">ScheduleRetrieveResponse</a></code>
82+
- <code title="get /agents/{agent_id}/schedules">client.agents.schedules.<a href="./src/agentex/resources/agents/schedules.py">list</a>(agent_id, \*\*<a href="src/agentex/types/agents/schedule_list_params.py">params</a>) -> <a href="./src/agentex/types/agents/schedule_list_response.py">ScheduleListResponse</a></code>
83+
- <code title="delete /agents/{agent_id}/schedules/{schedule_name}">client.agents.schedules.<a href="./src/agentex/resources/agents/schedules.py">delete</a>(schedule_name, \*, agent_id) -> <a href="./src/agentex/types/shared/delete_response.py">DeleteResponse</a></code>
84+
- <code title="post /agents/{agent_id}/schedules/{schedule_name}/pause">client.agents.schedules.<a href="./src/agentex/resources/agents/schedules.py">pause</a>(schedule_name, \*, agent_id, \*\*<a href="src/agentex/types/agents/schedule_pause_params.py">params</a>) -> <a href="./src/agentex/types/agents/schedule_pause_response.py">SchedulePauseResponse</a></code>
85+
- <code title="post /agents/{agent_id}/schedules/{schedule_name}/trigger">client.agents.schedules.<a href="./src/agentex/resources/agents/schedules.py">trigger</a>(schedule_name, \*, agent_id) -> <a href="./src/agentex/types/agents/schedule_trigger_response.py">ScheduleTriggerResponse</a></code>
86+
- <code title="post /agents/{agent_id}/schedules/{schedule_name}/unpause">client.agents.schedules.<a href="./src/agentex/resources/agents/schedules.py">unpause</a>(schedule_name, \*, agent_id, \*\*<a href="src/agentex/types/agents/schedule_unpause_params.py">params</a>) -> <a href="./src/agentex/types/agents/schedule_unpause_response.py">ScheduleUnpauseResponse</a></code>
4087

4188
# Tasks
4289

@@ -181,3 +228,19 @@ Methods:
181228

182229
- <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>
183230
- <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>
231+
232+
# Checkpoints
233+
234+
Types:
235+
236+
```python
237+
from agentex.types import CheckpointListResponse, CheckpointGetTupleResponse, CheckpointPutResponse
238+
```
239+
240+
Methods:
241+
242+
- <code title="post /checkpoints/list">client.checkpoints.<a href="./src/agentex/resources/checkpoints.py">list</a>(\*\*<a href="src/agentex/types/checkpoint_list_params.py">params</a>) -> <a href="./src/agentex/types/checkpoint_list_response.py">CheckpointListResponse</a></code>
243+
- <code title="post /checkpoints/delete-thread">client.checkpoints.<a href="./src/agentex/resources/checkpoints.py">delete_thread</a>(\*\*<a href="src/agentex/types/checkpoint_delete_thread_params.py">params</a>) -> None</code>
244+
- <code title="post /checkpoints/get-tuple">client.checkpoints.<a href="./src/agentex/resources/checkpoints.py">get_tuple</a>(\*\*<a href="src/agentex/types/checkpoint_get_tuple_params.py">params</a>) -> <a href="./src/agentex/types/checkpoint_get_tuple_response.py">Optional[CheckpointGetTupleResponse]</a></code>
245+
- <code title="post /checkpoints/put">client.checkpoints.<a href="./src/agentex/resources/checkpoints.py">put</a>(\*\*<a href="src/agentex/types/checkpoint_put_params.py">params</a>) -> <a href="./src/agentex/types/checkpoint_put_response.py">CheckpointPutResponse</a></code>
246+
- <code title="post /checkpoints/put-writes">client.checkpoints.<a href="./src/agentex/resources/checkpoints.py">put_writes</a>(\*\*<a href="src/agentex/types/checkpoint_put_writes_params.py">params</a>) -> None</code>

src/agentex/_client.py

Lines changed: 39 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,13 +35,14 @@
3535
)
3636

3737
if TYPE_CHECKING:
38-
from .resources import spans, tasks, agents, events, states, tracker, messages, deployment_history
38+
from .resources import spans, tasks, agents, events, states, tracker, messages, checkpoints, deployment_history
3939
from .resources.spans import SpansResource, AsyncSpansResource
4040
from .resources.tasks import TasksResource, AsyncTasksResource
41-
from .resources.agents import AgentsResource, AsyncAgentsResource
4241
from .resources.events import EventsResource, AsyncEventsResource
4342
from .resources.states import StatesResource, AsyncStatesResource
4443
from .resources.tracker import TrackerResource, AsyncTrackerResource
44+
from .resources.checkpoints import CheckpointsResource, AsyncCheckpointsResource
45+
from .resources.agents.agents import AgentsResource, AsyncAgentsResource
4546
from .resources.messages.messages import MessagesResource, AsyncMessagesResource
4647
from .resources.deployment_history import DeploymentHistoryResource, AsyncDeploymentHistoryResource
4748

@@ -195,6 +196,12 @@ def deployment_history(self) -> DeploymentHistoryResource:
195196

196197
return DeploymentHistoryResource(self)
197198

199+
@cached_property
200+
def checkpoints(self) -> CheckpointsResource:
201+
from .resources.checkpoints import CheckpointsResource
202+
203+
return CheckpointsResource(self)
204+
198205
@cached_property
199206
def with_raw_response(self) -> AgentexWithRawResponse:
200207
return AgentexWithRawResponse(self)
@@ -444,6 +451,12 @@ def deployment_history(self) -> AsyncDeploymentHistoryResource:
444451

445452
return AsyncDeploymentHistoryResource(self)
446453

454+
@cached_property
455+
def checkpoints(self) -> AsyncCheckpointsResource:
456+
from .resources.checkpoints import AsyncCheckpointsResource
457+
458+
return AsyncCheckpointsResource(self)
459+
447460
@cached_property
448461
def with_raw_response(self) -> AsyncAgentexWithRawResponse:
449462
return AsyncAgentexWithRawResponse(self)
@@ -615,6 +628,12 @@ def deployment_history(self) -> deployment_history.DeploymentHistoryResourceWith
615628

616629
return DeploymentHistoryResourceWithRawResponse(self._client.deployment_history)
617630

631+
@cached_property
632+
def checkpoints(self) -> checkpoints.CheckpointsResourceWithRawResponse:
633+
from .resources.checkpoints import CheckpointsResourceWithRawResponse
634+
635+
return CheckpointsResourceWithRawResponse(self._client.checkpoints)
636+
618637

619638
class AsyncAgentexWithRawResponse:
620639
_client: AsyncAgentex
@@ -670,6 +689,12 @@ def deployment_history(self) -> deployment_history.AsyncDeploymentHistoryResourc
670689

671690
return AsyncDeploymentHistoryResourceWithRawResponse(self._client.deployment_history)
672691

692+
@cached_property
693+
def checkpoints(self) -> checkpoints.AsyncCheckpointsResourceWithRawResponse:
694+
from .resources.checkpoints import AsyncCheckpointsResourceWithRawResponse
695+
696+
return AsyncCheckpointsResourceWithRawResponse(self._client.checkpoints)
697+
673698

674699
class AgentexWithStreamedResponse:
675700
_client: Agentex
@@ -725,6 +750,12 @@ def deployment_history(self) -> deployment_history.DeploymentHistoryResourceWith
725750

726751
return DeploymentHistoryResourceWithStreamingResponse(self._client.deployment_history)
727752

753+
@cached_property
754+
def checkpoints(self) -> checkpoints.CheckpointsResourceWithStreamingResponse:
755+
from .resources.checkpoints import CheckpointsResourceWithStreamingResponse
756+
757+
return CheckpointsResourceWithStreamingResponse(self._client.checkpoints)
758+
728759

729760
class AsyncAgentexWithStreamedResponse:
730761
_client: AsyncAgentex
@@ -780,6 +811,12 @@ def deployment_history(self) -> deployment_history.AsyncDeploymentHistoryResourc
780811

781812
return AsyncDeploymentHistoryResourceWithStreamingResponse(self._client.deployment_history)
782813

814+
@cached_property
815+
def checkpoints(self) -> checkpoints.AsyncCheckpointsResourceWithStreamingResponse:
816+
from .resources.checkpoints import AsyncCheckpointsResourceWithStreamingResponse
817+
818+
return AsyncCheckpointsResourceWithStreamingResponse(self._client.checkpoints)
819+
783820

784821
Client = Agentex
785822

src/agentex/resources/__init__.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,14 @@
5656
MessagesResourceWithStreamingResponse,
5757
AsyncMessagesResourceWithStreamingResponse,
5858
)
59+
from .checkpoints import (
60+
CheckpointsResource,
61+
AsyncCheckpointsResource,
62+
CheckpointsResourceWithRawResponse,
63+
AsyncCheckpointsResourceWithRawResponse,
64+
CheckpointsResourceWithStreamingResponse,
65+
AsyncCheckpointsResourceWithStreamingResponse,
66+
)
5967
from .deployment_history import (
6068
DeploymentHistoryResource,
6169
AsyncDeploymentHistoryResource,
@@ -114,4 +122,10 @@
114122
"AsyncDeploymentHistoryResourceWithRawResponse",
115123
"DeploymentHistoryResourceWithStreamingResponse",
116124
"AsyncDeploymentHistoryResourceWithStreamingResponse",
125+
"CheckpointsResource",
126+
"AsyncCheckpointsResource",
127+
"CheckpointsResourceWithRawResponse",
128+
"AsyncCheckpointsResourceWithRawResponse",
129+
"CheckpointsResourceWithStreamingResponse",
130+
"AsyncCheckpointsResourceWithStreamingResponse",
117131
]
Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2+
3+
from .agents import (
4+
AgentsResource,
5+
AsyncAgentsResource,
6+
AgentsResourceWithRawResponse,
7+
AsyncAgentsResourceWithRawResponse,
8+
AgentsResourceWithStreamingResponse,
9+
AsyncAgentsResourceWithStreamingResponse,
10+
)
11+
from .schedules import (
12+
SchedulesResource,
13+
AsyncSchedulesResource,
14+
SchedulesResourceWithRawResponse,
15+
AsyncSchedulesResourceWithRawResponse,
16+
SchedulesResourceWithStreamingResponse,
17+
AsyncSchedulesResourceWithStreamingResponse,
18+
)
19+
from .deployments import (
20+
DeploymentsResource,
21+
AsyncDeploymentsResource,
22+
DeploymentsResourceWithRawResponse,
23+
AsyncDeploymentsResourceWithRawResponse,
24+
DeploymentsResourceWithStreamingResponse,
25+
AsyncDeploymentsResourceWithStreamingResponse,
26+
)
27+
28+
__all__ = [
29+
"DeploymentsResource",
30+
"AsyncDeploymentsResource",
31+
"DeploymentsResourceWithRawResponse",
32+
"AsyncDeploymentsResourceWithRawResponse",
33+
"DeploymentsResourceWithStreamingResponse",
34+
"AsyncDeploymentsResourceWithStreamingResponse",
35+
"SchedulesResource",
36+
"AsyncSchedulesResource",
37+
"SchedulesResourceWithRawResponse",
38+
"AsyncSchedulesResourceWithRawResponse",
39+
"SchedulesResourceWithStreamingResponse",
40+
"AsyncSchedulesResourceWithStreamingResponse",
41+
"AgentsResource",
42+
"AsyncAgentsResource",
43+
"AgentsResourceWithRawResponse",
44+
"AsyncAgentsResourceWithRawResponse",
45+
"AgentsResourceWithStreamingResponse",
46+
"AsyncAgentsResourceWithStreamingResponse",
47+
]

0 commit comments

Comments
 (0)