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-62a09183a027c64707ad1b9d7a5ccc66c8abf3965e6075282cb5ab72f742a2b3.yml
 openapi_spec_hash: 47f43703822077a82e2edf82d4b7e0e5
-config_hash: dc388766b3eba23cc8e18aa81563182d
+config_hash: 5eaa2103a510f706c6dafdf62f809428

api.md

@@ -1,3 +1,9 @@
+# Shared Types
+
+```python
+from agentex.types import TaskMessageUpdate
+```
+
 # Agents
 
 Types:
@@ -13,7 +19,6 @@ from agentex.types import (
     DataDelta,
     TaskMessageContent,
     TaskMessageDelta,
-    TaskMessageUpdate,
     TextDelta,
     ToolRequestDelta,
     ToolResponseDelta,

src/agentex/types/__init__.py

@@ -7,6 +7,7 @@
 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

src/agentex/types/shared/__init__.py

@@ -0,0 +1,3 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .task_message_update import TaskMessageUpdate as TaskMessageUpdate

src/agentex/types/shared/task_message_update.py

@@ -0,0 +1,83 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Union, Optional
+from typing_extensions import Literal, Annotated, TypeAlias
+
+from ..._utils import PropertyInfo
+from ..._models import BaseModel
+from ..task_message import TaskMessage
+from ..task_message_delta import TaskMessageDelta
+from ..task_message_content import TaskMessageContent
+
+__all__ = [
+    "TaskMessageUpdate",
+    "StreamTaskMessageStart",
+    "StreamTaskMessageDelta",
+    "StreamTaskMessageFull",
+    "StreamTaskMessageDone",
+]
+
+
+class StreamTaskMessageStart(BaseModel):
+    content: TaskMessageContent
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["start"]] = None
+
+
+class StreamTaskMessageDelta(BaseModel):
+    delta: Optional[TaskMessageDelta] = None
+    """Delta for text updates"""
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["delta"]] = None
+
+
+class StreamTaskMessageFull(BaseModel):
+    content: TaskMessageContent
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["full"]] = None
+
+
+class StreamTaskMessageDone(BaseModel):
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["done"]] = None
+
+
+TaskMessageUpdate: TypeAlias = Annotated[
+    Union[StreamTaskMessageStart, StreamTaskMessageDelta, StreamTaskMessageFull, StreamTaskMessageDone],
+    PropertyInfo(discriminator="type"),
+]