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-1d08fb2290b5310c91801d7575d356628d372fd5434e15d3b9cead48eadb893f.yml
-openapi_spec_hash: 216a0edbf4e1a3cde23329d4f385faed
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-021b55c88964b7a5bfc9d692d32a52c6b0150445656d2407c4cb8e9dd1e5f100.yml
+openapi_spec_hash: ed92c0d5d6bed9cb5617f8a776ac42c9
 config_hash: 7661726e3cccf9f6349179841153601d

api.md

@@ -3,22 +3,7 @@
 Types:
 
 ```python
-from agentex.types import (
-    AcpType,
-    Agent,
-    AgentRpcParams,
-    AgentRpcRequest,
-    AgentRpcResponse,
-    AgentRpcResult,
-    DataDelta,
-    TaskMessageContent,
-    TaskMessageDelta,
-    TaskMessageUpdate,
-    TextDelta,
-    ToolRequestDelta,
-    ToolResponseDelta,
-    AgentListResponse,
-)
+from agentex.types import AcpType, Agent, AgentRpcRequest, AgentListResponse
 ```
 
 Methods:
@@ -28,8 +13,8 @@ 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>) -> <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>
+- <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>
 
 # Tasks
 
@@ -58,6 +43,7 @@ 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 AgentRpcParams, agent_rpc_params, agent_list_params, agent_rpc_by_name_params
+from ..types import 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,8 +20,6 @@
 )
 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"]
@@ -221,7 +219,7 @@ def rpc(
         agent_id: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: AgentRpcParams,
+        params: agent_rpc_params.Params,
         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.
@@ -230,13 +228,11 @@ def rpc(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AgentRpcResponse:
+    ) -> object:
         """
         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
@@ -261,15 +257,15 @@ def rpc(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=AgentRpcResponse,
+            cast_to=object,
         )
 
     def rpc_by_name(
         self,
         agent_name: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: AgentRpcParams,
+        params: agent_rpc_by_name_params.Params,
         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.
@@ -278,13 +274,11 @@ def rpc_by_name(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AgentRpcResponse:
+    ) -> object:
         """
         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
@@ -309,7 +303,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=AgentRpcResponse,
+            cast_to=object,
         )
 
 
@@ -507,7 +501,7 @@ async def rpc(
         agent_id: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: AgentRpcParams,
+        params: agent_rpc_params.Params,
         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.
@@ -516,13 +510,11 @@ async def rpc(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AgentRpcResponse:
+    ) -> object:
         """
         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
@@ -547,15 +539,15 @@ async def rpc(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=AgentRpcResponse,
+            cast_to=object,
         )
 
     async def rpc_by_name(
         self,
         agent_name: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: AgentRpcParams,
+        params: agent_rpc_by_name_params.Params,
         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.
@@ -564,13 +556,11 @@ async def rpc_by_name(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AgentRpcResponse:
+    ) -> object:
         """
         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
@@ -595,7 +585,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=AgentRpcResponse,
+            cast_to=object,
         )
 
 

src/agentex/resources/messages/batch.py

@@ -18,7 +18,6 @@
 )
 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
 
@@ -48,7 +47,7 @@ def with_streaming_response(self) -> BatchResourceWithStreamingResponse:
     def create(
         self,
         *,
-        contents: Iterable[TaskMessageContentParam],
+        contents: Iterable[batch_create_params.Content],
         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.
@@ -88,7 +87,7 @@ def update(
         self,
         *,
         task_id: str,
-        updates: Dict[str, TaskMessageContentParam],
+        updates: Dict[str, batch_update_params.Updates],
         # 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,
@@ -147,7 +146,7 @@ def with_streaming_response(self) -> AsyncBatchResourceWithStreamingResponse:
     async def create(
         self,
         *,
-        contents: Iterable[TaskMessageContentParam],
+        contents: Iterable[batch_create_params.Content],
         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.
@@ -187,7 +186,7 @@ async def update(
         self,
         *,
         task_id: str,
-        updates: Dict[str, TaskMessageContentParam],
+        updates: Dict[str, batch_update_params.Updates],
         # 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,7 +3,6 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import Literal
 
 import httpx
 
@@ -15,7 +14,7 @@
     BatchResourceWithStreamingResponse,
     AsyncBatchResourceWithStreamingResponse,
 )
-from ...types import message_list_params, message_create_params, message_update_params
+from ...types import StreamingStatus, 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
@@ -28,8 +27,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"]
 
@@ -61,9 +60,9 @@ def with_streaming_response(self) -> MessagesResourceWithStreamingResponse:
     def create(
         self,
         *,
-        content: TaskMessageContentParam,
+        content: message_create_params.Content,
         task_id: str,
-        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[StreamingStatus] | 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,
@@ -136,9 +135,9 @@ def update(
         self,
         message_id: str,
         *,
-        content: TaskMessageContentParam,
+        content: message_update_params.Content,
         task_id: str,
-        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[StreamingStatus] | 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,
@@ -246,9 +245,9 @@ def with_streaming_response(self) -> AsyncMessagesResourceWithStreamingResponse:
     async def create(
         self,
         *,
-        content: TaskMessageContentParam,
+        content: message_create_params.Content,
         task_id: str,
-        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[StreamingStatus] | 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,
@@ -321,9 +320,9 @@ async def update(
         self,
         message_id: str,
         *,
-        content: TaskMessageContentParam,
+        content: message_update_params.Content,
         task_id: str,
-        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[StreamingStatus] | 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,

src/agentex/types/__init__.py

@@ -8,38 +8,31 @@
 from .event import Event as Event
 from .state import State as State
 from .acp_type import AcpType as AcpType
-from .data_delta import DataDelta as DataDelta
-from .text_delta import TextDelta as TextDelta
 from .data_content import DataContent as DataContent
 from .task_message import TaskMessage as TaskMessage
 from .text_content import TextContent as TextContent
 from .message_style import MessageStyle as MessageStyle
 from .message_author import MessageAuthor as MessageAuthor
 from .agent_rpc_params import AgentRpcParams as AgentRpcParams
-from .agent_rpc_result import AgentRpcResult as AgentRpcResult
 from .span_list_params import SpanListParams as SpanListParams
+from .streaming_status import StreamingStatus as StreamingStatus
 from .agent_list_params import AgentListParams as AgentListParams
 from .event_list_params import EventListParams as EventListParams
 from .state_list_params import StateListParams as StateListParams
-from .agent_rpc_response import AgentRpcResponse as AgentRpcResponse
 from .agent_task_tracker import AgentTaskTracker as AgentTaskTracker
 from .data_content_param import DataContentParam as DataContentParam
 from .span_create_params import SpanCreateParams as SpanCreateParams
 from .span_list_response import SpanListResponse as SpanListResponse
 from .span_update_params import SpanUpdateParams as SpanUpdateParams
 from .task_list_response import TaskListResponse as TaskListResponse
-from .task_message_delta import TaskMessageDelta as TaskMessageDelta
 from .text_content_param import TextContentParam as TextContentParam
-from .tool_request_delta import ToolRequestDelta as ToolRequestDelta
 from .agent_list_response import AgentListResponse as AgentListResponse
 from .event_list_response import EventListResponse as EventListResponse
 from .message_list_params import MessageListParams as MessageListParams
 from .state_create_params import StateCreateParams as StateCreateParams
 from .state_list_response import StateListResponse as StateListResponse
 from .state_update_params import StateUpdateParams as StateUpdateParams
-from .tool_response_delta import ToolResponseDelta as ToolResponseDelta
 from .tracker_list_params import TrackerListParams as TrackerListParams
-from .task_message_content import TaskMessageContent as TaskMessageContent
 from .tool_request_content import ToolRequestContent as ToolRequestContent
 from .message_create_params import MessageCreateParams as MessageCreateParams
 from .message_list_response import MessageListResponse as MessageListResponse
@@ -48,6 +41,5 @@
 from .tracker_list_response import TrackerListResponse as TrackerListResponse
 from .tracker_update_params import TrackerUpdateParams as TrackerUpdateParams
 from .agent_rpc_by_name_params import AgentRpcByNameParams as AgentRpcByNameParams
-from .task_message_content_param import TaskMessageContentParam as TaskMessageContentParam
 from .tool_request_content_param import ToolRequestContentParam as ToolRequestContentParam
 from .tool_response_content_param import ToolResponseContentParam as ToolResponseContentParam