feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 34
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-021b55c88964b7a5bfc9d692d32a52c6b0150445656d2407c4cb8e9dd1e5f100.yml
-openapi_spec_hash: ed92c0d5d6bed9cb5617f8a776ac42c9
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-62a09183a027c64707ad1b9d7a5ccc66c8abf3965e6075282cb5ab72f742a2b3.yml
+openapi_spec_hash: 47f43703822077a82e2edf82d4b7e0e5
 config_hash: 2e4b423af3db79ebd8170c401ea9093a

api.md

@@ -1,9 +1,30 @@
+# Shared Types
+
+```python
+from agentex.types import TaskMessageUpdate
+```
+
 # Agents
 
 Types:
 
 ```python
-from agentex.types import AcpType, Agent, AgentRpcRequest, AgentListResponse
+from agentex.types import (
+    AcpType,
+    Agent,
+    AgentRpcParams,
+    AgentRpcRequest,
+    AgentRpcResponse,
+    AgentRpcResult,
+    DataDelta,
+    TaskMessageContent,
+    TaskMessageDelta,
+    TaskMessageUpdate,
+    TextDelta,
+    ToolRequestDelta,
+    ToolResponseDelta,
+    AgentListResponse,
+)
 ```
 
 Methods:
@@ -13,23 +34,23 @@ Methods:
 - <code title="delete /agents/{agent_id}">client.agents.<a href="./src/agentex/resources/agents.py">delete</a>(agent_id) -> <a href="./src/agentex/types/agent.py">Agent</a></code>
 - <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/agent.py">Agent</a></code>
 - <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>
-- <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>) -> object</code>
-- <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>) -> object</code>
+- <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>
+- <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>
 
 # Tasks
 
 Types:
 
 ```python
-from agentex.types import Task, TaskListResponse
+from agentex.types import Task, TaskListResponse, TaskDeleteResponse, TaskDeleteByNameResponse
 ```
 
 Methods:
 
 - <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.py">Task</a></code>
 - <code title="get /tasks">client.tasks.<a href="./src/agentex/resources/tasks.py">list</a>() -> <a href="./src/agentex/types/task_list_response.py">TaskListResponse</a></code>
-- <code title="delete /tasks/{task_id}">client.tasks.<a href="./src/agentex/resources/tasks.py">delete</a>(task_id) -> <a href="./src/agentex/types/task.py">Task</a></code>
-- <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/task.py">Task</a></code>
+- <code title="delete /tasks/{task_id}">client.tasks.<a href="./src/agentex/resources/tasks.py">delete</a>(task_id) -> <a href="./src/agentex/types/task_delete_response.py">TaskDeleteResponse</a></code>
+- <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/task_delete_by_name_response.py">TaskDeleteByNameResponse</a></code>
 - <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.py">Task</a></code>
 - <code title="get /tasks/{task_id}/stream">client.tasks.<a href="./src/agentex/resources/tasks.py">stream_events</a>(task_id) -> object</code>
 - <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>
@@ -43,7 +64,6 @@ from agentex.types import (
     DataContent,
     MessageAuthor,
     MessageStyle,
-    StreamingStatus,
     TaskMessage,
     TextContent,
     ToolRequestContent,

src/agentex/resources/agents.py

@@ -7,7 +7,7 @@
 
 import httpx
 
-from ..types import agent_rpc_params, agent_list_params, agent_rpc_by_name_params
+from ..types import AgentRpcParams, agent_rpc_params, agent_list_params, agent_rpc_by_name_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -20,6 +20,8 @@
 )
 from ..types.agent import Agent
 from .._base_client import make_request_options
+from ..types.agent_rpc_params import AgentRpcParams
+from ..types.agent_rpc_response import AgentRpcResponse
 from ..types.agent_list_response import AgentListResponse
 
 __all__ = ["AgentsResource", "AsyncAgentsResource"]
@@ -219,7 +221,7 @@ def rpc(
         agent_id: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: agent_rpc_params.Params,
+        params: AgentRpcParams,
         id: Union[int, str, None] | NotGiven = NOT_GIVEN,
         jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -228,11 +230,13 @@ def rpc(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> object:
+    ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique ID.
 
         Args:
+          params: The parameters for the agent RPC request
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -257,15 +261,15 @@ def rpc(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=AgentRpcResponse,
         )
 
     def rpc_by_name(
         self,
         agent_name: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: agent_rpc_by_name_params.Params,
+        params: AgentRpcParams,
         id: Union[int, str, None] | NotGiven = NOT_GIVEN,
         jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -274,11 +278,13 @@ def rpc_by_name(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> object:
+    ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique name.
 
         Args:
+          params: The parameters for the agent RPC request
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -303,7 +309,7 @@ def rpc_by_name(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=AgentRpcResponse,
         )
 
 
@@ -501,7 +507,7 @@ async def rpc(
         agent_id: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: agent_rpc_params.Params,
+        params: AgentRpcParams,
         id: Union[int, str, None] | NotGiven = NOT_GIVEN,
         jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -510,11 +516,13 @@ async def rpc(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> object:
+    ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique ID.
 
         Args:
+          params: The parameters for the agent RPC request
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -539,15 +547,15 @@ async def rpc(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=AgentRpcResponse,
         )
 
     async def rpc_by_name(
         self,
         agent_name: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: agent_rpc_by_name_params.Params,
+        params: AgentRpcParams,
         id: Union[int, str, None] | NotGiven = NOT_GIVEN,
         jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -556,11 +564,13 @@ async def rpc_by_name(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> object:
+    ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique name.
 
         Args:
+          params: The parameters for the agent RPC request
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -585,7 +595,7 @@ async def rpc_by_name(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=AgentRpcResponse,
         )
 
 

src/agentex/resources/messages/batch.py

@@ -18,6 +18,7 @@
 )
 from ..._base_client import make_request_options
 from ...types.messages import batch_create_params, batch_update_params
+from ...types.task_message_content_param import TaskMessageContentParam
 from ...types.messages.batch_create_response import BatchCreateResponse
 from ...types.messages.batch_update_response import BatchUpdateResponse
 
@@ -47,7 +48,7 @@ def with_streaming_response(self) -> BatchResourceWithStreamingResponse:
     def create(
         self,
         *,
-        contents: Iterable[batch_create_params.Content],
+        contents: Iterable[TaskMessageContentParam],
         task_id: str,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -87,7 +88,7 @@ def update(
         self,
         *,
         task_id: str,
-        updates: Dict[str, batch_update_params.Updates],
+        updates: Dict[str, TaskMessageContentParam],
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -146,7 +147,7 @@ def with_streaming_response(self) -> AsyncBatchResourceWithStreamingResponse:
     async def create(
         self,
         *,
-        contents: Iterable[batch_create_params.Content],
+        contents: Iterable[TaskMessageContentParam],
         task_id: str,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -186,7 +187,7 @@ async def update(
         self,
         *,
         task_id: str,
-        updates: Dict[str, batch_update_params.Updates],
+        updates: Dict[str, TaskMessageContentParam],
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,

src/agentex/resources/messages/messages.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from typing import Optional
+from typing_extensions import Literal
 
 import httpx
 
@@ -14,7 +15,7 @@
     BatchResourceWithStreamingResponse,
     AsyncBatchResourceWithStreamingResponse,
 )
-from ...types import StreamingStatus, message_list_params, message_create_params, message_update_params
+from ...types import message_list_params, message_create_params, message_update_params
 from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
@@ -27,8 +28,8 @@
 )
 from ..._base_client import make_request_options
 from ...types.task_message import TaskMessage
-from ...types.streaming_status import StreamingStatus
 from ...types.message_list_response import MessageListResponse
+from ...types.task_message_content_param import TaskMessageContentParam
 
 __all__ = ["MessagesResource", "AsyncMessagesResource"]
 
@@ -60,9 +61,9 @@ def with_streaming_response(self) -> MessagesResourceWithStreamingResponse:
     def create(
         self,
         *,
-        content: message_create_params.Content,
+        content: TaskMessageContentParam,
         task_id: str,
-        streaming_status: Optional[StreamingStatus] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -135,9 +136,9 @@ def update(
         self,
         message_id: str,
         *,
-        content: message_update_params.Content,
+        content: TaskMessageContentParam,
         task_id: str,
-        streaming_status: Optional[StreamingStatus] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -245,9 +246,9 @@ def with_streaming_response(self) -> AsyncMessagesResourceWithStreamingResponse:
     async def create(
         self,
         *,
-        content: message_create_params.Content,
+        content: TaskMessageContentParam,
         task_id: str,
-        streaming_status: Optional[StreamingStatus] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -320,9 +321,9 @@ async def update(
         self,
         message_id: str,
         *,
-        content: message_update_params.Content,
+        content: TaskMessageContentParam,
         task_id: str,
-        streaming_status: Optional[StreamingStatus] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,