feat(api): manual updates

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: c07c588fb8429fbf024189df62f20fa4
-config_hash: 5a41a91d658dffbd60d617eb02e945f6
+config_hash: 2e4b423af3db79ebd8170c401ea9093a

api.md

@@ -1,3 +1,9 @@
+# Shared Types
+
+```python
+from agentex.types import TaskMessageUpdate
+```
+
 # Agents
 
 Types:
@@ -6,10 +12,18 @@ Types:
 from agentex.types import (
     AcpType,
     Agent,
+    AgentRpcParams,
     AgentRpcRequest,
-    AgentListResponse,
     AgentRpcResponse,
-    AgentRpcByNameResponse,
+    AgentRpcResult,
+    DataDelta,
+    TaskMessageContent,
+    TaskMessageDelta,
+    TaskMessageUpdate,
+    TextDelta,
+    ToolRequestDelta,
+    ToolResponseDelta,
+    AgentListResponse,
 )
 ```
 
@@ -21,7 +35,7 @@ Methods:
 - <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_by_name_response.py">AgentRpcByNameResponse</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
 

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,9 +20,9 @@
 )
 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
-from ..types.agent_rpc_by_name_response import AgentRpcByNameResponse
 
 __all__ = ["AgentsResource", "AsyncAgentsResource"]
 
@@ -221,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.
@@ -269,7 +269,7 @@ def rpc_by_name(
         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.
@@ -278,7 +278,7 @@ def rpc_by_name(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AgentRpcByNameResponse:
+    ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique name.
 
@@ -309,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=AgentRpcByNameResponse,
+            cast_to=AgentRpcResponse,
         )
 
 
@@ -507,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.
@@ -555,7 +555,7 @@ async def rpc_by_name(
         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.
@@ -564,7 +564,7 @@ async def rpc_by_name(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AgentRpcByNameResponse:
+    ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique name.
 
@@ -595,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=AgentRpcByNameResponse,
+            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

@@ -29,6 +29,7 @@
 from ..._base_client import make_request_options
 from ...types.task_message import TaskMessage
 from ...types.message_list_response import MessageListResponse
+from ...types.task_message_content_param import TaskMessageContentParam
 
 __all__ = ["MessagesResource", "AsyncMessagesResource"]
 
@@ -60,7 +61,7 @@ def with_streaming_response(self) -> MessagesResourceWithStreamingResponse:
     def create(
         self,
         *,
-        content: message_create_params.Content,
+        content: TaskMessageContentParam,
         task_id: str,
         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.
@@ -135,7 +136,7 @@ def update(
         self,
         message_id: str,
         *,
-        content: message_update_params.Content,
+        content: TaskMessageContentParam,
         task_id: str,
         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.
@@ -245,7 +246,7 @@ def with_streaming_response(self) -> AsyncMessagesResourceWithStreamingResponse:
     async def create(
         self,
         *,
-        content: message_create_params.Content,
+        content: TaskMessageContentParam,
         task_id: str,
         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.
@@ -320,7 +321,7 @@ async def update(
         self,
         message_id: str,
         *,
-        content: message_update_params.Content,
+        content: TaskMessageContentParam,
         task_id: str,
         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.

src/agentex/types/__init__.py

@@ -7,13 +7,17 @@
 from .agent import Agent as Agent
 from .event import Event as Event
 from .state import State as State
+from .shared import TaskMessageUpdate as TaskMessageUpdate
 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 .agent_list_params import AgentListParams as AgentListParams
 from .event_list_params import EventListParams as EventListParams
@@ -25,14 +29,18 @@
 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
@@ -41,6 +49,6 @@
 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 .agent_rpc_by_name_response import AgentRpcByNameResponse as AgentRpcByNameResponse
+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

src/agentex/types/agent_rpc_by_name_params.py

@@ -2,85 +2,20 @@
 
 from __future__ import annotations
 
-from typing import Dict, Union, Optional
-from typing_extensions import Literal, Required, TypeAlias, TypedDict
+from typing import Union
+from typing_extensions import Literal, Required, TypedDict
 
-from .data_content_param import DataContentParam
-from .text_content_param import TextContentParam
-from .tool_request_content_param import ToolRequestContentParam
-from .tool_response_content_param import ToolResponseContentParam
+from .agent_rpc_params import AgentRpcParams
 
-__all__ = [
-    "AgentRpcByNameParams",
-    "Params",
-    "ParamsCreateTaskRequest",
-    "ParamsCancelTaskRequest",
-    "ParamsSendMessageRequest",
-    "ParamsSendMessageRequestContent",
-    "ParamsSendEventRequest",
-    "ParamsSendEventRequestContent",
-]
+__all__ = ["AgentRpcByNameParams"]
 
 
 class AgentRpcByNameParams(TypedDict, total=False):
     method: Required[Literal["event/send", "task/create", "message/send", "task/cancel"]]
 
-    params: Required[Params]
+    params: Required[AgentRpcParams]
     """The parameters for the agent RPC request"""
 
     id: Union[int, str, None]
 
     jsonrpc: Literal["2.0"]
-
-
-class ParamsCreateTaskRequest(TypedDict, total=False):
-    name: Optional[str]
-    """The name of the task to create"""
-
-    params: Optional[Dict[str, object]]
-    """The parameters for the task"""
-
-
-class ParamsCancelTaskRequest(TypedDict, total=False):
-    task_id: Optional[str]
-    """The ID of the task to cancel. Either this or task_name must be provided."""
-
-    task_name: Optional[str]
-    """The name of the task to cancel. Either this or task_id must be provided."""
-
-
-ParamsSendMessageRequestContent: TypeAlias = Union[
-    TextContentParam, DataContentParam, ToolRequestContentParam, ToolResponseContentParam
-]
-
-
-class ParamsSendMessageRequest(TypedDict, total=False):
-    content: Required[ParamsSendMessageRequestContent]
-    """The message that was sent to the agent"""
-
-    stream: bool
-    """Whether to stream the response message back to the client"""
-
-    task_id: Optional[str]
-    """The ID of the task that the message was sent to"""
-
-
-ParamsSendEventRequestContent: TypeAlias = Union[
-    TextContentParam, DataContentParam, ToolRequestContentParam, ToolResponseContentParam
-]
-
-
-class ParamsSendEventRequest(TypedDict, total=False):
-    content: Optional[ParamsSendEventRequestContent]
-    """The content to send to the event"""
-
-    task_id: Optional[str]
-    """The ID of the task that the event was sent to"""
-
-    task_name: Optional[str]
-    """The name of the task that the event was sent to"""
-
-
-Params: TypeAlias = Union[
-    ParamsCreateTaskRequest, ParamsCancelTaskRequest, ParamsSendMessageRequest, ParamsSendEventRequest
-]